athena-intelligence 0.1.41__tar.gz → 0.1.42__tar.gz

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/PKG-INFO

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

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/pyproject.toml

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

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/src/athena/__init__.py

@@ -18,6 +18,7 @@ from .types import (
     SqlResults,
     StatusEnum,
     StructuredParseResult,
+    ToolModels,
     Tools,
     UrlResult,
     ValidationError,
@@ -48,6 +49,7 @@ __all__ = [
     "StatusEnum",
     "StructuredParseInParsingModel",
     "StructuredParseResult",
+    "ToolModels",
     "Tools",
     "UnprocessableEntityError",
     "UrlResult",

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/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.41",
+            "X-Fern-SDK-Version": "0.1.42",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/src/athena/tools/client.py

@@ -14,6 +14,7 @@ from ..types.excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOu
 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:
     import pydantic.v1 as pydantic  # type: ignore
@@ -164,7 +165,7 @@ class ToolsClient:
     def tool_first_workflow(
         self,
         *,
-        model_name: str,
+        model: typing.Optional[ToolModels] = OMIT,
         tool_name: str,
         content: str,
         tool_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
@@ -172,7 +173,7 @@ class ToolsClient:
     ) -> ExcecuteToolFirstWorkflowOut:
         """
         Parameters:
-            - model_name: str.
+            - model: typing.Optional[ToolModels].
 
             - tool_name: str.
 
@@ -182,19 +183,22 @@ class ToolsClient:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from athena import ToolModels
         from athena.client import Athena
 
         client = Athena(
             api_key="YOUR_API_KEY",
         )
         client.tools.tool_first_workflow(
-            model_name="gpt-3.5-turbo",
+            model=ToolModels.MISTRAL_LARGE_0224,
             tool_name="tavily_search",
             content="summarize the website in one paragraph",
             tool_kwargs={"query": "website: www.athenaintelligence.ai"},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"model_name": model_name, "tool_name": tool_name, "content": content}
+        _request: typing.Dict[str, typing.Any] = {"tool_name": tool_name, "content": content}
+        if model is not OMIT:
+            _request["model"] = model
         if tool_kwargs is not OMIT:
             _request["tool_kwargs"] = tool_kwargs
         _response = self._client_wrapper.httpx_client.request(
@@ -374,7 +378,7 @@ class AsyncToolsClient:
     async def tool_first_workflow(
         self,
         *,
-        model_name: str,
+        model: typing.Optional[ToolModels] = OMIT,
         tool_name: str,
         content: str,
         tool_kwargs: typing.Optional[typing.Dict[str, typing.Any]] = OMIT,
@@ -382,7 +386,7 @@ class AsyncToolsClient:
     ) -> ExcecuteToolFirstWorkflowOut:
         """
         Parameters:
-            - model_name: str.
+            - model: typing.Optional[ToolModels].
 
             - tool_name: str.
 
@@ -392,19 +396,22 @@ class AsyncToolsClient:
 
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from athena import ToolModels
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
             api_key="YOUR_API_KEY",
         )
         await client.tools.tool_first_workflow(
-            model_name="gpt-3.5-turbo",
+            model=ToolModels.MISTRAL_LARGE_0224,
             tool_name="tavily_search",
             content="summarize the website in one paragraph",
             tool_kwargs={"query": "website: www.athenaintelligence.ai"},
         )
         """
-        _request: typing.Dict[str, typing.Any] = {"model_name": model_name, "tool_name": tool_name, "content": content}
+        _request: typing.Dict[str, typing.Any] = {"tool_name": tool_name, "content": content}
+        if model is not OMIT:
+            _request["model"] = model
         if tool_kwargs is not OMIT:
             _request["tool_kwargs"] = tool_kwargs
         _response = await self._client_wrapper.httpx_client.request(

{athena_intelligence-0.1.41 → athena_intelligence-0.1.42}/src/athena/types/__init__.py

@@ -17,6 +17,7 @@ from .snippet import Snippet
 from .sql_results import SqlResults
 from .status_enum import StatusEnum
 from .structured_parse_result import StructuredParseResult
+from .tool_models import ToolModels
 from .tools import Tools
 from .url_result import UrlResult
 from .validation_error import ValidationError
@@ -40,6 +41,7 @@ __all__ = [
     "SqlResults",
     "StatusEnum",
     "StructuredParseResult",
+    "ToolModels",
     "Tools",
     "UrlResult",
     "ValidationError",

athena_intelligence-0.1.42/src/athena/types/tool_models.py

@@ -0,0 +1,49 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class ToolModels(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    GPT_4_TURBO = "gpt-4-turbo"
+    GPT_4_TURBO_PREVIEW = "gpt-4-turbo-preview"
+    GPT_4 = "gpt-4"
+    MIXTRAL_SMALL_8_X_7_B_0211 = "mixtral-small-8x7b-0211"
+    MISTRAL_LARGE_0224 = "mistral-large-0224"
+    CLAUDE_3_OPUS_20240229 = "claude-3-opus-20240229"
+    CLAUDE_3_SONNET_20240229 = "claude-3-sonnet-20240229"
+    CLAUDE_3_HAIKU_20240307 = "claude-3-haiku-20240307"
+
+    def visit(
+        self,
+        gpt_4_turbo: typing.Callable[[], T_Result],
+        gpt_4_turbo_preview: typing.Callable[[], T_Result],
+        gpt_4: typing.Callable[[], T_Result],
+        mixtral_small_8_x_7_b_0211: typing.Callable[[], T_Result],
+        mistral_large_0224: 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],
+    ) -> T_Result:
+        if self is ToolModels.GPT_4_TURBO:
+            return gpt_4_turbo()
+        if self is ToolModels.GPT_4_TURBO_PREVIEW:
+            return gpt_4_turbo_preview()
+        if self is ToolModels.GPT_4:
+            return gpt_4()
+        if self is ToolModels.MIXTRAL_SMALL_8_X_7_B_0211:
+            return mixtral_small_8_x_7_b_0211()
+        if self is ToolModels.MISTRAL_LARGE_0224:
+            return mistral_large_0224()
+        if self is ToolModels.CLAUDE_3_OPUS_20240229:
+            return claude_3_opus_20240229()
+        if self is ToolModels.CLAUDE_3_SONNET_20240229:
+            return claude_3_sonnet_20240229()
+        if self is ToolModels.CLAUDE_3_HAIKU_20240307:
+            return claude_3_haiku_20240307()