llama-cloud 0.0.8__py3-none-any.whl → 0.0.9__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -9,6 +9,7 @@ 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_params import ChatParams
 from ...types.cloud_document import CloudDocument
 from ...types.cloud_document_create import CloudDocumentCreate
 from ...types.configured_transformation_item import ConfiguredTransformationItem
@@ -30,6 +31,7 @@ from ...types.pipeline_file import PipelineFile
 from ...types.pipeline_file_create import PipelineFileCreate
 from ...types.pipeline_type import PipelineType
 from ...types.preset_retrieval_params import PresetRetrievalParams
+from ...types.retrieval_mode import RetrievalMode
 from ...types.retrieve_results import RetrieveResults
 from .types.pipeline_file_update_custom_metadata_value import PipelineFileUpdateCustomMetadataValue
 
@@ -52,19 +54,22 @@ class PipelinesClient:
     def search_pipelines(
         self,
         *,
-        project_name: str,
+        project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
-            - project_name: str.
+            - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import LlamaCloud
@@ -73,7 +78,6 @@ class PipelinesClient:
             token="YOUR_TOKEN",
         )
         client.pipelines.search_pipelines(
-            project_name="string",
             pipeline_type=PipelineType.PLAYGROUND,
         )
         """
@@ -81,7 +85,12 @@ class PipelinesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
-                {"project_name": project_name, "pipeline_name": pipeline_name, "pipeline_type": pipeline_type}
+                {
+                    "project_name": project_name,
+                    "pipeline_name": pipeline_name,
+                    "pipeline_type": pipeline_type,
+                    "project_id": project_id,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -115,6 +124,7 @@ class PipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -133,6 +143,7 @@ class PipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -181,6 +192,7 @@ class PipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -199,6 +211,7 @@ class PipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -305,6 +318,7 @@ class PipelinesClient:
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import LlamaCloud
@@ -323,6 +337,7 @@ class PipelinesClient:
                     filters=[],
                     condition=FilterCondition.AND,
                 ),
+                retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
                 llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1069,6 +1084,8 @@ class PipelinesClient:
         rerank_top_n: typing.Optional[int] = OMIT,
         alpha: typing.Optional[float] = OMIT,
         search_filters: typing.Optional[MetadataFilters] = OMIT,
+        files_top_k: typing.Optional[int] = OMIT,
+        retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -1089,9 +1106,13 @@ class PipelinesClient:
 
             - search_filters: typing.Optional[MetadataFilters]. Search filters for retrieval.
 
+            - files_top_k: typing.Optional[int]. Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content).
+
+            - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
+
             - query: str. The query to retrieve against.
         ---
-        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -1103,6 +1124,7 @@ class PipelinesClient:
                 filters=[],
                 condition=FilterCondition.AND,
             ),
+            retrieval_mode=RetrievalMode.CHUNKS,
             query="string",
         )
         """
@@ -1119,6 +1141,10 @@ class PipelinesClient:
             _request["alpha"] = alpha
         if search_filters is not OMIT:
             _request["search_filters"] = search_filters
+        if files_top_k is not OMIT:
+            _request["files_top_k"] = files_top_k
+        if retrieval_mode is not OMIT:
+            _request["retrieval_mode"] = retrieval_mode
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/retrieve"),
@@ -1205,6 +1231,58 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def chat(self, pipeline_id: str, *, retrieval_parameters: PresetRetrievalParams, data: ChatParams) -> typing.Any:
+        """
+        Parameters:
+            - pipeline_id: str.
+
+            - retrieval_parameters: PresetRetrievalParams.
+
+            - data: ChatParams.
+        ---
+        from llama_cloud import (
+            ChatParams,
+            FilterCondition,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+        )
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.chat(
+            pipeline_id="string",
+            retrieval_parameters=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+            data=ChatParams(
+                messages=[],
+            ),
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/chat"),
+            json=jsonable_encoder({"retrieval_parameters": retrieval_parameters, "data": data}),
+            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 list_pipeline_documents(
         self, pipeline_id: str, *, skip: typing.Optional[int] = None, limit: typing.Optional[int] = None
     ) -> typing.List[CloudDocument]:
@@ -1446,19 +1524,22 @@ class AsyncPipelinesClient:
     async def search_pipelines(
         self,
         *,
-        project_name: str,
+        project_name: typing.Optional[str] = None,
         pipeline_name: typing.Optional[str] = None,
         pipeline_type: typing.Optional[PipelineType] = None,
+        project_id: typing.Optional[str] = None,
     ) -> typing.List[Pipeline]:
         """
         Search for pipelines by various parameters.
 
         Parameters:
