llama-cloud 0.1.30__py3-none-any.whl → 0.1.32__py3-none-any.whl

llama_cloud/resources/pipelines/client.py

@@ -410,6 +410,38 @@ class PipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def force_delete_pipeline(self, pipeline_id: str) -> None:
+        """
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import LlamaCloud
+
+        client = LlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        client.pipelines.force_delete_pipeline(
+            pipeline_id="string",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(
+                f"{self._client_wrapper.get_base_url()}/", f"api/v1/pipelines/{pipeline_id}/force-delete"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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 copy_pipeline(self, pipeline_id: str) -> Pipeline:
         """
         Copy a pipeline by ID.
@@ -2049,6 +2081,38 @@ class AsyncPipelinesClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def force_delete_pipeline(self, pipeline_id: str) -> None:
+        """
+        Parameters:
+            - pipeline_id: str.
+        ---
+        from llama_cloud.client import AsyncLlamaCloud
+
+        client = AsyncLlamaCloud(
+            token="YOUR_TOKEN",
+        )
+        await client.pipelines.force_delete_pipeline(
+            pipeline_id="string",
+        )
+        """
+        _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}/force-delete"
+            ),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return
+        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 copy_pipeline(self, pipeline_id: str) -> Pipeline:
         """
         Copy a pipeline by ID.

llama_cloud/types/__init__.py

@@ -15,10 +15,10 @@ from .advanced_mode_transform_config_segmentation_config import (
     AdvancedModeTransformConfigSegmentationConfig_None,
     AdvancedModeTransformConfigSegmentationConfig_Page,
 )
+from .agent_data import AgentData
 from .agent_deployment_list import AgentDeploymentList
 from .agent_deployment_summary import AgentDeploymentSummary
-from .app_schema_chat_chat_message import AppSchemaChatChatMessage
-from .app_schema_responses_message_role import AppSchemaResponsesMessageRole
+from .aggregate_group import AggregateGroup
 from .audio_block import AudioBlock
 from .auto_transform_config import AutoTransformConfig
 from .azure_open_ai_embedding import AzureOpenAiEmbedding
@@ -142,6 +142,13 @@ from .file_parse_public import FileParsePublic
 from .file_permission_info_value import FilePermissionInfoValue
 from .file_resource_info_value import FileResourceInfoValue
 from .filter_condition import FilterCondition
+from .filter_operation import FilterOperation
+from .filter_operation_eq import FilterOperationEq
+from .filter_operation_gt import FilterOperationGt
+from .filter_operation_gte import FilterOperationGte
+from .filter_operation_includes_item import FilterOperationIncludesItem
+from .filter_operation_lt import FilterOperationLt
+from .filter_operation_lte import FilterOperationLte
 from .filter_operator import FilterOperator
 from .free_credits_usage import FreeCreditsUsage
 from .gemini_embedding import GeminiEmbedding
@@ -181,7 +188,6 @@ from .llama_index_core_base_llms_types_chat_message_blocks_item import (
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image,
     LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text,
 )
-from .llama_index_core_base_llms_types_message_role import LlamaIndexCoreBaseLlmsTypesMessageRole
 from .llama_parse_parameters import LlamaParseParameters
 from .llama_parse_parameters_priority import LlamaParseParametersPriority
 from .llama_parse_supported_file_extensions import LlamaParseSupportedFileExtensions
@@ -190,13 +196,12 @@ from .llm_parameters import LlmParameters
 from .load_files_job_config import LoadFilesJobConfig
 from .managed_ingestion_status import ManagedIngestionStatus
 from .managed_ingestion_status_response import ManagedIngestionStatusResponse
-from .message import Message
 from .message_annotation import MessageAnnotation
+from .message_role import MessageRole
 from .metadata_filter import MetadataFilter
 from .metadata_filter_value import MetadataFilterValue
 from .metadata_filters import MetadataFilters
 from .metadata_filters_filters_item import MetadataFiltersFiltersItem
-from .model_configuration import ModelConfiguration
 from .node_relationship import NodeRelationship
 from .none_chunking_config import NoneChunkingConfig
 from .none_segmentation_config import NoneSegmentationConfig
@@ -215,6 +220,8 @@ from .paginated_jobs_history_with_metrics import PaginatedJobsHistoryWithMetrics
 from .paginated_list_cloud_documents_response import PaginatedListCloudDocumentsResponse
 from .paginated_list_pipeline_files_response import PaginatedListPipelineFilesResponse
 from .paginated_report_response import PaginatedReportResponse
+from .paginated_response_agent_data import PaginatedResponseAgentData
+from .paginated_response_aggregate_group import PaginatedResponseAggregateGroup
 from .parse_job_config import ParseJobConfig
 from .parse_job_config_priority import ParseJobConfigPriority
 from .parse_plan_level import ParsePlanLevel
@@ -328,13 +335,13 @@ from .role import Role
 from .schema_relax_mode import SchemaRelaxMode
 from .semantic_chunking_config import SemanticChunkingConfig
 from .sentence_chunking_config import SentenceChunkingConfig
+from .src_app_schema_chat_chat_message import SrcAppSchemaChatChatMessage
 from .status_enum import StatusEnum
 from .struct_mode import StructMode
 from .struct_parse_conf import StructParseConf
 from .supported_llm_model import SupportedLlmModel
 from .supported_llm_model_names import SupportedLlmModelNames
 from .text_block import TextBlock
-from .text_content_block import TextContentBlock
 from .text_node import TextNode
 from .text_node_relationships_value import TextNodeRelationshipsValue
 from .text_node_with_score import TextNodeWithScore
@@ -368,10 +375,10 @@ __all__ = [
     "AdvancedModeTransformConfigSegmentationConfig_Element",
     "AdvancedModeTransformConfigSegmentationConfig_None",
     "AdvancedModeTransformConfigSegmentationConfig_Page",
+    "AgentData",
     "AgentDeploymentList",
     "AgentDeploymentSummary",
-    "AppSchemaChatChatMessage",
-    "AppSchemaResponsesMessageRole",
+    "AggregateGroup",
     "AudioBlock",
     "AutoTransformConfig",
     "AzureOpenAiEmbedding",
@@ -491,6 +498,13 @@ __all__ = [
     "FilePermissionInfoValue",
     "FileResourceInfoValue",
     "FilterCondition",
+    "FilterOperation",
+    "FilterOperationEq",
+    "FilterOperationGt",
+    "FilterOperationGte",
+    "FilterOperationIncludesItem",
+    "FilterOperationLt",
+    "FilterOperationLte",
     "FilterOperator",
     "FreeCreditsUsage",
     "GeminiEmbedding",
@@ -526,7 +540,6 @@ __all__ = [
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Document",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Image",
     "LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem_Text",
-    "LlamaIndexCoreBaseLlmsTypesMessageRole",
     "LlamaParseParameters",
     "LlamaParseParametersPriority",
     "LlamaParseSupportedFileExtensions",
@@ -535,13 +548,12 @@ __all__ = [
     "LoadFilesJobConfig",
     "ManagedIngestionStatus",
     "ManagedIngestionStatusResponse",
-    "Message",
     "MessageAnnotation",
+    "MessageRole",
     "MetadataFilter",
     "MetadataFilterValue",
     "MetadataFilters",
     "MetadataFiltersFiltersItem",
-    "ModelConfiguration",
     "NodeRelationship",
     "NoneChunkingConfig",
     "NoneSegmentationConfig",
@@ -560,6 +572,8 @@ __all__ = [
     "PaginatedListCloudDocumentsResponse",
     "PaginatedListPipelineFilesResponse",
     "PaginatedReportResponse",
+    "PaginatedResponseAgentData",
+    "PaginatedResponseAggregateGroup",
     "ParseJobConfig",
     "ParseJobConfigPriority",
     "ParsePlanLevel",
@@ -663,13 +677,13 @@ __all__ = [
     "SchemaRelaxMode",
     "SemanticChunkingConfig",
     "SentenceChunkingConfig",
+    "SrcAppSchemaChatChatMessage",
     "StatusEnum",
     "StructMode",
     "StructParseConf",
     "SupportedLlmModel",
     "SupportedLlmModelNames",
     "TextBlock",
-    "TextContentBlock",
     "TextNode",
     "TextNodeRelationshipsValue",
     "TextNodeWithScore",

llama_cloud/types/{model_configuration.py → agent_data.py}

@@ -4,7 +4,6 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .supported_llm_model_names import SupportedLlmModelNames
 
 try:
     import pydantic
@@ -15,15 +14,17 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class ModelConfiguration(pydantic.BaseModel):
+class AgentData(pydantic.BaseModel):
     """
