athena-intelligence 0.1.94__tar.gz → 0.1.96__tar.gz

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/PKG-INFO

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

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/pyproject.toml

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

athena_intelligence-0.1.96/src/athena/__init__.py

@@ -0,0 +1,40 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .types import (
+    DataFrameParsingError,
+    DataFrameRequestOut,
+    DataFrameRequestOutColumnsItem,
+    DataFrameRequestOutDataItemItem,
+    DataFrameRequestOutIndexItem,
+    DataFrameUnknownFormatError,
+    FileFetchError,
+    HttpValidationError,
+    ValidationError,
+    ValidationErrorLocItem,
+)
+from .errors import InternalServerError, NotFoundError, UnprocessableEntityError, UnsupportedMediaTypeError
+from . import tools
+from .environment import AthenaEnvironment
+from .tools import ToolsDataFrameRequestColumnsItem
+from .version import __version__
+
+__all__ = [
+    "AthenaEnvironment",
+    "DataFrameParsingError",
+    "DataFrameRequestOut",
+    "DataFrameRequestOutColumnsItem",
+    "DataFrameRequestOutDataItemItem",
+    "DataFrameRequestOutIndexItem",
+    "DataFrameUnknownFormatError",
+    "FileFetchError",
+    "HttpValidationError",
+    "InternalServerError",
+    "NotFoundError",
+    "ToolsDataFrameRequestColumnsItem",
+    "UnprocessableEntityError",
+    "UnsupportedMediaTypeError",
+    "ValidationError",
+    "ValidationErrorLocItem",
+    "__version__",
+    "tools",
+]

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/src/athena/base_client.py

@@ -4,18 +4,9 @@ import typing
 
 import httpx
 
-from .chain.client import AsyncChainClient, ChainClient
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
-from .dataset.client import AsyncDatasetClient, DatasetClient
 from .environment import AthenaEnvironment
-from .message.client import AsyncMessageClient, MessageClient
-from .query.client import AsyncQueryClient, QueryClient
-from .report.client import AsyncReportClient, ReportClient
-from .search.client import AsyncSearchClient, SearchClient
-from .snippet.client import AsyncSnippetClient, SnippetClient
 from .tools.client import AsyncToolsClient, ToolsClient
-from .upload.client import AsyncUploadClient, UploadClient
-from .workflow.client import AsyncWorkflowClient, WorkflowClient
 
 
 class BaseAthena:
@@ -32,7 +23,7 @@ class BaseAthena:
 
 
 
-        Defaults to AthenaEnvironment.DEFAULT
+        Defaults to AthenaEnvironment.PRODUCTION
 
 
 
@@ -59,7 +50,7 @@ class BaseAthena:
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
         api_key: str,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -76,16 +67,7 @@ class BaseAthena:
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self.message = MessageClient(client_wrapper=self._client_wrapper)
-        self.dataset = DatasetClient(client_wrapper=self._client_wrapper)
-        self.snippet = SnippetClient(client_wrapper=self._client_wrapper)
-        self.report = ReportClient(client_wrapper=self._client_wrapper)
-        self.query = QueryClient(client_wrapper=self._client_wrapper)
-        self.search = SearchClient(client_wrapper=self._client_wrapper)
-        self.chain = ChainClient(client_wrapper=self._client_wrapper)
         self.tools = ToolsClient(client_wrapper=self._client_wrapper)
-        self.upload = UploadClient(client_wrapper=self._client_wrapper)
-        self.workflow = WorkflowClient(client_wrapper=self._client_wrapper)
 
 
 class AsyncBaseAthena:
@@ -102,7 +84,7 @@ class AsyncBaseAthena:
 
 
 
-        Defaults to AthenaEnvironment.DEFAULT
+        Defaults to AthenaEnvironment.PRODUCTION
 
 
 
@@ -129,7 +111,7 @@ class AsyncBaseAthena:
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
         api_key: str,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
@@ -146,16 +128,7 @@ class AsyncBaseAthena:
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
-        self.message = AsyncMessageClient(client_wrapper=self._client_wrapper)
-        self.dataset = AsyncDatasetClient(client_wrapper=self._client_wrapper)
-        self.snippet = AsyncSnippetClient(client_wrapper=self._client_wrapper)
-        self.report = AsyncReportClient(client_wrapper=self._client_wrapper)
-        self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
-        self.search = AsyncSearchClient(client_wrapper=self._client_wrapper)
-        self.chain = AsyncChainClient(client_wrapper=self._client_wrapper)
         self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
-        self.upload = AsyncUploadClient(client_wrapper=self._client_wrapper)
-        self.workflow = AsyncWorkflowClient(client_wrapper=self._client_wrapper)
 
 
 def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/src/athena/client.py

