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
@@ -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 .snippet import Snippet
9
9
 
10
10
 
@@ -20,8 +20,12 @@ class GetSnippetsResponse(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
@@ -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 .validation_error import ValidationError
9
9
 
10
10
 
@@ -16,8 +16,12 @@ class HttpValidationError(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
@@ -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 LangchainDocumentsRequestOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class LangchainDocumentsRequestOut(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
@@ -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 .document import Document
9
9
 
10
10
 
@@ -18,8 +18,12 @@ class MapReduceChainOut(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
@@ -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 MessageOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class MessageOut(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
@@ -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 .status_enum import StatusEnum
9
9
 
10
10
 
@@ -21,8 +21,12 @@ class MessageOutDto(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
@@ -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 PublishFormats(pydantic_v1.BaseModel):
@@ -17,8 +17,12 @@ class PublishFormats(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
@@ -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_model import FilterModel
9
9
  from .time_dimension_model import TimeDimensionModel
10
10
 
@@ -23,8 +23,12 @@ class QueryModel(pydantic_v1.BaseModel):
23
23
  return super().json(**kwargs_with_defaults)
24
24
 
25
25
  def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
26
- kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
- return super().dict(**kwargs_with_defaults)
26
+ kwargs_with_defaults_exclude_unset: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
27
+ kwargs_with_defaults_exclude_none: typing.Any = {"by_alias": True, "exclude_none": True, **kwargs}
28
+
29
+ return deep_union_pydantic_dicts(
30
+ super().dict(**kwargs_with_defaults_exclude_unset), super().dict(**kwargs_with_defaults_exclude_none)
31
+ )
28
32
 
29
33
  class Config:
30
34
  frozen = True
athena/types/report.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 Report(pydantic_v1.BaseModel):
@@ -18,8 +18,12 @@ class Report(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
@@ -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 ResearcherOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class ResearcherOut(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
@@ -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 SemanticQueryOut(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class SemanticQueryOut(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/snippet.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 Snippet(pydantic_v1.BaseModel):
@@ -21,8 +21,12 @@ class Snippet(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
@@ -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 SqlResults(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class SqlResults(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
@@ -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 StructuredParseResult(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class StructuredParseResult(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
@@ -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 TimeDimensionModel(pydantic_v1.BaseModel):
@@ -17,8 +17,12 @@ class TimeDimensionModel(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
@@ -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 .file_data_response import FileDataResponse
9
9
 
10
10
 
@@ -17,8 +17,12 @@ class UploadDocumentsOut(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
@@ -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 UrlResult(pydantic_v1.BaseModel):
@@ -15,8 +15,12 @@ class UrlResult(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
@@ -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 .validation_error_loc_item import ValidationErrorLocItem
9
9
 
10
10
 
@@ -18,8 +18,12 @@ class ValidationError(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
@@ -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 WorkflowStatusOut(pydantic_v1.BaseModel):
@@ -17,8 +17,12 @@ class WorkflowStatusOut(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/upload/client.py CHANGED
@@ -9,6 +9,7 @@ from ..core.api_error import ApiError
9
9
  from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
10
10
  from ..core.jsonable_encoder import jsonable_encoder
11
11
  from ..core.pydantic_utilities import pydantic_v1
12
+ from ..core.query_encoder import encode_query
12
13
  from ..core.remove_none_from_dict import remove_none_from_dict
13
14
  from ..core.request_options import RequestOptions
14
15
  from ..errors.unprocessable_entity_error import UnprocessableEntityError
@@ -52,8 +53,10 @@ class UploadClient:
52
53
  _response = self._client_wrapper.httpx_client.request(
53
54
  method="POST",
54
55
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/upload"),
55
- params=jsonable_encoder(
56
- request_options.get("additional_query_parameters") if request_options is not None else None
56
+ params=encode_query(
57
+ jsonable_encoder(
58
+ request_options.get("additional_query_parameters") if request_options is not None else None
59
+ )
57
60
  ),
58
61
  data=jsonable_encoder(remove_none_from_dict({}))
59
62
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -122,8 +125,10 @@ class AsyncUploadClient:
122
125
  _response = await self._client_wrapper.httpx_client.request(
123
126
  method="POST",
124
127
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/upload"),
125
- params=jsonable_encoder(
126
- request_options.get("additional_query_parameters") if request_options is not None else None
128
+ params=encode_query(
129
+ jsonable_encoder(
130
+ request_options.get("additional_query_parameters") if request_options is not None else None
131
+ )
127
132
  ),
128
133
  data=jsonable_encoder(remove_none_from_dict({}))
129
134
  if request_options is None or request_options.get("additional_body_parameters") is None
athena/workflow/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.unprocessable_entity_error import UnprocessableEntityError
@@ -52,8 +53,10 @@ class WorkflowClient:
52
53
  _response = self._client_wrapper.httpx_client.request(
53
54
  method="POST",
54
55
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/workflow/status"),
55
- params=jsonable_encoder(
56
- request_options.get("additional_query_parameters") if request_options is not None else None
56
+ params=encode_query(
57
+ jsonable_encoder(
58
+ request_options.get("additional_query_parameters") if request_options is not None else None
59
+ )
57
60
  ),
58
61
  json=jsonable_encoder({"workflow_id": workflow_id})
59
62
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -122,8 +125,10 @@ class AsyncWorkflowClient:
122
125
  _response = await self._client_wrapper.httpx_client.request(
123
126
  method="POST",
124
127
  url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/workflow/status"),
125
- params=jsonable_encoder(
126
- request_options.get("additional_query_parameters") if request_options is not None else None
128
+ params=encode_query(
129
+ jsonable_encoder(
130
+ request_options.get("additional_query_parameters") if request_options is not None else None
131
+ )
127
132
  ),
128
133
  json=jsonable_encoder({"workflow_id": workflow_id})
129
134
  if request_options is None or request_options.get("additional_body_parameters") is None
@@ -1,13 +1,23 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: athena-intelligence
3
- Version: 0.1.83
4
- Summary:
3
+ Version: 0.1.84
4
+ Summary: Athena Intelligence Python Library
5
5
  Requires-Python: >=3.8,<4.0
6
+ Classifier: Intended Audience :: Developers
7
+ Classifier: Operating System :: MacOS
8
+ Classifier: Operating System :: Microsoft :: Windows
9
+ Classifier: Operating System :: OS Independent
10
+ Classifier: Operating System :: POSIX
11
+ Classifier: Operating System :: POSIX :: Linux
12
+ Classifier: Programming Language :: Python
6
13
  Classifier: Programming Language :: Python :: 3
7
14
  Classifier: Programming Language :: Python :: 3.8
8
15
  Classifier: Programming Language :: Python :: 3.9
9
16
  Classifier: Programming Language :: Python :: 3.10
10
17
  Classifier: Programming Language :: Python :: 3.11
18
+ Classifier: Programming Language :: Python :: 3.12
19
+ Classifier: Topic :: Software Development :: Libraries :: Python Modules
20
+ Classifier: Typing :: Typed
11
21
  Requires-Dist: httpx (>=0.21.2)
12
22
  Requires-Dist: pydantic (>=1.9.2)
13
23
  Requires-Dist: python-magic (==0.4.27)