-    Configuration for the language model used in response generation.
+    API Result for a single agent data item
     """
 
-    model_name: typing.Optional[SupportedLlmModelNames] = pydantic.Field(
-        description="The name of the model to use for LLM completions."
-    )
-    temperature: typing.Optional[float] = pydantic.Field(description="The temperature to use for LLM completions.")
+    id: typing.Optional[str]
+    agent_slug: str
+    collection: typing.Optional[str]
+    data: typing.Dict[str, typing.Any]
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/agent_deployment_summary.py

@@ -17,7 +17,7 @@ except ImportError:
 class AgentDeploymentSummary(pydantic.BaseModel):
     id: str = pydantic.Field(description="Deployment ID. Prefixed with dpl-")
     project_id: str = pydantic.Field(description="Project ID")
-    app_slug: str = pydantic.Field(description="readable ID of the deployed app")
+    agent_slug: str = pydantic.Field(description="readable ID of the deployed app")
     thumbnail_url: typing.Optional[str]
     base_url: str = pydantic.Field(description="Base URL of the deployed app")
     display_name: str = pydantic.Field(description="Display name of the deployed app")

llama_cloud/types/{message.py → aggregate_group.py}

@@ -4,8 +4,6 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .app_schema_responses_message_role import AppSchemaResponsesMessageRole
-from .text_content_block import TextContentBlock
 
 try:
     import pydantic
@@ -16,13 +14,14 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class Message(pydantic.BaseModel):
-    role: AppSchemaResponsesMessageRole = pydantic.Field(
-        description="Role of the message in the conversation (system, user, assistant)"
-    )
-    blocks: typing.List[TextContentBlock] = pydantic.Field(
-        description="Content of the message. Can be an input or output content block."
-    )
+class AggregateGroup(pydantic.BaseModel):
+    """
+    API Result for a single group in the aggregate response
+    """
+
+    group_key: typing.Dict[str, typing.Any]
+    count: typing.Optional[int]
+    first_item: typing.Optional[typing.Dict[str, typing.Any]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/base_plan.py

@@ -35,6 +35,9 @@ class BasePlan(pydantic.BaseModel):
     is_payment_failed: typing.Optional[bool] = pydantic.Field(
         description="Whether the organization has a failed payment that requires support contact"
     )
+    failure_count: typing.Optional[int] = pydantic.Field(
+        description="The number of payment failures for this organization"
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/extract_mode.py

@@ -11,7 +11,6 @@ class ExtractMode(str, enum.Enum):
     BALANCED = "BALANCED"
     PREMIUM = "PREMIUM"
     MULTIMODAL = "MULTIMODAL"
-    ACCURATE = "ACCURATE"
 
     def visit(
         self,
@@ -19,7 +18,6 @@ class ExtractMode(str, enum.Enum):
         balanced: typing.Callable[[], T_Result],
         premium: typing.Callable[[], T_Result],
         multimodal: typing.Callable[[], T_Result],
-        accurate: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is ExtractMode.FAST:
             return fast()
@@ -29,5 +27,3 @@ class ExtractMode(str, enum.Enum):
             return premium()
         if self is ExtractMode.MULTIMODAL:
             return multimodal()
-        if self is ExtractMode.ACCURATE:
-            return accurate()

llama_cloud/types/filter_operation.py

@@ -0,0 +1,46 @@
+# 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 .filter_operation_eq import FilterOperationEq
+from .filter_operation_gt import FilterOperationGt
+from .filter_operation_gte import FilterOperationGte
+from .filter_operation_includes_item import FilterOperationIncludesItem
+from .filter_operation_lt import FilterOperationLt
+from .filter_operation_lte import FilterOperationLte
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FilterOperation(pydantic.BaseModel):
+    """
+    API request model for a filter comparison operation.
+    """
+
+    eq: typing.Optional[FilterOperationEq]
+    gt: typing.Optional[FilterOperationGt]
+    gte: typing.Optional[FilterOperationGte]
+    lt: typing.Optional[FilterOperationLt]
+    lte: typing.Optional[FilterOperationLte]
+    includes: typing.Optional[typing.List[typing.Optional[FilterOperationIncludesItem]]]
+
+    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/filter_operation_eq.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationEq = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/filter_operation_gt.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationGt = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/filter_operation_gte.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationGte = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/filter_operation_includes_item.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationIncludesItem = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/filter_operation_lt.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationLt = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/filter_operation_lte.py

@@ -0,0 +1,6 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+FilterOperationLte = typing.Union[float, int, str, dt.datetime]

llama_cloud/types/input_message.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .llama_index_core_base_llms_types_message_role import LlamaIndexCoreBaseLlmsTypesMessageRole
+from .message_role import MessageRole
 
 try:
     import pydantic
@@ -21,7 +21,7 @@ class InputMessage(pydantic.BaseModel):
     """
 
     id: typing.Optional[str] = pydantic.Field(description="ID of the message, if any. a UUID.")
