athena-intelligence 0.1.217__tar.gz → 0.1.219__tar.gz

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.217
+Version: 0.1.219
 Summary: Athena Intelligence Python Library
 Requires-Python: >=3.9,<4.0
 Classifier: Intended Audience :: Developers

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/pyproject.toml

@@ -3,7 +3,7 @@ name = "athena-intelligence"
 
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.217"
+version = "0.1.219"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/src/athena/__init__.py

@@ -18,6 +18,8 @@ from .types import (
     ChunkResultChunkId,
     Content,
     ConversationAssetInfo,
+    ConversationMessage,
+    ConversationResult,
     CreateNewSheetTabResponse,
     CustomAgentResponse,
     DataFrameRequestOut,
@@ -92,6 +94,8 @@ __all__ = [
     "Content",
     "ContentTooLargeError",
     "ConversationAssetInfo",
+    "ConversationMessage",
+    "ConversationResult",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/src/athena/core/client_wrapper.py

@@ -22,10 +22,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "athena-intelligence/0.1.217",
+            "User-Agent": "athena-intelligence/0.1.219",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.217",
+            "X-Fern-SDK-Version": "0.1.219",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/src/athena/types/__init__.py

@@ -15,6 +15,8 @@ from .chunk_result import ChunkResult
 from .chunk_result_chunk_id import ChunkResultChunkId
 from .content import Content
 from .conversation_asset_info import ConversationAssetInfo
+from .conversation_message import ConversationMessage
+from .conversation_result import ConversationResult
 from .create_new_sheet_tab_response import CreateNewSheetTabResponse
 from .custom_agent_response import CustomAgentResponse
 from .data_frame_request_out import DataFrameRequestOut
@@ -70,6 +72,8 @@ __all__ = [
     "ChunkResultChunkId",
     "Content",
     "ConversationAssetInfo",
+    "ConversationMessage",
+    "ConversationResult",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/src/athena/types/aop_execute_response_out.py

@@ -4,6 +4,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .conversation_result import ConversationResult
 
 
 class AopExecuteResponseOut(UniversalBaseModel):
@@ -31,6 +32,11 @@ class AopExecuteResponseOut(UniversalBaseModel):
     Base prompt of the AOP before user inputs were added
     """
 
+    conversation: ConversationResult = pydantic.Field()
+    """
+    The conversation result from the AOP execution
+    """
+
     enabled_tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
     """
     List of tools that were enabled for this execution
@@ -41,11 +47,6 @@ class AopExecuteResponseOut(UniversalBaseModel):
     Final prompt used for execution including user inputs
     """
 
-    response: typing.Dict[str, typing.Optional[typing.Any]] = pydantic.Field()
-    """
-    The execution response from the AOP
-    """
-
     status: str = pydantic.Field()
     """
     Status of the execution (e.g., 'submitted')

{athena_intelligence-0.1.217 → athena_intelligence-0.1.219}/src/athena/types/conversation_asset_info.py

@@ -4,6 +4,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .conversation_message import ConversationMessage
 
 
 class ConversationAssetInfo(UniversalBaseModel):
@@ -41,9 +42,9 @@ class ConversationAssetInfo(UniversalBaseModel):
     Last active channel for the conversation
     """
 
-    last_message: typing.Optional[str] = pydantic.Field(default=None)
+    last_message: typing.Optional[ConversationMessage] = pydantic.Field(default=None)
     """
-    Last message content
+    Last message in the conversation
     """
 
     linked_aops: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(
@@ -60,7 +61,7 @@ class ConversationAssetInfo(UniversalBaseModel):
     List of linked project assets
     """
 
-    messages: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(default=None)
+    messages: typing.Optional[typing.List[ConversationMessage]] = pydantic.Field(default=None)
     """
     Complete list of messages in the conversation from checkpoints
     """

athena_intelligence-0.1.219/src/athena/types/conversation_message.py

@@ -0,0 +1,42 @@
+# 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 ConversationMessage(UniversalBaseModel):
+    """
+    Model representing a single message in the conversation.
+    """
+
+    additional_kwargs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Additional message metadata
+    """
+
+    content: Content = pydantic.Field()
+    """
+    Content of the message - either a string or multimodal content parts
+    """
+
+    id: str = pydantic.Field()
+    """
+    Unique identifier for the message
+    """
+
+    role: str = pydantic.Field()
+    """
+    Role of the message sender (e.g., 'user', 'assistant')
+    """
+
+    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_intelligence-0.1.219/src/athena/types/conversation_result.py

@@ -0,0 +1,67 @@
+# 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 .conversation_message import ConversationMessage
+
+
+class ConversationResult(UniversalBaseModel):
+    """
+    Model representing the conversation result from AOP execution.
+    """
+
+    conversation_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    ID of the conversation asset
+    """
+
+    created_at: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    ISO timestamp when conversation was created
+    """
+
+    last_assistant_message: typing.Optional[ConversationMessage] = pydantic.Field(default=None)
+    """
+    The last message from the assistant
+    """
+
+    messages: typing.List[ConversationMessage] = pydantic.Field()
+    """
+    Complete list of messages in the conversation
+    """
+
+    messages_source: str = pydantic.Field()
+    """
+    Source of the messages (e.g., 'checkpoints')
+    """
+
+    metadata: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = pydantic.Field(default=None)
+    """
+    Additional conversation metadata
+    """
+
+    num_messages: int = pydantic.Field()
+    """
+    Total number of messages in the conversation
+    """
+
+    title: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Title of the conversation
+    """
+
+    updated_at: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    ISO timestamp when conversation was last updated
+    """
+
+    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