athena-intelligence 0.1.94__py3-none-any.whl → 0.1.96__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (67) hide show
  1. athena/__init__.py +1 -76
  2. athena/base_client.py +4 -31
  3. athena/client.py +4 -9
  4. athena/core/client_wrapper.py +1 -1
  5. athena/environment.py +1 -1
  6. athena/tools/client.py +61 -916
  7. athena/tools/types/tools_data_frame_request_columns_item.py +1 -1
  8. athena/types/__init__.py +0 -66
  9. athena/types/data_frame_parsing_error.py +1 -1
  10. athena/types/data_frame_request_out.py +7 -3
  11. athena/types/data_frame_unknown_format_error.py +1 -1
  12. {athena_intelligence-0.1.94.dist-info → athena_intelligence-0.1.96.dist-info}/METADATA +1 -1
  13. athena_intelligence-0.1.96.dist-info/RECORD +40 -0
  14. athena/chain/__init__.py +0 -2
  15. athena/chain/client.py +0 -302
  16. athena/dataset/__init__.py +0 -2
  17. athena/dataset/client.py +0 -124
  18. athena/message/__init__.py +0 -2
  19. athena/message/client.py +0 -249
  20. athena/polling_message_client.py +0 -125
  21. athena/query/__init__.py +0 -2
  22. athena/query/client.py +0 -123
  23. athena/report/__init__.py +0 -2
  24. athena/report/client.py +0 -125
  25. athena/search/__init__.py +0 -2
  26. athena/search/client.py +0 -171
  27. athena/snippet/__init__.py +0 -2
  28. athena/snippet/client.py +0 -208
  29. athena/types/athena_document_v_2_out.py +0 -29
  30. athena/types/convert_pdf_to_sheet_out.py +0 -30
  31. athena/types/dataset.py +0 -33
  32. athena/types/document.py +0 -47
  33. athena/types/excecute_tool_first_workflow_out.py +0 -29
  34. athena/types/file_data_response.py +0 -36
  35. athena/types/filter_model.py +0 -32
  36. athena/types/filter_operator.py +0 -73
  37. athena/types/firecrawl_scrape_url_data_reponse_dto.py +0 -32
  38. athena/types/firecrawl_scrape_url_metadata.py +0 -34
  39. athena/types/get_datasets_response.py +0 -34
  40. athena/types/get_snippet_out.py +0 -29
  41. athena/types/get_snippets_response.py +0 -34
  42. athena/types/langchain_documents_request_out.py +0 -29
  43. athena/types/llm_model.py +0 -141
  44. athena/types/map_reduce_chain_out.py +0 -32
  45. athena/types/message_out.py +0 -29
  46. athena/types/message_out_dto.py +0 -35
  47. athena/types/model.py +0 -93
  48. athena/types/publish_formats.py +0 -31
  49. athena/types/query_model.py +0 -40
  50. athena/types/report.py +0 -32
  51. athena/types/researcher_out.py +0 -29
  52. athena/types/semantic_query_out.py +0 -29
  53. athena/types/snippet.py +0 -35
  54. athena/types/sql_results.py +0 -29
  55. athena/types/status_enum.py +0 -29
  56. athena/types/structured_parse_result.py +0 -29
  57. athena/types/time_dimension_model.py +0 -33
  58. athena/types/tools.py +0 -57
  59. athena/types/upload_documents_out.py +0 -31
  60. athena/types/url_result.py +0 -29
  61. athena/types/workflow_status_out.py +0 -31
  62. athena/upload/__init__.py +0 -2
  63. athena/upload/client.py +0 -108
  64. athena/workflow/__init__.py +0 -2
  65. athena/workflow/client.py +0 -117
  66. athena_intelligence-0.1.94.dist-info/RECORD +0 -92
  67. {athena_intelligence-0.1.94.dist-info → athena_intelligence-0.1.96.dist-info}/WHEEL +0 -0
athena/__init__.py CHANGED
@@ -1,115 +1,40 @@
1
1
  # This file was auto-generated by Fern from our API Definition.
2
2
 
3
3
  from .types import (
4
- AthenaDocumentV2Out,
5
- ConvertPdfToSheetOut,
6
4
  DataFrameParsingError,
7
5
  DataFrameRequestOut,
8
6
  DataFrameRequestOutColumnsItem,
9
7
  DataFrameRequestOutDataItemItem,
10
8
  DataFrameRequestOutIndexItem,
11
9
  DataFrameUnknownFormatError,
12
- Dataset,
13
- Document,
14
- ExcecuteToolFirstWorkflowOut,
15
- FileDataResponse,
16
10
  FileFetchError,
17
- FilterModel,
18
- FilterOperator,
19
- FirecrawlScrapeUrlDataReponseDto,
20
- FirecrawlScrapeUrlMetadata,
21
- GetDatasetsResponse,
22
- GetSnippetOut,
23
- GetSnippetsResponse,
24
11
  HttpValidationError,
25
- LangchainDocumentsRequestOut,
26
- LlmModel,
27
- MapReduceChainOut,
28
- MessageOut,
29
- MessageOutDto,
30
- Model,
31
- PublishFormats,
32
- QueryModel,
33
- Report,
34
- ResearcherOut,
35
- SemanticQueryOut,
36
- Snippet,
37
- SqlResults,
38
- StatusEnum,
39
- StructuredParseResult,
40
- TimeDimensionModel,
41
- Tools,
42
- UploadDocumentsOut,
43
- UrlResult,
44
12
  ValidationError,
45
13
  ValidationErrorLocItem,
46
- WorkflowStatusOut,
47
14
  )
48
15
  from .errors import InternalServerError, NotFoundError, UnprocessableEntityError, UnsupportedMediaTypeError
49
- from . import chain, dataset, message, query, report, search, snippet, tools, upload, workflow
16
+ from . import tools
50
17
  from .environment import AthenaEnvironment
51
18
  from .tools import ToolsDataFrameRequestColumnsItem
52
19
  from .version import __version__
53
20
 
54
21
  __all__ = [
55
- "AthenaDocumentV2Out",
56
22
  "AthenaEnvironment",
57
- "ConvertPdfToSheetOut",
58
23
  "DataFrameParsingError",
59
24
  "DataFrameRequestOut",
60
25
  "DataFrameRequestOutColumnsItem",
61
26
  "DataFrameRequestOutDataItemItem",
62
27
  "DataFrameRequestOutIndexItem",
63
28
  "DataFrameUnknownFormatError",
64
- "Dataset",
65
- "Document",
66
- "ExcecuteToolFirstWorkflowOut",
67
- "FileDataResponse",
68
29
  "FileFetchError",
69
- "FilterModel",
70
- "FilterOperator",
71
- "FirecrawlScrapeUrlDataReponseDto",
72
- "FirecrawlScrapeUrlMetadata",
73
- "GetDatasetsResponse",
74
- "GetSnippetOut",
75
- "GetSnippetsResponse",
76
30
  "HttpValidationError",
77
31
  "InternalServerError",
78
- "LangchainDocumentsRequestOut",
79
- "LlmModel",
80
- "MapReduceChainOut",
81
- "MessageOut",
82
- "MessageOutDto",
83
- "Model",
84
32
  "NotFoundError",
85
- "PublishFormats",
86
- "QueryModel",
87
- "Report",
88
- "ResearcherOut",
89
- "SemanticQueryOut",
90
- "Snippet",
91
- "SqlResults",
92
- "StatusEnum",
93
- "StructuredParseResult",
94
- "TimeDimensionModel",
95
- "Tools",
96
33
  "ToolsDataFrameRequestColumnsItem",
97
34
  "UnprocessableEntityError",
98
35
  "UnsupportedMediaTypeError",
99
- "UploadDocumentsOut",
100
- "UrlResult",
101
36
  "ValidationError",
102
37
  "ValidationErrorLocItem",
103
- "WorkflowStatusOut",
104
38
  "__version__",
105
- "chain",
106
- "dataset",
107
- "message",
108
- "query",
109
- "report",
110
- "search",
111
- "snippet",
112
39
  "tools",
113
- "upload",
114
- "workflow",
115
40
  ]
athena/base_client.py CHANGED
@@ -4,18 +4,9 @@ import typing
4
4
 
5
5
  import httpx
6
6
 
7
- from .chain.client import AsyncChainClient, ChainClient
8
7
  from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
9
- from .dataset.client import AsyncDatasetClient, DatasetClient
10
8
  from .environment import AthenaEnvironment
11
- from .message.client import AsyncMessageClient, MessageClient
12
- from .query.client import AsyncQueryClient, QueryClient
13
- from .report.client import AsyncReportClient, ReportClient
14
- from .search.client import AsyncSearchClient, SearchClient
15
- from .snippet.client import AsyncSnippetClient, SnippetClient
16
9
  from .tools.client import AsyncToolsClient, ToolsClient
17
- from .upload.client import AsyncUploadClient, UploadClient
18
- from .workflow.client import AsyncWorkflowClient, WorkflowClient
19
10
 
20
11
 
21
12
  class BaseAthena:
@@ -32,7 +23,7 @@ class BaseAthena:
32
23
 
33
24
 
34
25
 
35
- Defaults to AthenaEnvironment.DEFAULT
26
+ Defaults to AthenaEnvironment.PRODUCTION
36
27
 
37
28
 
38
29
 
@@ -59,7 +50,7 @@ class BaseAthena:
59
50
  self,
60
51
  *,
61
52
  base_url: typing.Optional[str] = None,
62
- environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
53
+ environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
63
54
  api_key: str,
64
55
  timeout: typing.Optional[float] = None,
65
56
  follow_redirects: typing.Optional[bool] = True,
@@ -76,16 +67,7 @@ class BaseAthena:
76
67
  else httpx.Client(timeout=_defaulted_timeout),
77
68
  timeout=_defaulted_timeout,
78
69
  )
79
- self.message = MessageClient(client_wrapper=self._client_wrapper)
80
- self.dataset = DatasetClient(client_wrapper=self._client_wrapper)
81
- self.snippet = SnippetClient(client_wrapper=self._client_wrapper)
82
- self.report = ReportClient(client_wrapper=self._client_wrapper)
83
- self.query = QueryClient(client_wrapper=self._client_wrapper)
84
- self.search = SearchClient(client_wrapper=self._client_wrapper)
85
- self.chain = ChainClient(client_wrapper=self._client_wrapper)
86
70
  self.tools = ToolsClient(client_wrapper=self._client_wrapper)
87
- self.upload = UploadClient(client_wrapper=self._client_wrapper)
88
- self.workflow = WorkflowClient(client_wrapper=self._client_wrapper)
89
71
 
90
72
 
91
73
  class AsyncBaseAthena:
@@ -102,7 +84,7 @@ class AsyncBaseAthena:
102
84
 
103
85
 
104
86
 
105
- Defaults to AthenaEnvironment.DEFAULT
87
+ Defaults to AthenaEnvironment.PRODUCTION
106
88
 
107
89
 
108
90
 
@@ -129,7 +111,7 @@ class AsyncBaseAthena:
129
111
  self,
130
112
  *,
131
113
  base_url: typing.Optional[str] = None,
132
- environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
114
+ environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
133
115
  api_key: str,
134
116
  timeout: typing.Optional[float] = None,
135
117
  follow_redirects: typing.Optional[bool] = True,
@@ -146,16 +128,7 @@ class AsyncBaseAthena:
146
128
  else httpx.AsyncClient(timeout=_defaulted_timeout),
147
129
  timeout=_defaulted_timeout,
148
130
  )
149
- self.message = AsyncMessageClient(client_wrapper=self._client_wrapper)
150
- self.dataset = AsyncDatasetClient(client_wrapper=self._client_wrapper)
151
- self.snippet = AsyncSnippetClient(client_wrapper=self._client_wrapper)
152
- self.report = AsyncReportClient(client_wrapper=self._client_wrapper)
153
- self.query = AsyncQueryClient(client_wrapper=self._client_wrapper)
154
- self.search = AsyncSearchClient(client_wrapper=self._client_wrapper)
155
- self.chain = AsyncChainClient(client_wrapper=self._client_wrapper)
156
131
  self.tools = AsyncToolsClient(client_wrapper=self._client_wrapper)
