athena-intelligence 0.1.83__py3-none-any.whl → 0.1.84__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 (52) hide show
  1. athena/chain/client.py +17 -8
  2. athena/client.py +91 -4
  3. athena/core/__init__.py +4 -1
  4. athena/core/client_wrapper.py +1 -1
  5. athena/core/pydantic_utilities.py +16 -0
  6. athena/core/query_encoder.py +33 -0
  7. athena/dataset/client.py +27 -22
  8. athena/message/client.py +17 -8
  9. athena/query/client.py +9 -4
  10. athena/report/client.py +9 -4
  11. athena/search/client.py +9 -4
  12. athena/snippet/client.py +35 -26
  13. athena/tools/client.py +107 -74
  14. athena/types/convert_pdf_to_sheet_out.py +7 -3
  15. athena/types/data_frame_parsing_error.py +7 -3
  16. athena/types/data_frame_request_out.py +7 -3
  17. athena/types/data_frame_unknown_format_error.py +7 -3
  18. athena/types/dataset.py +7 -3
  19. athena/types/document.py +7 -3
  20. athena/types/excecute_tool_first_workflow_out.py +7 -3
  21. athena/types/file_data_response.py +7 -3
  22. athena/types/file_fetch_error.py +7 -3
  23. athena/types/filter_model.py +7 -3
  24. athena/types/firecrawl_scrape_url_data_reponse_dto.py +7 -3
  25. athena/types/firecrawl_scrape_url_metadata.py +7 -3
  26. athena/types/get_datasets_response.py +7 -3
  27. athena/types/get_snippet_out.py +7 -3
  28. athena/types/get_snippets_response.py +7 -3
  29. athena/types/http_validation_error.py +7 -3
  30. athena/types/langchain_documents_request_out.py +7 -3
  31. athena/types/map_reduce_chain_out.py +7 -3
  32. athena/types/message_out.py +7 -3
  33. athena/types/message_out_dto.py +7 -3
  34. athena/types/publish_formats.py +7 -3
  35. athena/types/query_model.py +7 -3
  36. athena/types/report.py +7 -3
  37. athena/types/researcher_out.py +7 -3
  38. athena/types/semantic_query_out.py +7 -3
  39. athena/types/snippet.py +7 -3
  40. athena/types/sql_results.py +7 -3
  41. athena/types/structured_parse_result.py +7 -3
  42. athena/types/time_dimension_model.py +7 -3
  43. athena/types/upload_documents_out.py +7 -3
  44. athena/types/url_result.py +7 -3
  45. athena/types/validation_error.py +7 -3
  46. athena/types/workflow_status_out.py +7 -3
  47. athena/upload/client.py +9 -4
  48. athena/workflow/client.py +9 -4
  49. {athena_intelligence-0.1.83.dist-info → athena_intelligence-0.1.84.dist-info}/METADATA +12 -2
  50. athena_intelligence-0.1.84.dist-info/RECORD +91 -0
  51. athena_intelligence-0.1.83.dist-info/RECORD +0 -90
  52. {athena_intelligence-0.1.83.dist-info → athena_intelligence-0.1.84.dist-info}/WHEEL +0 -0
athena/tools/client.py CHANGED
@@ -8,6 +8,7 @@ from ..core.api_error import ApiError
8
8
  from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
9
9
  from ..core.jsonable_encoder import jsonable_encoder
10
10
  from ..core.pydantic_utilities import pydantic_v1
11
+ from ..core.query_encoder import encode_query
11
12
  from ..core.remove_none_from_dict import remove_none_from_dict
12
13
  from ..core.request_options import RequestOptions
13
14
  from ..errors.internal_server_error import InternalServerError
