llama-cloud 0.0.8__py3-none-any.whl → 0.0.10__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

@@ -4,12 +4,15 @@ from .azure_open_ai_embedding import AzureOpenAiEmbedding
 from .base import Base
 from .base_prompt_template import BasePromptTemplate
 from .bedrock_embedding import BedrockEmbedding
+from .chat_data import ChatData
 from .chat_message import ChatMessage
 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_confluence_data_source import CloudConfluenceDataSource
 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
@@ -53,6 +56,7 @@ 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_job import ExtractionJob
 from .extraction_result import ExtractionResult
 from .extraction_result_data_value import ExtractionResultDataValue
 from .extraction_schema import ExtractionSchema
@@ -86,6 +90,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 +122,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 +134,9 @@ 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 .user_organization_delete import UserOrganizationDelete
 from .validation_error import ValidationError
 from .validation_error_loc_item import ValidationErrorLocItem
 
@@ -135,12 +145,15 @@ __all__ = [
     "Base",
     "BasePromptTemplate",
     "BedrockEmbedding",
+    "ChatData",
     "ChatMessage",
     "CloudAzStorageBlobDataSource",
     "CloudAzureAiSearchVectorStore",
     "CloudChromaVectorStore",
+    "CloudConfluenceDataSource",
     "CloudDocument",
     "CloudDocumentCreate",
+    "CloudJiraDataSource",
     "CloudNotionPageDataSource",
     "CloudOneDriveDataSource",
     "CloudPineconeVectorStore",
@@ -184,6 +197,7 @@ __all__ = [
     "EvalQuestion",
     "EvalQuestionCreate",
     "EvalQuestionResult",
+    "ExtractionJob",
     "ExtractionResult",
     "ExtractionResultDataValue",
     "ExtractionSchema",
@@ -217,6 +231,8 @@ __all__ = [
     "NodeParser",
     "ObjectType",
     "OpenAiEmbedding",
+    "Organization",
+    "OrganizationCreate",
     "ParserLanguages",
     "ParsingHistoryItem",
     "ParsingJob",
@@ -247,6 +263,7 @@ __all__ = [
     "PromptSpec",
     "PydanticProgramMode",
     "RelatedNodeInfo",
+    "RetrievalMode",
     "RetrieveResults",
     "SentenceSplitter",
     "SimpleFileNodeParser",
@@ -258,6 +275,9 @@ __all__ = [
     "TextNodeWithScore",
     "TokenTextSplitter",
     "TransformationCategoryNames",
+    "UserOrganization",
+    "UserOrganizationCreate",
+    "UserOrganizationDelete",
     "ValidationError",
     "ValidationErrorLocItem",
 ]

llama_cloud/types/chat_data.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 .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
+
+
+class ChatData(pydantic.BaseModel):
+    """
+    Base schema model for BaseComponent classes used in the platform.
+    Comes with special serialization logic for types used commonly in platform codebase.
+    """
+
+    retrieval_parameters: PresetRetrievalParams
+    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_azure_ai_search_vector_store.py

@@ -24,7 +24,7 @@ class CloudAzureAiSearchVectorStore(pydantic.BaseModel):
     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]]
+    filterable_metadata_field_keys: typing.Optional[typing.Dict[str, typing.Any]]
     embedding_dimension: typing.Optional[int]
     class_name: typing.Optional[str]
 

llama_cloud/types/cloud_confluence_data_source.py

@@ -0,0 +1,45 @@
+# 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 CloudConfluenceDataSource(pydantic.BaseModel):
+    """
+    Base component object to capture class names.
+    """
+
+    server_url: str = pydantic.Field(description="The server URL of the Confluence instance.")
+    authentication_mechanism: str = pydantic.Field(
+        description="Type of Authentication for connecting to Confluence APIs."
+    )
+    user_name: typing.Optional[str] = pydantic.Field(description="The username to use for authentication.")
+    api_token: typing.Optional[str] = pydantic.Field(description="The API token to use for authentication.")
+    space_key: typing.Optional[str] = pydantic.Field(description="The space key to read from.")
+    page_ids: typing.Optional[str] = pydantic.Field(description="The page IDs of the Confluence to read from.")
+    cql: typing.Optional[str] = pydantic.Field(description="The CQL query to use for fetching pages.")
+    label: typing.Optional[str] = pydantic.Field(description="The label to use for fetching pages.")
+    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_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/configurable_data_source_names.py

@@ -17,6 +17,8 @@ class ConfigurableDataSourceNames(str, enum.Enum):
     MICROSOFT_SHAREPOINT = "MICROSOFT_SHAREPOINT"
     SLACK = "SLACK"
     NOTION_PAGE = "NOTION_PAGE"
+    CONFLUENCE = "CONFLUENCE"
+    JIRA = "JIRA"
 
     def visit(
         self,
@@ -26,6 +28,8 @@ class ConfigurableDataSourceNames(str, enum.Enum):
         microsoft_sharepoint: typing.Callable[[], T_Result],
         slack: typing.Callable[[], T_Result],
         notion_page: typing.Callable[[], T_Result],
+        confluence: typing.Callable[[], T_Result],
+        jira: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ConfigurableDataSourceNames.S_3:
             return s_3()
@@ -39,3 +43,7 @@ class ConfigurableDataSourceNames(str, enum.Enum):
             return slack()
         if self is ConfigurableDataSourceNames.NOTION_PAGE:
             return notion_page()
+        if self is ConfigurableDataSourceNames.CONFLUENCE:
+            return confluence()
+        if self is ConfigurableDataSourceNames.JIRA:
+            return jira()

llama_cloud/types/data_source_component_one.py

@@ -3,6 +3,8 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_confluence_data_source import CloudConfluenceDataSource
+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
@@ -16,4 +18,6 @@ DataSourceComponentOne = typing.Union[
     CloudSharepointDataSource,
     CloudSlackDataSource,
     CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
+    CloudJiraDataSource,
 ]

llama_cloud/types/data_source_create_component_one.py

@@ -3,6 +3,8 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_confluence_data_source import CloudConfluenceDataSource
+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
@@ -16,4 +18,6 @@ DataSourceCreateComponentOne = typing.Union[
     CloudSharepointDataSource,
     CloudSlackDataSource,
     CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
+    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_job.py

@@ -0,0 +1,35 @@
+# 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 .file import File
+from .status_enum import StatusEnum
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractionJob(pydantic.BaseModel):
+    id: str = pydantic.Field(description="The id of the extraction job")
+    status: StatusEnum = pydantic.Field(description="The status of the extraction job")
+    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_schema.py

@@ -25,10 +25,9 @@ class ExtractionSchema(pydantic.BaseModel):
     updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
     name: str = pydantic.Field(description="The name of the extraction schema")
     project_id: str = pydantic.Field(description="The ID of the project that the extraction schema belongs to")
-    data_schema: typing.Dict[str, ExtractionSchemaDataSchemaValue] = pydantic.Field(
+    data_schema: typing.Optional[typing.Dict[str, ExtractionSchemaDataSchemaValue]] = pydantic.Field(
         description="The schema of the data"
     )
-    openai_api_key: str = pydantic.Field(description="The API key for the OpenAI API")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/llama_parse_parameters.py

@@ -33,6 +33,11 @@ class LlamaParseParameters(pydantic.BaseModel):
     page_separator: typing.Optional[str]
     bounding_box: typing.Optional[str]
     target_pages: typing.Optional[str]
+    use_vendor_multimodal_model: typing.Optional[str]
+    vendor_multimodal_model_name: typing.Optional[str]
+    vendor_multimodal_api_key: typing.Optional[str]
+    page_prefix: typing.Optional[str]
+    page_suffix: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/organization.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
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class Organization(pydantic.BaseModel):
+    """
+    Schema for an organization.
+    """
+
+    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")
+    name: str = pydantic.Field(description="A name for the organization.")
+
+    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/organization_create.py

@@ -0,0 +1,35 @@
+# 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 OrganizationCreate(pydantic.BaseModel):
+    """
+    Schema for creating an organization.
+    """
+
+    name: str = pydantic.Field(description="A name for the organization.")
+
+    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/pipeline.py

@@ -43,9 +43,6 @@ class Pipeline(pydantic.BaseModel):
     eval_parameters: typing.Optional[EvalExecutionParams] = pydantic.Field(
         description="Eval parameters for the pipeline."
     )
-    llama_parse_enabled: typing.Optional[bool] = pydantic.Field(
-        description="Whether to use LlamaParse during pipeline execution."
-    )
     llama_parse_parameters: typing.Optional[LlamaParseParameters] = pydantic.Field(
         description="Settings that can be configured for how to use LlamaParse to parse files within a LlamaCloud pipeline."
     )

llama_cloud/types/pipeline_create.py

@@ -40,9 +40,6 @@ class PipelineCreate(pydantic.BaseModel):
     eval_parameters: typing.Optional[EvalExecutionParams] = pydantic.Field(
         description="Eval parameters for the pipeline."
     )
-    llama_parse_enabled: typing.Optional[bool] = pydantic.Field(
-        description="Whether to use LlamaParse during pipeline execution."
-    )
     llama_parse_parameters: typing.Optional[LlamaParseParameters] = pydantic.Field(
         description="Settings that can be configured for how to use LlamaParse to parse files within a LlamaCloud pipeline."
     )

llama_cloud/types/pipeline_data_source_component_one.py

@@ -3,6 +3,8 @@
 import typing
 
 from .cloud_az_storage_blob_data_source import CloudAzStorageBlobDataSource
+from .cloud_confluence_data_source import CloudConfluenceDataSource
+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
@@ -16,4 +18,6 @@ PipelineDataSourceComponentOne = typing.Union[
     CloudSharepointDataSource,
     CloudSlackDataSource,
     CloudNotionPageDataSource,
+    CloudConfluenceDataSource,
+    CloudJiraDataSource,
 ]

llama_cloud/types/preset_retrieval_params.py

@@ -5,6 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .metadata_filters import MetadataFilters
+from .retrieval_mode import RetrievalMode
 
 try:
     import pydantic
@@ -28,6 +29,10 @@ class PresetRetrievalParams(pydantic.BaseModel):
         description="Alpha value for hybrid retrieval to determine the weights between dense and sparse retrieval. 0 is sparse retrieval and 1 is dense retrieval."
     )
     search_filters: typing.Optional[MetadataFilters] = pydantic.Field(description="Search filters for retrieval.")
+    files_top_k: typing.Optional[int] = pydantic.Field(
+        description="Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content)."
+    )
+    retrieval_mode: typing.Optional[RetrievalMode] = pydantic.Field(description="The retrieval mode for the query.")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/project.py

@@ -24,7 +24,7 @@ class Project(pydantic.BaseModel):
     created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
     updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
     ad_hoc_eval_dataset_id: typing.Optional[str]
-    user_id: str = pydantic.Field(description="The user ID of the project owner.")
+    organization_id: str = pydantic.Field(description="The Organization ID the project is under.")
     is_default: typing.Optional[bool] = pydantic.Field(
         description="Whether this project is the default project for the user."
     )