athena-intelligence 0.1.120__tar.gz → 0.1.122__tar.gz

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/PKG-INFO

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

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.120"
+version = "0.1.122"
 description = "Athena Intelligence Python Library"
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/src/athena/__init__.py

@@ -15,14 +15,6 @@ from .types import (
     GeneralAgentConfig,
     GeneralAgentConfigEnabledToolsItem,
     GeneralAgentRequest,
-    GeneralAgentRequestMessagesItem,
-    GeneralAgentRequestMessagesItemContent,
-    GeneralAgentRequestMessagesItemContentItem,
-    GeneralAgentRequestMessagesItemContentItemImageUrl,
-    GeneralAgentRequestMessagesItemContentItemText,
-    GeneralAgentRequestMessagesItemContentItem_ImageUrl,
-    GeneralAgentRequestMessagesItemContentItem_Text,
-    GeneralAgentRequestMessagesItemType,
     GeneralAgentResponse,
     ParentFolderError,
     ResearchAgentResponse,
@@ -64,14 +56,6 @@ __all__ = [
     "GeneralAgentConfig",
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",
-    "GeneralAgentRequestMessagesItem",
-    "GeneralAgentRequestMessagesItemContent",
-    "GeneralAgentRequestMessagesItemContentItem",
-    "GeneralAgentRequestMessagesItemContentItemImageUrl",
-    "GeneralAgentRequestMessagesItemContentItemText",
-    "GeneralAgentRequestMessagesItemContentItem_ImageUrl",
-    "GeneralAgentRequestMessagesItemContentItem_Text",
-    "GeneralAgentRequestMessagesItemType",
     "GeneralAgentResponse",
     "InternalServerError",
     "NotFoundError",

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/src/athena/agents/general/client.py

@@ -39,13 +39,7 @@ class GeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import Athena
 
         client = Athena(
@@ -58,10 +52,10 @@ class GeneralClient:
                         enabled_tools=[Tool.SEARCH],
                     ),
                     messages=[
-                        GeneralAgentRequestMessagesItem(
-                            content="Please call the search tool for AAPL news.",
-                            type=GeneralAgentRequestMessagesItemType.USER,
-                        )
+                        {
+                            "content": "Please call the search tool for AAPL news.",
+                            "type": "user",
+                        }
                     ],
                 )
             ],
@@ -103,13 +97,7 @@ class GeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import Athena
 
         client = Athena(
@@ -121,10 +109,10 @@ class GeneralClient:
                     enabled_tools=[Tool.SEARCH],
                 ),
                 messages=[
-                    GeneralAgentRequestMessagesItem(
-                        content="Please call the search tool for AAPL news.",
-                        type=GeneralAgentRequestMessagesItemType.USER,
-                    )
+                    {
+                        "content": "Please call the search tool for AAPL news.",
+                        "type": "user",
+                    }
                 ],
             ),
         )
@@ -162,13 +150,7 @@ class GeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import Athena
 
         client = Athena(
@@ -180,10 +162,10 @@ class GeneralClient:
                     enabled_tools=[Tool.SEARCH],
                 ),
                 messages=[
-                    GeneralAgentRequestMessagesItem(
-                        content="Please call the search tool for AAPL news.",
-                        type=GeneralAgentRequestMessagesItemType.USER,
-                    )
+                    {
+                        "content": "Please call the search tool for AAPL news.",
+                        "type": "user",
+                    }
                 ],
             ),
         )
@@ -230,13 +212,7 @@ class AsyncGeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
@@ -249,10 +225,10 @@ class AsyncGeneralClient:
                         enabled_tools=[Tool.SEARCH],
                     ),
                     messages=[
-                        GeneralAgentRequestMessagesItem(
-                            content="Please call the search tool for AAPL news.",
-                            type=GeneralAgentRequestMessagesItemType.USER,
-                        )
+                        {
+                            "content": "Please call the search tool for AAPL news.",
+                            "type": "user",
+                        }
                     ],
                 )
             ],
@@ -294,13 +270,7 @@ class AsyncGeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
@@ -312,10 +282,10 @@ class AsyncGeneralClient:
                     enabled_tools=[Tool.SEARCH],
                 ),
                 messages=[
-                    GeneralAgentRequestMessagesItem(
-                        content="Please call the search tool for AAPL news.",
-                        type=GeneralAgentRequestMessagesItemType.USER,
-                    )
+                    {
+                        "content": "Please call the search tool for AAPL news.",
+                        "type": "user",
+                    }
                 ],
             ),
         )
@@ -353,13 +323,7 @@ class AsyncGeneralClient:
 
         Examples
         --------
-        from athena import (
-            GeneralAgentConfig,
-            GeneralAgentRequest,
-            GeneralAgentRequestMessagesItem,
-            GeneralAgentRequestMessagesItemType,
-            Tool,
-        )
+        from athena import GeneralAgentConfig, GeneralAgentRequest, Tool
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
@@ -371,10 +335,10 @@ class AsyncGeneralClient:
                     enabled_tools=[Tool.SEARCH],
                 ),
                 messages=[
-                    GeneralAgentRequestMessagesItem(
-                        content="Please call the search tool for AAPL news.",
-                        type=GeneralAgentRequestMessagesItemType.USER,
-                    )
+                    {
+                        "content": "Please call the search tool for AAPL news.",
+                        "type": "user",
+                    }
                 ],
             ),
         )

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/src/athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.120",
+            "X-Fern-SDK-Version": "0.1.122",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/src/athena/types/__init__.py

@@ -14,18 +14,6 @@ from .file_too_large_error import FileTooLargeError
 from .general_agent_config import GeneralAgentConfig
 from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
 from .general_agent_request import GeneralAgentRequest
-from .general_agent_request_messages_item import GeneralAgentRequestMessagesItem
-from .general_agent_request_messages_item_content import GeneralAgentRequestMessagesItemContent
-from .general_agent_request_messages_item_content_item import (
-    GeneralAgentRequestMessagesItemContentItem,
-    GeneralAgentRequestMessagesItemContentItem_ImageUrl,
-    GeneralAgentRequestMessagesItemContentItem_Text,
-)
-from .general_agent_request_messages_item_content_item_image_url import (
-    GeneralAgentRequestMessagesItemContentItemImageUrl,
-)
-from .general_agent_request_messages_item_content_item_text import GeneralAgentRequestMessagesItemContentItemText
-from .general_agent_request_messages_item_type import GeneralAgentRequestMessagesItemType
 from .general_agent_response import GeneralAgentResponse
 from .parent_folder_error import ParentFolderError
 from .research_agent_response import ResearchAgentResponse
@@ -49,14 +37,6 @@ __all__ = [
     "GeneralAgentConfig",
     "GeneralAgentConfigEnabledToolsItem",
     "GeneralAgentRequest",
-    "GeneralAgentRequestMessagesItem",
-    "GeneralAgentRequestMessagesItemContent",
-    "GeneralAgentRequestMessagesItemContentItem",
-    "GeneralAgentRequestMessagesItemContentItemImageUrl",
-    "GeneralAgentRequestMessagesItemContentItemText",
-    "GeneralAgentRequestMessagesItemContentItem_ImageUrl",
-    "GeneralAgentRequestMessagesItemContentItem_Text",
-    "GeneralAgentRequestMessagesItemType",
     "GeneralAgentResponse",
     "ParentFolderError",
     "ResearchAgentResponse",

{athena_intelligence-0.1.120 → athena_intelligence-0.1.122}/src/athena/types/general_agent_request.py

@@ -6,7 +6,6 @@ import typing
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .general_agent_config import GeneralAgentConfig
-from .general_agent_request_messages_item import GeneralAgentRequestMessagesItem
 
 
 class GeneralAgentRequest(pydantic_v1.BaseModel):
@@ -15,7 +14,7 @@ class GeneralAgentRequest(pydantic_v1.BaseModel):
     """
 
     config: GeneralAgentConfig
-    messages: typing.List[GeneralAgentRequestMessagesItem] = pydantic_v1.Field()
+    messages: typing.List[typing.Any] = pydantic_v1.Field()
     """
     The messages to send to the agent. Each message should be a string (for text inputs) or a list of multimodal content parts.
     """

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item.py

@@ -1,32 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .general_agent_request_messages_item_content import GeneralAgentRequestMessagesItemContent
-from .general_agent_request_messages_item_type import GeneralAgentRequestMessagesItemType
-
-
-class GeneralAgentRequestMessagesItem(pydantic_v1.BaseModel):
-    content: GeneralAgentRequestMessagesItemContent
-    type: GeneralAgentRequestMessagesItemType
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item_content.py

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

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item_content_item.py

@@ -1,60 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class GeneralAgentRequestMessagesItemContentItem_Text(pydantic_v1.BaseModel):
-    text: str
-    type: typing.Literal["text"] = "text"
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-class GeneralAgentRequestMessagesItemContentItem_ImageUrl(pydantic_v1.BaseModel):
-    image_url: typing.Dict[str, str]
-    type: typing.Literal["image_url"] = "image_url"
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}
-
-
-GeneralAgentRequestMessagesItemContentItem = typing.Union[
-    GeneralAgentRequestMessagesItemContentItem_Text, GeneralAgentRequestMessagesItemContentItem_ImageUrl
-]

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item_content_item_image_url.py

@@ -1,29 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class GeneralAgentRequestMessagesItemContentItemImageUrl(pydantic_v1.BaseModel):
-    image_url: typing.Dict[str, str]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item_content_item_text.py

@@ -1,29 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-
-
-class GeneralAgentRequestMessagesItemContentItemText(pydantic_v1.BaseModel):
-    text: str
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
-
-        return deep_union_pydantic_dicts(
-            super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
-        )
-
-    class Config:
-        frozen = True
-        smart_union = True
-        extra = pydantic_v1.Extra.allow
-        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.120/src/athena/types/general_agent_request_messages_item_type.py

@@ -1,25 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import enum
-import typing
-
-T_Result = typing.TypeVar("T_Result")
-
-
-class GeneralAgentRequestMessagesItemType(str, enum.Enum):
-    USER = "user"
-    ASSISTANT = "assistant"
-    SYSTEM = "system"
-
-    def visit(
-        self,
-        user: typing.Callable[[], T_Result],
-        assistant: typing.Callable[[], T_Result],
-        system: typing.Callable[[], T_Result],
-    ) -> T_Result:
-        if self is GeneralAgentRequestMessagesItemType.USER:
-            return user()
-        if self is GeneralAgentRequestMessagesItemType.ASSISTANT:
-            return assistant()
-        if self is GeneralAgentRequestMessagesItemType.SYSTEM:
-            return system()