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

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,18 +5,21 @@ 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_gcs_data_source import CloudGcsDataSource
-from .cloud_google_drive_data_source import CloudGoogleDriveDataSource
+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
 from .cloud_postgres_vector_store import CloudPostgresVectorStore
 from .cloud_qdrant_vector_store import CloudQdrantVectorStore
 from .cloud_s_3_data_source import CloudS3DataSource
 from .cloud_sharepoint_data_source import CloudSharepointDataSource
+from .cloud_slack_data_source import CloudSlackDataSource
 from .cloud_weaviate_vector_store import CloudWeaviateVectorStore
 from .code_splitter import CodeSplitter
 from .cohere_embedding import CohereEmbedding
@@ -52,6 +55,10 @@ from .eval_llm_model_data import EvalLlmModelData
 from .eval_question import EvalQuestion
 from .eval_question_create import EvalQuestionCreate
 from .eval_question_result import EvalQuestionResult
+from .extraction_result import ExtractionResult
+from .extraction_result_data_value import ExtractionResultDataValue
+from .extraction_schema import ExtractionSchema
+from .extraction_schema_data_schema_value import ExtractionSchemaDataSchemaValue
 from .file import File
 from .file_resource_info_value import FileResourceInfoValue
 from .filter_condition import FilterCondition
@@ -81,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
@@ -111,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
@@ -122,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
 
@@ -131,18 +143,21 @@ __all__ = [
     "BasePromptTemplate",
     "BedrockEmbedding",
     "ChatMessage",
+    "ChatParams",
     "CloudAzStorageBlobDataSource",
+    "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
     "CloudDocument",
     "CloudDocumentCreate",
-    "CloudGcsDataSource",
-    "CloudGoogleDriveDataSource",
+    "CloudJiraDataSource",
+    "CloudNotionPageDataSource",
     "CloudOneDriveDataSource",
     "CloudPineconeVectorStore",
     "CloudPostgresVectorStore",
     "CloudQdrantVectorStore",
     "CloudS3DataSource",
     "CloudSharepointDataSource",
+    "CloudSlackDataSource",
     "CloudWeaviateVectorStore",
     "CodeSplitter",
     "CohereEmbedding",
@@ -178,6 +193,10 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtractionResult",
+    "ExtractionResultDataValue",
+    "ExtractionSchema",
+    "ExtractionSchemaDataSchemaValue",
     "File",
     "FileResourceInfoValue",
     "FilterCondition",
@@ -207,6 +226,8 @@ __all__ = [
     "NodeParser",
     "ObjectType",
     "OpenAiEmbedding",
+    "Organization",
+    "OrganizationCreate",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -237,6 +258,7 @@ __all__ = [
     "PromptSpec",
     "PydanticProgramMode",
     "RelatedNodeInfo",
+    "RetrievalMode",
     "RetrieveResults",
     "SentenceSplitter",
     "SimpleFileNodeParser",
@@ -248,6 +270,8 @@ __all__ = [
     "TextNodeWithScore",
     "TokenTextSplitter",
     "TransformationCategoryNames",
+    "UserOrganization",
+    "UserOrganizationCreate",
     "ValidationError",
     "ValidationErrorLocItem",
 ]

llama_cloud/types/azure_open_ai_embedding.py

@@ -62,6 +62,9 @@ class AzureOpenAiEmbedding(pydantic.BaseModel):
     )
     azure_endpoint: typing.Optional[str] = pydantic.Field(description="The Azure endpoint to use.")
     azure_deployment: typing.Optional[str] = pydantic.Field(description="The Azure deployment to use.")
+    use_azure_ad: bool = pydantic.Field(
+        description="Indicates if Microsoft Entra ID (former Azure AD) is used for token authentication"
+    )
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/{cloud_google_drive_data_source.py → chat_params.py}

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .chat_message import ChatMessage
 
 try:
     import pydantic
@@ -14,15 +15,13 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class CloudGoogleDriveDataSource(pydantic.BaseModel):
+class ChatParams(pydantic.BaseModel):
     """
-    Base component object to capture class names.
+    Base schema model for BaseComponent classes used in the platform.
+    Comes with special serialization logic for types used commonly in platform codebase.
     """
 
-    folder_id: str = pydantic.Field(description="The ID of the Google Drive folder to read from.")
-    service_account_key: typing.Dict[str, typing.Any] = pydantic.Field(
-        description="The service account key JSON to use for authentication."
-    )
+    messages: typing.List[ChatMessage]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_azure_ai_search_vector_store.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CloudAzureAiSearchVectorStore(pydantic.BaseModel):
+    """
+    Cloud Azure AI Search Vector Store.
+    """
+
+    supports_nested_metadata_filters: typing.Optional[bool]
+    search_service_api_key: str
+    search_service_endpoint: str
+    search_service_api_version: typing.Optional[str]
+    index_name: typing.Optional[str]
+    filterable_metadata_field_keys: typing.Optional[typing.List[str]]
+    embedding_dimension: typing.Optional[int]
+    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}

llama_cloud/types/cloud_jira_data_source.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CloudJiraDataSource(pydantic.BaseModel):
+    """
+    Cloud Jira Data Source integrating JiraReader.
+    """
+
+    email: typing.Optional[str] = pydantic.Field(description="The email address to use for authentication.")
+    api_token: typing.Optional[str] = pydantic.Field(
+        description="The API/ Access Token used for Basic, PAT and OAuth2 authentication."
+    )
+    server_url: typing.Optional[str] = pydantic.Field(description="The server url for Jira Cloud.")
+    cloud_id: typing.Optional[str] = pydantic.Field(description="The cloud ID, used in case of OAuth2.")
+    authentication_mechanism: str = pydantic.Field(description="Type of Authentication for connecting to Jira APIs.")
+    query: str = pydantic.Field(description="JQL (Jira Query Language) query to search.")
+    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}

