athena-intelligence 0.1.216__tar.gz → 0.1.218__tar.gz

{athena_intelligence-0.1.216 → athena_intelligence-0.1.218}/PKG-INFO

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

{athena_intelligence-0.1.216 → athena_intelligence-0.1.218}/pyproject.toml

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

{athena_intelligence-0.1.216 → athena_intelligence-0.1.218}/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.216 → athena_intelligence-0.1.218}/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.216",
+            "User-Agent": "athena-intelligence/0.1.218",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.216",
+            "X-Fern-SDK-Version": "0.1.218",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

{athena_intelligence-0.1.216 → athena_intelligence-0.1.218}/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.216 → athena_intelligence-0.1.218}/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.216 → athena_intelligence-0.1.218}/src/athena/types/conversation_asset_info.py

@@ -60,6 +60,11 @@ class ConversationAssetInfo(UniversalBaseModel):
     List of linked project assets
     """
 
+    messages: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(default=None)
+    """
+    Complete list of messages in the conversation from checkpoints
+    """
+
     model: typing.Optional[str] = pydantic.Field(default=None)
     """
     Model used in conversation

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

@@ -0,0 +1,41 @@
+# 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 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: str = pydantic.Field()
+    """
+    Content of the message
+    """
+
+    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.218/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