-            - project_name: str.
+            - project_name: typing.Optional[str].
 
             - pipeline_name: typing.Optional[str].
 
             - pipeline_type: typing.Optional[PipelineType].
+
+            - project_id: typing.Optional[str].
         ---
         from llama_cloud import PipelineType
         from llama_cloud.client import AsyncLlamaCloud
@@ -1467,7 +1548,6 @@ class AsyncPipelinesClient:
             token="YOUR_TOKEN",
         )
         await client.pipelines.search_pipelines(
-            project_name="string",
             pipeline_type=PipelineType.PLAYGROUND,
         )
         """
@@ -1475,7 +1555,12 @@ class AsyncPipelinesClient:
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/pipelines"),
             params=remove_none_from_dict(
-                {"project_name": project_name, "pipeline_name": pipeline_name, "pipeline_type": pipeline_type}
+                {
+                    "project_name": project_name,
+                    "pipeline_name": pipeline_name,
+                    "pipeline_type": pipeline_type,
+                    "project_id": project_id,
+                }
             ),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -1509,6 +1594,7 @@ class AsyncPipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1527,6 +1613,7 @@ class AsyncPipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1575,6 +1662,7 @@ class AsyncPipelinesClient:
             PipelineCreate,
             PipelineType,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1593,6 +1681,7 @@ class AsyncPipelinesClient:
                         filters=[],
                         condition=FilterCondition.AND,
                     ),
+                    retrieval_mode=RetrievalMode.CHUNKS,
                 ),
                 eval_parameters=EvalExecutionParams(
                     llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -1699,6 +1788,7 @@ class AsyncPipelinesClient:
             LlamaParseParameters,
             MetadataFilters,
             PresetRetrievalParams,
+            RetrievalMode,
             SupportedEvalLlmModelNames,
         )
         from llama_cloud.client import AsyncLlamaCloud
@@ -1717,6 +1807,7 @@ class AsyncPipelinesClient:
                     filters=[],
                     condition=FilterCondition.AND,
                 ),
+                retrieval_mode=RetrievalMode.CHUNKS,
             ),
             eval_parameters=EvalExecutionParams(
                 llm_model=SupportedEvalLlmModelNames.GPT_3_5_TURBO,
@@ -2465,6 +2556,8 @@ class AsyncPipelinesClient:
         rerank_top_n: typing.Optional[int] = OMIT,
         alpha: typing.Optional[float] = OMIT,
         search_filters: typing.Optional[MetadataFilters] = OMIT,
+        files_top_k: typing.Optional[int] = OMIT,
+        retrieval_mode: typing.Optional[RetrievalMode] = OMIT,
         query: str,
     ) -> RetrieveResults:
         """
@@ -2485,9 +2578,13 @@ class AsyncPipelinesClient:
 
             - search_filters: typing.Optional[MetadataFilters]. Search filters for retrieval.
 
+            - files_top_k: typing.Optional[int]. Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content).
+
+            - retrieval_mode: typing.Optional[RetrievalMode]. The retrieval mode for the query.
+
             - query: str. The query to retrieve against.
         ---
-        from llama_cloud import FilterCondition, MetadataFilters
+        from llama_cloud import FilterCondition, MetadataFilters, RetrievalMode
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -2499,6 +2596,7 @@ class AsyncPipelinesClient:
                 filters=[],
                 condition=FilterCondition.AND,
             ),