@@ -78,8 +79,10 @@ class ToolsClient:
78
79
  _response = self._client_wrapper.httpx_client.request(
79
80
  method="POST",
80
81
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/firecrawl/scrape-url"),
81
- params=jsonable_encoder(
82
- request_options.get("additional_query_parameters") if request_options is not None else None
82
+ params=encode_query(
83
+ jsonable_encoder(
84
+ request_options.get("additional_query_parameters") if request_options is not None else None
85
+ )
83
86
  ),
84
87
  json=jsonable_encoder(_request)
85
88
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -161,8 +164,10 @@ class ToolsClient:
161
164
  url=urllib.parse.urljoin(
162
165
  f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/langchain-documents"
163
166
  ),
164
- params=jsonable_encoder(
165
- request_options.get("additional_query_parameters") if request_options is not None else None
167
+ params=encode_query(
168
+ jsonable_encoder(
169
+ request_options.get("additional_query_parameters") if request_options is not None else None
170
+ )
166
171
  ),
167
172
  json=jsonable_encoder(_request)
168
173
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -249,21 +254,23 @@ class ToolsClient:
249
254
  _response = self._client_wrapper.httpx_client.request(
250
255
  method="GET",
251
256
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/data-frame"),
252
- params=jsonable_encoder(
253
- remove_none_from_dict(
254
- {
255
- "document_id": document_id,
256
- "row_limit": row_limit,
257
- "index_column": index_column,
258
- "columns": columns,
259
- "sheet_name": sheet_name,
260
- "separator": separator,
261
- **(
262
- request_options.get("additional_query_parameters", {})
263
- if request_options is not None
264
- else {}
265
- ),
266
- }
257
+ params=encode_query(
258
+ jsonable_encoder(
259
+ remove_none_from_dict(
260
+ {
261
+ "document_id": document_id,
262
+ "row_limit": row_limit,
263
+ "index_column": index_column,
264
+ "columns": columns,
265
+ "sheet_name": sheet_name,
266
+ "separator": separator,
267
+ **(
268
+ request_options.get("additional_query_parameters", {})
269
+ if request_options is not None
270
+ else {}
271
+ ),
272
+ }
273
+ )
267
274
  )
268
275
  ),
269
276
  headers=jsonable_encoder(
@@ -330,16 +337,18 @@ class ToolsClient:
330
337
  with self._client_wrapper.httpx_client.stream(
331
338
  method="GET",
332
339
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/raw-data"),
333
- params=jsonable_encoder(
334
- remove_none_from_dict(
335
- {
336
- "document_id": document_id,
337
- **(
338
- request_options.get("additional_query_parameters", {})
339
- if request_options is not None
340
- else {}
341
- ),
342
- }
340
+ params=encode_query(
341
+ jsonable_encoder(
342
+ remove_none_from_dict(
343
+ {
344
+ "document_id": document_id,
345
+ **(
346
+ request_options.get("additional_query_parameters", {})
347
+ if request_options is not None
348
+ else {}
349
+ ),
350
+ }
351
+ )
343
352
  )
344
353
  ),
345
354
  headers=jsonable_encoder(
@@ -424,8 +433,10 @@ class ToolsClient:
424
433
  _response = self._client_wrapper.httpx_client.request(
425
434
  method="POST",
426
435
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/first-agent"),
427
- params=jsonable_encoder(
428
- request_options.get("additional_query_parameters") if request_options is not None else None
436
+ params=encode_query(
437
+ jsonable_encoder(
438
+ request_options.get("additional_query_parameters") if request_options is not None else None
439
+ )
429
440
  ),
430
441
  json=jsonable_encoder(_request)
431
442
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -532,8 +543,10 @@ class ToolsClient:
532
543
  _response = self._client_wrapper.httpx_client.request(
533
544
  method="POST",
534
545
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/researcher"),
535
- params=jsonable_encoder(
536
- request_options.get("additional_query_parameters") if request_options is not None else None
546
+ params=encode_query(
547
+ jsonable_encoder(
548
+ request_options.get("additional_query_parameters") if request_options is not None else None
549
+ )
537
550
  ),
538
551
  json=jsonable_encoder(_request)
539
552
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -597,8 +610,10 @@ class ToolsClient:
597
610
  _response = self._client_wrapper.httpx_client.request(
598
611
  method="POST",
599
612
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/convert-pdf-to-sheet"),
600
- params=jsonable_encoder(
601
- request_options.get("additional_query_parameters") if request_options is not None else None
613
+ params=encode_query(
614
+ jsonable_encoder(
615
+ request_options.get("additional_query_parameters") if request_options is not None else None
616
+ )
602
617
  ),
603
618
  json=jsonable_encoder({"document_id": document_id})
604
619
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -689,8 +704,10 @@ class ToolsClient:
689
704
  _response = self._client_wrapper.httpx_client.request(
690
705
  method="POST",
691
706
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/strict-semantic-query"),
692
- params=jsonable_encoder(
693
- request_options.get("additional_query_parameters") if request_options is not None else None
707
+ params=encode_query(
708
+ jsonable_encoder(
709
+ request_options.get("additional_query_parameters") if request_options is not None else None
710
+ )
694
711
  ),
695
712
  json=jsonable_encoder(_request)
696
713
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -769,8 +786,10 @@ class AsyncToolsClient:
769
786
  _response = await self._client_wrapper.httpx_client.request(
770
787
  method="POST",
771
788
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/firecrawl/scrape-url"),
772
- params=jsonable_encoder(
773
- request_options.get("additional_query_parameters") if request_options is not None else None
789
+ params=encode_query(
790
+ jsonable_encoder(
791
+ request_options.get("additional_query_parameters") if request_options is not None else None
792
+ )
774
793
  ),
775
794
  json=jsonable_encoder(_request)
776
795
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -852,8 +871,10 @@ class AsyncToolsClient:
852
871
  url=urllib.parse.urljoin(
853
872
  f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/langchain-documents"
854
873
  ),
855
- params=jsonable_encoder(
856
- request_options.get("additional_query_parameters") if request_options is not None else None
874
+ params=encode_query(
875
+ jsonable_encoder(
876
+ request_options.get("additional_query_parameters") if request_options is not None else None
877
+ )
857
878
  ),
858
879
  json=jsonable_encoder(_request)
859
880
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -940,21 +961,23 @@ class AsyncToolsClient:
940
961
  _response = await self._client_wrapper.httpx_client.request(
941
962
  method="GET",
942
963
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/data-frame"),
943
- params=jsonable_encoder(
944
- remove_none_from_dict(
945
- {
946
- "document_id": document_id,
947
- "row_limit": row_limit,
948
- "index_column": index_column,
949
- "columns": columns,
950
- "sheet_name": sheet_name,
951
- "separator": separator,
952
- **(
953
- request_options.get("additional_query_parameters", {})
954
- if request_options is not None
955
- else {}
956
- ),
957
- }
964
+ params=encode_query(
965
+ jsonable_encoder(
966
+ remove_none_from_dict(
967
+ {
968
+ "document_id": document_id,
969
+ "row_limit": row_limit,
970
+ "index_column": index_column,
971
+ "columns": columns,
972
+ "sheet_name": sheet_name,
973
+ "separator": separator,
974
+ **(
975
+ request_options.get("additional_query_parameters", {})
976
+ if request_options is not None
977
+ else {}
978
+ ),
979
+ }
980
+ )
958
981
  )
959
982
  ),
960
983
  headers=jsonable_encoder(
@@ -1021,16 +1044,18 @@ class AsyncToolsClient:
1021
1044
  async with self._client_wrapper.httpx_client.stream(
1022
1045
  method="GET",
1023
1046
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/raw-data"),
1024
- params=jsonable_encoder(
1025
- remove_none_from_dict(
1026
- {
1027
- "document_id": document_id,
1028
- **(
1029
- request_options.get("additional_query_parameters", {})
1030
- if request_options is not None
1031
- else {}
1032
- ),
1033
- }
1047
+ params=encode_query(
1048
+ jsonable_encoder(
1049
+ remove_none_from_dict(
1050
+ {
1051
+ "document_id": document_id,
1052
+ **(
1053
+ request_options.get("additional_query_parameters", {})
1054
+ if request_options is not None
1055
+ else {}
1056
+ ),
1057
+ }
1058
+ )
1034
1059
  )
1035
1060
  ),
1036
1061
  headers=jsonable_encoder(
@@ -1115,8 +1140,10 @@ class AsyncToolsClient:
1115
1140
  _response = await self._client_wrapper.httpx_client.request(
1116
1141
  method="POST",
1117
1142
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/first-agent"),
1118
- params=jsonable_encoder(
1119
- request_options.get("additional_query_parameters") if request_options is not None else None
1143
+ params=encode_query(
1144
+ jsonable_encoder(
1145
+ request_options.get("additional_query_parameters") if request_options is not None else None
1146
+ )
1120
1147
  ),
1121
1148
  json=jsonable_encoder(_request)
1122
1149
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -1223,8 +1250,10 @@ class AsyncToolsClient:
1223
1250
  _response = await self._client_wrapper.httpx_client.request(
1224
1251
  method="POST",
1225
1252
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/researcher"),
1226
- params=jsonable_encoder(
1227
- request_options.get("additional_query_parameters") if request_options is not None else None
1253
+ params=encode_query(
1254
+ jsonable_encoder(
1255
+ request_options.get("additional_query_parameters") if request_options is not None else None
1256
+ )
1228
1257
  ),
1229
1258
  json=jsonable_encoder(_request)
1230
1259
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -1288,8 +1317,10 @@ class AsyncToolsClient:
1288
1317
  _response = await self._client_wrapper.httpx_client.request(
1289
1318
  method="POST",
1290
1319
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/convert-pdf-to-sheet"),
1291
- params=jsonable_encoder(
1292
- request_options.get("additional_query_parameters") if request_options is not None else None
1320
+ params=encode_query(
1321
+ jsonable_encoder(
1322
+ request_options.get("additional_query_parameters") if request_options is not None else None
1323
+ )
1293
1324
  ),
1294
1325
  json=jsonable_encoder({"document_id": document_id})
1295
1326
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -1380,8 +1411,10 @@ class AsyncToolsClient:
1380
1411
  _response = await self._client_wrapper.httpx_client.request(
1381
1412
  method="POST",
1382
1413
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/strict-semantic-query"),
1383
- params=jsonable_encoder(
1384
- request_options.get("additional_query_parameters") if request_options is not None else None
1414
+ params=encode_query(
1415
+ jsonable_encoder(
1416
+ request_options.get("additional_query_parameters") if request_options is not None else None
1417
+ )
1385
1418
  ),
1386
1419
  json=jsonable_encoder(_request)
1387
1420
  if request_options is None or request_options.get("additional_body_parameters") is None
athena/types/convert_pdf_to_sheet_out.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class ConvertPdfToSheetOut(pydantic_v1.BaseModel):
@@ -16,8 +16,12 @@ class ConvertPdfToSheetOut(pydantic_v1.BaseModel):
16
16
  return super().json(**kwargs_with_defaults)
17
17
 
18
18
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
19
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
- return super().dict(**kwargs_with_defaults)
19
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
21
+
22
+ return deep_union_pydantic_dicts(
23
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
24
+ )
21
25
 
22
26
  class Config:
23
27
  frozen = True
athena/types/data_frame_parsing_error.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class DataFrameParsingError(pydantic_v1.BaseModel):
@@ -16,8 +16,12 @@ class DataFrameParsingError(pydantic_v1.BaseModel):
16
16
  return super().json(**kwargs_with_defaults)
17
17
 
18
18
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
19
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
- return super().dict(**kwargs_with_defaults)
19
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
21
+
22
+ return deep_union_pydantic_dicts(
23
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
24
+ )
21
25
 
22
26
  class Config:
23
27
  frozen = True
athena/types/data_frame_request_out.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .data_frame_request_out_columns_item import DataFrameRequestOutColumnsItem
9
9
  from .data_frame_request_out_data_item_item import DataFrameRequestOutDataItemItem
10
10
  from .data_frame_request_out_index_item import DataFrameRequestOutIndexItem
@@ -20,8 +20,12 @@ class DataFrameRequestOut(pydantic_v1.BaseModel):
20
20
  return super().json(**kwargs_with_defaults)
21
21
 
22
22
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
23
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
24
- return super().dict(**kwargs_with_defaults)
23
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
24
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
25
+
26
+ return deep_union_pydantic_dicts(
27
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
28
+ )
25
29
 
26
30
  class Config:
27
31
  frozen = True
athena/types/data_frame_unknown_format_error.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class DataFrameUnknownFormatError(pydantic_v1.BaseModel):
@@ -17,8 +17,12 @@ class DataFrameUnknownFormatError(pydantic_v1.BaseModel):
17
17
  return super().json(**kwargs_with_defaults)
18
18
 
19
19
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
20
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
21
- return super().dict(**kwargs_with_defaults)
20
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
21
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
22
+
23
+ return deep_union_pydantic_dicts(
24
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
25
+ )
22
26
 
23
27
  class Config:
24
28
  frozen = True
athena/types/dataset.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class Dataset(pydantic_v1.BaseModel):
@@ -19,8 +19,12 @@ class Dataset(pydantic_v1.BaseModel):
19
19
  return super().json(**kwargs_with_defaults)
20
20
 
21
21
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
22
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
23
- return super().dict(**kwargs_with_defaults)
22
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
23
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
24
+
25
+ return deep_union_pydantic_dicts(
26
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
27
+ )
24
28
 
25
29
  class Config:
26
30
  frozen = True
athena/types/document.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class Document(pydantic_v1.BaseModel):
@@ -21,8 +21,12 @@ class Document(pydantic_v1.BaseModel):
21
21
  return super().json(**kwargs_with_defaults)
22
22
 
23
23
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
24
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
25
- return super().dict(**kwargs_with_defaults)
24
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
25
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
26
+
27
+ return deep_union_pydantic_dicts(
28
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
29
+ )
26
30
 
27
31
  class Config:
28
32
  frozen = True
athena/types/excecute_tool_first_workflow_out.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class ExcecuteToolFirstWorkflowOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class ExcecuteToolFirstWorkflowOut(pydantic_v1.BaseModel):
15
15
  return super().json(**kwargs_with_defaults)
16
16
 
17
17
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
18
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
19
- return super().dict(**kwargs_with_defaults)
18
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
19
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
20
+
21
+ return deep_union_pydantic_dicts(
22
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
23
+ )
20
24
 
21
25
  class Config:
22
26
  frozen = True
athena/types/file_data_response.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class FileDataResponse(pydantic_v1.BaseModel):
@@ -22,8 +22,12 @@ class FileDataResponse(pydantic_v1.BaseModel):
22
22
  return super().json(**kwargs_with_defaults)
23
23
 
24
24
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
25
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
26
- return super().dict(**kwargs_with_defaults)
25
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
26
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
27
+
28
+ return deep_union_pydantic_dicts(
29
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
30
+ )
27
31
 
28
32
  class Config:
29
33
  frozen = True
athena/types/file_fetch_error.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class FileFetchError(pydantic_v1.BaseModel):
@@ -16,8 +16,12 @@ class FileFetchError(pydantic_v1.BaseModel):
16
16
  return super().json(**kwargs_with_defaults)
17
17
 
18
18
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
19
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
- return super().dict(**kwargs_with_defaults)
19
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
20
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
21
+
22
+ return deep_union_pydantic_dicts(
23
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
24
+ )
21
25
 
22
26
  class Config:
23
27
  frozen = True
athena/types/filter_model.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .filter_operator import FilterOperator
9
9
 
10
10
 
@@ -18,8 +18,12 @@ class FilterModel(pydantic_v1.BaseModel):
18
18
  return super().json(**kwargs_with_defaults)
19
19
 
20
20
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
21
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
- return super().dict(**kwargs_with_defaults)
21
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
23
+
24
+ return deep_union_pydantic_dicts(
25
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
26
+ )
23
27
 
24
28
  class Config:
25
29
  frozen = True
athena/types/firecrawl_scrape_url_data_reponse_dto.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .firecrawl_scrape_url_metadata import FirecrawlScrapeUrlMetadata
9
9
 
10
10
 
@@ -18,8 +18,12 @@ class FirecrawlScrapeUrlDataReponseDto(pydantic_v1.BaseModel):
18
18
  return super().json(**kwargs_with_defaults)
19
19
 
20
20
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
21
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
- return super().dict(**kwargs_with_defaults)
21
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
23
+
24
+ return deep_union_pydantic_dicts(
25
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
26
+ )
23
27
 
24
28
  class Config:
25
29
  frozen = True
athena/types/firecrawl_scrape_url_metadata.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class FirecrawlScrapeUrlMetadata(pydantic_v1.BaseModel):
@@ -18,8 +18,12 @@ class FirecrawlScrapeUrlMetadata(pydantic_v1.BaseModel):
18
18
  return super().json(**kwargs_with_defaults)
19
19
 
20
20
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
21
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
- return super().dict(**kwargs_with_defaults)
21
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
22
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
23
+
24
+ return deep_union_pydantic_dicts(
25
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
26
+ )
23
27
 
24
28
  class Config:
25
29
  frozen = True
athena/types/get_datasets_response.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
  from .dataset import Dataset
9
9
 
10
10
 
@@ -20,8 +20,12 @@ class GetDatasetsResponse(pydantic_v1.BaseModel):
20
20
  return super().json(**kwargs_with_defaults)
21
21
 
22
22
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
23
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
24
- return super().dict(**kwargs_with_defaults)
23
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
24
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
25
+
26
+ return deep_union_pydantic_dicts(
27
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
28
+ )
25
29
 
26
30
  class Config:
27
31
  frozen = True
athena/types/get_snippet_out.py CHANGED
@@ -4,7 +4,7 @@ import datetime as dt
4
4
  import typing
5
5
 
6
6
  from ..core.datetime_utils import serialize_datetime
7
- from ..core.pydantic_utilities import pydantic_v1
7
+ from ..core.pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
8
8
 
9
9
 
10
10
  class GetSnippetOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class GetSnippetOut(pydantic_v1.BaseModel):
15
15
  return super().json(**kwargs_with_defaults)
16
16
 
17
17
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
18
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
19
- return super().dict(**kwargs_with_defaults)
18
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
19
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
20
+
21
+ return deep_union_pydantic_dicts(
22
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
23
+ )
20
24
 
21
25
  class Config:
22
26
  frozen = True