llama-cloud 0.1.6__py3-none-any.whl → 0.1.7a1__py3-none-any.whl

llama_cloud/types/pipeline_data_source_create.py

@@ -20,9 +20,7 @@ class PipelineDataSourceCreate(pydantic.BaseModel):
     """
 
     data_source_id: str = pydantic.Field(description="The ID of the data source.")
-    sync_interval: typing.Optional[float] = pydantic.Field(
-        description="The interval at which the data source should be synced."
-    )
+    sync_interval: typing.Optional[float]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_deployment.py

@@ -17,11 +17,11 @@ except ImportError:
 
 class PipelineDeployment(pydantic.BaseModel):
     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")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     status: ManagedIngestionStatus = pydantic.Field(description="Status of the pipeline deployment.")
-    started_at: typing.Optional[dt.datetime] = pydantic.Field(description="Time the pipeline deployment started.")
-    ended_at: typing.Optional[dt.datetime] = pydantic.Field(description="Time the pipeline deployment finished.")
+    started_at: typing.Optional[dt.datetime]
+    ended_at: 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/pipeline_file.py

@@ -24,33 +24,22 @@ class PipelineFile(pydantic.BaseModel):
     """
 
     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")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     name: 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.)")
+    external_file_id: typing.Optional[str]
+    file_size: typing.Optional[int]
+    file_type: typing.Optional[str]
     project_id: str = pydantic.Field(description="The ID of the project that the file belongs to")
-    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[PipelineFileResourceInfoValue]]] = pydantic.Field(
-        description="Resource information for the file"
-    )
-    permission_info: typing.Optional[
-        typing.Dict[str, typing.Optional[PipelineFilePermissionInfoValue]]
-    ] = 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"
-    )
-    file_id: typing.Optional[str] = pydantic.Field(description="The ID of the file")
+    last_modified_at: typing.Optional[dt.datetime]
+    resource_info: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileResourceInfoValue]]]
+    permission_info: typing.Optional[typing.Dict[str, typing.Optional[PipelineFilePermissionInfoValue]]]
+    data_source_id: typing.Optional[str]
+    file_id: typing.Optional[str]
     pipeline_id: str = pydantic.Field(description="The ID of the pipeline that the file is associated with")
-    custom_metadata: typing.Optional[
-        typing.Dict[str, typing.Optional[PipelineFileCustomMetadataValue]]
-    ] = pydantic.Field(description="Custom metadata for the file")
-    config_hash: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileConfigHashValue]]] = pydantic.Field(
-        description="Hashes for the configuration of the pipeline."
-    )
-    indexed_page_count: typing.Optional[int] = pydantic.Field(
-        description="The number of pages that have been indexed for this file"
-    )
+    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]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/pipeline_file_create.py

@@ -21,9 +21,7 @@ class PipelineFileCreate(pydantic.BaseModel):
     """
 
     file_id: str = pydantic.Field(description="The ID of the file")
-    custom_metadata: typing.Optional[
-        typing.Dict[str, typing.Optional[PipelineFileCreateCustomMetadataValue]]
-    ] = pydantic.Field(description="Custom metadata for the file")
+    custom_metadata: typing.Optional[typing.Dict[str, typing.Optional[PipelineFileCreateCustomMetadataValue]]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/playground_session.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .chat_message import ChatMessage
+from .app_schema_chat_chat_message import AppSchemaChatChatMessage
 from .llm_parameters import LlmParameters
 from .preset_retrieval_params import PresetRetrievalParams
 
@@ -23,8 +23,8 @@ class PlaygroundSession(pydantic.BaseModel):
     """
 
     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")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     pipeline_id: str
     user_id: str
     llm_params_id: str
@@ -33,7 +33,7 @@ class PlaygroundSession(pydantic.BaseModel):
     retrieval_params: typing.Optional[PresetRetrievalParams] = pydantic.Field(
         description="Preset retrieval parameters last used in this session."
     )
