athena-intelligence 0.1.217__py3-none-any.whl → 0.1.218__py3-none-any.whl

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

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/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/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/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

{athena_intelligence-0.1.217.dist-info → athena_intelligence-0.1.218.dist-info}/METADATA

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

{athena_intelligence-0.1.217.dist-info → athena_intelligence-0.1.218.dist-info}/RECORD

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=uYu0eT7L5Q3E6S2MxapA5hqgYRM9jL34MZ3ljdZhdBg,3737
+athena/__init__.py,sha256=nq2ST9gHq2d-VArwva_gmuTIhjweYS38muxDxPi5Qbw,3839
 athena/agents/__init__.py,sha256=dg7IOwE6-BQSx20JEhdc1VDHlbIHPy08x5fuww_Tvko,180
 athena/agents/client.py,sha256=A70jdG6spqLkPriU8-NCn0vOJvdc5f4SKoVZLOebZjQ,5975
 athena/agents/drive/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
@@ -24,7 +24,7 @@ athena/base_client.py,sha256=9OfmXHlNIFE_4Udzxn0Ue-H-sC2ps0AIm8ZRyuaDwXA,6738
 athena/client.py,sha256=lK3vVU3TF3YjPpiohpxcuRl8x_sSw8HmQ-uuDDeAT6I,22161
 athena/core/__init__.py,sha256=lTcqUPXcx4112yLDd70RAPeqq6tu3eFMe1pKOqkW9JQ,1562
 athena/core/api_error.py,sha256=44vPoTyWN59gonCIZMdzw7M1uspygiLnr3GNFOoVL2Q,614
-athena/core/client_wrapper.py,sha256=fLFlbg9xNO5yELpdXOcDq2IhaLOv3fzc4-XZ9pPswMA,2392
+athena/core/client_wrapper.py,sha256=cCT6LafzaetZcLGnZb9Dk5l93-T0eyaTqUsXUT1CxfI,2392
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=d4NNbX8XvXP32z8KpK2Xovv33nFfruIrpz0QWxlgpZk,2663
 athena/core/force_multipart.py,sha256=awxh5MtcRYe74ehY8U76jzv6fYM_w_D3Rur7KQQzSDk,429
@@ -76,10 +76,10 @@ athena/tools/tasks/client.py,sha256=c_YZ7OjQNJmPKbeeXJznXj3zo5CRFSv02dLupAdHAyo,
 athena/tools/tasks/raw_client.py,sha256=Mexzuf_HcRXWNlESDGQkHHv5tC2tAo-AX3PBSuSRO3U,3812
 athena/tools/types/__init__.py,sha256=ZI6REilQ0Xuocjd8iyNVfLvOsA2Ltlb8--px07wRSPg,217
 athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
-athena/types/__init__.py,sha256=P84dfrZwrzrFIka7jnJyadx4P8rQhhRtjT79CNSo3r4,4234
+athena/types/__init__.py,sha256=Tj5HNQv2HgwkikuAF3x3VwCORu0y2datS54XlbTWjso,4393
 athena/types/aop_async_execute_response_out.py,sha256=R3oy0RLyD-t4MQCnp3Ll13Ci3D5JUMhPyuetbhrONNs,1798
 athena/types/aop_execute_request_in.py,sha256=mEsMKyNN6e90gZra733lJDC6z0bZWdc72af3B-Z5aqE,889
-athena/types/aop_execute_response_out.py,sha256=sOaPWALEo5hWKSnoq_kXrHKtgqsJvgCWcOWpjCTtTCA,1816
+athena/types/aop_execute_response_out.py,sha256=ZRsVi8Ho0l2nBKUyE1uP-4F14FMuuLeap1SZGeVk8NI,1856
 athena/types/asset_content_request_out.py,sha256=RYlcY6j6Ju5EQL7UquDwyTe7uqxyuO8Bg8LCsv1YiUE,652
 athena/types/asset_node.py,sha256=3l7CUK2c_h4QT8ktSq0rFP9veF9G_V9mNe3NZlGl-xQ,804
 athena/types/asset_not_found_error.py,sha256=lIQpdTXCgbXRs21XCMhd_kB7fR6Y6Ep112mK4rejNx0,528
@@ -90,6 +90,8 @@ athena/types/chunk_result.py,sha256=hgrS4hMeuwTRpJ2YrMdrW_QWWWUQ82iYVVTuhFWm1X0,
 athena/types/chunk_result_chunk_id.py,sha256=pzJ6yL6NdUtseoeU4Kw2jlxSTMCVew2TrjhR1MbCuFg,124
 athena/types/content.py,sha256=sSPPkZkHZgA_rO6UyTnR2QBK5mOqUz2pZ--B86r5584,211
 athena/types/conversation_asset_info.py,sha256=SVGd4po14Tlzr8eGjB8UWdQofV8rGEuIhrwGFJ0IHxQ,2844
+athena/types/conversation_message.py,sha256=iH5m0YASJkdmxSUyKPd_d9NRKDxCFVZ7nlkClBoF8IE,1030
+athena/types/conversation_result.py,sha256=IY06B71I46CHoKyzmL-ebYRIr2qrZejt8nDC-bTlKCE,1817
 athena/types/create_new_sheet_tab_response.py,sha256=RF8iOL3mkSc3pY0pqQhvw9IdnncxDC_-XdSUhqPODsM,892
 athena/types/custom_agent_response.py,sha256=hzw1s7mcCI9V58l5OqK4Q59AGF_NctSx5scjJeVWckk,684
 athena/types/data_frame_request_out.py,sha256=wyVIEEI6mqSoH6SyXTQpzLCJOWwsAlUvG9iAVlNuNOU,1076
@@ -125,6 +127,6 @@ athena/types/text_content.py,sha256=tcVCPj3tHh5zQcTElr2tdCIjjfx3ZI63rKIlaG8vo64,
 athena/types/thread_status_response_out.py,sha256=UuSAvs9woL1i8RwvVRKsFUufN4A9jO3jsV47YMckvQU,1219
 athena/types/type.py,sha256=Gvs56nvBMPcQpOZkfPocGNNb7S05PuINianbT309QAQ,146
 athena/version.py,sha256=tnXYUugs9zF_pkVdem-QBorKSuhEOOuetkR57dADDxE,86
-athena_intelligence-0.1.217.dist-info/METADATA,sha256=28y1zmY7oq4v1PX4t6rmvJ1yPI8hTmQJrm--EbWFw_Y,5440
-athena_intelligence-0.1.217.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.217.dist-info/RECORD,,
+athena_intelligence-0.1.218.dist-info/METADATA,sha256=DoE9TfLg5U6j6mrRVPtFd4Iffr-sQT3t1egRJ5acQHY,5440
+athena_intelligence-0.1.218.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.218.dist-info/RECORD,,