-    role: LlamaIndexCoreBaseLlmsTypesMessageRole
+    role: MessageRole
     content: str
     data: typing.Optional[typing.Dict[str, typing.Any]]
     class_name: typing.Optional[str]

llama_cloud/types/legacy_parse_job_config.py

@@ -46,6 +46,9 @@ class LegacyParseJobConfig(pydantic.BaseModel):
     invalidate_cache: bool = pydantic.Field(alias="invalidateCache", description="Whether to invalidate the cache.")
     output_pdf_of_document: typing.Optional[bool] = pydantic.Field(alias="outputPDFOfDocument")
     outlined_table_extraction: typing.Optional[bool] = pydantic.Field(alias="outlinedTableExtraction")
+    merge_tables_across_pages_in_markdown: typing.Optional[bool] = pydantic.Field(
+        alias="mergeTablesAcrossPagesInMarkdown"
+    )
     save_images: typing.Optional[bool] = pydantic.Field(alias="saveImages")
     gpt_4_o: typing.Optional[bool] = pydantic.Field(alias="gpt4o", description="Whether to use GPT4o.")
     open_aiapi_key: str = pydantic.Field(alias="openAIAPIKey", description="The OpenAI API key.")

llama_cloud/types/llama_index_core_base_llms_types_chat_message.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .llama_index_core_base_llms_types_chat_message_blocks_item import LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem
-from .llama_index_core_base_llms_types_message_role import LlamaIndexCoreBaseLlmsTypesMessageRole
+from .message_role import MessageRole
 
 try:
     import pydantic