-    chat_messages: typing.Optional[typing.List[ChatMessage]] = pydantic.Field(
+    chat_messages: typing.Optional[typing.List[AppSchemaChatChatMessage]] = pydantic.Field(
         description="Chat message history for this session."
     )
 

llama_cloud/types/preset_retrieval_params.py

@@ -21,20 +21,14 @@ class PresetRetrievalParams(pydantic.BaseModel):
     Schema for the search params for an retrieval execution that can be preset for a pipeline.
     """
 
-    dense_similarity_top_k: typing.Optional[int] = pydantic.Field(description="Number of nodes for dense retrieval.")
-    dense_similarity_cutoff: typing.Optional[float] = pydantic.Field(
-        description="Minimum similarity score wrt query for retrieval"
-    )
-    sparse_similarity_top_k: typing.Optional[int] = pydantic.Field(description="Number of nodes for sparse retrieval.")
-    enable_reranking: typing.Optional[bool] = pydantic.Field(description="Enable reranking for retrieval")
-    rerank_top_n: typing.Optional[int] = pydantic.Field(description="Number of reranked nodes for returning.")
-    alpha: typing.Optional[float] = pydantic.Field(
-        description="Alpha value for hybrid retrieval to determine the weights between dense and sparse retrieval. 0 is sparse retrieval and 1 is dense retrieval."
-    )
-    search_filters: typing.Optional[MetadataFilters] = pydantic.Field(description="Search filters for retrieval.")
-    files_top_k: typing.Optional[int] = pydantic.Field(
-        description="Number of files to retrieve (only for retrieval mode files_via_metadata and files_via_content)."
-    )
+    dense_similarity_top_k: typing.Optional[int]
+    dense_similarity_cutoff: typing.Optional[float]
+    sparse_similarity_top_k: typing.Optional[int]
+    enable_reranking: typing.Optional[bool]
+    rerank_top_n: typing.Optional[int]
+    alpha: typing.Optional[float]
+    search_filters: typing.Optional[MetadataFilters]
+    files_top_k: typing.Optional[int]
     retrieval_mode: typing.Optional[RetrievalMode] = pydantic.Field(description="The retrieval mode for the query.")
     retrieve_image_nodes: typing.Optional[bool] = pydantic.Field(description="Whether to retrieve image nodes.")
     class_name: typing.Optional[str]

llama_cloud/types/presigned_url.py

@@ -21,9 +21,7 @@ class PresignedUrl(pydantic.BaseModel):
 
     url: str = pydantic.Field(description="A presigned URL for IO operations against a private file")
     expires_at: dt.datetime = pydantic.Field(description="The time at which the presigned URL expires")
-    form_fields: typing.Optional[typing.Dict[str, typing.Optional[str]]] = pydantic.Field(
-        description="Form fields for a presigned POST request"
-    )
+    form_fields: typing.Optional[typing.Dict[str, 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/progress_event.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 .progress_event_status import ProgressEventStatus
+from .report_event_type import ReportEventType
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ProgressEvent(pydantic.BaseModel):
+    """
+    Event for tracking progress of operations in workflows.
+    """
+
+    timestamp: typing.Optional[dt.datetime]
+    id: typing.Optional[str] = pydantic.Field(description="The ID of the event")
+    group_id: typing.Optional[str] = pydantic.Field(description="The ID of the group this event belongs to")
+    variant: ReportEventType
+    msg: str = pydantic.Field(description="The message to display to the user")
+    progress: typing.Optional[float]
+    status: typing.Optional[ProgressEventStatus] = pydantic.Field(description="Current status of the operation")
+    extra_detail: 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}
+        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/progress_event_status.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ProgressEventStatus(str, enum.Enum):
+    """
+    Current status of the operation
+    """
+
+    PENDING = "pending"
+    IN_PROGRESS = "in_progress"
+    COMPLETED = "completed"
+    ERROR = "error"
+
+    def visit(
+        self,
+        pending: typing.Callable[[], T_Result],
+        in_progress: typing.Callable[[], T_Result],
+        completed: typing.Callable[[], T_Result],
+        error: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is ProgressEventStatus.PENDING:
+            return pending()
+        if self is ProgressEventStatus.IN_PROGRESS:
+            return in_progress()
+        if self is ProgressEventStatus.COMPLETED:
+            return completed()
+        if self is ProgressEventStatus.ERROR:
+            return error()

llama_cloud/types/project.py

@@ -21,8 +21,8 @@ class Project(pydantic.BaseModel):
 
     name: str
     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")
+    created_at: typing.Optional[dt.datetime]
+    updated_at: typing.Optional[dt.datetime]
     ad_hoc_eval_dataset_id: typing.Optional[str]
     organization_id: str = pydantic.Field(description="The Organization ID the project is under.")
     is_default: typing.Optional[bool] = pydantic.Field(

llama_cloud/types/prompt_mixin_prompts.py

@@ -21,7 +21,7 @@ class PromptMixinPrompts(pydantic.BaseModel):
     """
 
     project_id: str = pydantic.Field(description="The ID of the project.")
-    id: typing.Optional[str] = pydantic.Field(description="The ID of the prompt set.")
+    id: typing.Optional[str]
     name: str = pydantic.Field(description="The name of the prompt set.")
     prompts: typing.List[PromptSpec] = pydantic.Field(description="The prompts.")
 

llama_cloud/types/prompt_spec.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .chat_message import ChatMessage
+from .app_schema_chat_chat_message import AppSchemaChatChatMessage
 
 try:
     import pydantic
@@ -19,10 +19,8 @@ class PromptSpec(pydantic.BaseModel):
     prompt_key: str = pydantic.Field(description="The key of the prompt in the PromptMixin.")
     prompt_class: str = pydantic.Field(description="The class of the prompt (PromptTemplate or ChatPromptTemplate).")
     prompt_type: str = pydantic.Field(description="The type of prompt.")
-    template: typing.Optional[str] = pydantic.Field(description="The template of the prompt.")
-    message_templates: typing.Optional[typing.List[ChatMessage]] = pydantic.Field(
-        description="The chat message templates of the prompt."
-    )
+    template: typing.Optional[str]
+    message_templates: typing.Optional[typing.List[AppSchemaChatChatMessage]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

llama_cloud/types/related_node_info.py

@@ -4,7 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
-from .object_type import ObjectType
+from .related_node_info_node_type import RelatedNodeInfoNodeType
 
 try:
     import pydantic
@@ -17,7 +17,7 @@ except ImportError:
 
 class RelatedNodeInfo(pydantic.BaseModel):
     node_id: str
-    node_type: typing.Optional[ObjectType]
+    node_type: typing.Optional[RelatedNodeInfoNodeType]
     metadata: typing.Optional[typing.Dict[str, typing.Any]]
     hash: typing.Optional[str]
     class_name: typing.Optional[str]

llama_cloud/types/related_node_info_node_type.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .object_type import ObjectType
+
+RelatedNodeInfoNodeType = typing.Union[ObjectType, str]

llama_cloud/types/report.py

@@ -0,0 +1,33 @@
+# 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 .report_block import ReportBlock
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class Report(pydantic.BaseModel):
+    id: str = pydantic.Field(description="The id of the report")
+    blocks: typing.Optional[typing.List[ReportBlock]] = pydantic.Field(description="The blocks of the report")
+
+    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_block.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
+from .text_node_with_score import TextNodeWithScore
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ReportBlock(pydantic.BaseModel):
+    idx: int = pydantic.Field(description="The index of the block")
+    template: str = pydantic.Field(description="The content of the block")
+    sources: typing.Optional[typing.List[TextNodeWithScore]] = pydantic.Field(description="The sources for the block")
+
+    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_block_dependency.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ReportBlockDependency(str, enum.Enum):
+    NONE = "none"
+    ALL = "all"
+    PREVIOUS = "previous"
+    NEXT = "next"
+
+    def visit(
+        self,
+        none: typing.Callable[[], T_Result],
+        all: typing.Callable[[], T_Result],
+        previous: typing.Callable[[], T_Result],
+        next: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is ReportBlockDependency.NONE:
+            return none()
+        if self is ReportBlockDependency.ALL:
+            return all()
+        if self is ReportBlockDependency.PREVIOUS:
+            return previous()
+        if self is ReportBlockDependency.NEXT:
+            return next()

llama_cloud/types/report_create_response.py

@@ -0,0 +1,31 @@
+# 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 ReportCreateResponse(pydantic.BaseModel):
+    id: str = pydantic.Field(description="The id of the report")
+
+    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_event_item.py

@@ -0,0 +1,40 @@
+# 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 .report_event_item_event_data import ReportEventItemEventData
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ReportEventItem(pydantic.BaseModel):
+    """
+    From backend schema
+    """
+
+    id: str = pydantic.Field(description="The id of the event")
+    report_id: str = pydantic.Field(description="The id of the report")
+    event_type: str = pydantic.Field(description="The type of the event")
+    event_data: ReportEventItemEventData = pydantic.Field(description="The data for the event")
+    timestamp: dt.datetime = pydantic.Field(description="The timestamp for the event")
+
+    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_event_item_event_data.py

@@ -0,0 +1,45 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+
+import typing
+
+import typing_extensions
+
+from .progress_event import ProgressEvent
+from .report_state_event import ReportStateEvent
+from .report_update_event import ReportUpdateEvent
+
+
+class ReportEventItemEventData_Progress(ProgressEvent):
+    type: typing_extensions.Literal["progress"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ReportEventItemEventData_ReportBlockUpdate(ReportUpdateEvent):
+    type: typing_extensions.Literal["report_block_update"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+class ReportEventItemEventData_ReportStateUpdate(ReportStateEvent):
+    type: typing_extensions.Literal["report_state_update"]
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+
+
+ReportEventItemEventData = typing.Union[
+    ReportEventItemEventData_Progress,
+    ReportEventItemEventData_ReportBlockUpdate,
+    ReportEventItemEventData_ReportStateUpdate,
+]

llama_cloud/types/report_event_type.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 ReportEventType(str, enum.Enum):
+    LOAD_TEMPLATE = "load_template"
+    EXTRACT_PLAN = "extract_plan"
+    SUMMARIZE = "summarize"
+    FILE_PROCESSING = "file_processing"
+    GENERATE_BLOCK = "generate_block"
+    EDITING = "editing"
+
+    def visit(
+        self,
+        load_template: typing.Callable[[], T_Result],
+        extract_plan: typing.Callable[[], T_Result],
+        summarize: typing.Callable[[], T_Result],
+        file_processing: typing.Callable[[], T_Result],
+        generate_block: typing.Callable[[], T_Result],
+        editing: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is ReportEventType.LOAD_TEMPLATE:
+            return load_template()
+        if self is ReportEventType.EXTRACT_PLAN:
+            return extract_plan()
+        if self is ReportEventType.SUMMARIZE:
+            return summarize()
+        if self is ReportEventType.FILE_PROCESSING:
+            return file_processing()
+        if self is ReportEventType.GENERATE_BLOCK:
+            return generate_block()
+        if self is ReportEventType.EDITING:
+            return editing()

llama_cloud/types/report_metadata.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 .report_state import ReportState
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ReportMetadata(pydantic.BaseModel):
+    """
+    Used to update the metadata of a report.
+    """
+
+    id: str = pydantic.Field(description="The id of the report")
+    name: str = pydantic.Field(description="The name of the report")
+    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")
+
+    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_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 .report_plan_block import ReportPlanBlock
+
+try:
+    import pydantic
+    if pydantic.__version__.startswith("1."):
+        raise ImportError
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ReportPlan(pydantic.BaseModel):
+    id: typing.Optional[str] = pydantic.Field(description="The id of the report plan")
+    blocks: typing.Optional[typing.List[ReportPlanBlock]] = pydantic.Field(description="The blocks of the report")
+    generated_at: typing.Optional[dt.datetime] = pydantic.Field(
+        description="The timestamp of when the plan was generated"
+    )
+
+    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}