llama-cloud 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

llama_cloud/types/data_sink.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .configurable_data_sink_names import ConfigurableDataSinkNames
-from .data_sink_component import DataSinkComponent
+from .data_sink_create_component import DataSinkCreateComponent
 
 try:
     import pydantic
@@ -22,11 +22,11 @@ class DataSink(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    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="The name of the data sink.")
     sink_type: ConfigurableDataSinkNames
-    component: DataSinkComponent
+    component: DataSinkCreateComponent = pydantic.Field(description="Component that implements the data sink")
     project_id: str
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/data_sink_create.py

@@ -23,7 +23,7 @@ class DataSinkCreate(pydantic.BaseModel):
 
     name: str = pydantic.Field(description="The name of the data sink.")
     sink_type: ConfigurableDataSinkNames
-    component: DataSinkCreateComponent
+    component: DataSinkCreateComponent = pydantic.Field(description="Component that implements the data sink")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/data_source.py

@@ -5,7 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .configurable_data_source_names import ConfigurableDataSourceNames
-from .data_source_component import DataSourceComponent
+from .data_source_create_component import DataSourceCreateComponent
 from .data_source_custom_metadata_value import DataSourceCustomMetadataValue
 
 try:
@@ -23,12 +23,14 @@ class DataSource(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    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="The name of the data source.")
     source_type: ConfigurableDataSourceNames
-    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceCustomMetadataValue]]]
-    component: DataSourceComponent
+    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceCustomMetadataValue]]] = pydantic.Field(
+        description="Custom metadata that will be present on all data loaded from the data source"
+    )
+    component: DataSourceCreateComponent = pydantic.Field(description="Component that implements the data source")
     project_id: str
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/data_source_create.py

@@ -24,8 +24,10 @@ class DataSourceCreate(pydantic.BaseModel):
 
     name: str = pydantic.Field(description="The name of the data source.")
     source_type: ConfigurableDataSourceNames
-    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceCreateCustomMetadataValue]]]
-    component: DataSourceCreateComponent
+    custom_metadata: typing.Optional[
+        typing.Dict[str, typing.Optional[DataSourceCreateCustomMetadataValue]]
+    ] = pydantic.Field(description="Custom metadata that will be present on all data loaded from the data source")
+    component: DataSourceCreateComponent = pydantic.Field(description="Component that implements the data source")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/embedding_model_config.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
+from .embedding_model_config_embedding_config import EmbeddingModelConfigEmbeddingConfig
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class EmbeddingModelConfig(pydantic.BaseModel):
+    """
+    Schema for an embedding model config.
+    """
+
+    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="The name of the embedding model config.")
+    embedding_config: EmbeddingModelConfigEmbeddingConfig = pydantic.Field(
+        description="The embedding configuration for the embedding model config."
+    )
+    project_id: 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/embedding_model_config_embedding_config.py

