llama-cloud 0.1.41__py3-none-any.whl → 1.0.0b4__py3-none-any.whl

llama_cloud/types/status_enum.py

@@ -1,37 +1,7 @@
-# This file was auto-generated by Fern from our API Definition.
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import enum
-import typing
+from typing_extensions import Literal, TypeAlias
 
-T_Result = typing.TypeVar("T_Result")
+__all__ = ["StatusEnum"]
 
-
-class StatusEnum(str, enum.Enum):
-    """
-    Enum for representing the status of a job
-    """
-
-    PENDING = "PENDING"
-    SUCCESS = "SUCCESS"
-    ERROR = "ERROR"
-    PARTIAL_SUCCESS = "PARTIAL_SUCCESS"
-    CANCELLED = "CANCELLED"
-
-    def visit(
-        self,
-        pending: typing.Callable[[], T_Result],
-        success: typing.Callable[[], T_Result],
-        error: typing.Callable[[], T_Result],
-        partial_success: typing.Callable[[], T_Result],
-        cancelled: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is StatusEnum.PENDING:
-            return pending()
-        if self is StatusEnum.SUCCESS:
-            return success()
-        if self is StatusEnum.ERROR:
-            return error()
-        if self is StatusEnum.PARTIAL_SUCCESS:
-            return partial_success()
-        if self is StatusEnum.CANCELLED:
-            return cancelled()
+StatusEnum: TypeAlias = Literal["PENDING", "SUCCESS", "ERROR", "PARTIAL_SUCCESS", "CANCELLED"]

llama_cloud/types/vertex_ai_embedding_config.py

@@ -1,34 +1,17 @@
-# This file was auto-generated by Fern from our API Definition.
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-import datetime as dt
-import typing
+from typing import Optional
+from typing_extensions import Literal
 
-from ..core.datetime_utils import serialize_datetime
+from .._models import BaseModel
 from .vertex_text_embedding import VertexTextEmbedding
 
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
+__all__ = ["VertexAIEmbeddingConfig"]
 
 
-class VertexAiEmbeddingConfig(pydantic.BaseModel):
-    component: typing.Optional[VertexTextEmbedding] = pydantic.Field(
-        description="Configuration for the VertexAI embedding model."
-    )
+class VertexAIEmbeddingConfig(BaseModel):
+    component: Optional[VertexTextEmbedding] = None
+    """Configuration for the VertexAI embedding model."""
 
-    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}
+    type: Optional[Literal["VERTEXAI_EMBEDDING"]] = None
+    """Type of the embedding model."""

llama_cloud/types/vertex_ai_embedding_config_param.py

@@ -0,0 +1,17 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, TypedDict
+
+from .vertex_text_embedding_param import VertexTextEmbeddingParam
+
+__all__ = ["VertexAIEmbeddingConfigParam"]
+
+
+class VertexAIEmbeddingConfigParam(TypedDict, total=False):
+    component: VertexTextEmbeddingParam
+    """Configuration for the VertexAI embedding model."""
+
+    type: Literal["VERTEXAI_EMBEDDING"]
+    """Type of the embedding model."""

llama_cloud/types/vertex_text_embedding.py

@@ -1,45 +1,47 @@
-# 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 .vertex_embedding_mode import VertexEmbeddingMode
-
-try:
-    import pydantic
-    if pydantic.__version__.startswith("1."):
-        raise ImportError
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class VertexTextEmbedding(pydantic.BaseModel):
-    model_name: typing.Optional[str] = pydantic.Field(description="The modelId of the VertexAI model to use.")
-    embed_batch_size: typing.Optional[int] = pydantic.Field(description="The batch size for embedding calls.")
-    num_workers: typing.Optional[int]
-    location: str = pydantic.Field(description="The default location to use when making API calls.")
-    project: str = pydantic.Field(description="The default GCP project to use when making Vertex API calls.")
-    embed_mode: typing.Optional[VertexEmbeddingMode] = pydantic.Field(description="The embedding mode to use.")
-    additional_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = pydantic.Field(
-        description="Additional kwargs for the Vertex."
-    )
-    client_email: typing.Optional[str]
-    token_uri: typing.Optional[str]
-    private_key_id: typing.Optional[str]
-    private_key: typing.Optional[str]
-    class_name: typing.Optional[str]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from typing_extensions import Literal
+
+from pydantic import Field as FieldInfo
+
+from .._models import BaseModel
+
+__all__ = ["VertexTextEmbedding"]
+
+
+class VertexTextEmbedding(BaseModel):
+    client_email: Optional[str] = None
+    """The client email for the VertexAI credentials."""
+
+    location: str
+    """The default location to use when making API calls."""
+
+    private_key: Optional[str] = None
+    """The private key for the VertexAI credentials."""
+
+    private_key_id: Optional[str] = None
+    """The private key ID for the VertexAI credentials."""
+
+    project: str
+    """The default GCP project to use when making Vertex API calls."""
+
+    token_uri: Optional[str] = None
+    """The token URI for the VertexAI credentials."""
+
+    additional_kwargs: Optional[Dict[str, object]] = None
+    """Additional kwargs for the Vertex."""
+
+    class_name: Optional[str] = None
+
+    embed_batch_size: Optional[int] = None
+    """The batch size for embedding calls."""
+
+    embed_mode: Optional[Literal["default", "classification", "clustering", "similarity", "retrieval"]] = None
+    """The embedding mode to use."""
+
+    api_model_name: Optional[str] = FieldInfo(alias="model_name", default=None)
+    """The modelId of the VertexAI model to use."""
+
+    num_workers: Optional[int] = None
+    """The number of workers to use for async embedding calls."""

llama_cloud/types/vertex_text_embedding_param.py

@@ -0,0 +1,45 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Optional
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["VertexTextEmbeddingParam"]
+
+
+class VertexTextEmbeddingParam(TypedDict, total=False):
+    client_email: Required[Optional[str]]
+    """The client email for the VertexAI credentials."""
+
+    location: Required[str]
+    """The default location to use when making API calls."""
+
+    private_key: Required[Optional[str]]
+    """The private key for the VertexAI credentials."""
+
+    private_key_id: Required[Optional[str]]
+    """The private key ID for the VertexAI credentials."""
+
+    project: Required[str]
+    """The default GCP project to use when making Vertex API calls."""
+
+    token_uri: Required[Optional[str]]
+    """The token URI for the VertexAI credentials."""
+
+    additional_kwargs: Dict[str, object]
+    """Additional kwargs for the Vertex."""
+
+    class_name: str
+
+    embed_batch_size: int
+    """The batch size for embedding calls."""
+
+    embed_mode: Literal["default", "classification", "clustering", "similarity", "retrieval"]
+    """The embedding mode to use."""
+
+    model_name: str
+    """The modelId of the VertexAI model to use."""
+
+    num_workers: Optional[int]
+    """The number of workers to use for async embedding calls."""