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.
- athena/chain/client.py +17 -8
- athena/client.py +91 -4
- athena/core/__init__.py +4 -1
- athena/core/client_wrapper.py +1 -1
- athena/core/pydantic_utilities.py +16 -0
- athena/core/query_encoder.py +33 -0
- athena/dataset/client.py +27 -22
- athena/message/client.py +17 -8
- athena/query/client.py +9 -4
- athena/report/client.py +9 -4
- athena/search/client.py +9 -4
- athena/snippet/client.py +35 -26
- athena/tools/client.py +107 -74
- athena/types/convert_pdf_to_sheet_out.py +7 -3
- athena/types/data_frame_parsing_error.py +7 -3
- athena/types/data_frame_request_out.py +7 -3
- athena/types/data_frame_unknown_format_error.py +7 -3
- athena/types/dataset.py +7 -3
- athena/types/document.py +7 -3
- athena/types/excecute_tool_first_workflow_out.py +7 -3
- athena/types/file_data_response.py +7 -3
- athena/types/file_fetch_error.py +7 -3
- athena/types/filter_model.py +7 -3
- athena/types/firecrawl_scrape_url_data_reponse_dto.py +7 -3
- athena/types/firecrawl_scrape_url_metadata.py +7 -3
- athena/types/get_datasets_response.py +7 -3
- athena/types/get_snippet_out.py +7 -3
- athena/types/get_snippets_response.py +7 -3
- athena/types/http_validation_error.py +7 -3
- athena/types/langchain_documents_request_out.py +7 -3
- athena/types/map_reduce_chain_out.py +7 -3
- athena/types/message_out.py +7 -3
- athena/types/message_out_dto.py +7 -3
- athena/types/publish_formats.py +7 -3
- athena/types/query_model.py +7 -3
- athena/types/report.py +7 -3
- athena/types/researcher_out.py +7 -3
- athena/types/semantic_query_out.py +7 -3
- athena/types/snippet.py +7 -3
- athena/types/sql_results.py +7 -3
- athena/types/structured_parse_result.py +7 -3
- athena/types/time_dimension_model.py +7 -3
- athena/types/upload_documents_out.py +7 -3
- athena/types/url_result.py +7 -3
- athena/types/validation_error.py +7 -3
- athena/types/workflow_status_out.py +7 -3
- athena/upload/client.py +9 -4
- athena/workflow/client.py +9 -4
- {athena_intelligence-0.1.83.dist-info → athena_intelligence-0.1.84.dist-info}/METADATA +12 -2
- athena_intelligence-0.1.84.dist-info/RECORD +91 -0
- athena_intelligence-0.1.83.dist-info/RECORD +0 -90
- {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
|
-
|
24
|
-
|
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
|
-
|
20
|
-
|
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
|
-
|
19
|
-
|
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
|
-
|
22
|
-
|
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/message_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 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
|
-
|
19
|
-
|
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/message_out_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 .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
|
-
|
25
|
-
|
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/publish_formats.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 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
|
-
|
21
|
-
|
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/query_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_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
|
-
|
27
|
-
|
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
|
-
|
22
|
-
|
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/researcher_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 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
|
-
|
19
|
-
|
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
|
-
|
19
|
-
|
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
|
-
|
25
|
-
|
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/sql_results.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 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
|
-
|
19
|
-
|
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
|
-
|
19
|
-
|
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
|
-
|
21
|
-
|
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
|
-
|
21
|
-
|
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/url_result.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 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
|
-
|
19
|
-
|
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/validation_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
|
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
|
-
|
22
|
-
|
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
|
-
|
21
|
-
|
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=
|
56
|
-
|
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=
|
126
|
-
|
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=
|
56
|
-
|
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=
|
126
|
-
|
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.
|
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)
|