athena-intelligence 0.1.332__tar.gz → 0.1.399__tar.gz

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/PKG-INFO

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

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/pyproject.toml

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

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/src/athena/__init__.py

@@ -63,6 +63,7 @@ if typing.TYPE_CHECKING:
         PromptMessage,
         PublicAssetOut,
         ResearchAgentResponse,
+        RunTaskResponse,
         SaveAssetRequestOut,
         Sheet,
         SheetOperationResponse,
@@ -157,6 +158,7 @@ _dynamic_imports: typing.Dict[str, str] = {
     "PublicAssetOut": ".types",
     "QueryExecuteRequestDatabaseAssetIds": ".query",
     "ResearchAgentResponse": ".types",
+    "RunTaskResponse": ".types",
     "SaveAssetRequestOut": ".types",
     "Sheet": ".types",
     "SheetOperationResponse": ".types",
@@ -270,6 +272,7 @@ __all__ = [
     "PublicAssetOut",
     "QueryExecuteRequestDatabaseAssetIds",
     "ResearchAgentResponse",
+    "RunTaskResponse",
     "SaveAssetRequestOut",
     "Sheet",
     "SheetOperationResponse",

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

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/src/athena/tools/__init__.py

@@ -9,7 +9,9 @@ if typing.TYPE_CHECKING:
     from .types import ToolsDataFrameRequestColumnsItem
     from . import calendar, email, sheets, structured_data_extractor, tasks
     from .sheets import UpdateSheetRangeRequestValuesItemItem
+    from .tasks import RunTaskRequestTaskType
 _dynamic_imports: typing.Dict[str, str] = {
+    "RunTaskRequestTaskType": ".tasks",
     "ToolsDataFrameRequestColumnsItem": ".types",
     "UpdateSheetRangeRequestValuesItemItem": ".sheets",
     "calendar": ".calendar",
@@ -42,6 +44,7 @@ def __dir__():
 
 
 __all__ = [
+    "RunTaskRequestTaskType",
     "ToolsDataFrameRequestColumnsItem",
     "UpdateSheetRangeRequestValuesItemItem",
     "calendar",

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/src/athena/tools/client.py

@@ -159,7 +159,6 @@ class ToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -171,8 +170,6 @@ class ToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -194,14 +191,12 @@ class ToolsClient:
         )
         client.tools.list_contents(
             asset_id="asset_id",
-            folder_id="folder_id",
             include_asset_details=True,
             include_system_files=True,
         )
         """
         _response = self._raw_client.list_contents(
             asset_id=asset_id,
-            folder_id=folder_id,
             include_asset_details=include_asset_details,
             include_system_files=include_system_files,
             request_options=request_options,
@@ -544,7 +539,6 @@ class AsyncToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -556,8 +550,6 @@ class AsyncToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -584,7 +576,6 @@ class AsyncToolsClient:
         async def main() -> None:
             await client.tools.list_contents(
                 asset_id="asset_id",
-                folder_id="folder_id",
                 include_asset_details=True,
                 include_system_files=True,
             )
@@ -594,7 +585,6 @@ class AsyncToolsClient:
         """
         _response = await self._raw_client.list_contents(
             asset_id=asset_id,
-            folder_id=folder_id,
             include_asset_details=include_asset_details,
             include_system_files=include_system_files,
             request_options=request_options,

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/src/athena/tools/raw_client.py

@@ -285,7 +285,6 @@ class RawToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -297,8 +296,6 @@ class RawToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -316,7 +313,6 @@ class RawToolsClient:
             method="GET",
             params={
                 "asset_id": asset_id,
-                "folder_id": folder_id,
                 "include_asset_details": include_asset_details,
                 "include_system_files": include_system_files,
             },
@@ -934,7 +930,6 @@ class AsyncRawToolsClient:
         self,
         *,
         asset_id: typing.Optional[str] = None,
-        folder_id: typing.Optional[str] = None,
         include_asset_details: typing.Optional[bool] = None,
         include_system_files: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
@@ -946,8 +941,6 @@ class AsyncRawToolsClient:
         ----------
         asset_id : typing.Optional[str]
 
-        folder_id : typing.Optional[str]
-
         include_asset_details : typing.Optional[bool]
 
         include_system_files : typing.Optional[bool]
@@ -965,7 +958,6 @@ class AsyncRawToolsClient:
             method="GET",
             params={
                 "asset_id": asset_id,
-                "folder_id": folder_id,
                 "include_asset_details": include_asset_details,
                 "include_system_files": include_system_files,
             },

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

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .types import RunTaskRequestTaskType
+_dynamic_imports: typing.Dict[str, str] = {"RunTaskRequestTaskType": ".types"}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["RunTaskRequestTaskType"]

athena_intelligence-0.1.399/src/athena/tools/tasks/client.py

@@ -0,0 +1,148 @@
+# 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.run_task_response import RunTaskResponse
+from .raw_client import AsyncRawTasksClient, RawTasksClient
+from .types.run_task_request_task_type import RunTaskRequestTaskType
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class TasksClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._raw_client = RawTasksClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawTasksClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTasksClient
+        """
+        return self._raw_client
+
+    def run_task(
+        self,
+        *,
+        task_id: str,
+        arguments: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        task_type: typing.Optional[RunTaskRequestTaskType] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> RunTaskResponse:
+        """
+        Run a [task](https://resources.athenaintel.com/docs/task-studio/home) and wait for the result.
+
+        Executes a Windmill script or flow synchronously. Server handles polling internally.
+
+        Parameters
+        ----------
+        task_id : str
+            The unique identifier (path) of the task. Example: 'f/public/databricks_describe_table'
+
+        arguments : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Arguments to pass to the task
+
+        task_type : typing.Optional[RunTaskRequestTaskType]
+            Type: 'script' or 'flow'
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        RunTaskResponse
+            Successful Response
+
+        Examples
+        --------
+        from athena import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.tasks.run_task(
+            task_id="task_id",
+        )
+        """
+        _response = self._raw_client.run_task(
+            task_id=task_id, arguments=arguments, task_type=task_type, request_options=request_options
+        )
+        return _response.data
+
+
+class AsyncTasksClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawTasksClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawTasksClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawTasksClient
+        """
+        return self._raw_client
+
+    async def run_task(
+        self,
+        *,
+        task_id: str,
+        arguments: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        task_type: typing.Optional[RunTaskRequestTaskType] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> RunTaskResponse:
+        """
+        Run a [task](https://resources.athenaintel.com/docs/task-studio/home) and wait for the result.
+
+        Executes a Windmill script or flow synchronously. Server handles polling internally.
+
+        Parameters
+        ----------
+        task_id : str
+            The unique identifier (path) of the task. Example: 'f/public/databricks_describe_table'
+
+        arguments : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Arguments to pass to the task
+
+        task_type : typing.Optional[RunTaskRequestTaskType]
+            Type: 'script' or 'flow'
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        RunTaskResponse
+            Successful Response
+
+        Examples
+        --------
+        import asyncio
+
+        from athena import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+
+
+        async def main() -> None:
+            await client.tools.tasks.run_task(
+                task_id="task_id",
+            )
+
+
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.run_task(
+            task_id=task_id, arguments=arguments, task_type=task_type, request_options=request_options
+        )
+        return _response.data

athena_intelligence-0.1.399/src/athena/tools/tasks/raw_client.py

@@ -0,0 +1,170 @@
+# 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.pydantic_utilities import parse_obj_as
+from ...core.request_options import RequestOptions
+from ...errors.unprocessable_entity_error import UnprocessableEntityError
+from ...types.run_task_response import RunTaskResponse
+from .types.run_task_request_task_type import RunTaskRequestTaskType
+
+# this is used as the default value for optional parameters
+OMIT = typing.cast(typing.Any, ...)
+
+
+class RawTasksClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def run_task(
+        self,
+        *,
+        task_id: str,
+        arguments: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        task_type: typing.Optional[RunTaskRequestTaskType] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[RunTaskResponse]:
+        """
+        Run a [task](https://resources.athenaintel.com/docs/task-studio/home) and wait for the result.
+
+        Executes a Windmill script or flow synchronously. Server handles polling internally.
+
+        Parameters
+        ----------
+        task_id : str
+            The unique identifier (path) of the task. Example: 'f/public/databricks_describe_table'
+
+        arguments : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Arguments to pass to the task
+
+        task_type : typing.Optional[RunTaskRequestTaskType]
+            Type: 'script' or 'flow'
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        HttpResponse[RunTaskResponse]
+            Successful Response
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run",
+            method="POST",
+            json={
+                "arguments": arguments,
+                "task_id": task_id,
+                "task_type": task_type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    RunTaskResponse,
+                    parse_obj_as(
+                        type_=RunTaskResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return HttpResponse(response=_response, data=_data)
+            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(),
+                        ),
+                    ),
+                )
+            _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 AsyncRawTasksClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def run_task(
+        self,
+        *,
+        task_id: str,
+        arguments: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = OMIT,
+        task_type: typing.Optional[RunTaskRequestTaskType] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[RunTaskResponse]:
+        """
+        Run a [task](https://resources.athenaintel.com/docs/task-studio/home) and wait for the result.
+
+        Executes a Windmill script or flow synchronously. Server handles polling internally.
+
+        Parameters
+        ----------
+        task_id : str
+            The unique identifier (path) of the task. Example: 'f/public/databricks_describe_table'
+
+        arguments : typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]]
+            Arguments to pass to the task
+
+        task_type : typing.Optional[RunTaskRequestTaskType]
+            Type: 'script' or 'flow'
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[RunTaskResponse]
+            Successful Response
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/tasks/run",
+            method="POST",
+            json={
+                "arguments": arguments,
+                "task_id": task_id,
+                "task_type": task_type,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    RunTaskResponse,
+                    parse_obj_as(
+                        type_=RunTaskResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            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(),
+                        ),
+                    ),
+                )
+            _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.399/src/athena/tools/tasks/types/__init__.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+# isort: skip_file
+
+import typing
+from importlib import import_module
+
+if typing.TYPE_CHECKING:
+    from .run_task_request_task_type import RunTaskRequestTaskType
+_dynamic_imports: typing.Dict[str, str] = {"RunTaskRequestTaskType": ".run_task_request_task_type"}
+
+
+def __getattr__(attr_name: str) -> typing.Any:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(f"No {attr_name} found in _dynamic_imports for module name -> {__name__}")
+    try:
+        module = import_module(module_name, __package__)
+        if module_name == f".{attr_name}":
+            return module
+        else:
+            return getattr(module, attr_name)
+    except ImportError as e:
+        raise ImportError(f"Failed to import {attr_name} from {module_name}: {e}") from e
+    except AttributeError as e:
+        raise AttributeError(f"Failed to get {attr_name} from {module_name}: {e}") from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)
+
+
+__all__ = ["RunTaskRequestTaskType"]

athena_intelligence-0.1.399/src/athena/tools/tasks/types/run_task_request_task_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+RunTaskRequestTaskType = typing.Union[typing.Literal["script", "flow"], typing.Any]

{athena_intelligence-0.1.332 → athena_intelligence-0.1.399}/src/athena/types/__init__.py

@@ -62,6 +62,7 @@ if typing.TYPE_CHECKING:
     from .prompt_message import PromptMessage
     from .public_asset_out import PublicAssetOut
     from .research_agent_response import ResearchAgentResponse
+    from .run_task_response import RunTaskResponse
     from .save_asset_request_out import SaveAssetRequestOut
     from .sheet import Sheet
     from .sheet_operation_response import SheetOperationResponse
@@ -132,6 +133,7 @@ _dynamic_imports: typing.Dict[str, str] = {
     "PromptMessage": ".prompt_message",
     "PublicAssetOut": ".public_asset_out",
     "ResearchAgentResponse": ".research_agent_response",
+    "RunTaskResponse": ".run_task_response",
     "SaveAssetRequestOut": ".save_asset_request_out",
     "Sheet": ".sheet",
     "SheetOperationResponse": ".sheet_operation_response",
@@ -226,6 +228,7 @@ __all__ = [
     "PromptMessage",
     "PublicAssetOut",
     "ResearchAgentResponse",
+    "RunTaskResponse",
     "SaveAssetRequestOut",
     "Sheet",
     "SheetOperationResponse",

athena_intelligence-0.1.399/src/athena/types/run_task_response.py

@@ -0,0 +1,46 @@
+# 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 RunTaskResponse(UniversalBaseModel):
+    """
+    Response model for task execution.
+    """
+
+    error: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Error message if failed
+    """
+
+    error_type: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Error type: 'timeout', 'execution_error', 'not_found', 'validation_error'
+    """
+
+    job_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Windmill job ID for debugging
+    """
+
+    result: typing.Optional[typing.Optional[typing.Any]] = pydantic.Field(default=None)
+    """
+    The task result
+    """
+
+    success: bool = pydantic.Field()
+    """
+    Whether execution succeeded
+    """
+
+    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.332/src/athena/tools/tasks/__init__.py

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