athena-intelligence 0.1.126__py3-none-any.whl → 0.1.184__py3-none-any.whl

athena/types/__init__.py

@@ -1,11 +1,16 @@
 # This file was auto-generated by Fern from our API Definition.
 
+# isort: skip_file
+
+from .asset_content_request_out import AssetContentRequestOut
 from .asset_node import AssetNode
 from .asset_not_found_error import AssetNotFoundError
+from .asset_screenshot_response_out import AssetScreenshotResponseOut
 from .chunk import Chunk
 from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, ChunkContentItem_Text
 from .chunk_result import ChunkResult
 from .chunk_result_chunk_id import ChunkResultChunkId
+from .content import Content
 from .custom_agent_response import CustomAgentResponse
 from .data_frame_request_out import DataFrameRequestOut
 from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
@@ -21,26 +26,39 @@ from .general_agent_config import GeneralAgentConfig
 from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
 from .general_agent_request import GeneralAgentRequest
 from .general_agent_response import GeneralAgentResponse
+from .general_agent_response_message import GeneralAgentResponseMessage
+from .general_agent_response_message_kwargs import GeneralAgentResponseMessageKwargs
+from .id import Id
 from .image_url_content import ImageUrlContent
+from .input_message import InputMessage
+from .input_message_content_item import (
+    InputMessageContentItem,
+    InputMessageContentItem_ImageUrl,
+    InputMessageContentItem_Text,
+)
+from .paginated_assets_out import PaginatedAssetsOut
 from .parent_folder_error import ParentFolderError
 from .prompt_message import PromptMessage
+from .public_asset_out import PublicAssetOut
 from .research_agent_response import ResearchAgentResponse
 from .save_asset_request_out import SaveAssetRequestOut
 from .sql_agent_response import SqlAgentResponse
 from .structured_data_extractor_response import StructuredDataExtractorResponse
 from .text_content import TextContent