@@ -8,7 +8,6 @@ from typing_extensions import TypeVar, ParamSpec
 
 from .base_client import BaseAthena, AsyncBaseAthena
 from .environment import AthenaEnvironment
-from .polling_message_client import MessagePollingClient, AsyncMessagePollingClient
 from .tools.client import AsyncToolsClient, ToolsClient
 from .types.data_frame_request_out import DataFrameRequestOut
 
@@ -138,7 +137,7 @@ class Athena(BaseAthena):
 
 
 
-        Defaults to AthenaEnvironment.DEFAULT
+        Defaults to AthenaEnvironment.PRODUCTION
 
 
 
@@ -164,7 +163,7 @@ class Athena(BaseAthena):
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
         api_key: typing.Optional[str] = None,
         timeout: typing.Optional[float] = 60,
         httpx_client: typing.Optional[httpx.Client] = None
@@ -184,8 +183,6 @@ class Athena(BaseAthena):
             timeout=timeout,
             httpx_client=httpx_client,
         )
-        self.message = MessagePollingClient(
-            client_wrapper=self._client_wrapper)
         self.tools = WrappedToolsClient(client_wrapper=self._client_wrapper)
 
 
@@ -204,7 +201,7 @@ class AsyncAthena(AsyncBaseAthena):
 
 
 
-        Defaults to AthenaEnvironment.DEFAULT
+        Defaults to AthenaEnvironment.PRODUCTION
 
 
 
@@ -230,7 +227,7 @@ class AsyncAthena(AsyncBaseAthena):
         self,
         *,
         base_url: typing.Optional[str] = None,
-        environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
+        environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
         api_key: typing.Optional[str] = None,
         timeout: typing.Optional[float] = 60,
         httpx_client: typing.Optional[httpx.AsyncClient] = None
@@ -250,8 +247,6 @@ class AsyncAthena(AsyncBaseAthena):
             timeout=timeout,
             httpx_client=httpx_client,
         )
-        self.message = AsyncMessagePollingClient(
-            client_wrapper=self._client_wrapper)
         self.tools = WrappedAsyncToolsClient(client_wrapper=self._client_wrapper)
 
 

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/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.94",
+            "X-Fern-SDK-Version": "0.1.96",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/src/athena/environment.py

@@ -4,4 +4,4 @@ import enum
 
 
 class AthenaEnvironment(enum.Enum):
-    DEFAULT = "https://api.athenaintelligence.ai"
+    PRODUCTION = "https://api.athenaintel.com"

athena_intelligence-0.1.96/src/athena/tools/client.py

