Skip to content

XAIFilesClient

autogen.beta.config.xai.files.XAIFilesClient #

XAIFilesClient(config)

Files API client for xAI.

Source code in autogen/beta/config/xai/files.py 
20
21
22
23
24
25
26
27
def __init__(self, config: "XAIConfig") -> None:
    self._client = AsyncClient(
        api_key=config.api_key,
        api_host=config.api_host,
        timeout=config.timeout,
        metadata=config.metadata,
        channel_options=config.channel_options,
    )

upload async #

upload(data, filename, purpose=None)
Source code in autogen/beta/config/xai/files.py 
29
30
31
32
33
34
35
36
37
38
async def upload(self, data: bytes, filename: str, purpose: str | None = None) -> UploadedFile:
    result = await self._client.files.upload(data, filename=filename)
    return UploadedFile(
        file_id=result.id,
        filename=result.filename,
        provider=FileProvider.XAI,
        bytes_count=result.size,
        purpose=purpose,
        created_at=_created_at_to_float(result.created_at),
    )

read async #

read(file_id)
Source code in autogen/beta/config/xai/files.py 
40
41
42
43
44
45
46
async def read(self, file_id: str) -> FileContent:
    metadata = await self._client.files.get(file_id)
    data = await self._client.files.content(file_id)
    return FileContent(
        name=metadata.filename,
        data=data,
    )

list async #

list()
Source code in autogen/beta/config/xai/files.py 
48
49
50
51
52
53
54
55
56
57
58
59
async def list(self) -> list[UploadedFile]:
    result = await self._client.files.list()
    return [
        UploadedFile(
            file_id=f.id,
            filename=f.filename,
            provider=FileProvider.XAI,
            bytes_count=f.size,
            created_at=_created_at_to_float(f.created_at),
        )
        for f in result.data
    ]

delete async #

delete(file_id)
Source code in autogen/beta/config/xai/files.py 
61
62
async def delete(self, file_id: str) -> None:
    await self._client.files.delete(file_id)