llama-cloud 0.1.5__py3-none-any.whl → 0.1.7__py3-none-any.whl

llama_cloud/resources/chat_apps/client.py

@@ -0,0 +1,630 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+import urllib.parse
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.jsonable_encoder import jsonable_encoder
+from ...core.remove_none_from_dict import remove_none_from_dict
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.chat_app import ChatApp
+from ...types.chat_app_response import ChatAppResponse
+from ...types.http_validation_error import HttpValidationError
+from ...types.input_message import InputMessage
+from ...types.llm_parameters import LlmParameters
+from ...types.preset_retrieval_params import PresetRetrievalParams
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class ChatAppsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def get_chat_apps_api_v_1_apps_get(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[ChatAppResponse]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.get_chat_apps_api_v_1_apps_get()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[ChatAppResponse], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def create_chat_app_api_v_1_apps_post(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: str,
+        pipeline_id: str,
+        llm_config: LlmParameters,
+        retrieval_config: PresetRetrievalParams,
+        chat_app_create_project_id: str,
+    ) -> ChatApp:
+        """
+        Create a new chat app.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - name: str.
+
+            - pipeline_id: str.
+
+            - llm_config: LlmParameters.
+
+            - retrieval_config: PresetRetrievalParams.
+
+            - chat_app_create_project_id: str.
+        ---
+        from llama_cloud import (
+            FilterCondition,
+            LlmParameters,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+            SupportedLlmModelNames,
+        )
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.create_chat_app_api_v_1_apps_post(
+            name="string",
+            pipeline_id="string",
+            llm_config=LlmParameters(
+                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+            ),
+            retrieval_config=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+            chat_app_create_project_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(
+                {
+                    "name": name,
+                    "pipeline_id": pipeline_id,
+                    "llm_config": llm_config,
+                    "retrieval_config": retrieval_config,
+                    "project_id": chat_app_create_project_id,
+                }
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def get_chat_app(self, id: str) -> ChatApp:
+        """
+        Get a chat app by ID.
+
+        Parameters:
+            - id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.get_chat_app(
+            id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def delete_chat_app(self, id: str) -> typing.Any:
+        """
+        Parameters:
+            - id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.delete_chat_app(
+            id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def update_chat_app(
+        self,
+        id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: typing.Optional[str] = OMIT,
+        llm_config: typing.Optional[LlmParameters] = OMIT,
+        retrieval_config: typing.Optional[PresetRetrievalParams] = OMIT,
+    ) -> ChatApp:
+        """
+        Update a chat app.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - name: typing.Optional[str].
+
+            - llm_config: typing.Optional[LlmParameters].
+
+            - retrieval_config: typing.Optional[PresetRetrievalParams].
+        ---
+        from llama_cloud import (
+            FilterCondition,
+            LlmParameters,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+            SupportedLlmModelNames,
+        )
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.update_chat_app(
+            id="string",
+            llm_config=LlmParameters(
+                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+            ),
+            retrieval_config=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if name is not OMIT:
+            _request["name"] = name
+        if llm_config is not OMIT:
+            _request["llm_config"] = llm_config
+        if retrieval_config is not OMIT:
+            _request["retrieval_config"] = retrieval_config
+        _response = self._client_wrapper.httpx_client.request(
+            "PATCH",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    def chat_with_chat_app(self, id: str, *, messages: typing.Optional[typing.List[InputMessage]] = OMIT) -> typing.Any:
+        """
+        Chat with a chat app.
+
+        Parameters:
+            - id: str.
+
+            - messages: typing.Optional[typing.List[InputMessage]].
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.chat_apps.chat_with_chat_app(
+            id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if messages is not OMIT:
+            _request["messages"] = messages
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}/chat"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncChatAppsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def get_chat_apps_api_v_1_apps_get(
+        self, *, project_id: typing.Optional[str] = None, organization_id: typing.Optional[str] = None
+    ) -> typing.List[ChatAppResponse]:
+        """
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.get_chat_apps_api_v_1_apps_get()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.List[ChatAppResponse], _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def create_chat_app_api_v_1_apps_post(
+        self,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: str,
+        pipeline_id: str,
+        llm_config: LlmParameters,
+        retrieval_config: PresetRetrievalParams,
+        chat_app_create_project_id: str,
+    ) -> ChatApp:
+        """
+        Create a new chat app.
+
+        Parameters:
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - name: str.
+
+            - pipeline_id: str.
+
+            - llm_config: LlmParameters.
+
+            - retrieval_config: PresetRetrievalParams.
+
+            - chat_app_create_project_id: str.
+        ---
+        from llama_cloud import (
+            FilterCondition,
+            LlmParameters,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+            SupportedLlmModelNames,
+        )
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.create_chat_app_api_v_1_apps_post(
+            name="string",
+            pipeline_id="string",
+            llm_config=LlmParameters(
+                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+            ),
+            retrieval_config=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+            chat_app_create_project_id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/apps"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(
+                {
+                    "name": name,
+                    "pipeline_id": pipeline_id,
+                    "llm_config": llm_config,
+                    "retrieval_config": retrieval_config,
+                    "project_id": chat_app_create_project_id,
+                }
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def get_chat_app(self, id: str) -> ChatApp:
+        """
+        Get a chat app by ID.
+
+        Parameters:
+            - id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.get_chat_app(
+            id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "GET",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def delete_chat_app(self, id: str) -> typing.Any:
+        """
+        Parameters:
+            - id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.delete_chat_app(
+            id="string",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "DELETE",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def update_chat_app(
+        self,
+        id: str,
+        *,
+        project_id: typing.Optional[str] = None,
+        organization_id: typing.Optional[str] = None,
+        name: typing.Optional[str] = OMIT,
+        llm_config: typing.Optional[LlmParameters] = OMIT,
+        retrieval_config: typing.Optional[PresetRetrievalParams] = OMIT,
+    ) -> ChatApp:
+        """
+        Update a chat app.
+
+        Parameters:
+            - id: str.
+
+            - project_id: typing.Optional[str].
+
+            - organization_id: typing.Optional[str].
+
+            - name: typing.Optional[str].
+
+            - llm_config: typing.Optional[LlmParameters].
+
+            - retrieval_config: typing.Optional[PresetRetrievalParams].
+        ---
+        from llama_cloud import (
+            FilterCondition,
+            LlmParameters,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+            SupportedLlmModelNames,
+        )
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.update_chat_app(
+            id="string",
+            llm_config=LlmParameters(
+                model_name=SupportedLlmModelNames.GPT_3_5_TURBO,
+            ),
+            retrieval_config=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if name is not OMIT:
+            _request["name"] = name
+        if llm_config is not OMIT:
+            _request["llm_config"] = llm_config
+        if retrieval_config is not OMIT:
+            _request["retrieval_config"] = retrieval_config
+        _response = await self._client_wrapper.httpx_client.request(
+            "PATCH",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}"),
+            params=remove_none_from_dict({"project_id": project_id, "organization_id": organization_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ChatApp, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def chat_with_chat_app(
+        self, id: str, *, messages: typing.Optional[typing.List[InputMessage]] = OMIT
+    ) -> typing.Any:
+        """
+        Chat with a chat app.
+
+        Parameters:
+            - id: str.
+
+            - messages: typing.Optional[typing.List[InputMessage]].
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.chat_apps.chat_with_chat_app(
+            id="string",
+        )
+        """
+        _request: typing.Dict[str, typing.Any] = {}
+        if messages is not OMIT:
+            _request["messages"] = messages
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/apps/{id}/chat"),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

llama_cloud/resources/data_sinks/client.py

@@ -213,7 +213,7 @@ class DataSinksClient:
 
             - sink_type: ConfigurableDataSinkNames.
 
-            - component: typing.Optional[DataSinkUpdateComponent].
+            - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
         ---
         from llama_cloud import ConfigurableDataSinkNames
         from llama_cloud.client import LlamaCloud
@@ -467,7 +467,7 @@ class AsyncDataSinksClient:
 
             - sink_type: ConfigurableDataSinkNames.
 
-            - component: typing.Optional[DataSinkUpdateComponent].
+            - component: typing.Optional[DataSinkUpdateComponent]. Component that implements the data sink
         ---
         from llama_cloud import ConfigurableDataSinkNames
         from llama_cloud.client import AsyncLlamaCloud

llama_cloud/resources/data_sources/client.py

@@ -218,7 +218,7 @@ class DataSourcesClient:
 
             - custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceUpdateCustomMetadataValue]]].
 
-            - component: typing.Optional[DataSourceUpdateComponent].
+            - component: typing.Optional[DataSourceUpdateComponent]. Component that implements the data source
         ---
         from llama_cloud import ConfigurableDataSourceNames
         from llama_cloud.client import LlamaCloud
@@ -478,7 +478,7 @@ class AsyncDataSourcesClient:
 
             - custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceUpdateCustomMetadataValue]]].
 
-            - component: typing.Optional[DataSourceUpdateComponent].
+            - component: typing.Optional[DataSourceUpdateComponent]. Component that implements the data source
         ---
         from llama_cloud import ConfigurableDataSourceNames
         from llama_cloud.client import AsyncLlamaCloud

llama_cloud/resources/embedding_model_configs/client.py

@@ -63,7 +63,7 @@ class EmbeddingModelConfigsClient:
 
     def create_embedding_model_config(
         self, *, project_id: str, name: str, embedding_config: EmbeddingModelConfigCreateEmbeddingConfig
-    ) -> typing.Any:
+    ) -> EmbeddingModelConfig:
         """
         Create a new embedding model configuration within a specified project.
 
@@ -83,7 +83,7 @@ class EmbeddingModelConfigsClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(EmbeddingModelConfig, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -230,7 +230,7 @@ class AsyncEmbeddingModelConfigsClient:
 
     async def create_embedding_model_config(
         self, *, project_id: str, name: str, embedding_config: EmbeddingModelConfigCreateEmbeddingConfig
-    ) -> typing.Any:
+    ) -> EmbeddingModelConfig:
         """
         Create a new embedding model configuration within a specified project.
 
@@ -250,7 +250,7 @@ class AsyncEmbeddingModelConfigsClient:
             timeout=60,
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(typing.Any, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(EmbeddingModelConfig, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try: