athena-intelligence 0.1.206__tar.gz → 0.1.207__tar.gz

{athena_intelligence-0.1.206 → athena_intelligence-0.1.207}/PKG-INFO

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

{athena_intelligence-0.1.206 → athena_intelligence-0.1.207}/pyproject.toml

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

{athena_intelligence-0.1.206 → athena_intelligence-0.1.207}/src/athena/__init__.py

@@ -17,6 +17,7 @@ from .types import (
     ChunkResult,
     ChunkResultChunkId,
     Content,
+    ConversationAssetInfo,
     CreateNewSheetTabResponse,
     CustomAgentResponse,
     DataFrameRequestOut,
@@ -51,6 +52,7 @@ from .types import (
     SqlAgentResponse,
     StructuredDataExtractorResponse,
     TextContent,
+    ThreadStatusResponseOut,
     Type,
 )
 from .errors import (
@@ -62,7 +64,7 @@ from .errors import (
     UnprocessableEntityError,
     UnsupportedMediaTypeError,
 )
-from . import agents, aop, assets, query, tools
+from . import agents, aop, assets, query, threads, tools
 from .client import AsyncAthena, Athena
 from .environment import AthenaEnvironment
 from .query import QueryExecuteRequestDatabaseAssetIds
@@ -89,6 +91,7 @@ __all__ = [
     "ChunkResultChunkId",
     "Content",
     "ContentTooLargeError",
+    "ConversationAssetInfo",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",
@@ -126,6 +129,7 @@ __all__ = [
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "TextContent",
+    "ThreadStatusResponseOut",
     "ToolsDataFrameRequestColumnsItem",
     "Type",
     "UnauthorizedError",
@@ -136,5 +140,6 @@ __all__ = [
     "aop",
     "assets",
     "query",
+    "threads",
     "tools",
 ]

{athena_intelligence-0.1.206 → athena_intelligence-0.1.207}/src/athena/base_client.py

@@ -9,6 +9,7 @@ from .assets.client import AssetsClient, AsyncAssetsClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from .environment import AthenaEnvironment
 from .query.client import AsyncQueryClient, QueryClient
+from .threads.client import AsyncThreadsClient, ThreadsClient
 from .tools.client import AsyncToolsClient, ToolsClient
 
 
@@ -81,6 +82,7 @@ class BaseAthena:
         self.aop = AopClient(client_wrapper=self._client_wrapper)
         self.assets = AssetsClient(client_wrapper=self._client_wrapper)
         self.query = QueryClient(client_wrapper=self._client_wrapper)
+        self.threads = ThreadsClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
 
 
@@ -153,6 +155,7 @@ class AsyncBaseAthena:
         self.aop = AsyncAopClient(client_wrapper=self._client_wrapper)
         self.assets = AsyncAssetsClient(client_wrapper=self._client_wrapper)
         self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
+        self.threads = AsyncThreadsClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
 
 

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

athena_intelligence-0.1.207/src/athena/threads/client.py

@@ -0,0 +1,114 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from ..core.request_options import RequestOptions
+from ..types.thread_status_response_out import ThreadStatusResponseOut
+from .raw_client import AsyncRawThreadsClient, RawThreadsClient
+
+
+class ThreadsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawThreadsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawThreadsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawThreadsClient
+        """
+        return self._raw_client
+
+    def get_status(
+        self, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> ThreadStatusResponseOut:
+        """
+        Check the status of a thread execution by thread ID. Returns thread status and associated conversation asset information for tracking progress.
+
+        Parameters
+        ----------
+        thread_id : str
+            The unique thread ID to check status for
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ThreadStatusResponseOut
+            Thread status retrieved successfully
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.threads.get_status(
+            thread_id="thread_id",
+        )
+        """
+        _response = self._raw_client.get_status(thread_id, request_options=request_options)
+        return _response.data
+
+
+class AsyncThreadsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawThreadsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawThreadsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawThreadsClient
+        """
+        return self._raw_client
+
+    async def get_status(
+        self, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> ThreadStatusResponseOut:
+        """
+        Check the status of a thread execution by thread ID. Returns thread status and associated conversation asset information for tracking progress.
+
+        Parameters
+        ----------
+        thread_id : str
+            The unique thread ID to check status for
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ThreadStatusResponseOut
+            Thread status retrieved successfully
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.threads.get_status(
+                thread_id="thread_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_status(thread_id, request_options=request_options)
+        return _response.data

athena_intelligence-0.1.207/src/athena/threads/raw_client.py

@@ -0,0 +1,193 @@
+# 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.http_response import AsyncHttpResponse, HttpResponse
+from ..core.jsonable_encoder import jsonable_encoder
+from ..core.pydantic_utilities import parse_obj_as
+from ..core.request_options import RequestOptions
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.unauthorized_error import UnauthorizedError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.asset_not_found_error import AssetNotFoundError
+from ..types.thread_status_response_out import ThreadStatusResponseOut
+
+
+class RawThreadsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def get_status(
+        self, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> HttpResponse[ThreadStatusResponseOut]:
+        """
+        Check the status of a thread execution by thread ID. Returns thread status and associated conversation asset information for tracking progress.
+
+        Parameters
+        ----------
+        thread_id : str
+            The unique thread ID to check status for
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[ThreadStatusResponseOut]
+            Thread status retrieved successfully
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            f"api/v0/threads/{jsonable_encoder(thread_id)}/status",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ThreadStatusResponseOut,
+                    parse_obj_as(
+                        type_=ThreadStatusResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
+
+
+class AsyncRawThreadsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def get_status(
+        self, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[ThreadStatusResponseOut]:
+        """
+        Check the status of a thread execution by thread ID. Returns thread status and associated conversation asset information for tracking progress.
+
+        Parameters
+        ----------
+        thread_id : str
+            The unique thread ID to check status for
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[ThreadStatusResponseOut]
+            Thread status retrieved successfully
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"api/v0/threads/{jsonable_encoder(thread_id)}/status",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    ThreadStatusResponseOut,
+                    parse_obj_as(
+                        type_=ThreadStatusResponseOut,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        AssetNotFoundError,
+                        parse_obj_as(
+                            type_=AssetNotFoundError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
+        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)

athena_intelligence-0.1.207/src/athena/tools/tasks/__init__.py

@@ -0,0 +1,4 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+

{athena_intelligence-0.1.206 → athena_intelligence-0.1.207}/src/athena/types/__init__.py

@@ -14,6 +14,7 @@ from .chunk_content_item import ChunkContentItem, ChunkContentItem_ImageUrl, Chu
 from .chunk_result import ChunkResult
 from .chunk_result_chunk_id import ChunkResultChunkId
 from .content import Content
+from .conversation_asset_info import ConversationAssetInfo
 from .create_new_sheet_tab_response import CreateNewSheetTabResponse
 from .custom_agent_response import CustomAgentResponse
 from .data_frame_request_out import DataFrameRequestOut
@@ -50,6 +51,7 @@ from .sheet_operation_response import SheetOperationResponse
 from .sql_agent_response import SqlAgentResponse
 from .structured_data_extractor_response import StructuredDataExtractorResponse
 from .text_content import TextContent
+from .thread_status_response_out import ThreadStatusResponseOut
 from .type import Type
 
 __all__ = [
@@ -67,6 +69,7 @@ __all__ = [
     "ChunkResult",
     "ChunkResultChunkId",
     "Content",
+    "ConversationAssetInfo",
     "CreateNewSheetTabResponse",
     "CustomAgentResponse",
     "DataFrameRequestOut",
@@ -101,5 +104,6 @@ __all__ = [
     "SqlAgentResponse",
     "StructuredDataExtractorResponse",
     "TextContent",
+    "ThreadStatusResponseOut",
     "Type",
 ]

athena_intelligence-0.1.207/src/athena/types/conversation_asset_info.py

@@ -0,0 +1,100 @@
+# 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 ConversationAssetInfo(UniversalBaseModel):
+    """
+    Conversation asset information associated with a thread.
+    """
+
+    agent: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Agent configuration used in conversation
+    """
+
+    conversation_asset_id: str = pydantic.Field()
+    """
+    ID of the conversation asset
+    """
+
+    created_at: str = pydantic.Field()
+    """
+    ISO timestamp when conversation was created
+    """
+
+    created_by: str = pydantic.Field()
+    """
+    User ID who created the conversation
+    """
+
+    last_channel: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Last active channel for the conversation
+    """
+
+    last_message: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Last message content
+    """
+
+    linked_aops: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(
+        default=None
+    )
+    """
+    List of linked AOP assets
+    """
+
+    linked_projects: typing.Optional[typing.List[typing.Dict[str, typing.Optional[typing.Any]]]] = pydantic.Field(
+        default=None
+    )
+    """
+    List of linked project assets
+    """
+
+    model: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Model used in conversation
+    """
+
+    num_messages: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of messages in the conversation
+    """
+
+    start_channel: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Channel where conversation was started
+    """
+
+    state: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Current state of the conversation (e.g., 'running', 'completed')
+    """
+
+    title: str = pydantic.Field()
+    """
+    Title of the conversation
+    """
+
+    updated_at: str = pydantic.Field()
+    """
+    ISO timestamp when conversation was last updated
+    """
+
+    workspace_id: str = pydantic.Field()
+    """
+    Workspace ID where conversation exists
+    """
+
+    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.207/src/athena/types/thread_status_response_out.py

@@ -0,0 +1,47 @@
+# 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_asset_info import ConversationAssetInfo
+
+
+class ThreadStatusResponseOut(UniversalBaseModel):
+    """
+    Response model for thread status check.
+    """
+
+    conversation_asset: typing.Optional[ConversationAssetInfo] = pydantic.Field(default=None)
+    """
+    Information about the associated conversation asset
+    """
+
+    created_at: str = pydantic.Field()
+    """
+    ISO timestamp when thread was created
+    """
+
+    status: str = pydantic.Field()
+    """
+    Current status of the thread (e.g., 'running', 'completed', 'failed')
+    """
+
+    thread_id: str = pydantic.Field()
+    """
+    The thread ID that was checked
+    """
+
+    updated_at: str = pydantic.Field()
+    """
+    ISO timestamp when thread 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