llama-cloud 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

llama_cloud/types/__init__.py

@@ -73,6 +73,7 @@ from .data_source_create_custom_metadata_value import DataSourceCreateCustomMeta
 from .data_source_custom_metadata_value import DataSourceCustomMetadataValue
 from .data_source_definition import DataSourceDefinition
 from .edit_suggestion import EditSuggestion
+from .edit_suggestion_blocks_item import EditSuggestionBlocksItem
 from .element_segmentation_config import ElementSegmentationConfig
 from .embedding_model_config import EmbeddingModelConfig
 from .embedding_model_config_embedding_config import (
@@ -110,7 +111,8 @@ from .extract_agent_data_schema_value import ExtractAgentDataSchemaValue
 from .extract_config import ExtractConfig
 from .extract_job import ExtractJob
 from .extract_job_create import ExtractJobCreate
-from .extract_job_create_data_schema_override_value import ExtractJobCreateDataSchemaOverrideValue
+from .extract_job_create_data_schema_override import ExtractJobCreateDataSchemaOverride
+from .extract_job_create_data_schema_override_zero_value import ExtractJobCreateDataSchemaOverrideZeroValue
 from .extract_mode import ExtractMode
 from .extract_resultset import ExtractResultset
 from .extract_resultset_data import ExtractResultsetData
@@ -118,10 +120,15 @@ from .extract_resultset_data_item_value import ExtractResultsetDataItemValue
 from .extract_resultset_data_zero_value import ExtractResultsetDataZeroValue
 from .extract_resultset_extraction_metadata_value import ExtractResultsetExtractionMetadataValue
 from .extract_run import ExtractRun
+from .extract_run_data import ExtractRunData
+from .extract_run_data_item_value import ExtractRunDataItemValue
 from .extract_run_data_schema_value import ExtractRunDataSchemaValue
-from .extract_run_data_value import ExtractRunDataValue
+from .extract_run_data_zero_value import ExtractRunDataZeroValue
 from .extract_run_extraction_metadata_value import ExtractRunExtractionMetadataValue
+from .extract_schema_validate_response import ExtractSchemaValidateResponse
+from .extract_schema_validate_response_data_schema_value import ExtractSchemaValidateResponseDataSchemaValue
 from .extract_state import ExtractState
+from .extract_target import ExtractTarget
 from .file import File
 from .file_permission_info_value import FilePermissionInfoValue
 from .file_resource_info_value import FileResourceInfoValue
@@ -240,6 +247,7 @@ from .pipeline_type import PipelineType
 from .plan import Plan
 from .playground_session import PlaygroundSession
 from .pooling import Pooling
+from .preset_composite_retrieval_params import PresetCompositeRetrievalParams
 from .preset_retrieval_params import PresetRetrievalParams
 from .presigned_url import PresignedUrl
 from .progress_event import ProgressEvent
@@ -263,6 +271,7 @@ from .report_event_item_event_data import (
     ReportEventItemEventData_ReportStateUpdate,
 )
 from .report_event_type import ReportEventType
+from .report_file_info import ReportFileInfo
 from .report_metadata import ReportMetadata
 from .report_plan import ReportPlan
 from .report_plan_block import ReportPlanBlock
@@ -373,6 +382,7 @@ __all__ = [
     "DataSourceCustomMetadataValue",
     "DataSourceDefinition",
     "EditSuggestion",
+    "EditSuggestionBlocksItem",
     "ElementSegmentationConfig",
     "EmbeddingModelConfig",
     "EmbeddingModelConfigEmbeddingConfig",
@@ -406,7 +416,8 @@ __all__ = [
     "ExtractConfig",
     "ExtractJob",
     "ExtractJobCreate",
-    "ExtractJobCreateDataSchemaOverrideValue",
+    "ExtractJobCreateDataSchemaOverride",
+    "ExtractJobCreateDataSchemaOverrideZeroValue",
     "ExtractMode",
     "ExtractResultset",
     "ExtractResultsetData",
@@ -414,10 +425,15 @@ __all__ = [
     "ExtractResultsetDataZeroValue",
     "ExtractResultsetExtractionMetadataValue",
     "ExtractRun",
+    "ExtractRunData",
+    "ExtractRunDataItemValue",
     "ExtractRunDataSchemaValue",
-    "ExtractRunDataValue",
+    "ExtractRunDataZeroValue",
     "ExtractRunExtractionMetadataValue",
+    "ExtractSchemaValidateResponse",
+    "ExtractSchemaValidateResponseDataSchemaValue",
     "ExtractState",
+    "ExtractTarget",
     "File",
     "FilePermissionInfoValue",
     "FileResourceInfoValue",
@@ -528,6 +544,7 @@ __all__ = [
     "Plan",
     "PlaygroundSession",
     "Pooling",
+    "PresetCompositeRetrievalParams",
     "PresetRetrievalParams",
     "PresignedUrl",
     "ProgressEvent",
@@ -549,6 +566,7 @@ __all__ = [
     "ReportEventItemEventData_ReportBlockUpdate",
     "ReportEventItemEventData_ReportStateUpdate",
     "ReportEventType",
+    "ReportFileInfo",
     "ReportMetadata",
     "ReportPlan",
     "ReportPlanBlock",

llama_cloud/types/chat_app.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .llm_parameters import LlmParameters
-from .preset_retrieval_params import PresetRetrievalParams
+from .preset_composite_retrieval_params import PresetCompositeRetrievalParams
 
 try:
     import pydantic
@@ -21,14 +21,16 @@ class ChatApp(pydantic.BaseModel):
     Schema for a chat app
     """
 
-    id: str
-    name: str
-    pipeline_id: str
-    project_id: str
-    llm_config: LlmParameters
-    retrieval_config: PresetRetrievalParams
-    created_at: dt.datetime
-    updated_at: dt.datetime
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    name: str = pydantic.Field(description="Name of the chat app")
+    retriever_id: str = pydantic.Field(description="ID of the retriever to use for the chat app")
+    llm_config: LlmParameters = pydantic.Field(description="Configuration for the LLM model to use for the chat app")
+    retrieval_config: PresetCompositeRetrievalParams = pydantic.Field(
+        description="Configuration for the retrieval model to use for the chat app"
+    )
+    project_id: str = pydantic.Field(description="ID of the project the chat app belongs to")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/chat_app_response.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .llm_parameters import LlmParameters
-from .preset_retrieval_params import PresetRetrievalParams
+from .preset_composite_retrieval_params import PresetCompositeRetrievalParams
 
 try:
     import pydantic
@@ -17,15 +17,17 @@ except ImportError:
 
 
 class ChatAppResponse(pydantic.BaseModel):
-    id: str
-    name: str
-    pipeline_id: str
-    project_id: str
-    llm_config: LlmParameters
-    retrieval_config: PresetRetrievalParams
-    created_at: dt.datetime
-    updated_at: dt.datetime
-    pipeline_name: str
+    id: str = pydantic.Field(description="Unique identifier")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
+    name: str = pydantic.Field(description="Name of the chat app")
+    retriever_id: str = pydantic.Field(description="ID of the retriever to use for the chat app")
+    llm_config: LlmParameters = pydantic.Field(description="Configuration for the LLM model to use for the chat app")
+    retrieval_config: PresetCompositeRetrievalParams = pydantic.Field(
+        description="Configuration for the retrieval model to use for the chat app"
+    )
+    project_id: str = pydantic.Field(description="ID of the project the chat app belongs to")
+    retriever_name: str
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/cloud_mongo_db_atlas_vector_search.py

@@ -35,6 +35,7 @@ class CloudMongoDbAtlasVectorSearch(pydantic.BaseModel):
     collection_name: str
     vector_index_name: typing.Optional[str]
     fulltext_index_name: typing.Optional[str]
+    embedding_dimension: typing.Optional[int]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/edit_suggestion.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .report_block import ReportBlock
+from .edit_suggestion_blocks_item import EditSuggestionBlocksItem
 
 try:
     import pydantic
@@ -21,9 +21,8 @@ class EditSuggestion(pydantic.BaseModel):
     """
 
     justification: str
-    start_line: int
-    end_line: int
-    blocks: typing.List[ReportBlock]
+    blocks: typing.List[EditSuggestionBlocksItem]
+    removed_indices: typing.Optional[typing.List[int]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/edit_suggestion_blocks_item.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .report_block import ReportBlock
+from .report_plan_block import ReportPlanBlock
+
+EditSuggestionBlocksItem = typing.Union[ReportBlock, ReportPlanBlock]

llama_cloud/types/extract_config.py

@@ -5,6 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .extract_mode import ExtractMode
+from .extract_target import ExtractTarget
 
 try:
     import pydantic
@@ -20,6 +21,7 @@ class ExtractConfig(pydantic.BaseModel):
     Additional parameters for the extraction agent.
     """
 
+    extraction_target: typing.Optional[ExtractTarget] = pydantic.Field(description="The extraction target specified.")
     extraction_mode: typing.Optional[ExtractMode] = pydantic.Field(description="The extraction mode specified.")
     handle_missing: typing.Optional[bool] = pydantic.Field(
         description="Whether to handle missing fields in the schema."

llama_cloud/types/extract_job_create.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .extract_config import ExtractConfig
-from .extract_job_create_data_schema_override_value import ExtractJobCreateDataSchemaOverrideValue
+from .extract_job_create_data_schema_override import ExtractJobCreateDataSchemaOverride
 
 try:
     import pydantic
@@ -23,7 +23,9 @@ class ExtractJobCreate(pydantic.BaseModel):
 
     extraction_agent_id: str = pydantic.Field(description="The id of the extraction agent")
     file_id: str = pydantic.Field(description="The id of the file")
-    data_schema_override: typing.Optional[typing.Dict[str, typing.Optional[ExtractJobCreateDataSchemaOverrideValue]]]
+    data_schema_override: typing.Optional[ExtractJobCreateDataSchemaOverride] = pydantic.Field(
+        description="The data schema to override the extraction agent's data schema with"
+    )
     config_override: typing.Optional[ExtractConfig]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/extract_job_create_data_schema_override.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_job_create_data_schema_override_zero_value import ExtractJobCreateDataSchemaOverrideZeroValue
+
+ExtractJobCreateDataSchemaOverride = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractJobCreateDataSchemaOverrideZeroValue]], str
+]

llama_cloud/types/{extract_job_create_data_schema_override_value.py → extract_job_create_data_schema_override_zero_value.py}

@@ -2,6 +2,6 @@
 
 import typing
 
-ExtractJobCreateDataSchemaOverrideValue = typing.Union[
+ExtractJobCreateDataSchemaOverrideZeroValue = typing.Union[
     typing.Dict[str, typing.Any], typing.List[typing.Any], str, int, float, bool
 ]

llama_cloud/types/extract_mode.py

@@ -7,11 +7,11 @@ T_Result = typing.TypeVar("T_Result")
 
 
 class ExtractMode(str, enum.Enum):
-    PER_DOC = "PER_DOC"
-    PER_PAGE = "PER_PAGE"
+    FAST = "FAST"
+    ACCURATE = "ACCURATE"
 
-    def visit(self, per_doc: typing.Callable[[], T_Result], per_page: typing.Callable[[], T_Result]) -> T_Result:
-        if self is ExtractMode.PER_DOC:
-            return per_doc()
-        if self is ExtractMode.PER_PAGE:
-            return per_page()
+    def visit(self, fast: typing.Callable[[], T_Result], accurate: typing.Callable[[], T_Result]) -> T_Result:
+        if self is ExtractMode.FAST:
+            return fast()
+        if self is ExtractMode.ACCURATE:
+            return accurate()

llama_cloud/types/extract_run.py

@@ -5,8 +5,8 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .extract_config import ExtractConfig
+from .extract_run_data import ExtractRunData
 from .extract_run_data_schema_value import ExtractRunDataSchemaValue
-from .extract_run_data_value import ExtractRunDataValue
 from .extract_run_extraction_metadata_value import ExtractRunExtractionMetadataValue
 from .extract_state import ExtractState
 from .file import File
@@ -37,7 +37,7 @@ class ExtractRun(pydantic.BaseModel):
     status: ExtractState = pydantic.Field(description="The status of the extraction run")
     error: typing.Optional[str]
     job_id: typing.Optional[str]
-    data: typing.Optional[typing.Dict[str, typing.Optional[ExtractRunDataValue]]]
+    data: typing.Optional[ExtractRunData] = pydantic.Field(description="The data extracted from the file")
     extraction_metadata: typing.Optional[typing.Dict[str, typing.Optional[ExtractRunExtractionMetadataValue]]]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/extract_run_data.py

@@ -0,0 +1,11 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .extract_run_data_item_value import ExtractRunDataItemValue
+from .extract_run_data_zero_value import ExtractRunDataZeroValue
+
+ExtractRunData = typing.Union[
+    typing.Dict[str, typing.Optional[ExtractRunDataZeroValue]],
+    typing.List[typing.Dict[str, typing.Optional[ExtractRunDataItemValue]]],
+]

llama_cloud/types/extract_run_data_item_value.py

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

llama_cloud/types/extract_run_data_zero_value.py

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

llama_cloud/types/extract_schema_validate_response.py

@@ -0,0 +1,32 @@
+# 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 .extract_schema_validate_response_data_schema_value import ExtractSchemaValidateResponseDataSchemaValue
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractSchemaValidateResponse(pydantic.BaseModel):
+    data_schema: typing.Dict[str, typing.Optional[ExtractSchemaValidateResponseDataSchemaValue]]
+
+    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/extract_schema_validate_response_data_schema_value.py

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

llama_cloud/types/extract_target.py

@@ -0,0 +1,17 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ExtractTarget(str, enum.Enum):
+    PER_DOC = "PER_DOC"
+    PER_PAGE = "PER_PAGE"
+
+    def visit(self, per_doc: typing.Callable[[], T_Result], per_page: typing.Callable[[], T_Result]) -> T_Result:
+        if self is ExtractTarget.PER_DOC:
+            return per_doc()
+        if self is ExtractTarget.PER_PAGE:
+            return per_page()

llama_cloud/types/llama_extract_settings.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .llama_parse_parameters import LlamaParseParameters
 
 try:
     import pydantic
@@ -25,11 +26,17 @@ class LlamaExtractSettings(pydantic.BaseModel):
     max_file_size: typing.Optional[int] = pydantic.Field(
         description="The maximum file size (in bytes) allowed for the document."
     )
-    max_num_pages: typing.Optional[int] = pydantic.Field(
+    max_tokens: typing.Optional[int] = pydantic.Field(
+        description="The maximum number of tokens allowed for the document."
+    )
+    max_pages: typing.Optional[int] = pydantic.Field(
         description="The maximum number of pages allowed for the document."
     )
     extraction_prompt: typing.Optional[str] = pydantic.Field(description="The prompt to use for the extraction.")
     error_handling_prompt: typing.Optional[str] = pydantic.Field(description="The prompt to use for error handling.")
+    llama_parse_params: typing.Optional[LlamaParseParameters] = pydantic.Field(
+        description="LlamaParse related settings."
+    )
 
     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

@@ -82,6 +82,15 @@ class LlamaParseParameters(pydantic.BaseModel):
     complemental_formatting_instruction: typing.Optional[str]
     content_guideline_instruction: typing.Optional[str]
     spreadsheet_extract_sub_tables: typing.Optional[bool]
+    job_timeout_in_seconds: typing.Optional[float]
+    job_timeout_extra_time_per_page_in_seconds: typing.Optional[float]
+    strict_mode_image_extraction: typing.Optional[bool]
+    strict_mode_image_ocr: typing.Optional[bool]
+    strict_mode_reconstruction: typing.Optional[bool]
+    strict_mode_buggy_font: typing.Optional[bool]
+    ignore_document_elements_for_layout_detection: typing.Optional[bool]
+    output_tables_as_html: typing.Optional[bool] = pydantic.Field(alias="output_tables_as_HTML")
+    internal_is_screenshot_job: typing.Optional[bool]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/plan.py

@@ -15,6 +15,8 @@ except ImportError:
 
 
 class Plan(pydantic.BaseModel):
+    id: str = pydantic.Field(description="The ID of the plan")
+    name: typing.Optional[str]
     total_users: typing.Optional[int]
     total_indexes: typing.Optional[int]
     total_indexed_pages: typing.Optional[int]
@@ -25,6 +27,8 @@ class Plan(pydantic.BaseModel):
     allowed_external_index: typing.Optional[bool] = pydantic.Field(
         description="If is allowed to use external data sources or sinks in indexes"
     )
+    starting_on: typing.Optional[dt.datetime]
+    ending_before: 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/preset_composite_retrieval_params.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 .composite_retrieval_mode import CompositeRetrievalMode
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class PresetCompositeRetrievalParams(pydantic.BaseModel):
+    mode: typing.Optional[CompositeRetrievalMode] = pydantic.Field(description="The mode of composite retrieval.")
+    rerank_top_n: typing.Optional[int] = pydantic.Field(
+        description="The number of nodes to retrieve after reranking over retrieved nodes from all retrieval tools."
+    )
+
+    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/report_file_info.py

@@ -0,0 +1,37 @@
+# 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 ReportFileInfo(pydantic.BaseModel):
+    """
+    Information about a file stored in S3
+    """
+
+    original_name: str = pydantic.Field(description="Original filename uploaded by user")
+    s_3_path: str = pydantic.Field(alias="s3_path", description="Path to file in S3")
+
+    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
+        allow_population_by_field_name = True
+        json_encoders = {dt.datetime: serialize_datetime}

llama_cloud/types/report_metadata.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .report_file_info import ReportFileInfo
 from .report_state import ReportState
 
 try:
@@ -25,7 +26,7 @@ class ReportMetadata(pydantic.BaseModel):
     report_metadata: typing.Dict[str, typing.Any] = pydantic.Field(description="The metadata for the report")
     state: ReportState = pydantic.Field(description="The state of the report")
     input_files: typing.Optional[typing.List[str]]
-    template_file: typing.Optional[str]
+    template_file: typing.Optional[ReportFileInfo]
     template_text: typing.Optional[str]
     template_instructions: typing.Optional[str]
 

llama_cloud/types/supported_llm_model_names.py

@@ -12,8 +12,14 @@ class SupportedLlmModelNames(str, enum.Enum):
     GPT_4_TURBO = "GPT_4_TURBO"
     GPT_4_O = "GPT_4O"
     GPT_4_O_MINI = "GPT_4O_MINI"
-    AZURE_OPENAI = "AZURE_OPENAI"
+    AZURE_OPENAI_GPT_3_5_TURBO = "AZURE_OPENAI_GPT_3_5_TURBO"
+    AZURE_OPENAI_GPT_4_O = "AZURE_OPENAI_GPT_4O"
+    AZURE_OPENAI_GPT_4_O_MINI = "AZURE_OPENAI_GPT_4O_MINI"
+    AZURE_OPENAI_GPT_4 = "AZURE_OPENAI_GPT_4"
+    AZURE_OPENAI_TEXT_EMBEDDING_3_LARGE = "AZURE_OPENAI_TEXT_EMBEDDING_3_LARGE"
     CLAUDE_3_5_SONNET = "CLAUDE_3_5_SONNET"
+    BEDROCK_CLAUDE_3_5_SONNET = "BEDROCK_CLAUDE_3_5_SONNET"
+    VERTEX_AI_CLAUDE_3_5_SONNET = "VERTEX_AI_CLAUDE_3_5_SONNET"
 
     def visit(
         self,
@@ -22,8 +28,14 @@ class SupportedLlmModelNames(str, enum.Enum):
         gpt_4_turbo: typing.Callable[[], T_Result],
         gpt_4_o: typing.Callable[[], T_Result],
         gpt_4_o_mini: typing.Callable[[], T_Result],
-        azure_openai: typing.Callable[[], T_Result],
+        azure_openai_gpt_3_5_turbo: typing.Callable[[], T_Result],
+        azure_openai_gpt_4_o: typing.Callable[[], T_Result],
+        azure_openai_gpt_4_o_mini: typing.Callable[[], T_Result],
+        azure_openai_gpt_4: typing.Callable[[], T_Result],
+        azure_openai_text_embedding_3_large: typing.Callable[[], T_Result],
         claude_3_5_sonnet: typing.Callable[[], T_Result],
+        bedrock_claude_3_5_sonnet: typing.Callable[[], T_Result],
+        vertex_ai_claude_3_5_sonnet: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is SupportedLlmModelNames.GPT_3_5_TURBO:
             return gpt_3_5_turbo()
@@ -35,7 +47,19 @@ class SupportedLlmModelNames(str, enum.Enum):
             return gpt_4_o()
         if self is SupportedLlmModelNames.GPT_4_O_MINI:
             return gpt_4_o_mini()
-        if self is SupportedLlmModelNames.AZURE_OPENAI:
-            return azure_openai()
+        if self is SupportedLlmModelNames.AZURE_OPENAI_GPT_3_5_TURBO:
+            return azure_openai_gpt_3_5_turbo()
+        if self is SupportedLlmModelNames.AZURE_OPENAI_GPT_4_O:
+            return azure_openai_gpt_4_o()
+        if self is SupportedLlmModelNames.AZURE_OPENAI_GPT_4_O_MINI:
+            return azure_openai_gpt_4_o_mini()
+        if self is SupportedLlmModelNames.AZURE_OPENAI_GPT_4:
+            return azure_openai_gpt_4()
+        if self is SupportedLlmModelNames.AZURE_OPENAI_TEXT_EMBEDDING_3_LARGE:
+            return azure_openai_text_embedding_3_large()
         if self is SupportedLlmModelNames.CLAUDE_3_5_SONNET:
             return claude_3_5_sonnet()
+        if self is SupportedLlmModelNames.BEDROCK_CLAUDE_3_5_SONNET:
+            return bedrock_claude_3_5_sonnet()
+        if self is SupportedLlmModelNames.VERTEX_AI_CLAUDE_3_5_SONNET:
+            return vertex_ai_claude_3_5_sonnet()

{llama_cloud-0.1.9.dist-info → llama_cloud-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: llama-cloud
-Version: 0.1.9
+Version: 0.1.11
 Summary: 
 License: MIT
 Author: Logan Markewich