-from .tool import Tool
 from .type import Type
 
 __all__ = [
+    "AssetContentRequestOut",
     "AssetNode",
     "AssetNotFoundError",
+    "AssetScreenshotResponseOut",
     "Chunk",
     "ChunkContentItem",
     "ChunkContentItem_ImageUrl",
     "ChunkContentItem_Text",
     "ChunkResult",
     "ChunkResultChunkId",
+    "Content",
     "CustomAgentResponse",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
@@ -56,14 +74,22 @@ __all__ = [
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",
     "GeneralAgentResponse",
+    "GeneralAgentResponseMessage",
+    "GeneralAgentResponseMessageKwargs",
+    "Id",
     "ImageUrlContent",
+    "InputMessage",
+    "InputMessageContentItem",
+    "InputMessageContentItem_ImageUrl",
+    "InputMessageContentItem_Text",
+    "PaginatedAssetsOut",
     "ParentFolderError",
     "PromptMessage",
+    "PublicAssetOut",
     "ResearchAgentResponse",
     "SaveAssetRequestOut",
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "TextContent",
-    "Tool",
     "Type",
 ]

athena/types/asset_content_request_out.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class AssetContentRequestOut(UniversalBaseModel):
+    """
+    Response model with asset content.
+    """
+
+    content: str = pydantic.Field()
+    """
+    The content of the asset
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

athena/types/asset_node.py

@@ -1,11 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from ..core.pydantic_utilities import update_forward_refs
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class AssetNode(UniversalBaseModel):

athena/types/asset_not_found_error.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class AssetNotFoundError(UniversalBaseModel):

athena/types/asset_screenshot_response_out.py

@@ -0,0 +1,43 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+
+
+class AssetScreenshotResponseOut(UniversalBaseModel):
+    """
+    Response model with screenshot data.
+    """
+
+    base_64_image: typing_extensions.Annotated[str, FieldMetadata(alias="base64_image")] = pydantic.Field()
+    """
+    Base64-encoded image data
+    """
+
+    message: str = pydantic.Field()
+    """
+    Status message
+    """
+
+    page_number: int = pydantic.Field()
+    """
+    The page number that was captured
+    """
+
+    total_pages: int = pydantic.Field()
+    """
+    Total number of pages in the document
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

athena/types/chunk.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .chunk_content_item import ChunkContentItem
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chunk_content_item import ChunkContentItem
 
 
 class Chunk(UniversalBaseModel):

athena/types/chunk_content_item.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class ChunkContentItem_Text(UniversalBaseModel):

athena/types/chunk_result.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .chunk_result_chunk_id import ChunkResultChunkId
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .chunk_result_chunk_id import ChunkResultChunkId
 
 
 class ChunkResult(UniversalBaseModel):

athena/types/content.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from .input_message_content_item import InputMessageContentItem
+
+Content = typing.Union[str, typing.List[InputMessageContentItem]]

athena/types/custom_agent_response.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class CustomAgentResponse(UniversalBaseModel):

athena/types/data_frame_request_out.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
 from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
 from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class DataFrameRequestOut(UniversalBaseModel):

athena/types/data_frame_unknown_format_error.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class DataFrameUnknownFormatError(UniversalBaseModel):

athena/types/document_chunk.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class DocumentChunk(UniversalBaseModel):

athena/types/drive_agent_response.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class DriveAgentResponse(UniversalBaseModel):

athena/types/file_chunk_request_out.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .document_chunk import DocumentChunk
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .document_chunk import DocumentChunk
 
 
 class FileChunkRequestOut(UniversalBaseModel):

athena/types/file_too_large_error.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
 import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class FileTooLargeError(UniversalBaseModel):

athena/types/folder_response.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+from __future__ import annotations
+
 import typing
-from .asset_node import AssetNode
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel, update_forward_refs
 
 
 class FolderResponse(UniversalBaseModel):
@@ -13,7 +14,7 @@ class FolderResponse(UniversalBaseModel):
     """
 
     structure_tree_ascii: str
-    tree_data: typing.Dict[str, AssetNode]
+    tree_data: typing.Dict[str, "AssetNode"]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
@@ -23,3 +24,8 @@ class FolderResponse(UniversalBaseModel):
             frozen = True
             smart_union = True
             extra = pydantic.Extra.allow
+
+
+from .asset_node import AssetNode  # noqa: E402, F401, I001
+
+update_forward_refs(FolderResponse)

athena/types/general_agent_config.py

@@ -1,10 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
 
 
 class GeneralAgentConfig(UniversalBaseModel):
@@ -14,6 +14,7 @@ class GeneralAgentConfig(UniversalBaseModel):
 
     enabled_tools: typing.Optional[typing.List[GeneralAgentConfigEnabledToolsItem]] = None
     knowledge_base_asset_ids: typing.Optional[typing.List[str]] = None
+    model: typing.Optional[str] = None
     system_prompt: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:

athena/types/general_agent_config_enabled_tools_item.py

@@ -1,6 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from .tool import Tool
 
-GeneralAgentConfigEnabledToolsItem = typing.Union[Tool, str]
+GeneralAgentConfigEnabledToolsItem = typing.Union[typing.Literal["search"], typing.Literal["browse"], str]

athena/types/general_agent_request.py

@@ -1,10 +1,11 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .general_agent_config import GeneralAgentConfig
 import typing
+
 import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .general_agent_config import GeneralAgentConfig
+from .input_message import InputMessage
 
 
 class GeneralAgentRequest(UniversalBaseModel):
@@ -12,12 +13,22 @@ class GeneralAgentRequest(UniversalBaseModel):
     A chat request for the Athena SDK.
     """
 
+    channel: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The channel through which the request is being made.
+    """
+
     config: GeneralAgentConfig
-    messages: typing.List[typing.Optional[typing.Any]] = pydantic.Field()
+    messages: typing.List[InputMessage] = pydantic.Field()
     """
     The messages to send to the agent. Each message should be a string (for text inputs) or a list of multimodal content parts.
     """
 
+    thread_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional thread ID for conversation persistence. If not provided, a new thread will be created.
+    """
+
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:

athena/types/general_agent_response.py

@@ -1,9 +1,10 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .general_agent_response_message import GeneralAgentResponseMessage
 
 
 class GeneralAgentResponse(UniversalBaseModel):
@@ -11,7 +12,7 @@ class GeneralAgentResponse(UniversalBaseModel):
     The response from the agent.
     """
 
-    messages: typing.List[typing.Dict[str, typing.Optional[typing.Any]]]
+    messages: typing.List[GeneralAgentResponseMessage]
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

athena/types/general_agent_response_message.py

@@ -0,0 +1,100 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+from .content import Content
+from .general_agent_response_message_kwargs import GeneralAgentResponseMessageKwargs
+from .id import Id
+
+
+class GeneralAgentResponseMessage(UniversalBaseModel):
+    """
+    A response message from the agent.
+    """
+
+    additional_kwargs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Additional keyword arguments for the message.
+    """
+
+    content: typing.Optional[Content] = pydantic.Field(default=None)
+    """
+    The content of the message, can be string or list of content parts.
+    """
+
+    id: typing.Optional[Id] = pydantic.Field(default=None)
+    """
+    Unique identifier for the message or LangChain class identifier.
+    """
+
+    kwargs_: typing_extensions.Annotated[
+        typing.Optional[GeneralAgentResponseMessageKwargs], FieldMetadata(alias="kwargs")
+    ] = pydantic.Field(default=None)
+    """
+    Structured kwargs field containing known message data.
+    """
+
+    langchain_id: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    LangChain class identifier
+    """
+
+    lc: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    LangChain version marker
+    """
+
+    message_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    The actual message identifier (when id contains LangChain class path).
+    """
+
+    name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Name associated with the message.
+    """
+
+    response_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Metadata about the response generation.
+    """
+
+    role: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Role of the message sender (e.g., 'ai', 'human', 'system', 'tool'). Automatically populated from kwargs.type for convenience.
+    """
+
+    tool_call_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    ID of the tool call this message responds to (for tool messages).
+    """
+
+    tool_calls: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(
+        default=None
+    )
+    """
+    Tool calls made by the message (for AI messages).
+    """
+
+    type: str = pydantic.Field()
+    """
+    The type of the message (e.g., 'human', 'ai', 'system', 'tool').
+    """
+
+    usage_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Usage metadata including token counts.
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

athena/types/general_agent_response_message_kwargs.py

@@ -0,0 +1,74 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .content import Content
+
+
+class GeneralAgentResponseMessageKwargs(UniversalBaseModel):
+    """
+    Structured kwargs data for response messages with basic required fields.
+    """
+
+    additional_kwargs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Additional keyword arguments for the message.
+    """
+
+    content: typing.Optional[Content] = pydantic.Field(default=None)
+    """
+    The content of the response message, matches main response content.
+    """
+
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Unique identifier for the message.
+    """
+
+    name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Name associated with the message.
+    """
+
+    response_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Metadata about the response generation.
+    """
+
+    role: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Role of the message sender.
+    """
+
+    tool_call_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    ID of the tool call this message responds to (for tool messages).
+    """
+
+    tool_calls: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(
+        default=None
+    )
+    """
+    Tool calls made by the message (for AI messages).
+    """
+
+    type: str = pydantic.Field()
+    """
+    The type of the message (e.g., 'human', 'ai', 'system', 'tool').
+    """
+
+    usage_metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Usage metadata including token counts.
+    """
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

athena/types/{tool.py → id.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-Tool = typing.Union[typing.Literal["search", "browse"], typing.Any]
+Id = typing.Union[str, typing.List[str]]

athena/types/image_url_content.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class ImageUrlContent(UniversalBaseModel):