llama-cloud 0.1.12__py3-none-any.whl → 0.1.14__py3-none-any.whl

llama_cloud/types/base_plan_metronome_plan_type.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 BasePlanMetronomePlanType(str, enum.Enum):
+    PLAN = "plan"
+    CONTRACT = "contract"
+
+    def visit(self, plan: typing.Callable[[], T_Result], contract: typing.Callable[[], T_Result]) -> T_Result:
+        if self is BasePlanMetronomePlanType.PLAN:
+            return plan()
+        if self is BasePlanMetronomePlanType.CONTRACT:
+            return contract()

llama_cloud/types/base_plan_name.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BasePlanName(str, enum.Enum):
+    FREE = "free"
+    LLAMA_PARSE = "llama_parse"
+    ENTERPRISE = "enterprise"
+    UNKNOWN = "unknown"
+    FREE_CONTRACT = "free_contract"
+    PRO = "pro"
+    ENTERPRISE_CONTRACT = "enterprise_contract"
+    ENTERPRISE_POC = "enterprise_poc"
+
+    def visit(
+        self,
+        free: typing.Callable[[], T_Result],
+        llama_parse: typing.Callable[[], T_Result],
+        enterprise: typing.Callable[[], T_Result],
+        unknown: typing.Callable[[], T_Result],
+        free_contract: typing.Callable[[], T_Result],
+        pro: typing.Callable[[], T_Result],
+        enterprise_contract: typing.Callable[[], T_Result],
+        enterprise_poc: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BasePlanName.FREE:
+            return free()
+        if self is BasePlanName.LLAMA_PARSE:
+            return llama_parse()
+        if self is BasePlanName.ENTERPRISE:
+            return enterprise()
+        if self is BasePlanName.UNKNOWN:
+            return unknown()
+        if self is BasePlanName.FREE_CONTRACT:
+            return free_contract()
+        if self is BasePlanName.PRO:
+            return pro()
+        if self is BasePlanName.ENTERPRISE_CONTRACT:
+            return enterprise_contract()
+        if self is BasePlanName.ENTERPRISE_POC:
+            return enterprise_poc()

llama_cloud/types/base_plan_plan_frequency.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class BasePlanPlanFrequency(str, enum.Enum):
+    MONTHLY = "MONTHLY"
+    QUARTERLY = "QUARTERLY"
+    ANNUAL = "ANNUAL"
+
+    def visit(
+        self,
+        monthly: typing.Callable[[], T_Result],
+        quarterly: typing.Callable[[], T_Result],
+        annual: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is BasePlanPlanFrequency.MONTHLY:
+            return monthly()
+        if self is BasePlanPlanFrequency.QUARTERLY:
+            return quarterly()
+        if self is BasePlanPlanFrequency.ANNUAL:
+            return annual()

llama_cloud/types/billing_period.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
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class BillingPeriod(pydantic.BaseModel):
+    start_date: dt.datetime
+    end_date: dt.datetime
+
+    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/{base.py → credit_type.py}

@@ -14,7 +14,10 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class Base(pydantic.BaseModel):
+class CreditType(pydantic.BaseModel):
+    id: str
+    name: 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)

llama_cloud/types/data_source.py

@@ -29,6 +29,7 @@ class DataSource(pydantic.BaseModel):
     source_type: ConfigurableDataSourceNames
     custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[DataSourceCustomMetadataValue]]]
     component: DataSourceComponent = pydantic.Field(description="Component that implements the data source")
+    version_metadata: typing.Optional[typing.Dict[str, typing.Any]]
     project_id: str
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/eval_dataset_job_record.py

@@ -6,7 +6,6 @@ import typing
 import typing_extensions
 
 from ..core.datetime_utils import serialize_datetime
-from .base import Base
 from .eval_dataset_job_params import EvalDatasetJobParams
 from .status_enum import StatusEnum
 
@@ -43,7 +42,7 @@ class EvalDatasetJobRecord(pydantic.BaseModel):
     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[typing.Any]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/extract_agent_create.py