@@ -21,7 +21,7 @@ class LlamaIndexCoreBaseLlmsTypesChatMessage(pydantic.BaseModel):
     Chat message.
     """
 
-    role: typing.Optional[LlamaIndexCoreBaseLlmsTypesMessageRole]
+    role: typing.Optional[MessageRole]
     additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]]
     blocks: typing.Optional[typing.List[LlamaIndexCoreBaseLlmsTypesChatMessageBlocksItem]]
 

llama_cloud/types/llama_parse_parameters.py

@@ -34,6 +34,7 @@ class LlamaParseParameters(pydantic.BaseModel):
     disable_image_extraction: typing.Optional[bool]
     invalidate_cache: typing.Optional[bool]
     outlined_table_extraction: typing.Optional[bool]
+    merge_tables_across_pages_in_markdown: typing.Optional[bool]
     output_pdf_of_document: typing.Optional[bool]
     do_not_cache: typing.Optional[bool]
     fast_mode: typing.Optional[bool]

llama_cloud/types/{llama_index_core_base_llms_types_message_role.py → message_role.py}

@@ -6,7 +6,7 @@ import typing
 T_Result = typing.TypeVar("T_Result")
 
 
-class LlamaIndexCoreBaseLlmsTypesMessageRole(str, enum.Enum):
+class MessageRole(str, enum.Enum):
     """
     Message role.
     """
@@ -31,19 +31,19 @@ class LlamaIndexCoreBaseLlmsTypesMessageRole(str, enum.Enum):
         chatbot: typing.Callable[[], T_Result],
         model: typing.Callable[[], T_Result],
     ) -> T_Result:
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.SYSTEM:
+        if self is MessageRole.SYSTEM:
             return system()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.DEVELOPER:
+        if self is MessageRole.DEVELOPER:
             return developer()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.USER:
+        if self is MessageRole.USER:
             return user()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.ASSISTANT:
+        if self is MessageRole.ASSISTANT:
             return assistant()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.FUNCTION:
+        if self is MessageRole.FUNCTION:
             return function()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.TOOL:
+        if self is MessageRole.TOOL:
             return tool()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.CHATBOT:
+        if self is MessageRole.CHATBOT:
             return chatbot()
-        if self is LlamaIndexCoreBaseLlmsTypesMessageRole.MODEL:
+        if self is MessageRole.MODEL:
             return model()

llama_cloud/types/{text_content_block.py → paginated_response_agent_data.py}

@@ -3,9 +3,8 @@
 import datetime as dt
 import typing
 
-import typing_extensions
-
 from ..core.datetime_utils import serialize_datetime
+from .agent_data import AgentData
 
 try:
     import pydantic
@@ -16,9 +15,10 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class TextContentBlock(pydantic.BaseModel):
-    type: typing.Optional[typing_extensions.Literal["text"]]
-    content: typing.Optional[str] = pydantic.Field(description="Content of the text block")
+class PaginatedResponseAgentData(pydantic.BaseModel):
+    items: typing.List[AgentData] = pydantic.Field(description="The list of items.")
+    next_page_token: typing.Optional[str]
+    total_size: typing.Optional[int]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/paginated_response_aggregate_group.py

@@ -0,0 +1,34 @@
+# 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 .aggregate_group import AggregateGroup
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class PaginatedResponseAggregateGroup(pydantic.BaseModel):
+    items: typing.List[AggregateGroup] = pydantic.Field(description="The list of items.")
+    next_page_token: typing.Optional[str]
+    total_size: typing.Optional[int]
+
+    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/parse_job_config.py

@@ -36,6 +36,7 @@ class ParseJobConfig(pydantic.BaseModel):
     disable_image_extraction: typing.Optional[bool]
     invalidate_cache: typing.Optional[bool]
     outlined_table_extraction: typing.Optional[bool]
+    merge_tables_across_pages_in_markdown: typing.Optional[bool]
     output_pdf_of_document: typing.Optional[bool]
     do_not_cache: typing.Optional[bool]
     fast_mode: typing.Optional[bool]

llama_cloud/types/playground_session.py

@@ -4,9 +4,9 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .app_schema_chat_chat_message import AppSchemaChatChatMessage
 from .llm_parameters import LlmParameters
 from .preset_retrieval_params import PresetRetrievalParams
+from .src_app_schema_chat_chat_message import SrcAppSchemaChatChatMessage
 
 try:
     import pydantic
@@ -33,7 +33,7 @@ class PlaygroundSession(pydantic.BaseModel):
     retrieval_params: typing.Optional[PresetRetrievalParams] = pydantic.Field(
         description="Preset retrieval parameters last used in this session."
     )
-    chat_messages: typing.Optional[typing.List[AppSchemaChatChatMessage]] = pydantic.Field(
+    chat_messages: typing.Optional[typing.List[SrcAppSchemaChatChatMessage]] = pydantic.Field(
         description="Chat message history for this session."
     )
 

llama_cloud/types/role.py

@@ -24,7 +24,6 @@ class Role(pydantic.BaseModel):
     created_at: typing.Optional[dt.datetime]
     updated_at: typing.Optional[dt.datetime]
     name: str = pydantic.Field(description="A name for the role.")
-    organization_id: typing.Optional[str]
     permissions: typing.List[Permission] = pydantic.Field(description="The actual permissions of the role.")
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/{app_schema_chat_chat_message.py → src_app_schema_chat_chat_message.py}

@@ -4,8 +4,8 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .llama_index_core_base_llms_types_message_role import LlamaIndexCoreBaseLlmsTypesMessageRole
 from .message_annotation import MessageAnnotation
+from .message_role import MessageRole
 
 try:
     import pydantic
@@ -16,13 +16,13 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class AppSchemaChatChatMessage(pydantic.BaseModel):
+class SrcAppSchemaChatChatMessage(pydantic.BaseModel):
     id: str
     index: int = pydantic.Field(description="The index of the message in the chat.")
     annotations: typing.Optional[typing.List[MessageAnnotation]] = pydantic.Field(
         description="Retrieval annotations for the message."
     )
-    role: LlamaIndexCoreBaseLlmsTypesMessageRole = pydantic.Field(description="The role of the message.")
+    role: MessageRole = pydantic.Field(description="The role of the message.")
     content: typing.Optional[str]
     additional_kwargs: typing.Optional[typing.Dict[str, str]] = pydantic.Field(
         description="Additional arguments passed to the model"

llama_cloud/types/user_organization_role.py

@@ -26,7 +26,6 @@ class UserOrganizationRole(pydantic.BaseModel):
     user_id: str = pydantic.Field(description="The user's ID.")
     organization_id: str = pydantic.Field(description="The organization's ID.")
     project_ids: typing.Optional[typing.List[str]]
-    role_id: str = pydantic.Field(description="The role's ID.")
     role: Role = pydantic.Field(description="The role.")
 
     def json(self, **kwargs: typing.Any) -> str: