athena-intelligence 0.1.43__tar.gz → 0.1.45__tar.gz

{athena_intelligence-0.1.43 → athena_intelligence-0.1.45}/PKG-INFO

@@ -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 → athena_intelligence-0.1.45}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.43"
+version = "0.1.45"
 description = ""
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.43 → athena_intelligence-0.1.45}/src/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_intelligence-0.1.43 → athena_intelligence-0.1.45}/src/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_intelligence-0.1.43 → athena_intelligence-0.1.45}/src/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_intelligence-0.1.43 → athena_intelligence-0.1.45}/src/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_intelligence-0.1.43 → athena_intelligence-0.1.45}/src/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/src/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_intelligence-0.1.43/src/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}