llama-cloud 0.1.10__tar.gz → 0.1.11__tar.gz

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/PKG-INFO

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

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/llama_cloud/__init__.py

@@ -70,6 +70,7 @@ from .types import (
     DataSourceCustomMetadataValue,
     DataSourceDefinition,
     EditSuggestion,
+    EditSuggestionBlocksItem,
     ElementSegmentationConfig,
     EmbeddingModelConfig,
     EmbeddingModelConfigEmbeddingConfig,
@@ -120,6 +121,7 @@ from .types import (
     ExtractSchemaValidateResponse,
     ExtractSchemaValidateResponseDataSchemaValue,
     ExtractState,
+    ExtractTarget,
     File,
     FilePermissionInfoValue,
     FileResourceInfoValue,
@@ -417,6 +419,7 @@ __all__ = [
     "DataSourceUpdateComponent",
     "DataSourceUpdateCustomMetadataValue",
     "EditSuggestion",
+    "EditSuggestionBlocksItem",
     "ElementSegmentationConfig",
     "EmbeddingModelConfig",
     "EmbeddingModelConfigCreateEmbeddingConfig",
@@ -481,6 +484,7 @@ __all__ = [
     "ExtractSchemaValidateResponse",
     "ExtractSchemaValidateResponseDataSchemaValue",
     "ExtractState",
+    "ExtractTarget",
     "File",
     "FileCreateFromUrlResourceInfoValue",
     "FileCreatePermissionInfoValue",

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/llama_cloud/resources/llama_extract/client.py

@@ -88,7 +88,7 @@ class LlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode
+        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -97,7 +97,8 @@ class LlamaExtractClient:
         client.llama_extract.create_extraction_agent(
             name="string",
             config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
+                extraction_target=ExtractTarget.PER_DOC,
+                extraction_mode=ExtractMode.FAST,
             ),
         )
         """
@@ -234,7 +235,7 @@ class LlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode
+        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -243,7 +244,8 @@ class LlamaExtractClient:
         client.llama_extract.update_extraction_agent(
             extraction_agent_id="string",
             config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
+                extraction_target=ExtractTarget.PER_DOC,
+                extraction_mode=ExtractMode.FAST,
             ),
         )
         """
@@ -336,7 +338,12 @@ class LlamaExtractClient:
         Parameters:
             - request: ExtractJobCreate.
         ---
-        from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
+        from llama_cloud import (
+            ExtractConfig,
+            ExtractJobCreate,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -347,7 +354,8 @@ class LlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
         )
@@ -412,6 +420,7 @@ class LlamaExtractClient:
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
+            ExtractTarget,
             LlamaExtractSettings,
             LlamaParseParameters,
         )
@@ -425,7 +434,8 @@ class LlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
             extract_settings=LlamaExtractSettings(
@@ -458,7 +468,12 @@ class LlamaExtractClient:
         Parameters:
             - request: ExtractJobCreate.
         ---
-        from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
+        from llama_cloud import (
+            ExtractConfig,
+            ExtractJobCreate,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import LlamaCloud
 
         client = LlamaCloud(
@@ -469,7 +484,8 @@ class LlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
         )
@@ -504,6 +520,7 @@ class LlamaExtractClient:
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
+            ExtractTarget,
             LlamaExtractSettings,
             LlamaParseParameters,
         )
@@ -517,7 +534,8 @@ class LlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
             extract_settings=LlamaExtractSettings(
@@ -725,7 +743,7 @@ class AsyncLlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode
+        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -734,7 +752,8 @@ class AsyncLlamaExtractClient:
         await client.llama_extract.create_extraction_agent(
             name="string",
             config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
+                extraction_target=ExtractTarget.PER_DOC,
+                extraction_mode=ExtractMode.FAST,
             ),
         )
         """
@@ -871,7 +890,7 @@ class AsyncLlamaExtractClient:
 
             - config: ExtractConfig. The configuration parameters for the extraction agent.
         ---
-        from llama_cloud import ExtractConfig, ExtractMode
+        from llama_cloud import ExtractConfig, ExtractMode, ExtractTarget
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -880,7 +899,8 @@ class AsyncLlamaExtractClient:
         await client.llama_extract.update_extraction_agent(
             extraction_agent_id="string",
             config=ExtractConfig(
-                extraction_mode=ExtractMode.PER_DOC,
+                extraction_target=ExtractTarget.PER_DOC,
+                extraction_mode=ExtractMode.FAST,
             ),
         )
         """
@@ -973,7 +993,12 @@ class AsyncLlamaExtractClient:
         Parameters:
             - request: ExtractJobCreate.
         ---
-        from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
+        from llama_cloud import (
+            ExtractConfig,
+            ExtractJobCreate,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -984,7 +1009,8 @@ class AsyncLlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
         )
@@ -1049,6 +1075,7 @@ class AsyncLlamaExtractClient:
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
+            ExtractTarget,
             LlamaExtractSettings,
             LlamaParseParameters,
         )
@@ -1062,7 +1089,8 @@ class AsyncLlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
             extract_settings=LlamaExtractSettings(
@@ -1095,7 +1123,12 @@ class AsyncLlamaExtractClient:
         Parameters:
             - request: ExtractJobCreate.
         ---
-        from llama_cloud import ExtractConfig, ExtractJobCreate, ExtractMode
+        from llama_cloud import (
+            ExtractConfig,
+            ExtractJobCreate,
+            ExtractMode,
+            ExtractTarget,
+        )
         from llama_cloud.client import AsyncLlamaCloud
 
         client = AsyncLlamaCloud(
@@ -1106,7 +1139,8 @@ class AsyncLlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
         )
@@ -1141,6 +1175,7 @@ class AsyncLlamaExtractClient:
             ExtractConfig,
             ExtractJobCreate,
             ExtractMode,
+            ExtractTarget,
             LlamaExtractSettings,
             LlamaParseParameters,
         )
@@ -1154,7 +1189,8 @@ class AsyncLlamaExtractClient:
                 extraction_agent_id="string",
                 file_id="string",
                 config_override=ExtractConfig(
-                    extraction_mode=ExtractMode.PER_DOC,
+                    extraction_target=ExtractTarget.PER_DOC,
+                    extraction_mode=ExtractMode.FAST,
                 ),
             ),
             extract_settings=LlamaExtractSettings(

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/llama_cloud/resources/parsing/client.py

@@ -100,6 +100,96 @@ class ParsingClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def screenshot(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        file: typing.Optional[str] = OMIT,
+        do_not_cache: bool,
+        http_proxy: str,
+        input_s_3_path: str,
+        input_s_3_region: str,
+        input_url: str,
+        invalidate_cache: bool,
+        max_pages: typing.Optional[int] = OMIT,
+        output_s_3_path_prefix: str,
+        output_s_3_region: str,
+        target_pages: str,
+        webhook_url: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - file: typing.Optional[str].
+
+            - do_not_cache: bool.
+
+            - http_proxy: str.
+
+            - input_s_3_path: str.
+
+            - input_s_3_region: str.
+
+            - input_url: str.
+
+            - invalidate_cache: bool.
+
+            - max_pages: typing.Optional[int].
+
+            - output_s_3_path_prefix: str.
+
+            - output_s_3_region: str.
+
+            - target_pages: str.
+
+            - webhook_url: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "do_not_cache": do_not_cache,
+            "http_proxy": http_proxy,
+            "input_s3_path": input_s_3_path,
+            "input_s3_region": input_s_3_region,
+            "input_url": input_url,
+            "invalidate_cache": invalidate_cache,
+            "output_s3_path_prefix": output_s_3_path_prefix,
+            "output_s3_region": output_s_3_region,
+            "target_pages": target_pages,
+            "webhook_url": webhook_url,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+        }
+        if file is not OMIT:
+            _request["file"] = file
+        if max_pages is not OMIT:
+            _request["max_pages"] = max_pages
+        _response = self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/parsing/screenshot"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     def upload_file(
         self,
         *,
@@ -957,6 +1047,96 @@ class AsyncParsingClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    async def screenshot(
+        self,
+        *,
+        organization_id: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        file: typing.Optional[str] = OMIT,
+        do_not_cache: bool,
+        http_proxy: str,
+        input_s_3_path: str,
+        input_s_3_region: str,
+        input_url: str,
+        invalidate_cache: bool,
+        max_pages: typing.Optional[int] = OMIT,
+        output_s_3_path_prefix: str,
+        output_s_3_region: str,
+        target_pages: str,
+        webhook_url: str,
+        job_timeout_in_seconds: float,
+        job_timeout_extra_time_per_page_in_seconds: float,
+    ) -> ParsingJob:
+        """
+        Parameters:
+            - organization_id: typing.Optional[str].
+
+            - project_id: typing.Optional[str].
+
+            - file: typing.Optional[str].
+
+            - do_not_cache: bool.
+
+            - http_proxy: str.
+
+            - input_s_3_path: str.
+
+            - input_s_3_region: str.
+
+            - input_url: str.
+
+            - invalidate_cache: bool.
+
+            - max_pages: typing.Optional[int].
+
+            - output_s_3_path_prefix: str.
+
+            - output_s_3_region: str.
+
+            - target_pages: str.
+
+            - webhook_url: str.
+
+            - job_timeout_in_seconds: float.
+
+            - job_timeout_extra_time_per_page_in_seconds: float.
+        """
+        _request: typing.Dict[str, typing.Any] = {
+            "do_not_cache": do_not_cache,
+            "http_proxy": http_proxy,
+            "input_s3_path": input_s_3_path,
+            "input_s3_region": input_s_3_region,
+            "input_url": input_url,
+            "invalidate_cache": invalidate_cache,
+            "output_s3_path_prefix": output_s_3_path_prefix,
+            "output_s3_region": output_s_3_region,
+            "target_pages": target_pages,
+            "webhook_url": webhook_url,
+            "job_timeout_in_seconds": job_timeout_in_seconds,
+            "job_timeout_extra_time_per_page_in_seconds": job_timeout_extra_time_per_page_in_seconds,
+        }
+        if file is not OMIT:
+            _request["file"] = file
+        if max_pages is not OMIT:
+            _request["max_pages"] = max_pages
+        _response = await self._client_wrapper.httpx_client.request(
+            "POST",
+            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v1/parsing/screenshot"),
+            params=remove_none_from_dict({"organization_id": organization_id, "project_id": project_id}),
+            json=jsonable_encoder(_request),
+            headers=self._client_wrapper.get_headers(),
+            timeout=60,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic.parse_obj_as(ParsingJob, _response.json())  # type: ignore
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
     async def upload_file(
         self,
         *,

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/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 (
@@ -127,6 +128,7 @@ from .extract_run_extraction_metadata_value import ExtractRunExtractionMetadataV
 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
@@ -380,6 +382,7 @@ __all__ = [
     "DataSourceCustomMetadataValue",
     "DataSourceDefinition",
     "EditSuggestion",
+    "EditSuggestionBlocksItem",
     "ElementSegmentationConfig",
     "EmbeddingModelConfig",
     "EmbeddingModelConfigEmbeddingConfig",
@@ -430,6 +433,7 @@ __all__ = [
     "ExtractSchemaValidateResponse",
     "ExtractSchemaValidateResponseDataSchemaValue",
     "ExtractState",
+    "ExtractTarget",
     "File",
     "FilePermissionInfoValue",
     "FileResourceInfoValue",

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/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-0.1.11/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-0.1.10 → llama_cloud-0.1.11}/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-0.1.11/llama_cloud/types/extract_mode.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 ExtractMode(str, enum.Enum):
+    FAST = "FAST"
+    ACCURATE = "ACCURATE"
+
+    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-0.1.10/llama_cloud/types/extract_mode.py → llama_cloud-0.1.11/llama_cloud/types/extract_target.py

@@ -6,12 +6,12 @@ import typing
 T_Result = typing.TypeVar("T_Result")
 
 
-class ExtractMode(str, enum.Enum):
+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 ExtractMode.PER_DOC:
+        if self is ExtractTarget.PER_DOC:
             return per_doc()
-        if self is ExtractMode.PER_PAGE:
+        if self is ExtractTarget.PER_PAGE:
             return per_page()

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/llama_cloud/types/llama_extract_settings.py

@@ -26,7 +26,10 @@ 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.")

{llama_cloud-0.1.10 → llama_cloud-0.1.11}/llama_cloud/types/llama_parse_parameters.py

@@ -90,6 +90,7 @@ class LlamaParseParameters(pydantic.BaseModel):
     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-0.1.11/llama_cloud/types/supported_llm_model_names.py

@@ -0,0 +1,65 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class SupportedLlmModelNames(str, enum.Enum):
+    GPT_3_5_TURBO = "GPT_3_5_TURBO"
+    GPT_4 = "GPT_4"
+    GPT_4_TURBO = "GPT_4_TURBO"
+    GPT_4_O = "GPT_4O"
+    GPT_4_O_MINI = "GPT_4O_MINI"
+    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,
+        gpt_3_5_turbo: typing.Callable[[], T_Result],
+        gpt_4: typing.Callable[[], T_Result],
+        gpt_4_turbo: typing.Callable[[], T_Result],
+        gpt_4_o: typing.Callable[[], T_Result],
+        gpt_4_o_mini: 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()
+        if self is SupportedLlmModelNames.GPT_4:
+            return gpt_4()
+        if self is SupportedLlmModelNames.GPT_4_TURBO:
+            return gpt_4_turbo()
+        if self is SupportedLlmModelNames.GPT_4_O:
+            return gpt_4_o()
+        if self is SupportedLlmModelNames.GPT_4_O_MINI:
+            return gpt_4_o_mini()
+        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.10 → llama_cloud-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "llama-cloud"
-version = "0.1.10"
+version = "0.1.11"
 description = ""
 authors = ["Logan Markewich <logan@runllama.ai>"]
 readme = "README.md"