athena-intelligence 0.1.45__py3-none-any.whl → 0.1.49__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/__init__.py +12 -3
- athena/base_client.py +27 -6
- athena/chain/__init__.py +0 -3
- athena/chain/client.py +42 -44
- athena/core/__init__.py +2 -0
- athena/core/client_wrapper.py +14 -6
- athena/core/http_client.py +8 -3
- athena/core/jsonable_encoder.py +7 -11
- athena/core/pydantic_utilities.py +12 -0
- athena/dataset/client.py +15 -15
- athena/message/client.py +33 -25
- athena/query/client.py +15 -15
- athena/report/client.py +15 -15
- athena/search/client.py +15 -15
- athena/snippet/client.py +15 -15
- athena/tasks/__init__.py +2 -0
- athena/tasks/client.py +191 -0
- athena/tools/client.py +178 -25
- athena/types/__init__.py +8 -0
- athena/types/dataset.py +3 -6
- athena/types/document.py +31 -0
- athena/types/excecute_tool_first_workflow_out.py +3 -6
- athena/types/firecrawl_scrape_url_data_reponse_dto.py +3 -6
- athena/types/firecrawl_scrape_url_metadata.py +5 -7
- athena/types/get_datasets_response.py +3 -6
- athena/types/get_snippets_response.py +3 -6
- athena/types/http_validation_error.py +3 -6
- athena/types/langchain_documents_request_out.py +26 -0
- athena/types/llm_model.py +93 -0
- athena/types/message_out.py +3 -6
- athena/types/message_out_dto.py +3 -6
- athena/types/model.py +0 -4
- athena/types/plan_execute_out.py +32 -0
- athena/types/report.py +3 -6
- athena/types/snippet.py +3 -6
- athena/types/sql_results.py +3 -6
- athena/types/structured_parse_result.py +3 -6
- athena/types/url_result.py +3 -6
- athena/types/validation_error.py +3 -6
- athena/version.py +4 -0
- {athena_intelligence-0.1.45.dist-info → athena_intelligence-0.1.49.dist-info}/METADATA +1 -1
- athena_intelligence-0.1.49.dist-info/RECORD +65 -0
- athena/chain/types/__init__.py +0 -5
- athena/chain/types/structured_parse_in_parsing_model.py +0 -53
- athena_intelligence-0.1.45.dist-info/RECORD +0 -59
- {athena_intelligence-0.1.45.dist-info → athena_intelligence-0.1.49.dist-info}/WHEEL +0 -0
athena/types/message_out.py
CHANGED
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class MessageOut(pydantic.BaseModel):
|
10
|
+
class MessageOut(pydantic_v1.BaseModel):
|
15
11
|
id: str
|
16
12
|
|
17
13
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -25,4 +21,5 @@ class MessageOut(pydantic.BaseModel):
|
|
25
21
|
class Config:
|
26
22
|
frozen = True
|
27
23
|
smart_union = True
|
24
|
+
extra = pydantic_v1.Extra.allow
|
28
25
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/message_out_dto.py
CHANGED
@@ -4,15 +4,11 @@ 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
8
|
from .status_enum import StatusEnum
|
8
9
|
|
9
|
-
try:
|
10
|
-
import pydantic.v1 as pydantic # type: ignore
|
11
|
-
except ImportError:
|
12
|
-
import pydantic # type: ignore
|
13
10
|
|
14
|
-
|
15
|
-
class MessageOutDto(pydantic.BaseModel):
|
11
|
+
class MessageOutDto(pydantic_v1.BaseModel):
|
16
12
|
id: str
|
17
13
|
conversation_id: str
|
18
14
|
logs: typing.Optional[str] = None
|
@@ -31,4 +27,5 @@ class MessageOutDto(pydantic.BaseModel):
|
|
31
27
|
class Config:
|
32
28
|
frozen = True
|
33
29
|
smart_union = True
|
30
|
+
extra = pydantic_v1.Extra.allow
|
34
31
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/model.py
CHANGED
@@ -18,7 +18,6 @@ class Model(str, enum.Enum):
|
|
18
18
|
MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
|
19
19
|
MISTRAL_LARGE_0224 = "mistral-large-0224"
|
20
20
|
MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
|
21
|
-
LLAMA_V_270_B_CHAT = "llama-v2-70b-chat"
|
22
21
|
LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
|
23
22
|
LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
|
24
23
|
CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
|
@@ -36,7 +35,6 @@ class Model(str, enum.Enum):
|
|
36
35
|
mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
|
37
36
|
mistral_large_0224: typing.Callable[[], T_Result],
|
38
37
|
mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
|
39
|
-
llama_v_270_b_chat: typing.Callable[[], T_Result],
|
40
38
|
llama_v_370_b_instruct: typing.Callable[[], T_Result],
|
41
39
|
llama_v_38_b_instruct: typing.Callable[[], T_Result],
|
42
40
|
claude_3_opus_20240229: typing.Callable[[], T_Result],
|
@@ -59,8 +57,6 @@ class Model(str, enum.Enum):
|
|
59
57
|
return mistral_large_0224()
|
60
58
|
if self is Model.MIXTRAL_8_X_22_B_INSTRUCT:
|
61
59
|
return mixtral_8_x_22_b_instruct()
|
62
|
-
if self is Model.LLAMA_V_270_B_CHAT:
|
63
|
-
return llama_v_270_b_chat()
|
64
60
|
if self is Model.LLAMA_V_370_B_INSTRUCT:
|
65
61
|
return llama_v_370_b_instruct()
|
66
62
|
if self is Model.LLAMA_V_38_B_INSTRUCT:
|
@@ -0,0 +1,32 @@
|
|
1
|
+
# This file was auto-generated by Fern from our API Definition.
|
2
|
+
|
3
|
+
import datetime as dt
|
4
|
+
import typing
|
5
|
+
|
6
|
+
from ..core.datetime_utils import serialize_datetime
|
7
|
+
from ..core.pydantic_utilities import pydantic_v1
|
8
|
+
|
9
|
+
|
10
|
+
class PlanExecuteOut(pydantic_v1.BaseModel):
|
11
|
+
input: str
|
12
|
+
instructions: str
|
13
|
+
input_data: str
|
14
|
+
output_format: str
|
15
|
+
plan: typing.List[str]
|
16
|
+
past_steps: typing.List[typing.List[typing.Any]]
|
17
|
+
final_output: str
|
18
|
+
done: bool
|
19
|
+
|
20
|
+
def json(self, **kwargs: typing.Any) -> str:
|
21
|
+
kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
|
22
|
+
return super().json(**kwargs_with_defaults)
|
23
|
+
|
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)
|
27
|
+
|
28
|
+
class Config:
|
29
|
+
frozen = True
|
30
|
+
smart_union = True
|
31
|
+
extra = pydantic_v1.Extra.allow
|
32
|
+
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/report.py
CHANGED
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class Report(pydantic.BaseModel):
|
10
|
+
class Report(pydantic_v1.BaseModel):
|
15
11
|
id: str
|
16
12
|
name: typing.Optional[str] = None
|
17
13
|
workspace_access: bool
|
@@ -28,4 +24,5 @@ class Report(pydantic.BaseModel):
|
|
28
24
|
class Config:
|
29
25
|
frozen = True
|
30
26
|
smart_union = True
|
27
|
+
extra = pydantic_v1.Extra.allow
|
31
28
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/snippet.py
CHANGED
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class Snippet(pydantic.BaseModel):
|
10
|
+
class Snippet(pydantic_v1.BaseModel):
|
15
11
|
id: str
|
16
12
|
title: typing.Optional[str] = None
|
17
13
|
description: typing.Optional[str] = None
|
@@ -31,4 +27,5 @@ class Snippet(pydantic.BaseModel):
|
|
31
27
|
class Config:
|
32
28
|
frozen = True
|
33
29
|
smart_union = True
|
30
|
+
extra = pydantic_v1.Extra.allow
|
34
31
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/sql_results.py
CHANGED
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class SqlResults(pydantic.BaseModel):
|
10
|
+
class SqlResults(pydantic_v1.BaseModel):
|
15
11
|
result: typing.Optional[typing.Any] = None
|
16
12
|
|
17
13
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -25,4 +21,5 @@ class SqlResults(pydantic.BaseModel):
|
|
25
21
|
class Config:
|
26
22
|
frozen = True
|
27
23
|
smart_union = True
|
24
|
+
extra = pydantic_v1.Extra.allow
|
28
25
|
json_encoders = {dt.datetime: serialize_datetime}
|
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class StructuredParseResult(pydantic.BaseModel):
|
10
|
+
class StructuredParseResult(pydantic_v1.BaseModel):
|
15
11
|
result: typing.Dict[str, typing.Any]
|
16
12
|
|
17
13
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -25,4 +21,5 @@ class StructuredParseResult(pydantic.BaseModel):
|
|
25
21
|
class Config:
|
26
22
|
frozen = True
|
27
23
|
smart_union = True
|
24
|
+
extra = pydantic_v1.Extra.allow
|
28
25
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/url_result.py
CHANGED
@@ -4,14 +4,10 @@ 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
8
|
|
8
|
-
try:
|
9
|
-
import pydantic.v1 as pydantic # type: ignore
|
10
|
-
except ImportError:
|
11
|
-
import pydantic # type: ignore
|
12
9
|
|
13
|
-
|
14
|
-
class UrlResult(pydantic.BaseModel):
|
10
|
+
class UrlResult(pydantic_v1.BaseModel):
|
15
11
|
result: typing.Dict[str, typing.Any]
|
16
12
|
|
17
13
|
def json(self, **kwargs: typing.Any) -> str:
|
@@ -25,4 +21,5 @@ class UrlResult(pydantic.BaseModel):
|
|
25
21
|
class Config:
|
26
22
|
frozen = True
|
27
23
|
smart_union = True
|
24
|
+
extra = pydantic_v1.Extra.allow
|
28
25
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/types/validation_error.py
CHANGED
@@ -4,15 +4,11 @@ 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
8
|
from .validation_error_loc_item import ValidationErrorLocItem
|
8
9
|
|
9
|
-
try:
|
10
|
-
import pydantic.v1 as pydantic # type: ignore
|
11
|
-
except ImportError:
|
12
|
-
import pydantic # type: ignore
|
13
10
|
|
14
|
-
|
15
|
-
class ValidationError(pydantic.BaseModel):
|
11
|
+
class ValidationError(pydantic_v1.BaseModel):
|
16
12
|
loc: typing.List[ValidationErrorLocItem]
|
17
13
|
msg: str
|
18
14
|
type: str
|
@@ -28,4 +24,5 @@ class ValidationError(pydantic.BaseModel):
|
|
28
24
|
class Config:
|
29
25
|
frozen = True
|
30
26
|
smart_union = True
|
27
|
+
extra = pydantic_v1.Extra.allow
|
31
28
|
json_encoders = {dt.datetime: serialize_datetime}
|
athena/version.py
ADDED
@@ -0,0 +1,65 @@
|
|
1
|
+
athena/__init__.py,sha256=EHglML-qUfPnwk5zsnJ79TrVz1wl_CqT-_JkrTqX6AQ,1547
|
2
|
+
athena/base_client.py,sha256=aniFuLr3wzh9PChzWZrkKV3ivSjOnS58p9g11dMXzgI,6860
|
3
|
+
athena/chain/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
4
|
+
athena/chain/client.py,sha256=BPvzeoDzOyAgSWCAVWS35xQKBv39gVOfAhO7B1B7h34,8026
|
5
|
+
athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
|
6
|
+
athena/core/__init__.py,sha256=1pNSKkwyQvMl_F0wohBqmoQAITptg3zlvCwsoSSzy7c,853
|
7
|
+
athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
|
8
|
+
athena/core/client_wrapper.py,sha256=e5bqcaJcyraFOz98rl3FUz0SWdriGivXLYrNzL6Gqvk,1495
|
9
|
+
athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
|
10
|
+
athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
|
11
|
+
athena/core/http_client.py,sha256=5ok6hqgZDJhg57EHvMnr0BBaHdG50QxFPKaCZ9aVWTc,5059
|
12
|
+
athena/core/jsonable_encoder.py,sha256=L6G68Py6gEo8n87rXwlkLPUyzHvXftEBjJZNb2tCuOA,3742
|
13
|
+
athena/core/pydantic_utilities.py,sha256=eCOGHdLoaxK_6QSFYKdN0LWJo3atTUXT3VPyqJh3rxM,329
|
14
|
+
athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
|
15
|
+
athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
|
16
|
+
athena/dataset/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
17
|
+
athena/dataset/client.py,sha256=C43dgxtfFhlpB5_l4WMqGZp-sCNFo66CsTDLi1cAz2c,6182
|
18
|
+
athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
|
19
|
+
athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
|
20
|
+
athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
|
21
|
+
athena/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
22
|
+
athena/message/client.py,sha256=7HdDU7d2W0Yr8_9i4Bk4uo7r6aprB7O7wTbRkJCQFj8,12498
|
23
|
+
athena/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
|
24
|
+
athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
25
|
+
athena/query/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
26
|
+
athena/query/client.py,sha256=FprFNth7SOF6Irc3YUsSZO8DVqTkfOZDhaIfBlXQUSw,6338
|
27
|
+
athena/report/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
28
|
+
athena/report/client.py,sha256=rSNGKHf2wImbOmY06jXClMK3S5B4sQPCvQOPxrb9lXk,6623
|
29
|
+
athena/search/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
30
|
+
athena/search/client.py,sha256=8ukNlqy1Wv-uci1tJ7uz-lhOF6OK7_kF31kNO1MGpok,7610
|
31
|
+
athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
32
|
+
athena/snippet/client.py,sha256=7wVWdr4Ox6e5dAyEZIxuP0FUO4GHYOHMjJzwHvkyhcU,6182
|
33
|
+
athena/tasks/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
34
|
+
athena/tasks/client.py,sha256=VtpBe-EzyOhjIOFwcN4Lx4cWZ7DZ4_Qj9Eyz43DBaUQ,8214
|
35
|
+
athena/tools/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
36
|
+
athena/tools/client.py,sha256=Dv_XRrhnkmaqfhs1oon31gJl0UvSjMQ3Sy5o7QqH1ac,20114
|
37
|
+
athena/types/__init__.py,sha256=JZReH-Lqa3W6C8VcOToGaT-HqdII_SNLQp-VpzIK0oI,1722
|
38
|
+
athena/types/dataset.py,sha256=ShFYop4Pj-pscWrjWZQFboUmK5TDX3NzP0xNRZimpp8,994
|
39
|
+
athena/types/document.py,sha256=evK_-wGk07kB8y5xyPMFCgqDbItuxCAawdUN20b6zFg,1061
|
40
|
+
athena/types/excecute_tool_first_workflow_out.py,sha256=T4GxP3yzTY3XkumdpUdXbn8Tx_iNc1exed8N2SnwV2w,875
|
41
|
+
athena/types/firecrawl_scrape_url_data_reponse_dto.py,sha256=-MkjjhzRTpuyoypLmiGtvH01TjeoVQxpX-HsALUSFUM,1001
|
42
|
+
athena/types/firecrawl_scrape_url_metadata.py,sha256=kIKb0mMGxw7-49GSsagfx6AperguHDKOvODGPjFtOxU,1143
|
43
|
+
athena/types/get_datasets_response.py,sha256=kbv8BI2nEo34-HJZV33dPhKWKrA1FiIS_OUkUYJj1ZQ,969
|
44
|
+
athena/types/get_snippets_response.py,sha256=pgwYqmddU5shKeVaE4RQSFN9SLsVAeQp3sqIkQlvzoU,969
|
45
|
+
athena/types/http_validation_error.py,sha256=u4t-1U0DI0u3Zj_Oz7AmGmpL4sqBzoS_5nZHImWQbmM,953
|
46
|
+
athena/types/langchain_documents_request_out.py,sha256=9XPPEMY6zB2m5f4u4LnO2dmveKtOpAkzxKvENEJYz2c,919
|
47
|
+
athena/types/llm_model.py,sha256=b1W7CtiusS5Du3jLugdvpd7fUQ5GhcIU-2YxK4UAqvs,4101
|
48
|
+
athena/types/message_out.py,sha256=HJZizmFH7crD3OHm0fdTy3189F2gv5qR8aaUbTTfWFI,845
|
49
|
+
athena/types/message_out_dto.py,sha256=1G8srlYaIYmoYRstLKm97xZGxK87DK57CiO9hYnt3gQ,1031
|
50
|
+
athena/types/model.py,sha256=73jxfZPxAtK0iFex4bluPRrfpF5sY3Y6VFa9iALykBw,2991
|
51
|
+
athena/types/plan_execute_out.py,sha256=g9RrUTu32KYm-r8acfZ8Gl4-jTOvFJ_Du1Rxv3YokcE,1034
|
52
|
+
athena/types/report.py,sha256=km2CgCbHBXQQbPai1y5sGlsQpO7WAlUVvdsRC_7f4KI,926
|
53
|
+
athena/types/snippet.py,sha256=Mrc92_hBABJQjCSToAA-FgwhvO-Jn8Kjm-lYI6aMlUY,1106
|
54
|
+
athena/types/sql_results.py,sha256=ExPFds4vZ425AxGt0jhykbPhOjkplZPGQwVKb0LHg_g,880
|
55
|
+
athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,672
|
56
|
+
athena/types/structured_parse_result.py,sha256=fph7KrT_X_2BKDCOFN1UEufeaMmpSEvT0Oi6aM-e3kU,885
|
57
|
+
athena/types/tool_models.py,sha256=ltl4RX0QhFIj_wsKLZMDip72_Ko7P-7v0aL9HACXVFo,1829
|
58
|
+
athena/types/tools.py,sha256=mhRkKAwlsDud-fFOhsx2T3hBD-FAtuCnGHyU9cLPcGU,1422
|
59
|
+
athena/types/url_result.py,sha256=lIgnQeyKy_UfFFPe7HMrrRzb-SK089RxcKcKN9Q3DNQ,873
|
60
|
+
athena/types/validation_error.py,sha256=yqombbKLBSzTPFn6CJH_hbo7tpS68T3JvMdd7kBtO1g,972
|
61
|
+
athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
|
62
|
+
athena/version.py,sha256=8aYAOJtVLaJLpRp6mTiEIhnl8gXA7yE0aDtZ-3mKQ4k,87
|
63
|
+
athena_intelligence-0.1.49.dist-info/METADATA,sha256=IA03tyg22B0NrD6XUNu_2kWv3TC1_owKwgQ1jBQiSKo,4738
|
64
|
+
athena_intelligence-0.1.49.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
|
65
|
+
athena_intelligence-0.1.49.dist-info/RECORD,,
|
athena/chain/types/__init__.py
DELETED
@@ -1,53 +0,0 @@
|
|
1
|
-
# This file was auto-generated by Fern from our API Definition.
|
2
|
-
|
3
|
-
import enum
|
4
|
-
import typing
|
5
|
-
|
6
|
-
T_Result = typing.TypeVar("T_Result")
|
7
|
-
|
8
|
-
|
9
|
-
class StructuredParseInParsingModel(str, enum.Enum):
|
10
|
-
"""
|
11
|
-
The model to be used for parsing.
|
12
|
-
"""
|
13
|
-
|
14
|
-
GPT_4_TURBO = "gpt-4-turbo"
|
15
|
-
GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
|
16
|
-
GPT_4 = "gpt-4"
|
17
|
-
GPT_35_TURBO = "gpt-3.5-turbo"
|
18
|
-
MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
|
19
|
-
MISTRAL_LARGE_0224 = "mistral-large-0224"
|
20
|
-
CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
|
21
|
-
CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
|
22
|
-
CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
|
23
|
-
|
24
|
-
def visit(
|
25
|
-
self,
|
26
|
-
gpt_4_turbo: typing.Callable[[], T_Result],
|
27
|
-
gpt_4_turbo_preview: typing.Callable[[], T_Result],
|
28
|
-
gpt_4: typing.Callable[[], T_Result],
|
29
|
-
gpt_35_turbo: typing.Callable[[], T_Result],
|
30
|
-
mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
|
31
|
-
mistral_large_0224: typing.Callable[[], T_Result],
|
32
|
-
claude_3_opus_20240229: typing.Callable[[], T_Result],
|
33
|
-
claude_3_sonnet_20240229: typing.Callable[[], T_Result],
|
34
|
-
claude_3_haiku_20240307: typing.Callable[[], T_Result],
|
35
|
-
) -> T_Result:
|
36
|
-
if self is StructuredParseInParsingModel.GPT_4_TURBO:
|
37
|
-
return gpt_4_turbo()
|
38
|
-
if self is StructuredParseInParsingModel.GPT_4_TURBO_PREVIEW:
|
39
|
-
return gpt_4_turbo_preview()
|
40
|
-
if self is StructuredParseInParsingModel.GPT_4:
|
41
|
-
return gpt_4()
|
42
|
-
if self is StructuredParseInParsingModel.GPT_35_TURBO:
|
43
|
-
return gpt_35_turbo()
|
44
|
-
if self is StructuredParseInParsingModel.MIXTRAL_SMALL_8_X_7_B_0211:
|
45
|
-
return mixtral_small_8_x_7_b_0211()
|
46
|
-
if self is StructuredParseInParsingModel.MISTRAL_LARGE_0224:
|
47
|
-
return mistral_large_0224()
|
48
|
-
if self is StructuredParseInParsingModel.CLAUDE_3_OPUS_20240229:
|
49
|
-
return claude_3_opus_20240229()
|
50
|
-
if self is StructuredParseInParsingModel.CLAUDE_3_SONNET_20240229:
|
51
|
-
return claude_3_sonnet_20240229()
|
52
|
-
if self is StructuredParseInParsingModel.CLAUDE_3_HAIKU_20240307:
|
53
|
-
return claude_3_haiku_20240307()
|
@@ -1,59 +0,0 @@
|
|
1
|
-
athena/__init__.py,sha256=LTNvMytqGxQgKMYkUeJ3D6V12qy2wz9PQE9AzPREBd4,1389
|
2
|
-
athena/base_client.py,sha256=tvX2FKc9AnNBm0k64njjUCOt7dPxBBn6BpClJpcPqgc,5416
|
3
|
-
athena/chain/__init__.py,sha256=I1CBCogKCvJBIPdXXFQWdGJyFs6pdp0_dp6rnybd5mI,159
|
4
|
-
athena/chain/client.py,sha256=SdlDtNr1QBbDgZGBAFW-GNT0FqPdhvAUcrQaHcGuUds,8333
|
5
|
-
athena/chain/types/__init__.py,sha256=s4rY--H5yj6slggsUnRQNrKBHZ3QGE9jQWopTCQOFpg,187
|
6
|
-
athena/chain/types/structured_parse_in_parsing_model.py,sha256=tr6DLP2v71IUScCOeLrUjUtXrAr3WlaW0sSD3ns2r-Q,2203
|
7
|
-
athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
|
8
|
-
athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
|
9
|
-
athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
|
10
|
-
athena/core/client_wrapper.py,sha256=KS50tFev6ZS1bMDm5hVdz1JOR4qnONQeHMVTZI8e6jY,1198
|
11
|
-
athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
|
12
|
-
athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
|
13
|
-
athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
|
14
|
-
athena/core/jsonable_encoder.py,sha256=MTYkDov2EryHgee4QM46uZiBOuOXK9KTHlBdBwU-CpU,3799
|
15
|
-
athena/core/remove_none_from_dict.py,sha256=8m91FC3YuVem0Gm9_sXhJ2tGvP33owJJdrqCLEdowGw,330
|
16
|
-
athena/core/request_options.py,sha256=-3QoOMMHI2exIyHH6Q2MD7rpo_6w-i6zMAy0nqWTN8c,1420
|
17
|
-
athena/dataset/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
18
|
-
athena/dataset/client.py,sha256=ilxpF0K9AlnGAQjrumdqZ1V1Wu3ARoM-06iDE5gQiCQ,6087
|
19
|
-
athena/environment.py,sha256=D_CljQlUahhEi9smvMslj_5Y8gMFO6D0fRCL0ydRLuM,165
|
20
|
-
athena/errors/__init__.py,sha256=pbbVUFtB9LCocA1RMWMMF_RKjsy5YkOKX5BAuE49w6g,170
|
21
|
-
athena/errors/unprocessable_entity_error.py,sha256=FvR7XPlV3Xx5nu8HNlmLhBRdk4so_gCHjYT5PyZe6sM,313
|
22
|
-
athena/message/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
23
|
-
athena/message/client.py,sha256=NT0E6j-LsjZygrjEzn-Bcy8T_Q_ayNK0jR6z3hQjcnI,12185
|
24
|
-
athena/polling_message_client.py,sha256=dmmycImvog2niEFFPo4rE5xMJHUlq9NqAr4xlFK6_Os,3998
|
25
|
-
athena/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
26
|
-
athena/query/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
27
|
-
athena/query/client.py,sha256=UOx-Bq-xFFm-sTMTmJjWGrC6q_7vhVno3nYzmi81xwI,6243
|
28
|
-
athena/report/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
29
|
-
athena/report/client.py,sha256=sGJDrgk_E1SPleRYNhvspmsz-G3FQwMW-3alFzZPquE,6528
|
30
|
-
athena/search/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
31
|
-
athena/search/client.py,sha256=zhE86fKD61gHQ1w45DYfWVHIZ1APGK6yUrWAHYi0OVM,7515
|
32
|
-
athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
33
|
-
athena/snippet/client.py,sha256=D0rSpm6ql9cnUj-mMe3z8OHRgRQQuk3bBW2CZSRnyp4,6087
|
34
|
-
athena/tools/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
|
35
|
-
athena/tools/client.py,sha256=1a23IVikMANN_8GdewBC_5bOqeJlYX2Bg35_e3PSxh0,13328
|
36
|
-
athena/types/__init__.py,sha256=vZkpWtCrgzHJOYK8Qds37aeBX6dDXgYGZc7Ur9RPGtk,1450
|
37
|
-
athena/types/dataset.py,sha256=70OJPxKBAYu7xthGEgrUolSdyLqiyh6X49INw1oN0sA,1014
|
38
|
-
athena/types/excecute_tool_first_workflow_out.py,sha256=tGL6pNN4uhL_knWf-SQc-Z1IPJFNVsAmYtDI-VjXS2s,895
|
39
|
-
athena/types/firecrawl_scrape_url_data_reponse_dto.py,sha256=LbJY-SC_WNQG3nbswG8NTfVl_u_tpV1HO7Y3MWCk5gc,1021
|
40
|
-
athena/types/firecrawl_scrape_url_metadata.py,sha256=sqHTtq5_5vujsMixJBDJULPK7MrvxEpB2wGPwC-bTdM,1128
|
41
|
-
athena/types/get_datasets_response.py,sha256=BCdT8yTLfOsXeyFadlyoas4zzseFWGPAdGpkgkOuaD8,989
|
42
|
-
athena/types/get_snippets_response.py,sha256=Lpn7bHJLpPQozN93unCV-8eByAAfz1MhQWR3G3Z1vl4,989
|
43
|
-
athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
|
44
|
-
athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
|
45
|
-
athena/types/message_out_dto.py,sha256=qgRibRbDNOWVnVGP7Rribh9WdoCT2CSiPUXeIWECqq4,1051
|
46
|
-
athena/types/model.py,sha256=odPPEr-0PAivpX5bhlCIEje66pIHA_x1keFJBTEqGu0,3180
|
47
|
-
athena/types/report.py,sha256=QVaqVfHMAV3s9_V2CqjIEMcRrbJhD8zmi82vrk2A8x0,946
|
48
|
-
athena/types/snippet.py,sha256=POIVJNV9iQxiVegB_qwQx-PZPPSyoIPhyxTsueNVUGA,1126
|
49
|
-
athena/types/sql_results.py,sha256=pNH32nyf1bzoYJs3FgHctLdLO02oOjyGgLkHACACB6k,900
|
50
|
-
athena/types/status_enum.py,sha256=0UZbhdAx215GHC-U53RS98mYHtn1N3On4VBe4j02Qtc,672
|
51
|
-
athena/types/structured_parse_result.py,sha256=7I-w06OmtxXFY01k7FXFSNPe5PpM3z54xNUbs62lSv0,905
|
52
|
-
athena/types/tool_models.py,sha256=ltl4RX0QhFIj_wsKLZMDip72_Ko7P-7v0aL9HACXVFo,1829
|
53
|
-
athena/types/tools.py,sha256=mhRkKAwlsDud-fFOhsx2T3hBD-FAtuCnGHyU9cLPcGU,1422
|
54
|
-
athena/types/url_result.py,sha256=zajsW46qJnD6GPimb5kHkUncjqBfzHUlGOcKuUGMX-E,893
|
55
|
-
athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
|
56
|
-
athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
|
57
|
-
athena_intelligence-0.1.45.dist-info/METADATA,sha256=BY3eecOb508WNjpFOadAQkLg422Q0ON8I1bbHqV_AQo,4738
|
58
|
-
athena_intelligence-0.1.45.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
|
59
|
-
athena_intelligence-0.1.45.dist-info/RECORD,,
|
File without changes
|