llama_cloud/types/{cloud_gcs_data_source.py → cloud_notion_page_data_source.py}

@@ -14,16 +14,14 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class CloudGcsDataSource(pydantic.BaseModel):
+class CloudNotionPageDataSource(pydantic.BaseModel):
     """
     Base component object to capture class names.
     """
 
-    bucket: str = pydantic.Field(description="The name of the GCS bucket to read from.")
-    prefix: typing.Optional[str] = pydantic.Field(description="The prefix of the GCS objects to read from.")
-    service_account_key: typing.Dict[str, typing.Any] = pydantic.Field(
-        description="The service account key JSON to use for authentication."
-    )
+    integration_token: str = pydantic.Field(description="The integration token to use for authentication.")
+    database_ids: typing.Optional[str] = pydantic.Field(description="The Notion Database Id to read content from.")
+    page_ids: typing.Optional[str] = pydantic.Field(description="The Page ID's of the Notion to read from.")
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/cloud_sharepoint_data_source.py

@@ -22,6 +22,7 @@ class CloudSharepointDataSource(pydantic.BaseModel):
     site_name: str = pydantic.Field(description="The name of the SharePoint site to download from.")
     folder_path: typing.Optional[str] = pydantic.Field(description="The path of the Sharepoint folder to read from.")
     folder_id: typing.Optional[str] = pydantic.Field(description="The ID of the Sharepoint folder to read from.")
+    drive_name: typing.Optional[str] = pydantic.Field(description="The name of the Sharepoint drive to read from.")
     client_id: str = pydantic.Field(description="The client ID to use for authentication.")
     client_secret: str = pydantic.Field(description="The client secret to use for authentication.")
     tenant_id: str = pydantic.Field(description="The tenant ID to use for authentication.")

