athena-intelligence 0.1.43__py3-none-any.whl → 0.1.45__py3-none-any.whl

athena/__init__.py

@@ -2,14 +2,12 @@
 
 from .types import (
     Dataset,
-    Document,
     ExcecuteToolFirstWorkflowOut,
     FirecrawlScrapeUrlDataReponseDto,
     FirecrawlScrapeUrlMetadata,
     GetDatasetsResponse,
     GetSnippetsResponse,
     HttpValidationError,
-    LangchainDocumentsRequestOut,
     MessageOut,
     MessageOutDto,
     Model,
@@ -32,14 +30,12 @@ from .environment import AthenaEnvironment
 __all__ = [
     "AthenaEnvironment",
     "Dataset",
-    "Document",
     "ExcecuteToolFirstWorkflowOut",
     "FirecrawlScrapeUrlDataReponseDto",
     "FirecrawlScrapeUrlMetadata",
     "GetDatasetsResponse",
     "GetSnippetsResponse",
     "HttpValidationError",
-    "LangchainDocumentsRequestOut",
     "MessageOut",
     "MessageOutDto",
     "Model",

athena/core/client_wrapper.py

@@ -16,7 +16,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.43",
+            "X-Fern-SDK-Version": "0.1.45",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

athena/tools/client.py

@@ -13,7 +13,6 @@ from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOut
 from ..types.firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlDataReponseDto
 from ..types.http_validation_error import HttpValidationError
-from ..types.langchain_documents_request_out import LangchainDocumentsRequestOut
 from ..types.tool_models import ToolModels
 
 try:
@@ -92,76 +91,6 @@ class ToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def langchain_documents(
-        self,
-        *,
-        document_id: str,
-        pagination_limit: typing.Optional[int] = OMIT,
-        pagination_offset: typing.Optional[int] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> LangchainDocumentsRequestOut:
-        """
-        Parameters:
-            - document_id: str.
-
-            - pagination_limit: typing.Optional[int].
-
-            - pagination_offset: typing.Optional[int].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import Athena
-
-        client = Athena(
-            api_key="YOUR_API_KEY",
-        )
-        client.tools.langchain_documents(
-            document_id="doc_9249292-d118-42d3-95b4-00eccfe0754f",
-            pagination_limit=250,
-            pagination_offset=0,
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {"document_id": document_id}
-        if pagination_limit is not OMIT:
-            _request["pagination_limit"] = pagination_limit
-        if pagination_offset is not OMIT:
-            _request["pagination_offset"] = pagination_offset
-        _response = self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/langchain-documents"),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-            retries=0,
-            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(LangchainDocumentsRequestOut, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
     def tool_first_workflow(
         self,
         *,
@@ -305,76 +234,6 @@ class AsyncToolsClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def langchain_documents(
-        self,
-        *,
-        document_id: str,
-        pagination_limit: typing.Optional[int] = OMIT,
-        pagination_offset: typing.Optional[int] = OMIT,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> LangchainDocumentsRequestOut:
-        """
-        Parameters:
-            - document_id: str.
-
-            - pagination_limit: typing.Optional[int].
-
-            - pagination_offset: typing.Optional[int].
-
-            - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
-        ---
-        from athena.client import AsyncAthena
-
-        client = AsyncAthena(
-            api_key="YOUR_API_KEY",
-        )
-        await client.tools.langchain_documents(
-            document_id="doc_9249292-d118-42d3-95b4-00eccfe0754f",
-            pagination_limit=250,
-            pagination_offset=0,
-        )
-        """
-        _request: typing.Dict[str, typing.Any] = {"document_id": document_id}
-        if pagination_limit is not OMIT:
-            _request["pagination_limit"] = pagination_limit
-        if pagination_offset is not OMIT:
-            _request["pagination_offset"] = pagination_offset
-        _response = await self._client_wrapper.httpx_client.request(
-            "POST",
-            urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/file/langchain-documents"),
-            params=jsonable_encoder(
-                request_options.get("additional_query_parameters") if request_options is not None else None
-            ),
-            json=jsonable_encoder(_request)
-            if request_options is None or request_options.get("additional_body_parameters") is None
-            else {
-                **jsonable_encoder(_request),
-                **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
-            },
-            headers=jsonable_encoder(
-                remove_none_from_dict(
-                    {
-                        **self._client_wrapper.get_headers(),
-                        **(request_options.get("additional_headers", {}) if request_options is not None else {}),
-                    }
-                )
-            ),
-            timeout=request_options.get("timeout_in_seconds")
-            if request_options is not None and request_options.get("timeout_in_seconds") is not None
-            else 60,
-            retries=0,
-            max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
-        )
-        if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(LangchainDocumentsRequestOut, _response.json())  # type: ignore
-        if _response.status_code == 422:
-            raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
-        try:
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
     async def tool_first_workflow(
         self,
         *,

athena/types/__init__.py

@@ -1,14 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .dataset import Dataset
-from .document import Document
 from .excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOut
 from .firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlDataReponseDto
 from .firecrawl_scrape_url_metadata import FirecrawlScrapeUrlMetadata
 from .get_datasets_response import GetDatasetsResponse
 from .get_snippets_response import GetSnippetsResponse
 from .http_validation_error import HttpValidationError
-from .langchain_documents_request_out import LangchainDocumentsRequestOut
 from .message_out import MessageOut
 from .message_out_dto import MessageOutDto
 from .model import Model
@@ -25,14 +23,12 @@ from .validation_error_loc_item import ValidationErrorLocItem
 
 __all__ = [
     "Dataset",
-    "Document",
     "ExcecuteToolFirstWorkflowOut",
     "FirecrawlScrapeUrlDataReponseDto",
     "FirecrawlScrapeUrlMetadata",
     "GetDatasetsResponse",
     "GetSnippetsResponse",
     "HttpValidationError",
-    "LangchainDocumentsRequestOut",
     "MessageOut",
     "MessageOutDto",
     "Model",

athena/types/model.py

@@ -17,8 +17,10 @@ class Model(str, enum.Enum):
     GPT_4 = "gpt-4"
     MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
     MISTRAL_LARGE_0224 = "mistral-large-0224"
-    MIXTRAL_8_X_22_B_INSTRUCT_HF = "mixtral-8x22b-instruct-hf"
+    MIXTRAL_8_X_22_B_INSTRUCT = "mixtral-8x22b-instruct"
     LLAMA_V_270_B_CHAT = "llama-v2-70b-chat"
+    LLAMA_V_370_B_INSTRUCT = "llama-v3-70b-instruct"
+    LLAMA_V_38_B_INSTRUCT = "llama-v3-8b-instruct"
     CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
     CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
     CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
@@ -33,8 +35,10 @@ class Model(str, enum.Enum):
         gpt_4: typing.Callable[[], T_Result],
         mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
         mistral_large_0224: typing.Callable[[], T_Result],
-        mixtral_8_x_22_b_instruct_hf: typing.Callable[[], T_Result],
+        mixtral_8_x_22_b_instruct: typing.Callable[[], T_Result],
         llama_v_270_b_chat: typing.Callable[[], T_Result],
+        llama_v_370_b_instruct: typing.Callable[[], T_Result],
+        llama_v_38_b_instruct: typing.Callable[[], T_Result],
         claude_3_opus_20240229: typing.Callable[[], T_Result],
         claude_3_sonnet_20240229: typing.Callable[[], T_Result],
         claude_3_haiku_20240307: typing.Callable[[], T_Result],
@@ -53,10 +57,14 @@ class Model(str, enum.Enum):
             return mixtral_small_8_x_7_b_0211()
         if self is Model.MISTRAL_LARGE_0224:
             return mistral_large_0224()
-        if self is Model.MIXTRAL_8_X_22_B_INSTRUCT_HF:
-            return mixtral_8_x_22_b_instruct_hf()
+        if self is Model.MIXTRAL_8_X_22_B_INSTRUCT:
+            return mixtral_8_x_22_b_instruct()
         if self is Model.LLAMA_V_270_B_CHAT:
             return llama_v_270_b_chat()
+        if self is Model.LLAMA_V_370_B_INSTRUCT:
+            return llama_v_370_b_instruct()
+        if self is Model.LLAMA_V_38_B_INSTRUCT:
+            return llama_v_38_b_instruct()
         if self is Model.CLAUDE_3_OPUS_20240229:
             return claude_3_opus_20240229()
         if self is Model.CLAUDE_3_SONNET_20240229:

{athena_intelligence-0.1.43.dist-info → athena_intelligence-0.1.45.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.43
+Version: 0.1.45
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

{athena_intelligence-0.1.43.dist-info → athena_intelligence-0.1.45.dist-info}/RECORD

@@ -1,4 +1,4 @@
-athena/__init__.py,sha256=uXIyAvCSPx3g5Y96sk21HkSjNLz4Hi4XIKncoAxNT90,1489
+athena/__init__.py,sha256=LTNvMytqGxQgKMYkUeJ3D6V12qy2wz9PQE9AzPREBd4,1389
 athena/base_client.py,sha256=tvX2FKc9AnNBm0k64njjUCOt7dPxBBn6BpClJpcPqgc,5416
 athena/chain/__init__.py,sha256=I1CBCogKCvJBIPdXXFQWdGJyFs6pdp0_dp6rnybd5mI,159
 athena/chain/client.py,sha256=SdlDtNr1QBbDgZGBAFW-GNT0FqPdhvAUcrQaHcGuUds,8333
@@ -7,7 +7,7 @@ athena/chain/types/structured_parse_in_parsing_model.py,sha256=tr6DLP2v71IUScCOe
 athena/client.py,sha256=8QypiDlbZ0C1YsJh6GzhylLVCZXDQc1MCJTURo2_vvI,3576
 athena/core/__init__.py,sha256=RWfyDqkzWsf8e3VGc3NV60MovfJbg5XWzNFGB2DZ0hA,790
 athena/core/api_error.py,sha256=RE8LELok2QCjABadECTvtDp7qejA1VmINCh6TbqPwSE,426
-athena/core/client_wrapper.py,sha256=WWy7BCqaS38CcCzs1CNfIvhD1W8xtO92v0Lg6QN5oaI,1198
+athena/core/client_wrapper.py,sha256=KS50tFev6ZS1bMDm5hVdz1JOR4qnONQeHMVTZI8e6jY,1198
 athena/core/datetime_utils.py,sha256=nBys2IsYrhPdszxGKCNRPSOCwa-5DWOHG95FB8G9PKo,1047
 athena/core/file.py,sha256=sy1RUGZ3aJYuw998bZytxxo6QdgKmlnlgBaMvwEKCGg,1480
 athena/core/http_client.py,sha256=LI0yP3jUyE0Ue7oyBcI9nyo1pljOwh9Y5ycTeIpKwOg,4882
@@ -32,20 +32,18 @@ athena/search/client.py,sha256=zhE86fKD61gHQ1w45DYfWVHIZ1APGK6yUrWAHYi0OVM,7515
 athena/snippet/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
 athena/snippet/client.py,sha256=D0rSpm6ql9cnUj-mMe3z8OHRgRQQuk3bBW2CZSRnyp4,6087
 athena/tools/__init__.py,sha256=FTtvy8EDg9nNNg9WCatVgKTRYV8-_v1roeGPAKoa_pw,65
-athena/tools/client.py,sha256=qk9PJ70meZVKZXReLEsxp1woUMpZ6UPOcuLYqHhMt1Q,19643
-athena/types/__init__.py,sha256=ClUrCuQeInMWgv8vO1AGSV0rQEm51YABwBPvB250fno,1607
+athena/tools/client.py,sha256=1a23IVikMANN_8GdewBC_5bOqeJlYX2Bg35_e3PSxh0,13328
+athena/types/__init__.py,sha256=vZkpWtCrgzHJOYK8Qds37aeBX6dDXgYGZc7Ur9RPGtk,1450
 athena/types/dataset.py,sha256=70OJPxKBAYu7xthGEgrUolSdyLqiyh6X49INw1oN0sA,1014
-athena/types/document.py,sha256=CAByS_smWjl-edYLCCc4fkdhZYXSVUKQqEamjbhpgU0,1081
 athena/types/excecute_tool_first_workflow_out.py,sha256=tGL6pNN4uhL_knWf-SQc-Z1IPJFNVsAmYtDI-VjXS2s,895
 athena/types/firecrawl_scrape_url_data_reponse_dto.py,sha256=LbJY-SC_WNQG3nbswG8NTfVl_u_tpV1HO7Y3MWCk5gc,1021
 athena/types/firecrawl_scrape_url_metadata.py,sha256=sqHTtq5_5vujsMixJBDJULPK7MrvxEpB2wGPwC-bTdM,1128
 athena/types/get_datasets_response.py,sha256=BCdT8yTLfOsXeyFadlyoas4zzseFWGPAdGpkgkOuaD8,989
 athena/types/get_snippets_response.py,sha256=Lpn7bHJLpPQozN93unCV-8eByAAfz1MhQWR3G3Z1vl4,989
 athena/types/http_validation_error.py,sha256=Fcv_CTMMrLvCeTHjF0n5xf5tskMDgt-J6H9gp654eQw,973
-athena/types/langchain_documents_request_out.py,sha256=P5Qq7BkCWILORC3yOkvt8pssdYgd_vnTTANNtrWF-l8,939
 athena/types/message_out.py,sha256=uvZY_Podv2XccEk8CICug9I_S2hFJTSzCBwcHiauW7A,865
 athena/types/message_out_dto.py,sha256=qgRibRbDNOWVnVGP7Rribh9WdoCT2CSiPUXeIWECqq4,1051
-athena/types/model.py,sha256=kKzxm9vlaJoMbdE0SQtBBvsqeL7PnlOOhxuCdfvjtdw,2782
+athena/types/model.py,sha256=odPPEr-0PAivpX5bhlCIEje66pIHA_x1keFJBTEqGu0,3180
 athena/types/report.py,sha256=QVaqVfHMAV3s9_V2CqjIEMcRrbJhD8zmi82vrk2A8x0,946
 athena/types/snippet.py,sha256=POIVJNV9iQxiVegB_qwQx-PZPPSyoIPhyxTsueNVUGA,1126
 athena/types/sql_results.py,sha256=pNH32nyf1bzoYJs3FgHctLdLO02oOjyGgLkHACACB6k,900
@@ -56,6 +54,6 @@ athena/types/tools.py,sha256=mhRkKAwlsDud-fFOhsx2T3hBD-FAtuCnGHyU9cLPcGU,1422
 athena/types/url_result.py,sha256=zajsW46qJnD6GPimb5kHkUncjqBfzHUlGOcKuUGMX-E,893
 athena/types/validation_error.py,sha256=2JhGNJouo8QpfrMBoT_JCwYSn1nFN2Nnq0p9uPLDH-U,992
 athena/types/validation_error_loc_item.py,sha256=LAtjCHIllWRBFXvAZ5QZpp7CPXjdtN9EB7HrLVo6EP0,128
-athena_intelligence-0.1.43.dist-info/METADATA,sha256=-CdRL5jywIpooZZp_UiHFFZiQYMsPh30ft8jrKKfGHQ,4738
-athena_intelligence-0.1.43.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-athena_intelligence-0.1.43.dist-info/RECORD,,
+athena_intelligence-0.1.45.dist-info/METADATA,sha256=BY3eecOb508WNjpFOadAQkLg422Q0ON8I1bbHqV_AQo,4738
+athena_intelligence-0.1.45.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+athena_intelligence-0.1.45.dist-info/RECORD,,

athena/types/document.py

@@ -1,34 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class Document(pydantic.BaseModel):
-    """
-    Class for storing a piece of text and associated metadata.
-    """
-
-    page_content: str
-    metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
-    type: typing.Optional[typing.Literal["Document"]] = None
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}

athena/types/langchain_documents_request_out.py

@@ -1,29 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import datetime as dt
-import typing
-
-from ..core.datetime_utils import serialize_datetime
-from .document import Document
-
-try:
-    import pydantic.v1 as pydantic  # type: ignore
-except ImportError:
-    import pydantic  # type: ignore
-
-
-class LangchainDocumentsRequestOut(pydantic.BaseModel):
-    documents: typing.List[Document]
-
-    def json(self, **kwargs: typing.Any) -> str:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().json(**kwargs_with_defaults)
-
-    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
-        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
-        return super().dict(**kwargs_with_defaults)
-
-    class Config:
-        frozen = True
-        smart_union = True
-        json_encoders = {dt.datetime: serialize_datetime}