@@ -0,0 +1,39 @@
+# 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_agent_create_data_schema import ExtractAgentCreateDataSchema
+from .extract_config import ExtractConfig
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractAgentCreate(pydantic.BaseModel):
+    """
+    Settings for creating an extraction agent.
+    """
+
+    name: str = pydantic.Field(description="The name of the extraction schema")
+    data_schema: ExtractAgentCreateDataSchema = pydantic.Field(description="The schema of the data.")
+    config: ExtractConfig = pydantic.Field(description="The configuration parameters for the extraction agent.")
+
+    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_agent_update.py

@@ -0,0 +1,38 @@
+# 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_agent_update_data_schema import ExtractAgentUpdateDataSchema
+from .extract_config import ExtractConfig
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractAgentUpdate(pydantic.BaseModel):
+    """
+    Settings for updating an extraction schema.
+    """
+
+    data_schema: ExtractAgentUpdateDataSchema = pydantic.Field(description="The schema of the data")
+    config: ExtractConfig = pydantic.Field(description="The configuration parameters for the extraction agent.")
+
+    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_request.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_request_data_schema import ExtractSchemaValidateRequestDataSchema
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ExtractSchemaValidateRequest(pydantic.BaseModel):
+    data_schema: ExtractSchemaValidateRequestDataSchema
+
+    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/free_credits_usage.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
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class FreeCreditsUsage(pydantic.BaseModel):
+    starting_balance: int
+    remaining_balance: int
+    grant_name: str
+    expires_at: dt.datetime
+
+    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_record.py

@@ -4,7 +4,6 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .base import Base
 from .job_names import JobNames
 from .status_enum import StatusEnum
 
@@ -26,7 +25,7 @@ class JobRecord(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[Base]
+    parameters: typing.Optional[typing.Any]
     session_id: typing.Optional[str]
     correlation_id: typing.Optional[str]
     parent_job_execution_id: typing.Optional[str]
@@ -41,7 +40,7 @@ class JobRecord(pydantic.BaseModel):
     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[typing.Any]
 
     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

@@ -5,6 +5,7 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from .parser_languages import ParserLanguages
+from .parsing_mode import ParsingMode
 
 try:
     import pydantic
@@ -24,6 +25,8 @@ class LlamaParseParameters(pydantic.BaseModel):
     parsing_instruction: typing.Optional[str]
     disable_ocr: typing.Optional[bool]
     annotate_links: typing.Optional[bool]
+    adaptive_long_table: typing.Optional[bool]
+    compact_markdown_table: typing.Optional[bool]
     disable_reconstruction: typing.Optional[bool]
     disable_image_extraction: typing.Optional[bool]
     invalidate_cache: typing.Optional[bool]
@@ -31,6 +34,7 @@ class LlamaParseParameters(pydantic.BaseModel):
     do_not_cache: typing.Optional[bool]
     fast_mode: typing.Optional[bool]
     skip_diagonal_text: typing.Optional[bool]
+    preserve_layout_alignment_across_pages: typing.Optional[bool]
     gpt_4_o_mode: typing.Optional[bool] = pydantic.Field(alias="gpt4o_mode")
     gpt_4_o_api_key: typing.Optional[str] = pydantic.Field(alias="gpt4o_api_key")
     do_not_unroll_columns: typing.Optional[bool]
@@ -48,6 +52,7 @@ class LlamaParseParameters(pydantic.BaseModel):
     target_pages: typing.Optional[str]
     use_vendor_multimodal_model: typing.Optional[bool]
     vendor_multimodal_model_name: typing.Optional[str]
+    model: typing.Optional[str]
     vendor_multimodal_api_key: typing.Optional[str]
     page_prefix: typing.Optional[str]
     page_suffix: typing.Optional[str]
@@ -91,6 +96,10 @@ class LlamaParseParameters(pydantic.BaseModel):
     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]