+            retrieval_mode=RetrievalMode.CHUNKS,
             query="string",
         )
         """
@@ -2515,6 +2613,10 @@ class AsyncPipelinesClient:
             _request["alpha"] = alpha
         if search_filters is not OMIT:
             _request["search_filters"] = search_filters
+        if files_top_k is not OMIT:
+            _request["files_top_k"] = files_top_k
+        if retrieval_mode is not OMIT:
+            _request["retrieval_mode"] = retrieval_mode
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/retrieve"),
@@ -2601,6 +2703,60 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def chat(
+        self, pipeline_id: str, *, retrieval_parameters: PresetRetrievalParams, data: ChatParams
+    ) -> typing.Any:
+        """
+        Parameters:
+            - pipeline_id: str.
+
+            - retrieval_parameters: PresetRetrievalParams.
+
+            - data: ChatParams.
+        ---
+        from llama_cloud import (
+            ChatParams,
+            FilterCondition,
+            MetadataFilters,
+            PresetRetrievalParams,
+            RetrievalMode,
+        )
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.chat(
+            pipeline_id="string",
+            retrieval_parameters=PresetRetrievalParams(
+                search_filters=MetadataFilters(
+                    filters=[],
+                    condition=FilterCondition.AND,
+                ),
+                retrieval_mode=RetrievalMode.CHUNKS,
+            ),
+            data=ChatParams(
+                messages=[],
+            ),
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/chat"),
+            json=jsonable_encoder({"retrieval_parameters": retrieval_parameters, "data": data}),
+            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 list_pipeline_documents(
         self, pipeline_id: str, *, skip: typing.Optional[int] = None, limit: typing.Optional[int] = None
     ) -> typing.List[CloudDocument]:

llama_cloud/resources/projects/client.py

@@ -34,11 +34,15 @@ class ProjectsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def list_projects(self, *, project_name: typing.Optional[str] = None) -> typing.List[Project]:
+    def list_projects(
+        self, *, organization_id: typing.Optional[str] = None, project_name: typing.Optional[str] = None
+    ) -> typing.List[Project]:
         """
         List projects or get one by name
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - project_name: typing.Optional[str].
         ---
         from llama_cloud.client import LlamaCloud
@@ -51,7 +55,7 @@ class ProjectsClient:
         _response = self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
-            params=remove_none_from_dict({"project_name": project_name}),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_name": project_name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -65,11 +69,13 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def create_project(self, *, request: ProjectCreate) -> Project:
+    def create_project(self, *, organization_id: typing.Optional[str] = None, request: ProjectCreate) -> Project:
         """
         Create a new project.
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - request: ProjectCreate.
         ---
         from llama_cloud import ProjectCreate
@@ -87,6 +93,7 @@ class ProjectsClient:
         _response = self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -101,12 +108,14 @@ class ProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def upsert_project(self, *, request: ProjectCreate) -> Project:
+    def upsert_project(self, *, organization_id: typing.Optional[str] = None, request: ProjectCreate) -> Project:
         """
         Upsert a project.
         Updates if a project with the same name already exists. Otherwise, creates a new project.
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - request: ProjectCreate.
         ---
         from llama_cloud import ProjectCreate
@@ -124,6 +133,7 @@ class ProjectsClient:
         _response = self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -620,11 +630,15 @@ class AsyncProjectsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def list_projects(self, *, project_name: typing.Optional[str] = None) -> typing.List[Project]:
+    async def list_projects(
+        self, *, organization_id: typing.Optional[str] = None, project_name: typing.Optional[str] = None
+    ) -> typing.List[Project]:
         """
         List projects or get one by name
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - project_name: typing.Optional[str].
         ---
         from llama_cloud.client import AsyncLlamaCloud
@@ -637,7 +651,7 @@ class AsyncProjectsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "GET",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
-            params=remove_none_from_dict({"project_name": project_name}),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_name": project_name}),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
         )
@@ -651,11 +665,13 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def create_project(self, *, request: ProjectCreate) -> Project:
+    async def create_project(self, *, organization_id: typing.Optional[str] = None, request: ProjectCreate) -> Project:
         """
         Create a new project.
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - request: ProjectCreate.
         ---
         from llama_cloud import ProjectCreate
@@ -673,6 +689,7 @@ class AsyncProjectsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "POST",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,
@@ -687,12 +704,14 @@ class AsyncProjectsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def upsert_project(self, *, request: ProjectCreate) -> Project:
+    async def upsert_project(self, *, organization_id: typing.Optional[str] = None, request: ProjectCreate) -> Project:
         """
         Upsert a project.
         Updates if a project with the same name already exists. Otherwise, creates a new project.
 
         Parameters:
