athena-intelligence 0.1.225__py3-none-any.whl → 0.1.226__py3-none-any.whl

athena/__init__.py

@@ -17,8 +17,6 @@ from .types import (
     ChunkResultChunkId,
     Content,
     ConversationAssetInfo,
-    ConversationMessage,
-    ConversationResult,
     CreateNewSheetTabResponse,
     CustomAgentResponse,
     DataFrameRequestOut,
@@ -92,8 +90,6 @@ __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.225",
+            "User-Agent": "athena-intelligence/0.1.226",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.225",
+            "X-Fern-SDK-Version": "0.1.226",
             **(self.get_custom_headers() or {}),
         }
         headers["X-API-KEY"] = self.api_key

athena/types/__init__.py

@@ -14,8 +14,6 @@ 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,8 +68,6 @@ __all__ = [
     "ChunkResultChunkId",
     "Content",
     "ConversationAssetInfo",
-    "ConversationMessage",
-    "ConversationResult",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",

athena/types/aop_async_execute_response_out.py

@@ -31,11 +31,6 @@ class AopAsyncExecuteResponseOut(UniversalBaseModel):
     Base prompt of the AOP before user inputs were added
     """
 
-    enabled_tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
-    """
-    List of tools that were enabled for this execution
-    """
-
     final_prompt: str = pydantic.Field()
     """
     Final prompt used for execution including user inputs

athena/types/aop_execute_response_out.py

@@ -4,7 +4,6 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
-from .conversation_result import ConversationResult
 
 
 class AopExecuteResponseOut(UniversalBaseModel):
@@ -32,14 +31,14 @@ class AopExecuteResponseOut(UniversalBaseModel):
     Base prompt of the AOP before user inputs were added
     """
 
-    conversation: ConversationResult = pydantic.Field()
+    conversation: typing.Optional[typing.Optional[typing.Any]] = pydantic.Field(default=None)
     """
     The conversation result from the AOP execution
     """
 
-    enabled_tools: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    error: typing.Optional[str] = pydantic.Field(default=None)
     """
-    List of tools that were enabled for this execution
+    Error message if execution failed
     """
 
     final_prompt: str = pydantic.Field()

athena/types/conversation_asset_info.py

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

{athena_intelligence-0.1.225.dist-info → athena_intelligence-0.1.226.dist-info}/METADATA

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

{athena_intelligence-0.1.225.dist-info → athena_intelligence-0.1.226.dist-info}/RECORD

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=m4d9K_0cNfspmKY-t6QI6Iy35Wg_kfrw7SWmiIsQ-hs,3789
+athena/__init__.py,sha256=XUBXN4ZyAqw-HPDFdhWdx3EjdsReRFNdUr5zjWX4N4o,3687
 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=kaJkHrHC6nsRRkkc6bkSV-hF8XfAHH1wrxlxQZx5dXs,2392
+athena/core/client_wrapper.py,sha256=UswueTk0Tj8KOnmyXEcydVOBib1IgFR_xCgRe5cB-ZI,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=phDCY6o04mrhCG4P_MjxohSPUM5K8skioaxusKXcApg,4313
-athena/types/aop_async_execute_response_out.py,sha256=R3oy0RLyD-t4MQCnp3Ll13Ci3D5JUMhPyuetbhrONNs,1798
+athena/types/__init__.py,sha256=AwyXvqPl4A6ji4C11HVfG0gdfsUQrhfA7K4FJLZPpTY,4154
+athena/types/aop_async_execute_response_out.py,sha256=Ggs9j5JvLCMaVwaHNyY3whA5gcUffLqc4PwuxYWI-38,1642
 athena/types/aop_execute_request_in.py,sha256=mEsMKyNN6e90gZra733lJDC6z0bZWdc72af3B-Z5aqE,889
-athena/types/aop_execute_response_out.py,sha256=ZRsVi8Ho0l2nBKUyE1uP-4F14FMuuLeap1SZGeVk8NI,1856
+athena/types/aop_execute_response_out.py,sha256=226NGju-pFVGE-zrwNGGiVB1BqkqtGJdMVsdqQVtnxk,1804
 athena/types/asset_content_request_out.py,sha256=RYlcY6j6Ju5EQL7UquDwyTe7uqxyuO8Bg8LCsv1YiUE,652
 athena/types/asset_node.py,sha256=3l7CUK2c_h4QT8ktSq0rFP9veF9G_V9mNe3NZlGl-xQ,804
 athena/types/asset_screenshot_response_out.py,sha256=tiAt9xnpXWMRUnblWzUEJbXYzB412kdqKkb_02OFFGE,1069
@@ -88,9 +88,7 @@ athena/types/chunk_content_item.py,sha256=nKP8lq4AbbAZEKY7bRKOc7sDvqfyslCBCn8Cl_
 athena/types/chunk_result.py,sha256=hgrS4hMeuwTRpJ2YrMdrW_QWWWUQ82iYVVTuhFWm1X0,734
 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=SbsWJuGwic6nliEu9ykW-q2A7SfnRt4U3e1curB8Qok,2900
-athena/types/conversation_message.py,sha256=BMTX63uNvJbWTyGtTGCnqtSW3bAcy-NR1RYjvchvofA,1109
-athena/types/conversation_result.py,sha256=IY06B71I46CHoKyzmL-ebYRIr2qrZejt8nDC-bTlKCE,1817
+athena/types/conversation_asset_info.py,sha256=Ym2CGyWN73rUDGAHmyCNNUPCN2xbYc57lvX5qrQMoN0,2862
 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
@@ -126,6 +124,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.225.dist-info/METADATA,sha256=Ul5huPAMJsF6SmlTw-xH5zvQhPpL4KaaobCqySH8sgo,5440
-athena_intelligence-0.1.225.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.225.dist-info/RECORD,,
+athena_intelligence-0.1.226.dist-info/METADATA,sha256=u3C420jebCV3hyxqS_UEg8MVNRrC1dn1HJeZkvTQHD4,5440
+athena_intelligence-0.1.226.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.226.dist-info/RECORD,,

athena/types/conversation_message.py

@@ -1,42 +0,0 @@
-# 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/types/conversation_result.py

@@ -1,67 +0,0 @@
-# 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