+    parse_mode: typing.Optional[ParsingMode]
+    system_prompt: typing.Optional[str]
+    system_prompt_append: typing.Optional[str]
+    user_prompt: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/llm_parameters.py

@@ -22,6 +22,7 @@ class LlmParameters(pydantic.BaseModel):
     system_prompt: typing.Optional[str]
     temperature: typing.Optional[float]
     use_chain_of_thought_reasoning: typing.Optional[bool]
+    use_citation: typing.Optional[bool]
     class_name: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:

llama_cloud/types/page_screenshot_metadata.py

@@ -18,6 +18,7 @@ class PageScreenshotMetadata(pydantic.BaseModel):
     page_index: int = pydantic.Field(description="The index of the page for which the screenshot is taken (0-indexed)")
     file_id: str = pydantic.Field(description="The ID of the file that the page screenshot was taken from")
     image_size: int = pydantic.Field(description="The size of the image in bytes")
+    metadata: 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/paginated_list_cloud_documents_response.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 .cloud_document import CloudDocument
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class PaginatedListCloudDocumentsResponse(pydantic.BaseModel):
+    documents: typing.List[CloudDocument] = pydantic.Field(description="The documents to list")
+    limit: int = pydantic.Field(description="The limit of the documents")
+    offset: int = pydantic.Field(description="The offset of the documents")
+    total_count: int = pydantic.Field(description="The total number of documents")
+
+    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/parsing_mode.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ParsingMode(str, enum.Enum):
+    """
+    Enum for representing the mode of parsing to be used
+    """
+
+    PARSE_PAGE_WITHOUT_LLM = "parse_page_without_llm"
+    PARSE_PAGE_WITH_LLM = "parse_page_with_llm"
+    PARSE_PAGE_WITH_LVM = "parse_page_with_lvm"
+    PARSE_PAGE_WITH_AGENT = "parse_page_with_agent"
+    PARSE_DOCUMENT_WITH_LLM = "parse_document_with_llm"
+
+    def visit(
+        self,
+        parse_page_without_llm: typing.Callable[[], T_Result],
+        parse_page_with_llm: typing.Callable[[], T_Result],
+        parse_page_with_lvm: typing.Callable[[], T_Result],
+        parse_page_with_agent: typing.Callable[[], T_Result],
+        parse_document_with_llm: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is ParsingMode.PARSE_PAGE_WITHOUT_LLM:
+            return parse_page_without_llm()
+        if self is ParsingMode.PARSE_PAGE_WITH_LLM:
+            return parse_page_with_llm()
+        if self is ParsingMode.PARSE_PAGE_WITH_LVM:
+            return parse_page_with_lvm()
+        if self is ParsingMode.PARSE_PAGE_WITH_AGENT:
+            return parse_page_with_agent()
+        if self is ParsingMode.PARSE_DOCUMENT_WITH_LLM:
+            return parse_document_with_llm()

llama_cloud/types/pipeline_data_source.py

@@ -29,6 +29,7 @@ class PipelineDataSource(pydantic.BaseModel):
     source_type: ConfigurableDataSourceNames
     custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[PipelineDataSourceCustomMetadataValue]]]
     component: PipelineDataSourceComponent = pydantic.Field(description="Component that implements the data source")
+    version_metadata: typing.Optional[typing.Dict[str, typing.Any]]
     project_id: str
     data_source_id: str = pydantic.Field(description="The ID of the data source.")
     pipeline_id: str = pydantic.Field(description="The ID of the pipeline.")

llama_cloud/types/pipeline_file.py

@@ -40,6 +40,7 @@ class PipelineFile(pydantic.BaseModel):
     custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileCustomMetadataValue]]]
     config_hash: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileConfigHashValue]]]
     indexed_page_count: typing.Optional[int]