llama_cloud/types/cloud_slack_data_source.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class CloudSlackDataSource(pydantic.BaseModel):
+    """
+    Base component object to capture class names.
+    """
+
+    slack_token: str = pydantic.Field(description="Slack Bot Token.")
+    channel_ids: typing.Optional[str] = pydantic.Field(description="Slack Channel.")
+    latest_date: typing.Optional[str] = pydantic.Field(description="Latest date.")
+    earliest_date: typing.Optional[str] = pydantic.Field(description="Earliest date.")
+    earliest_date_timestamp: typing.Optional[float] = pydantic.Field(description="Earliest date timestamp.")
+    latest_date_timestamp: typing.Optional[float] = pydantic.Field(description="Latest date timestamp.")
+    channel_patterns: typing.Optional[str] = pydantic.Field(description="Slack Channel name pattern.")
+    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}

llama_cloud/types/configurable_data_sink_names.py

@@ -16,6 +16,7 @@ class ConfigurableDataSinkNames(str, enum.Enum):
     POSTGRES = "POSTGRES"
     QDRANT = "QDRANT"
     WEAVIATE = "WEAVIATE"
+    AZUREAI_SEARCH = "AZUREAI_SEARCH"
 
     def visit(
         self,
@@ -24,6 +25,7 @@ class ConfigurableDataSinkNames(str, enum.Enum):
         postgres: typing.Callable[[], T_Result],
         qdrant: typing.Callable[[], T_Result],
         weaviate: typing.Callable[[], T_Result],
+        azureai_search: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ConfigurableDataSinkNames.CHROMA:
             return chroma()
@@ -35,3 +37,5 @@ class ConfigurableDataSinkNames(str, enum.Enum):
             return qdrant()
         if self is ConfigurableDataSinkNames.WEAVIATE:
             return weaviate()
+        if self is ConfigurableDataSinkNames.AZUREAI_SEARCH:
+            return azureai_search()

llama_cloud/types/configurable_data_source_names.py

@@ -13,29 +13,33 @@ class ConfigurableDataSourceNames(str, enum.Enum):
 
     S_3 = "S3"
     AZURE_STORAGE_BLOB = "AZURE_STORAGE_BLOB"
-    GCS = "GCS"
-    GOOGLE_DRIVE = "GOOGLE_DRIVE"
     MICROSOFT_ONEDRIVE = "MICROSOFT_ONEDRIVE"
     MICROSOFT_SHAREPOINT = "MICROSOFT_SHAREPOINT"
+    SLACK = "SLACK"
+    NOTION_PAGE = "NOTION_PAGE"
+    JIRA = "JIRA"
 
     def visit(
         self,
         s_3: typing.Callable[[], T_Result],
         azure_storage_blob: typing.Callable[[], T_Result],
-        gcs: typing.Callable[[], T_Result],
-        google_drive: typing.Callable[[], T_Result],
         microsoft_onedrive: typing.Callable[[], T_Result],
         microsoft_sharepoint: typing.Callable[[], T_Result],
+        slack: typing.Callable[[], T_Result],
+        notion_page: typing.Callable[[], T_Result],
+        jira: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ConfigurableDataSourceNames.S_3:
             return s_3()
         if self is ConfigurableDataSourceNames.AZURE_STORAGE_BLOB:
             return azure_storage_blob()
-        if self is ConfigurableDataSourceNames.GCS:
-            return gcs()
-        if self is ConfigurableDataSourceNames.GOOGLE_DRIVE:
-            return google_drive()
         if self is ConfigurableDataSourceNames.MICROSOFT_ONEDRIVE:
             return microsoft_onedrive()
         if self is ConfigurableDataSourceNames.MICROSOFT_SHAREPOINT:
             return microsoft_sharepoint()
+        if self is ConfigurableDataSourceNames.SLACK:
+            return slack()
+        if self is ConfigurableDataSourceNames.NOTION_PAGE:
+            return notion_page()
+        if self is ConfigurableDataSourceNames.JIRA:
+            return jira()

llama_cloud/types/data_sink_component_one.py

@@ -2,6 +2,7 @@
 
 import typing
 
+from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_pinecone_vector_store import CloudPineconeVectorStore
 from .cloud_postgres_vector_store import CloudPostgresVectorStore
@@ -14,4 +15,5 @@ DataSinkComponentOne = typing.Union[
     CloudPostgresVectorStore,
     CloudQdrantVectorStore,
     CloudWeaviateVectorStore,
+    CloudAzureAiSearchVectorStore,
 ]

llama_cloud/types/data_sink_create_component_one.py

@@ -2,6 +2,7 @@
 
 import typing
 
+from .cloud_azure_ai_search_vector_store import CloudAzureAiSearchVectorStore
 from .cloud_chroma_vector_store import CloudChromaVectorStore
 from .cloud_pinecone_vector_store import CloudPineconeVectorStore
 from .cloud_postgres_vector_store import CloudPostgresVectorStore
@@ -14,4 +15,5 @@ DataSinkCreateComponentOne = typing.Union[
     CloudPostgresVectorStore,
     CloudQdrantVectorStore,
     CloudWeaviateVectorStore,
+    CloudAzureAiSearchVectorStore,
 ]

llama_cloud/types/data_source_component_one.py

@@ -3,17 +3,19 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
-from .cloud_gcs_data_source import CloudGcsDataSource
-from .cloud_google_drive_data_source import CloudGoogleDriveDataSource
+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_s_3_data_source import CloudS3DataSource
 from .cloud_sharepoint_data_source import CloudSharepointDataSource
+from .cloud_slack_data_source import CloudSlackDataSource
 
 DataSourceComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
-    CloudGcsDataSource,
-    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
+    CloudSlackDataSource,
+    CloudNotionPageDataSource,
+    CloudJiraDataSource,
 ]

llama_cloud/types/data_source_create_component_one.py

@@ -3,17 +3,19 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
-from .cloud_gcs_data_source import CloudGcsDataSource
-from .cloud_google_drive_data_source import CloudGoogleDriveDataSource
+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_s_3_data_source import CloudS3DataSource
 from .cloud_sharepoint_data_source import CloudSharepointDataSource
+from .cloud_slack_data_source import CloudSlackDataSource
 
 DataSourceCreateComponentOne = typing.Union[
     CloudS3DataSource,
     CloudAzStorageBlobDataSource,
-    CloudGcsDataSource,
-    CloudGoogleDriveDataSource,
     CloudOneDriveDataSource,
     CloudSharepointDataSource,
+    CloudSlackDataSource,
+    CloudNotionPageDataSource,
+    CloudJiraDataSource,
 ]

llama_cloud/types/eval_dataset_job_record.py

@@ -38,6 +38,7 @@ class EvalDatasetJobRecord(pydantic.BaseModel):
         description="The correlation ID for this job. Used for tracking the job across services."
     )
     parent_job_execution_id: typing.Optional[str] = pydantic.Field(description="The ID of the parent job execution.")
+    user_id: typing.Optional[str] = pydantic.Field(description="The ID of the user that created this job")
     created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
     id: typing.Optional[str] = pydantic.Field(description="Unique identifier")
     status: StatusEnum

llama_cloud/types/extraction_result.py

@@ -0,0 +1,42 @@
+# 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 .extraction_result_data_value import ExtractionResultDataValue
+from .file import File
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractionResult(pydantic.BaseModel):
+    """
+    Schema for an extraction result.
+    """
+
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
+    schema_id: str = pydantic.Field(description="The id of the schema")
+    data: typing.Dict[str, ExtractionResultDataValue] = pydantic.Field(description="The data extracted from the file")
+    file: File = pydantic.Field(description="The file that the extract was extracted from")
+
+    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}

llama_cloud/types/extraction_result_data_value.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+ExtractionResultDataValue = typing.Union[typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool]