athena-intelligence 0.1.119__py3-none-any.whl → 0.1.120__py3-none-any.whl

athena/tools/tasks/client.py

@@ -0,0 +1,87 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from json.decoder import JSONDecodeError
+
+from ...core.api_error import ApiError
+from ...core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ...core.pydantic_utilities import pydantic_v1
+from ...core.request_options import RequestOptions
+
+
+class TasksClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def run_task(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Run a [task](https://resources.athenaintel.com/docs/task-studio/home).
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.tasks.run_task()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncTasksClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def run_task(self, *, request_options: typing.Optional[RequestOptions] = None) -> typing.Any:
+        """
+        Coming soon! Run a [task](https://resources.athenaintel.com/docs/task-studio/home).
+
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.Any
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.tasks.run_task()
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run", method="POST", request_options=request_options
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(typing.Any, _response.json())  # type: ignore
+        try:
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)

athena/types/__init__.py

@@ -1,39 +1,67 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .asset_not_found_error import AssetNotFoundError
-from .athena_assistant_config import AthenaAssistantConfig
-from .athena_assistant_config_enabled_tools_item import AthenaAssistantConfigEnabledToolsItem
-from .athena_assistant_reponse import AthenaAssistantReponse
-from .athena_assistant_request import AthenaAssistantRequest
+from .custom_agent_response import CustomAgentResponse
 from .data_frame_request_out import DataFrameRequestOut
 from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
 from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
 from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
 from .data_frame_unknown_format_error import DataFrameUnknownFormatError
 from .document_chunk import DocumentChunk
+from .drive_agent_response import DriveAgentResponse
 from .file_chunk_request_out import FileChunkRequestOut
 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
 from .save_asset_request_out import SaveAssetRequestOut
+from .sql_agent_response import SqlAgentResponse
 from .structured_data_extractor_reponse import StructuredDataExtractorReponse
 from .tool import Tool
 
 __all__ = [
     "AssetNotFoundError",
-    "AthenaAssistantConfig",
-    "AthenaAssistantConfigEnabledToolsItem",
-    "AthenaAssistantReponse",
-    "AthenaAssistantRequest",
+    "CustomAgentResponse",
     "DataFrameRequestOut",
     "DataFrameRequestOutColumnsItem",
     "DataFrameRequestOutDataItemItem",
     "DataFrameRequestOutIndexItem",
     "DataFrameUnknownFormatError",
     "DocumentChunk",
+    "DriveAgentResponse",
     "FileChunkRequestOut",
     "FileTooLargeError",
+    "GeneralAgentConfig",
+    "GeneralAgentConfigEnabledToolsItem",
+    "GeneralAgentRequest",
+    "GeneralAgentRequestMessagesItem",
+    "GeneralAgentRequestMessagesItemContent",
+    "GeneralAgentRequestMessagesItemContentItem",
+    "GeneralAgentRequestMessagesItemContentItemImageUrl",
+    "GeneralAgentRequestMessagesItemContentItemText",
+    "GeneralAgentRequestMessagesItemContentItem_ImageUrl",
+    "GeneralAgentRequestMessagesItemContentItem_Text",
+    "GeneralAgentRequestMessagesItemType",
+    "GeneralAgentResponse",
     "ParentFolderError",
+    "ResearchAgentResponse",
     "SaveAssetRequestOut",
+    "SqlAgentResponse",
     "StructuredDataExtractorReponse",
     "Tool",
 ]

athena/types/custom_agent_response.py

@@ -0,0 +1,32 @@
+# 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 CustomAgentResponse(pydantic_v1.BaseModel):
+    result: typing.Dict[str, typing.Any] = pydantic_v1.Field()
+    """
+    The agent's response. Format depends on the specific agent implementation.
+    """
+
+    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/types/drive_agent_response.py

@@ -0,0 +1,32 @@
+# 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 DriveAgentResponse(pydantic_v1.BaseModel):
+    result: typing.Dict[str, typing.Any] = pydantic_v1.Field()
+    """
+    Results of the drive operation
+    """
+
+    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/types/{athena_assistant_config.py → general_agent_config.py}

@@ -5,15 +5,15 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .athena_assistant_config_enabled_tools_item import AthenaAssistantConfigEnabledToolsItem
+from .general_agent_config_enabled_tools_item import GeneralAgentConfigEnabledToolsItem
 
 
-class AthenaAssistantConfig(pydantic_v1.BaseModel):
+class GeneralAgentConfig(pydantic_v1.BaseModel):
     """
-    Configurable fields for the general agent.
+    Configurable fields for the agent.
     """
 
-    enabled_tools: typing.Optional[typing.List[AthenaAssistantConfigEnabledToolsItem]] = None
+    enabled_tools: typing.Optional[typing.List[GeneralAgentConfigEnabledToolsItem]] = None
     knowledge_base_asset_ids: typing.Optional[typing.List[str]] = None
     system_prompt: typing.Optional[str] = None
 

athena/types/{athena_assistant_config_enabled_tools_item.py → general_agent_config_enabled_tools_item.py}

@@ -4,4 +4,4 @@ import typing
 
 from .tool import Tool
 
-AthenaAssistantConfigEnabledToolsItem = typing.Union[Tool, str]
+GeneralAgentConfigEnabledToolsItem = typing.Union[Tool, str]

athena/types/general_agent_request.py

@@ -0,0 +1,39 @@
+# 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_config import GeneralAgentConfig
+from .general_agent_request_messages_item import GeneralAgentRequestMessagesItem
+
+
+class GeneralAgentRequest(pydantic_v1.BaseModel):
+    """
+    A chat request for the Athena SDK.
+    """
+
+    config: GeneralAgentConfig
+    messages: typing.List[GeneralAgentRequestMessagesItem] = 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.
+    """
+
+    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/types/general_agent_request_messages_item.py

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

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

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

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

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

@@ -0,0 +1,25 @@
+# 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()

athena/types/{athena_assistant_reponse.py → general_agent_response.py}

@@ -7,9 +7,9 @@ from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 
 
-class AthenaAssistantReponse(pydantic_v1.BaseModel):
+class GeneralAgentResponse(pydantic_v1.BaseModel):
     """
-    The response from the assistant.
+    The response from the agent.
     """
 
     messages: typing.List[typing.Dict[str, typing.Any]]

athena/types/research_agent_response.py

@@ -0,0 +1,32 @@
+# 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 ResearchAgentResponse(pydantic_v1.BaseModel):
+    findings: typing.Dict[str, typing.Any] = pydantic_v1.Field()
+    """
+    Research findings and compiled information
+    """
+
+    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/types/{athena_assistant_request.py → sql_agent_response.py}

@@ -5,18 +5,17 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
-from .athena_assistant_config import AthenaAssistantConfig
 
 
-class AthenaAssistantRequest(pydantic_v1.BaseModel):
+class SqlAgentResponse(pydantic_v1.BaseModel):
+    metadata: typing.Dict[str, typing.Any] = pydantic_v1.Field()
     """
-    A chat request for the Athena SDK.
+    Additional metadata about the generated query
     """
 
-    config: AthenaAssistantConfig
-    messages: typing.List[typing.Any] = pydantic_v1.Field()
+    query_asset_id: str = pydantic_v1.Field()
     """
-    The messages to send to the assistant. Each message should be a string (for text inputs) or a list of multimodal content parts.
+    The asset ID of the generated SQL query object
     """
 
     def json(self, **kwargs: typing.Any) -> str:

{athena_intelligence-0.1.119.dist-info → athena_intelligence-0.1.120.dist-info}/METADATA

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

{athena_intelligence-0.1.119.dist-info → athena_intelligence-0.1.120.dist-info}/RECORD

@@ -1,15 +1,19 @@
-athena/__init__.py,sha256=qe7vfVyUgh4lzmUyzvYArfPNjvyahJyAUaQe8UQJIhw,1862
-athena/agents/__init__.py,sha256=0DbSuLxvqIRHKaYvdfLF0ylhq9oJ1uxhiW5h-FSj4Yc,184
-athena/agents/athena_assistant/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/agents/athena_assistant/client.py,sha256=izneKMmYkypaTiPcqq3zf7n0XyuV9fQjmtAqKIdZeNs,12444
-athena/agents/client.py,sha256=ER9TWMqF-MlsG5L3QVq1PfY-n_5Ns17dEnd9dIjRcNs,1010
-athena/agents/structured_data_extractor/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/agents/structured_data_extractor/client.py,sha256=7X-xSDUhfzO8U51Y0tFEzKl_9qYS8PAF_0Rg1FbCj90,6064
+athena/__init__.py,sha256=MACR2vDOM0HwmGVSVhh42Ip3lHd9l8JjRKIo5Aevx5Y,2836
+athena/agents/__init__.py,sha256=I6MO2O_hb6KLa8oDHbGNSAhcPE-dsrX6LMcAEhsg3PQ,160
+athena/agents/client.py,sha256=aI8rNhXBSVJ-hvjnIoCK9sKvHB0e95Zkn-3YpXOKFrY,6721
+athena/agents/drive/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/agents/drive/client.py,sha256=k--gUzUSi6L5kpmKHi7eX1UnlEq4kG-QI9phAZ5srC0,4727
+athena/agents/general/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/agents/general/client.py,sha256=y9shUOac8R_pu_yd0V-4cMq68qGSjqTiX8V-igYmQPA,14366
+athena/agents/research/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/agents/research/client.py,sha256=mo63cUvLw8AC9I29WI4Ym_jOsvRsco1wRqsNp0rtCEc,4745
+athena/agents/sql/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/agents/sql/client.py,sha256=1r2aT4JLxVCeL0FzEBeHmttwerdRmJOxVHim2_CHkvg,4767
 athena/base_client.py,sha256=-kVdOlIibBz48lxWratdQAzT7fTvZsORvOMF3KoPDPw,5647
 athena/client.py,sha256=4PUPrBPCMTFpHR1yuKVR5eC1AYBl_25SMf6ZH82JHB0,19039
 athena/core/__init__.py,sha256=UFXpYzcGxWQUucU1TkjOQ9mGWN3A5JohluOIWVYKU4I,973
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=rAKPSQX1LSvfr5YZWrZIfliuFdTYM1oxCHv3hjCwfRc,1806
+athena/core/client_wrapper.py,sha256=WBlIIwsQ_x87kLK92PgqwXkZgqmn8LFo38pUgumr8xQ,1806
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=Z4NuAsJD-51yqmoME17O5sxwx5orSp1wsnd6bPyKcgA,17768
@@ -32,29 +36,47 @@ athena/query/__init__.py,sha256=EIeVs3aDVNAqLyWetSrQldzc2yUdJkf_cJXyrLdMDj0,171
 athena/query/client.py,sha256=GYvBV7YFYjL3gyskeF6C8BJZlLvU7R045gJ5kGiYkvg,9043
 athena/query/types/__init__.py,sha256=WX-Or2h5NY2sv93ojrZsHcmiFHGYdqd0yxNo-5iGHR4,206
 athena/query/types/query_execute_request_database_asset_ids.py,sha256=aoVl5Xb34Q27hYGuVTnByGIxtHkL67wAwzXh7eJctew,154
-athena/tools/__init__.py,sha256=3n7oOoMebo06MAQqYRE2CX9Q0fTNnKBYE0cTlh1MPkM,165
-athena/tools/client.py,sha256=lYzvUj_QW-b3ighcXV29LUo59HpM4V9ACe2cbtoHhHg,20209
+athena/tools/__init__.py,sha256=DREW2sa5Z-Rj8v1LQ-tNhflI-EO_oDc6NCmF5v0LWeU,288
+athena/tools/calendar/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/tools/calendar/client.py,sha256=hKWzWyl1GwFG69oX3tektwNfy2sV5Lt6PRy9vTyPLOo,5283
+athena/tools/client.py,sha256=9ec2gnf3z_vhr3EqT_-ZksevTDtFP1jftPY4os0Ty3Q,21166
+athena/tools/email/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/tools/email/client.py,sha256=epUkV5af3eilVgRR81SFZAf29JuhEWKMkdMuN6qDLUM,7593
+athena/tools/structured_data_extractor/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/tools/structured_data_extractor/client.py,sha256=yEF-wVyRhebkSlb0r_kcvs4P7TZSlfPTwG-GdZAclvQ,6272
+athena/tools/tasks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
+athena/tools/tasks/client.py,sha256=5kT6ulh2YDIbNYiv-knBjtF-ST7p0dUvZyrd7t5O61s,2975
 athena/tools/types/__init__.py,sha256=cA-ZQm6veQAP3_vKu9KkZpISsQqgTBN_Z--FGY1c2iA,197
 athena/tools/types/tools_data_frame_request_columns_item.py,sha256=GA1FUlTV_CfSc-KToTAwFf4Exl0rr4fsweVZupztjw0,138
-athena/types/__init__.py,sha256=TmkqEsOHEaNq68rlKbwTphj8YnpJdJvrogdjg9mNVKs,1645
+athena/types/__init__.py,sha256=9BVgfnLmiuBnB0DrTg3y91mBEQ8fAdo3l4hELyoBtzA,3075
 athena/types/asset_not_found_error.py,sha256=ZcgqRuzvO4Z8vVVxwtDB-QtKhpVIVV3hqQuJeUoOoJE,1121
-athena/types/athena_assistant_config.py,sha256=vfzhIaY2T68u44iyJTEvURAGtX77vvyPhm9KA5ys7L4,1477
-athena/types/athena_assistant_config_enabled_tools_item.py,sha256=JWS0KbgzoO4lEX0gTyjnePy7oe47i3uXuTT3vl1lE7o,168
-athena/types/athena_assistant_reponse.py,sha256=tBYSAmaJ3AtKUvg9lzTHLj1guHKuvVOzBDGvKGzG7fE,1218
-athena/types/athena_assistant_request.py,sha256=nFj5n7uBNShFRvkcQPhpu2xJsTx24qA20knmjsE07XI,1465
+athena/types/custom_agent_response.py,sha256=_Vm_fJq4cETtOawBW7p0cvH4Jmle26lHQZ73A8MdLX0,1263
 athena/types/data_frame_request_out.py,sha256=1CEBe-baDQi0uz_EgMw0TKGYXGj6KV44cL3ViRTZLKM,1669
 athena/types/data_frame_request_out_columns_item.py,sha256=9cjzciFv6C8n8Griytt_q_8ovkzHViS5tvUcMDfkfKE,143
 athena/types/data_frame_request_out_data_item_item.py,sha256=KMTJRr-1bdKDNMbUericCliwRoPHLGRV-n2bJtxdRW0,144
 athena/types/data_frame_request_out_index_item.py,sha256=bW7oe912trpkYKodj-I_AiTXXy61yWzliekcsUZkZE0,141
 athena/types/data_frame_unknown_format_error.py,sha256=lbgAUArEgIYZt3P5Ji4Clp-xyXnKJR7-v5VzXwKu0J8,1168
 athena/types/document_chunk.py,sha256=deXiiMA_U5EabUh1Fg2AB4ElYuc5OsTnrU7JwLApJmA,1227
+athena/types/drive_agent_response.py,sha256=UMKF43e5WScH0a9ITuxjwGWzAzvXAl1OsfRVeXSyfjk,1218
 athena/types/file_chunk_request_out.py,sha256=Ju0I_UpSitjQ-XqSIRvvg2XA6QtfCLZClPq5PUqmPNg,1258
 athena/types/file_too_large_error.py,sha256=AinkrcgR7lcTILAD8RX0x48P3GlSoAh1OihxMvSvRuo,1120
+athena/types/general_agent_config.py,sha256=FaswWVsDTsL5Fs9Tlx4zSK1S8OrsFnzruEt7l72XlGA,1457
+athena/types/general_agent_config_enabled_tools_item.py,sha256=6gYaU7uIDJbgygtBKLdYL-VbPxxbEcxwRsT8VaW5vN8,165
+athena/types/general_agent_request.py,sha256=9HEZZFKJZsmqBRKb_-EawJ0H_AZ22Tl9apLjMCuAB34,1551
+athena/types/general_agent_request_messages_item.py,sha256=4sFvUZ3YeR5ce7R7EKRSz6RHjq0CWQk7Jip7_9t2t7Y,1401
+athena/types/general_agent_request_messages_item_content.py,sha256=kG1NX_QA_o-HEQPDVxI6-h9xkrfTyKfMqZ2bE7ipnTw,302
+athena/types/general_agent_request_messages_item_content_item.py,sha256=U_g0R0bAxdKhdUmlThA9BH8vKq2qTJ1FmtL_iQQtRrk,2387
+athena/types/general_agent_request_messages_item_content_item_image_url.py,sha256=JNdVoLBMzm6gl_OOFajseZIAzu5_tLms76UL1e7WRTk,1173
+athena/types/general_agent_request_messages_item_content_item_text.py,sha256=GfSBWQ7igY3uQlsBHKmVlHXBrcSP1rVScTDPy9om62k,1146
+athena/types/general_agent_request_messages_item_type.py,sha256=dDSQcu9xtKZdNjDux2xSQd7AO9WdrOFmi8rHFU6GTbY,725
+athena/types/general_agent_response.py,sha256=9BxqXzchSti5O0Ch_WJkvmkawkBhpH03QlZIbKdYbAY,1212
 athena/types/parent_folder_error.py,sha256=ZMF-i3mZY6Mu1n5uQ60Q3mIIfehlWuXtgFUkSYspkx8,1120
+athena/types/research_agent_response.py,sha256=-1mX4M0IEWDFH3alSZdtuhZHSerjWYJQkn74r3Dp26g,1235
 athena/types/save_asset_request_out.py,sha256=5bpBaUV3oeuL_hz4s07c-6MQHkn4cBsyxgT_SD5oi6I,1193
+athena/types/sql_agent_response.py,sha256=DmeG0HPZkPT_gTrtkroVZluGZIV9McB8wmME2iT8PB0,1347
 athena/types/structured_data_extractor_reponse.py,sha256=Fa-54k20UIqHKDsmUpF4wZpaiTb_J35XysY1GZUf63w,1306
 athena/types/tool.py,sha256=6H2BFZiBgQOtYUAwSYBeGZKhwev17IEwnIjgmno6dZw,436
 athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
-athena_intelligence-0.1.119.dist-info/METADATA,sha256=-kKig1kesG7Av6k1YBHSqxuob32bk0x8lid2_u4Sv1U,5274
-athena_intelligence-0.1.119.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.119.dist-info/RECORD,,
+athena_intelligence-0.1.120.dist-info/METADATA,sha256=27jXrmT86Ay3sxkrgx-zHJRM3BOPi8wA-ubld5DxLHs,5274
+athena_intelligence-0.1.120.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.120.dist-info/RECORD,,