157
- self.upload = AsyncUploadClient(client_wrapper=self._client_wrapper)
158
- self.workflow = AsyncWorkflowClient(client_wrapper=self._client_wrapper)
159
132
 
160
133
 
161
134
  def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AthenaEnvironment) -> str:
athena/client.py CHANGED
@@ -8,7 +8,6 @@ from typing_extensions import TypeVar, ParamSpec
8
8
 
9
9
  from .base_client import BaseAthena, AsyncBaseAthena
10
10
  from .environment import AthenaEnvironment
11
- from .polling_message_client import MessagePollingClient, AsyncMessagePollingClient
12
11
  from .tools.client import AsyncToolsClient, ToolsClient
13
12
  from .types.data_frame_request_out import DataFrameRequestOut
14
13
 
@@ -138,7 +137,7 @@ class Athena(BaseAthena):
138
137
 
139
138
 
140
139
 
141
- Defaults to AthenaEnvironment.DEFAULT
140
+ Defaults to AthenaEnvironment.PRODUCTION
142
141
 
143
142
 
144
143
 
@@ -164,7 +163,7 @@ class Athena(BaseAthena):
164
163
  self,
165
164
  *,
166
165
  base_url: typing.Optional[str] = None,
167
- environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
166
+ environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
168
167
  api_key: typing.Optional[str] = None,
169
168
  timeout: typing.Optional[float] = 60,
170
169
  httpx_client: typing.Optional[httpx.Client] = None
@@ -184,8 +183,6 @@ class Athena(BaseAthena):
184
183
  timeout=timeout,
185
184
  httpx_client=httpx_client,
186
185
  )
187
- self.message = MessagePollingClient(
188
- client_wrapper=self._client_wrapper)
189
186
  self.tools = WrappedToolsClient(client_wrapper=self._client_wrapper)
190
187
 
191
188
 
@@ -204,7 +201,7 @@ class AsyncAthena(AsyncBaseAthena):
204
201
 
205
202
 
206
203
 
207
- Defaults to AthenaEnvironment.DEFAULT
204
+ Defaults to AthenaEnvironment.PRODUCTION
208
205
 
209
206
 
210
207
 
@@ -230,7 +227,7 @@ class AsyncAthena(AsyncBaseAthena):
230
227
  self,
231
228
  *,
232
229
  base_url: typing.Optional[str] = None,
233
- environment: AthenaEnvironment = AthenaEnvironment.DEFAULT,
230
+ environment: AthenaEnvironment = AthenaEnvironment.PRODUCTION,
234
231
  api_key: typing.Optional[str] = None,
235
232
  timeout: typing.Optional[float] = 60,
236
233
  httpx_client: typing.Optional[httpx.AsyncClient] = None
@@ -250,8 +247,6 @@ class AsyncAthena(AsyncBaseAthena):
250
247
  timeout=timeout,
251
248
  httpx_client=httpx_client,
252
249
  )
253
- self.message = AsyncMessagePollingClient(
254
- client_wrapper=self._client_wrapper)
255
250
  self.tools = WrappedAsyncToolsClient(client_wrapper=self._client_wrapper)
256
251
 
257
252
 
athena/core/client_wrapper.py CHANGED
@@ -17,7 +17,7 @@ class BaseClientWrapper:
17
17
  headers: typing.Dict[str, str] = {
18
18
  "X-Fern-Language": "Python",
19
19
  "X-Fern-SDK-Name": "athena-intelligence",
20
- "X-Fern-SDK-Version": "0.1.94",
20
+ "X-Fern-SDK-Version": "0.1.96",
21
21
  }
22
22
  headers["X-API-KEY"] = self.api_key
23
23
  return headers
athena/environment.py CHANGED
@@ -4,4 +4,4 @@ import enum
4
4
 
5
5
 
6
6
  class AthenaEnvironment(enum.Enum):
7
- DEFAULT = "https://api.athenaintelligence.ai"
7
+ PRODUCTION = "https://api.athenaintel.com"