+            - organization_id: typing.Optional[str].
+
             - request: ProjectCreate.
         ---
         from llama_cloud import ProjectCreate
@@ -710,6 +729,7 @@ class AsyncProjectsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "PUT",
             urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/projects"),
+            params=remove_none_from_dict({"organization_id": organization_id}),
             json=jsonable_encoder(request),
             headers=self._client_wrapper.get_headers(),
             timeout=60,

llama_cloud/types/__init__.py

@@ -5,11 +5,13 @@ from .base import Base
 from .base_prompt_template import BasePromptTemplate
 from .bedrock_embedding import BedrockEmbedding
 from .chat_message import ChatMessage
+from .chat_params import ChatParams
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
 from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_document import CloudDocument
 from .cloud_document_create import CloudDocumentCreate
+from .cloud_jira_data_source import CloudJiraDataSource
 from .cloud_notion_page_data_source import CloudNotionPageDataSource
 from .cloud_one_drive_data_source import CloudOneDriveDataSource
 from .cloud_pinecone_vector_store import CloudPineconeVectorStore
@@ -86,6 +88,8 @@ from .metric_result import MetricResult
 from .node_parser import NodeParser
 from .object_type import ObjectType
 from .open_ai_embedding import OpenAiEmbedding
+from .organization import Organization
+from .organization_create import OrganizationCreate
 from .parser_languages import ParserLanguages
 from .parsing_history_item import ParsingHistoryItem
 from .parsing_job import ParsingJob
@@ -116,6 +120,7 @@ from .prompt_mixin_prompts import PromptMixinPrompts
 from .prompt_spec import PromptSpec
 from .pydantic_program_mode import PydanticProgramMode
 from .related_node_info import RelatedNodeInfo
+from .retrieval_mode import RetrievalMode
 from .retrieve_results import RetrieveResults
 from .sentence_splitter import SentenceSplitter
 from .simple_file_node_parser import SimpleFileNodeParser
@@ -127,6 +132,8 @@ from .text_node_relationships_value import TextNodeRelationshipsValue
 from .text_node_with_score import TextNodeWithScore
 from .token_text_splitter import TokenTextSplitter
 from .transformation_category_names import TransformationCategoryNames
+from .user_organization import UserOrganization
+from .user_organization_create import UserOrganizationCreate
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
 
@@ -136,11 +143,13 @@ __all__ = [
     "BasePromptTemplate",
     "BedrockEmbedding",
     "ChatMessage",
+    "ChatParams",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudJiraDataSource",
     "CloudNotionPageDataSource",
     "CloudOneDriveDataSource",
     "CloudPineconeVectorStore",
@@ -217,6 +226,8 @@ __all__ = [
     "NodeParser",
     "ObjectType",
     "OpenAiEmbedding",
+    "Organization",
+    "OrganizationCreate",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -247,6 +258,7 @@ __all__ = [
     "PromptSpec",
     "PydanticProgramMode",
     "RelatedNodeInfo",
+    "RetrievalMode",
     "RetrieveResults",
     "SentenceSplitter",
     "SimpleFileNodeParser",
@@ -258,6 +270,8 @@ __all__ = [
     "TextNodeWithScore",
     "TokenTextSplitter",
     "TransformationCategoryNames",
+    "UserOrganization",
+    "UserOrganizationCreate",
     "ValidationError",
     "ValidationErrorLocItem",
 ]

llama_cloud/types/chat_params.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .chat_message import ChatMessage
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ChatParams(pydantic.BaseModel):
+    """
+    Base schema model for BaseComponent classes used in the platform.
+    Comes with special serialization logic for types used commonly in platform codebase.
+    """
+
+    messages: typing.List[ChatMessage]
+    class_name: typing.Optional[str]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}