@@ -0,0 +1,89 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
+from .bedrock_embedding_config import BedrockEmbeddingConfig
+from .cohere_embedding_config import CohereEmbeddingConfig
+from .gemini_embedding_config import GeminiEmbeddingConfig
+from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
+from .open_ai_embedding_config import OpenAiEmbeddingConfig
+from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
+
+
+class EmbeddingModelConfigEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["AZURE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
+    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
+    type: typing_extensions.Literal["COHERE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
+    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
+    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
+    type: typing_extensions.Literal["VERTEXAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+EmbeddingModelConfigEmbeddingConfig = typing.Union[
+    EmbeddingModelConfigEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigEmbeddingConfig_VertexaiEmbedding,
+]

llama_cloud/types/embedding_model_config_update.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 .embedding_model_config_update_embedding_config import EmbeddingModelConfigUpdateEmbeddingConfig
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class EmbeddingModelConfigUpdate(pydantic.BaseModel):
+    name: typing.Optional[str] = pydantic.Field(description="The name of the embedding model config.")
+    embedding_config: typing.Optional[EmbeddingModelConfigUpdateEmbeddingConfig] = pydantic.Field(
+        description="The embedding configuration for the embedding model config."
+    )
+
+    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/embedding_model_config_update_embedding_config.py

@@ -0,0 +1,89 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .azure_open_ai_embedding_config import AzureOpenAiEmbeddingConfig
+from .bedrock_embedding_config import BedrockEmbeddingConfig
+from .cohere_embedding_config import CohereEmbeddingConfig
+from .gemini_embedding_config import GeminiEmbeddingConfig
+from .hugging_face_inference_api_embedding_config import HuggingFaceInferenceApiEmbeddingConfig
+from .open_ai_embedding_config import OpenAiEmbeddingConfig
+from .vertex_ai_embedding_config import VertexAiEmbeddingConfig
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding(AzureOpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["AZURE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding(BedrockEmbeddingConfig):
+    type: typing_extensions.Literal["BEDROCK_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding(CohereEmbeddingConfig):
+    type: typing_extensions.Literal["COHERE_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding(GeminiEmbeddingConfig):
+    type: typing_extensions.Literal["GEMINI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding(HuggingFaceInferenceApiEmbeddingConfig):
+    type: typing_extensions.Literal["HUGGINGFACE_API_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding(OpenAiEmbeddingConfig):
+    type: typing_extensions.Literal["OPENAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding(VertexAiEmbeddingConfig):
+    type: typing_extensions.Literal["VERTEXAI_EMBEDDING"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+EmbeddingModelConfigUpdateEmbeddingConfig = typing.Union[
+    EmbeddingModelConfigUpdateEmbeddingConfig_AzureEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_BedrockEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_CohereEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_GeminiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_HuggingfaceApiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_OpenaiEmbedding,
+    EmbeddingModelConfigUpdateEmbeddingConfig_VertexaiEmbedding,
+]

llama_cloud/types/eval_dataset.py

@@ -21,8 +21,8 @@ class EvalDataset(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    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="The name of the EvalDataset.")
     project_id: str
 

llama_cloud/types/eval_dataset_job_record.py

@@ -28,21 +28,27 @@ class EvalDatasetJobRecord(pydantic.BaseModel):
     partitions: typing.Dict[str, str] = pydantic.Field(
         description="The partitions for this execution. Used for determining where to save job output."
     )
-    parameters: typing.Optional[EvalDatasetJobParams]
-    session_id: typing.Optional[str]
-    correlation_id: typing.Optional[str]
-    parent_job_execution_id: typing.Optional[str]
-    user_id: typing.Optional[str]
+    parameters: typing.Optional[EvalDatasetJobParams] = pydantic.Field(
+        description="Additional input parameters for the eval execution."
+    )
+    session_id: typing.Optional[str] = pydantic.Field(
+        description="The upstream request ID that created this job. Used for tracking the job across services."
+    )
+    correlation_id: typing.Optional[str] = pydantic.Field(
+        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
     error_code: typing.Optional[str]
     error_message: typing.Optional[str]
-    attempts: typing.Optional[int]
+    attempts: typing.Optional[int] = pydantic.Field(description="The number of times this job has been attempted")
     started_at: typing.Optional[dt.datetime]
     ended_at: typing.Optional[dt.datetime]
     updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
-    data: typing.Optional[Base]
+    data: typing.Optional[Base] = pydantic.Field(description="Additional metadata for the job execution.")
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/eval_execution_params_override.py

@@ -20,8 +20,12 @@ class EvalExecutionParamsOverride(pydantic.BaseModel):
     Schema for the params override for an eval execution.
     """
 
-    llm_model: typing.Optional[SupportedLlmModelNames]
-    qa_prompt_tmpl: typing.Optional[str]
+    llm_model: typing.Optional[SupportedLlmModelNames] = pydantic.Field(
+        description="The LLM model to use within eval execution."
+    )
+    qa_prompt_tmpl: typing.Optional[str] = pydantic.Field(
+        description="The template to use for the question answering prompt."
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/eval_question.py

@@ -16,8 +16,8 @@ except ImportError:
 
 class EvalQuestion(pydantic.BaseModel):
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
     content: str = pydantic.Field(description="The content of the question.")
     eval_dataset_id: str
     eval_dataset_index: int = pydantic.Field(

llama_cloud/types/extraction_result.py

@@ -22,8 +22,8 @@ class ExtractionResult(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
     schema_id: str = pydantic.Field(description="The id of the schema")
     data: typing.Dict[str, typing.Optional[ExtractionResultDataValue]] = pydantic.Field(
         description="The data extracted from the file"

llama_cloud/types/extraction_schema.py

@@ -21,11 +21,13 @@ class ExtractionSchema(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    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="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.Optional[typing.Dict[str, typing.Optional[ExtractionSchemaDataSchemaValue]]]
+    data_schema: typing.Optional[typing.Dict[str, typing.Optional[ExtractionSchemaDataSchemaValue]]] = pydantic.Field(
+        description="The schema of the data"
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/file.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .file_permission_info_value import FilePermissionInfoValue
 from .file_resource_info_value import FileResourceInfoValue
 
 try:
@@ -21,15 +22,22 @@ class File(pydantic.BaseModel):
     """
 
     id: str = pydantic.Field(description="Unique identifier")
-    created_at: typing.Optional[dt.datetime]
-    updated_at: typing.Optional[dt.datetime]
+    created_at: typing.Optional[dt.datetime] = pydantic.Field(description="Creation datetime")
+    updated_at: typing.Optional[dt.datetime] = pydantic.Field(description="Update datetime")
     name: str
-    file_size: typing.Optional[int]
-    file_type: typing.Optional[str]
+    file_size: typing.Optional[int] = pydantic.Field(description="Size of the file in bytes")
+    file_type: typing.Optional[str] = pydantic.Field(description="File type (e.g. pdf, docx, etc.)")
     project_id: str = pydantic.Field(description="The ID of the project that the file belongs to")
-    last_modified_at: typing.Optional[dt.datetime]
-    resource_info: typing.Optional[typing.Dict[str, typing.Optional[FileResourceInfoValue]]]
-    data_source_id: typing.Optional[str]
+    last_modified_at: typing.Optional[dt.datetime] = pydantic.Field(description="The last modified time of the file")
+    resource_info: typing.Optional[typing.Dict[str, typing.Optional[FileResourceInfoValue]]] = pydantic.Field(
+        description="Resource information for the file"
+    )
+    permission_info: typing.Optional[typing.Dict[str, typing.Optional[FilePermissionInfoValue]]] = pydantic.Field(
+        description="Permission information for the file"
+    )
+    data_source_id: typing.Optional[str] = pydantic.Field(
+        description="The ID of the data source that the file belongs to"
+    )
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/file_permission_info_value.py

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

llama_cloud/types/filter_operator.py

@@ -36,7 +36,7 @@ class FilterOperator(str, enum.Enum):
         in_: typing.Callable[[], T_Result],
         nin: typing.Callable[[], T_Result],
         any: typing.Callable[[], T_Result],
-        all: typing.Callable[[], T_Result],
+        all_: typing.Callable[[], T_Result],
         text_match: typing.Callable[[], T_Result],
         contains: typing.Callable[[], T_Result],
         is_empty: typing.Callable[[], T_Result],
@@ -60,7 +60,7 @@ class FilterOperator(str, enum.Enum):
         if self is FilterOperator.ANY:
             return any()
         if self is FilterOperator.ALL:
-            return all()
+            return all_()
         if self is FilterOperator.TEXT_MATCH:
             return text_match()
         if self is FilterOperator.CONTAINS:

llama_cloud/types/gemini_embedding.py

@@ -17,12 +17,16 @@ except ImportError:
 class GeminiEmbedding(pydantic.BaseModel):
     model_name: typing.Optional[str] = pydantic.Field(description="The modelId of the Gemini model to use.")
     embed_batch_size: typing.Optional[int] = pydantic.Field(description="The batch size for embedding calls.")
-    num_workers: typing.Optional[int]
-    title: typing.Optional[str]
-    task_type: typing.Optional[str]
-    api_key: typing.Optional[str]
-    api_base: typing.Optional[str]
-    transport: typing.Optional[str]
+    num_workers: typing.Optional[int] = pydantic.Field(
+        description="The number of workers to use for async embedding calls."
+    )
+    title: typing.Optional[str] = pydantic.Field(
+        description="Title is only applicable for retrieval_document tasks, and is used to represent a document title. For other tasks, title is invalid."
+    )
+    task_type: typing.Optional[str] = pydantic.Field(description="The task for embedding model.")
+    api_key: typing.Optional[str] = pydantic.Field(description="API key to access the model. Defaults to None.")
+    api_base: typing.Optional[str] = pydantic.Field(description="API base to access the model. Defaults to None.")
+    transport: typing.Optional[str] = pydantic.Field(description="Transport to access the model. Defaults to None.")
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/hugging_face_inference_api_embedding.py

@@ -4,8 +4,8 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .hugging_face_inference_api_embedding_token import HuggingFaceInferenceApiEmbeddingToken
 from .pooling import Pooling
+from .token import Token
 
 try:
     import pydantic
@@ -17,19 +17,35 @@ except ImportError:
 
 
 class HuggingFaceInferenceApiEmbedding(pydantic.BaseModel):
-    model_name: typing.Optional[str]
+    model_name: typing.Optional[str] = pydantic.Field(
+        description="Hugging Face model name. If None, the task will be used."
+    )
     embed_batch_size: typing.Optional[int] = pydantic.Field(description="The batch size for embedding calls.")
-    num_workers: typing.Optional[int]
-    pooling: typing.Optional[Pooling]
-    query_instruction: typing.Optional[str]
-    text_instruction: typing.Optional[str]
-    token: typing.Optional[HuggingFaceInferenceApiEmbeddingToken] = pydantic.Field(
+    num_workers: typing.Optional[int] = pydantic.Field(
+        description="The number of workers to use for async embedding calls."
+    )
+    pooling: typing.Optional[Pooling] = pydantic.Field(
+        description="Pooling strategy. If None, the model's default pooling is used."
+    )
+    query_instruction: typing.Optional[str] = pydantic.Field(
+        description="Instruction to prepend during query embedding."
+    )
+    text_instruction: typing.Optional[str] = pydantic.Field(description="Instruction to prepend during text embedding.")
+    token: typing.Optional[Token] = pydantic.Field(
         description="Hugging Face token. Will default to the locally saved token. Pass token=False if you don’t want to send your token to the server."
     )
-    timeout: typing.Optional[float]
-    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]]
-    cookies: typing.Optional[typing.Dict[str, typing.Optional[str]]]
-    task: typing.Optional[str]
+    timeout: typing.Optional[float] = pydantic.Field(
+        description="The maximum number of seconds to wait for a response from the server. Loading a new model in Inference API can take up to several minutes. Defaults to None, meaning it will loop until the server is available."
+    )
+    headers: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(
+        description="Additional headers to send to the server. By default only the authorization and user-agent headers are sent. Values in this dictionary will override the default values."
+    )
+    cookies: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(
+        description="Additional cookies to send to the server."
+    )
+    task: typing.Optional[str] = pydantic.Field(
+        description="Optional task to pick Hugging Face's recommended model, used when model_name is left as default of None."
+    )
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/input_message.py

@@ -23,7 +23,9 @@ class InputMessage(pydantic.BaseModel):
     id: str = pydantic.Field(description="ID of the message, if any. a UUID.")
     role: MessageRole
     content: str
-    data: typing.Optional[typing.Dict[str, typing.Any]]
+    data: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
+        description="Additional data to be stored with the message."
+    )
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/interval_usage_and_plan.py

@@ -0,0 +1,36 @@
+# 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 .plan import Plan
+from .usage import Usage
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class IntervalUsageAndPlan(pydantic.BaseModel):
+    start_window: typing.Optional[dt.datetime]
+    end_window: typing.Optional[dt.datetime]
+    plan: typing.Optional[Plan]
+    usage: typing.Optional[Usage]
+
+    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/job_name_mapping.py

@@ -18,6 +18,7 @@ class JobNameMapping(str, enum.Enum):
     PARSE = "PARSE"
     TRANSFORM = "TRANSFORM"
     INGESTION = "INGESTION"
+    METADATA_UPDATE = "METADATA_UPDATE"
 
     def visit(
         self,
@@ -28,6 +29,7 @@ class JobNameMapping(str, enum.Enum):
         parse: typing.Callable[[], T_Result],
         transform: typing.Callable[[], T_Result],
         ingestion: typing.Callable[[], T_Result],
+        metadata_update: typing.Callable[[], T_Result],
     ) -> T_Result:
         if self is JobNameMapping.MANAGED_INGESTION:
             return managed_ingestion()
@@ -43,3 +45,5 @@ class JobNameMapping(str, enum.Enum):
             return transform()
         if self is JobNameMapping.INGESTION:
             return ingestion()
+        if self is JobNameMapping.METADATA_UPDATE:
+            return metadata_update()