diff --git a/src/letta_client/models/client.py b/src/letta_client/models/client.py index 4b6e4e12..c6a31992 100644 --- a/src/letta_client/models/client.py +++ b/src/letta_client/models/client.py @@ -4,6 +4,7 @@ from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper from ..core.request_options import RequestOptions +from ..types.embedding_config import EmbeddingConfig from ..types.llm_config import LlmConfig from ..types.provider_category import ProviderCategory from ..types.provider_type import ProviderType @@ -75,7 +76,7 @@ def list( ) return _response.data - def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> None: + def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[EmbeddingConfig]: """ Parameters ---------- @@ -84,7 +85,7 @@ def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions Returns ------- - None + typing.List[EmbeddingConfig] Examples -------- @@ -172,7 +173,7 @@ async def main() -> None: ) return _response.data - async def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> None: + async def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.List[EmbeddingConfig]: """ Parameters ---------- @@ -181,7 +182,7 @@ async def listembeddingmodels(self, *, request_options: typing.Optional[RequestO Returns ------- - None + typing.List[EmbeddingConfig] Examples -------- diff --git a/src/letta_client/models/raw_client.py b/src/letta_client/models/raw_client.py index 10386b18..9c654d50 100644 --- a/src/letta_client/models/raw_client.py +++ b/src/letta_client/models/raw_client.py @@ -9,6 +9,7 @@ from ..core.request_options import RequestOptions from ..core.unchecked_base_model import construct_type from ..errors.unprocessable_entity_error import UnprocessableEntityError +from ..types.embedding_config import EmbeddingConfig from ..types.http_validation_error import HttpValidationError from ..types.llm_config import LlmConfig from ..types.provider_category import ProviderCategory @@ -82,7 +83,7 @@ def list( raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json) - def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[None]: + def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions] = None) -> HttpResponse[typing.List[EmbeddingConfig]]: """ Parameters ---------- @@ -91,16 +92,23 @@ def listembeddingmodels(self, *, request_options: typing.Optional[RequestOptions Returns ------- - HttpResponse[None] + HttpResponse[typing.List[EmbeddingConfig]] """ _response = self._client_wrapper.httpx_client.request( - "v1/models/embeddings", + "v1/models/embedding", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: - return HttpResponse(response=_response, data=None) + _data = typing.cast( + typing.List[EmbeddingConfig], + construct_type( + type_=typing.List[EmbeddingConfig], # type: ignore + object_=_response.json(), + ), + ) + return HttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) @@ -176,7 +184,7 @@ async def list( async def listembeddingmodels( self, *, request_options: typing.Optional[RequestOptions] = None - ) -> AsyncHttpResponse[None]: + ) -> AsyncHttpResponse[typing.List[EmbeddingConfig]]: """ Parameters ---------- @@ -185,16 +193,23 @@ async def listembeddingmodels( Returns ------- - AsyncHttpResponse[None] + AsyncHttpResponse[typing.List[EmbeddingConfig]] """ _response = await self._client_wrapper.httpx_client.request( - "v1/models/embeddings", + "v1/models/embedding", method="GET", request_options=request_options, ) try: if 200 <= _response.status_code < 300: - return AsyncHttpResponse(response=_response, data=None) + _data = typing.cast( + typing.List[EmbeddingConfig], + construct_type( + type_=typing.List[EmbeddingConfig], # type: ignore + object_=_response.json(), + ), + ) + return AsyncHttpResponse(response=_response, data=_data) _response_json = _response.json() except JSONDecodeError: raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text) diff --git a/tests/custom/test_client.py b/tests/custom/test_client.py index faaacd64..0e66b865 100644 --- a/tests/custom/test_client.py +++ b/tests/custom/test_client.py @@ -41,7 +41,7 @@ def test_create_agent(client) -> None: label="human", ) ], - model="anthropic/claude-3-5-sonnet-20241022", + model="anthropic/claude-sonnet-4-20250514", embedding="openai/text-embedding-ada-002", ) assert agent is not None