@@ -0,0 +1,299 @@
+# 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
+from ..errors.internal_server_error import InternalServerError
+from ..errors.not_found_error import NotFoundError
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..errors.unsupported_media_type_error import UnsupportedMediaTypeError
+from ..types.data_frame_parsing_error import DataFrameParsingError
+from ..types.data_frame_request_out import DataFrameRequestOut
+from ..types.data_frame_unknown_format_error import DataFrameUnknownFormatError
+from ..types.file_fetch_error import FileFetchError
+from ..types.http_validation_error import HttpValidationError
+from .types.tools_data_frame_request_columns_item import ToolsDataFrameRequestColumnsItem
+
+
+class ToolsClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def data_frame(
+        self,
+        *,
+        document_id: str,
+        row_limit: typing.Optional[int] = None,
+        index_column: typing.Optional[int] = None,
+        columns: typing.Optional[
+            typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
+        ] = None,
+        sheet_name: typing.Optional[str] = None,
+        separator: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Parameters
+        ----------
+        document_id : str
+
+        row_limit : typing.Optional[int]
+
+        index_column : typing.Optional[int]
+
+        columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
+            should be a list of strings or a list of integers
+
+        sheet_name : typing.Optional[str]
+            only for excel files
+
+        separator : typing.Optional[str]
+            only for csv files
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.data_frame(
+            document_id="document_id",
+        )
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "api/v0/tools/file/data-frame",
+            method="GET",
+            params={
+                "document_id": document_id,
+                "row_limit": row_limit,
+                "index_column": index_column,
+                "columns": columns,
+                "sheet_name": sheet_name,
+                "separator": separator,
+            },
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 415:
+            raise UnsupportedMediaTypeError(
+                pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _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)
+
+    def raw_data(
+        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.Iterator[bytes]:
+        """
+        Get the raw file data for given asset.
+
+        Parameters
+        ----------
+        document_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.Iterator[bytes]
+            Stream the file in original format.
+
+        Examples
+        --------
+        from athena.client import Athena
+
+        client = Athena(
+            api_key="YOUR_API_KEY",
+        )
+        client.tools.raw_data(
+            document_id="string",
+        )
+        """
+        with self._client_wrapper.httpx_client.stream(
+            "api/v0/tools/file/raw-data",
+            method="GET",
+            params={"document_id": document_id},
+            request_options=request_options,
+        ) as _response:
+            if 200 <= _response.status_code < 300:
+                for _chunk in _response.iter_bytes():
+                    yield _chunk
+                return
+            _response.read()
+            if _response.status_code == 404:
+                raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    pydantic_v1.parse_obj_as(HttpValidationError, _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 AsyncToolsClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def data_frame(
+        self,
+        *,
+        document_id: str,
+        row_limit: typing.Optional[int] = None,
+        index_column: typing.Optional[int] = None,
+        columns: typing.Optional[
+            typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]
+        ] = None,
+        sheet_name: typing.Optional[str] = None,
+        separator: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None
+    ) -> DataFrameRequestOut:
+        """
+        Parameters
+        ----------
+        document_id : str
+
+        row_limit : typing.Optional[int]
+
+        index_column : typing.Optional[int]
+
+        columns : typing.Optional[typing.Union[ToolsDataFrameRequestColumnsItem, typing.Sequence[ToolsDataFrameRequestColumnsItem]]]
+            should be a list of strings or a list of integers
+
+        sheet_name : typing.Optional[str]
+            only for excel files
+
+        separator : typing.Optional[str]
+            only for csv files
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DataFrameRequestOut
+            Successful Response
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.data_frame(
+            document_id="document_id",
+        )
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "api/v0/tools/file/data-frame",
+            method="GET",
+            params={
+                "document_id": document_id,
+                "row_limit": row_limit,
+                "index_column": index_column,
+                "columns": columns,
+                "sheet_name": sheet_name,
+                "separator": separator,
+            },
+            request_options=request_options,
+        )
+        if 200 <= _response.status_code < 300:
+            return pydantic_v1.parse_obj_as(DataFrameRequestOut, _response.json())  # type: ignore
+        if _response.status_code == 404:
+            raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+        if _response.status_code == 415:
+            raise UnsupportedMediaTypeError(
+                pydantic_v1.parse_obj_as(DataFrameUnknownFormatError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 422:
+            raise UnprocessableEntityError(
+                pydantic_v1.parse_obj_as(HttpValidationError, _response.json())  # type: ignore
+            )
+        if _response.status_code == 500:
+            raise InternalServerError(pydantic_v1.parse_obj_as(DataFrameParsingError, _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)
+
+    async def raw_data(
+        self, *, document_id: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.AsyncIterator[bytes]:
+        """
+        Get the raw file data for given asset.
+
+        Parameters
+        ----------
+        document_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Yields
+        ------
+        typing.AsyncIterator[bytes]
+            Stream the file in original format.
+
+        Examples
+        --------
+        from athena.client import AsyncAthena
+
+        client = AsyncAthena(
+            api_key="YOUR_API_KEY",
+        )
+        await client.tools.raw_data(
+            document_id="string",
+        )
+        """
+        async with self._client_wrapper.httpx_client.stream(
+            "api/v0/tools/file/raw-data",
+            method="GET",
+            params={"document_id": document_id},
+            request_options=request_options,
+        ) as _response:
+            if 200 <= _response.status_code < 300:
+                async for _chunk in _response.aiter_bytes():
+                    yield _chunk
+                return
+            await _response.aread()
+            if _response.status_code == 404:
+                raise NotFoundError(pydantic_v1.parse_obj_as(FileFetchError, _response.json()))  # type: ignore
+            if _response.status_code == 422:
+                raise UnprocessableEntityError(
+                    pydantic_v1.parse_obj_as(HttpValidationError, _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_intelligence-0.1.94 → athena_intelligence-0.1.96}/src/athena/tools/types/tools_data_frame_request_columns_item.py

@@ -2,4 +2,4 @@
 
 import typing
 
-ToolsDataFrameRequestColumnsItem = typing.Union[int, str]
+ToolsDataFrameRequestColumnsItem = typing.Union[str, int]

athena_intelligence-0.1.96/src/athena/types/__init__.py

@@ -0,0 +1,25 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from .data_frame_parsing_error import DataFrameParsingError
+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 .file_fetch_error import FileFetchError
+from .http_validation_error import HttpValidationError
+from .validation_error import ValidationError
+from .validation_error_loc_item import ValidationErrorLocItem
+
+__all__ = [
+    "DataFrameParsingError",
+    "DataFrameRequestOut",
+    "DataFrameRequestOutColumnsItem",
+    "DataFrameRequestOutDataItemItem",
+    "DataFrameRequestOutIndexItem",
+    "DataFrameUnknownFormatError",
+    "FileFetchError",
+    "HttpValidationError",
+    "ValidationError",
+    "ValidationErrorLocItem",
+]

{athena_intelligence-0.1.94 → athena_intelligence-0.1.96}/src/athena/types/data_frame_request_out.py

@@ -11,9 +11,13 @@ from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
 
 
 class DataFrameRequestOut(pydantic_v1.BaseModel):
-    columns: typing.List[DataFrameRequestOutColumnsItem]
-    index: typing.List[DataFrameRequestOutIndexItem]
-    data: typing.List[typing.List[DataFrameRequestOutDataItemItem]]
+    """
+    Response model with JSON dataframe representation.
+    """
+
+    columns: typing.List[typing.Optional[DataFrameRequestOutColumnsItem]]
+    data: typing.List[typing.List[typing.Optional[DataFrameRequestOutDataItemItem]]]
+    index: typing.List[typing.Optional[DataFrameRequestOutIndexItem]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}

athena_intelligence-0.1.94/src/athena/__init__.py

@@ -1,115 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from .types import (
-    AthenaDocumentV2Out,
-    ConvertPdfToSheetOut,
-    DataFrameParsingError,
-    DataFrameRequestOut,
-    DataFrameRequestOutColumnsItem,
-    DataFrameRequestOutDataItemItem,
-    DataFrameRequestOutIndexItem,
-    DataFrameUnknownFormatError,
-    Dataset,
-    Document,
-    ExcecuteToolFirstWorkflowOut,
-    FileDataResponse,
-    FileFetchError,
-    FilterModel,
-    FilterOperator,
-    FirecrawlScrapeUrlDataReponseDto,
-    FirecrawlScrapeUrlMetadata,
-    GetDatasetsResponse,
-    GetSnippetOut,
-    GetSnippetsResponse,
-    HttpValidationError,
-    LangchainDocumentsRequestOut,
-    LlmModel,
-    MapReduceChainOut,
-    MessageOut,
-    MessageOutDto,
-    Model,
-    PublishFormats,
-    QueryModel,
-    Report,
-    ResearcherOut,
-    SemanticQueryOut,
-    Snippet,
-    SqlResults,
-    StatusEnum,
-    StructuredParseResult,
-    TimeDimensionModel,
-    Tools,
-    UploadDocumentsOut,
-    UrlResult,
-    ValidationError,
-    ValidationErrorLocItem,
-    WorkflowStatusOut,
-)
-from .errors import InternalServerError, NotFoundError, UnprocessableEntityError, UnsupportedMediaTypeError
-from . import chain, dataset, message, query, report, search, snippet, tools, upload, workflow
-from .environment import AthenaEnvironment
-from .tools import ToolsDataFrameRequestColumnsItem
-from .version import __version__
-
-__all__ = [
-    "AthenaDocumentV2Out",
-    "AthenaEnvironment",
-    "ConvertPdfToSheetOut",
-    "DataFrameParsingError",
-    "DataFrameRequestOut",
-    "DataFrameRequestOutColumnsItem",
-    "DataFrameRequestOutDataItemItem",
-    "DataFrameRequestOutIndexItem",
-    "DataFrameUnknownFormatError",
-    "Dataset",
-    "Document",
-    "ExcecuteToolFirstWorkflowOut",
-    "FileDataResponse",
-    "FileFetchError",
-    "FilterModel",
-    "FilterOperator",
-    "FirecrawlScrapeUrlDataReponseDto",
-    "FirecrawlScrapeUrlMetadata",
-    "GetDatasetsResponse",
-    "GetSnippetOut",
-    "GetSnippetsResponse",
-    "HttpValidationError",
-    "InternalServerError",
-    "LangchainDocumentsRequestOut",
-    "LlmModel",
-    "MapReduceChainOut",
-    "MessageOut",
-    "MessageOutDto",
-    "Model",
-    "NotFoundError",
-    "PublishFormats",
-    "QueryModel",
-    "Report",
-    "ResearcherOut",
-    "SemanticQueryOut",
-    "Snippet",
-    "SqlResults",
-    "StatusEnum",
-    "StructuredParseResult",
-    "TimeDimensionModel",
-    "Tools",
-    "ToolsDataFrameRequestColumnsItem",
-    "UnprocessableEntityError",
-    "UnsupportedMediaTypeError",
-    "UploadDocumentsOut",
-    "UrlResult",
-    "ValidationError",
-    "ValidationErrorLocItem",
-    "WorkflowStatusOut",
-    "__version__",
-    "chain",
-    "dataset",
-    "message",
-    "query",
-    "report",
-    "search",
-    "snippet",
-    "tools",
-    "upload",
-    "workflow",
-]