+    status: typing.Optional[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/plan_limits.py

@@ -0,0 +1,52 @@
+# 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 PlanLimits(pydantic.BaseModel):
+    allow_pay_as_you_go: bool = pydantic.Field(description="Whether usage is allowed after credit grants are exhausted")
+    subscription_cost_usd: int
+    max_monthly_invoice_total_usd: typing.Optional[int]
+    max_concurrent_parse_jobs_premium: typing.Optional[int]
+    max_concurrent_parse_jobs_other: typing.Optional[int]
+    max_extraction_agents: typing.Optional[int]
+    max_extraction_runs: typing.Optional[int]
+    max_extraction_jobs: typing.Optional[int]
+    max_pages_per_index: typing.Optional[int]
+    max_files_per_index: typing.Optional[int]
+    max_indexes: typing.Optional[int]
+    max_concurrent_index_jobs: typing.Optional[int]
+    max_data_sources: typing.Optional[int]
+    max_embedding_models: typing.Optional[int]
+    max_data_sinks: typing.Optional[int]
+    max_published_agents: typing.Optional[int]
+    max_report_agent_sessions: typing.Optional[int]
+    max_users: typing.Optional[int]
+    max_organizations: typing.Optional[int]
+    max_projects: typing.Optional[int]
+    mfa_enabled: bool
+    sso_enabled: bool
+
+    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/recurring_credit_grant.py

@@ -0,0 +1,44 @@
+# 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 .credit_type import CreditType
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class RecurringCreditGrant(pydantic.BaseModel):
+    name: str
+    credit_amount: int
+    credit_type: CreditType
+    product_id: str = pydantic.Field(
+        description="The ID of the product in Metronome used to represent the credit grant"
+    )
+    priority: float
+    rollover_fraction: float = pydantic.Field(
+        description="The fraction of the credit that will roll over to the next period, between 0 and 1"
+    )
+    periods_duration: typing.Optional[float] = pydantic.Field(
+        description="How many billing periods the credit grant will last for"
+    )
+
+    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/usage.py

@@ -4,6 +4,8 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .free_credits_usage import FreeCreditsUsage
+from .usage_active_alerts_item import UsageActiveAlertsItem
 
 try:
     import pydantic
@@ -19,13 +21,12 @@ class Usage(pydantic.BaseModel):
     Response model; use UsageSubmission for tracking
     """
 
+    active_free_credits_usage: typing.Optional[typing.List[FreeCreditsUsage]]
     total_users: typing.Optional[int]
     total_indexes: typing.Optional[int]
     total_indexed_pages: typing.Optional[int]
-    extract_pages: typing.Optional[int]
-    parse_pages: typing.Optional[int]
-    index_pages: typing.Optional[int]
-    credits: typing.Optional[int]
+    active_alerts: typing.Optional[typing.List[UsageActiveAlertsItem]]
+    current_invoice_total_usd_cents: 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/usage_active_alerts_item.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class UsageActiveAlertsItem(str, enum.Enum):
+    PLAN_SPEND_LIMIT_EXCEEDED = "plan_spend_limit_exceeded"
+    CONFIGURED_SPEND_LIMIT_EXCEEDED = "configured_spend_limit_exceeded"
+    FREE_CREDITS_EXHAUSTED = "free_credits_exhausted"
+
+    def visit(
+        self,
+        plan_spend_limit_exceeded: typing.Callable[[], T_Result],
+        configured_spend_limit_exceeded: typing.Callable[[], T_Result],
+        free_credits_exhausted: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is UsageActiveAlertsItem.PLAN_SPEND_LIMIT_EXCEEDED:
+            return plan_spend_limit_exceeded()
+        if self is UsageActiveAlertsItem.CONFIGURED_SPEND_LIMIT_EXCEEDED:
+            return configured_spend_limit_exceeded()
+        if self is UsageActiveAlertsItem.FREE_CREDITS_EXHAUSTED:
+            return free_credits_exhausted()

llama_cloud/types/{interval_usage_and_plan.py → usage_and_plan.py}

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .plan import Plan
+from .base_plan import BasePlan
 from .usage import Usage
 
 try:
@@ -16,11 +16,9 @@ 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]
+class UsageAndPlan(pydantic.BaseModel):
+    plan: BasePlan
+    usage: Usage
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}