airtrain 0.1.61__py3-none-any.whl → 0.1.62__py3-none-any.whl

airtrain/__init__.py

@@ -5,7 +5,7 @@ This library provides a flexible framework for building AI agents
 that can complete complex tasks using AI models, skills, and tools.
 """
 
-__version__ = "0.1.61"
+__version__ = "0.1.62"
 
 import sys
 
@@ -17,19 +17,18 @@ from .integrations import (
     # OpenAI
     OpenAICredentials,
     OpenAIChatSkill,
-    OpenAICompletionSkill,
     # Anthropic
     AnthropicCredentials,
     AnthropicChatSkill,
     # Together.ai
     TogetherAICredentials,
-    TogetherChatSkill,
+    TogetherAIChatSkill,
     # Fireworks
     FireworksCredentials,
     FireworksChatSkill,
     # Google
     GeminiCredentials,
-    GeminiChatSkill,
+    GoogleChatSkill,
     # Search
     ExaCredentials,
     ExaSearchSkill,
@@ -88,19 +87,18 @@ __all__ = [
     # OpenAI Integration
     "OpenAICredentials",
     "OpenAIChatSkill",
-    "OpenAICompletionSkill",
     # Anthropic Integration
     "AnthropicCredentials",
     "AnthropicChatSkill",
     # Together Integration
     "TogetherAICredentials",
-    "TogetherChatSkill",
+    "TogetherAIChatSkill",
     # Fireworks Integration
     "FireworksCredentials",
     "FireworksChatSkill",
     # Google Integration
     "GeminiCredentials",
-    "GeminiChatSkill",
+    "GoogleChatSkill",
     # Search Integration
     "ExaCredentials",
     "ExaSearchSkill",

airtrain/integrations/__init__.py

@@ -3,7 +3,7 @@
 # Credentials imports
 from .openai.credentials import OpenAICredentials
 from .aws.credentials import AWSCredentials
-from .google.credentials import GoogleCloudCredentials
+from .google.credentials import GoogleCloudCredentials, GeminiCredentials
 from .anthropic.credentials import AnthropicCredentials
 from .groq.credentials import GroqCredentials
 from .together.credentials import TogetherAICredentials
@@ -11,6 +11,8 @@ from .ollama.credentials import OllamaCredentials
 from .sambanova.credentials import SambanovaCredentials
 from .cerebras.credentials import CerebrasCredentials
 from .perplexity.credentials import PerplexityCredentials
+from .fireworks.credentials import FireworksCredentials
+from .search.exa.credentials import ExaCredentials
 
 # Skills imports
 from .openai.skills import OpenAIChatSkill, OpenAIParserSkill
@@ -23,6 +25,9 @@ from .ollama.skills import OllamaChatSkill
 from .sambanova.skills import SambanovaChatSkill
 from .cerebras.skills import CerebrasChatSkill
 from .perplexity.skills import PerplexityChatSkill, PerplexityStreamingChatSkill
+from .fireworks.skills import FireworksChatSkill
+from .search.exa.skills import ExaSearchSkill
+from .search.exa import ExaSearchInputSchema, ExaSearchOutputSchema
 
 # Model configurations
 from .openai.models_config import OPENAI_MODELS, OpenAIModelConfig
@@ -48,6 +53,9 @@ __all__ = [
     "SambanovaCredentials",
     "CerebrasCredentials",
     "PerplexityCredentials",
+    "FireworksCredentials",
+    "GeminiCredentials",
+    "ExaCredentials",
     # Skills
     "OpenAIChatSkill",
     "OpenAIParserSkill",
@@ -61,6 +69,10 @@ __all__ = [
     "CerebrasChatSkill",
     "PerplexityChatSkill",
     "PerplexityStreamingChatSkill",
+    "FireworksChatSkill",
+    "ExaSearchSkill",
+    "ExaSearchInputSchema",
+    "ExaSearchOutputSchema",
     # Model configurations
     "OPENAI_MODELS",
     "OpenAIModelConfig",

airtrain/integrations/combined/list_models_factory.py

@@ -17,8 +17,8 @@ from airtrain.integrations.cerebras.credentials import CerebrasCredentials
 from airtrain.integrations.sambanova.credentials import SambanovaCredentials
 from airtrain.integrations.perplexity.credentials import PerplexityCredentials
 
-# Import Perplexity list models
-from airtrain.integrations.perplexity.list_models import PerplexityListModelsSkill
+# Remove this import to avoid circular dependency
+# from airtrain.integrations.perplexity.list_models import PerplexityListModelsSkill
 
 
 # Generic list models input schema
@@ -175,7 +175,8 @@ class ListModelsSkillFactory:
         "groq": GroqListModelsSkill,
         "cerebras": CerebrasListModelsSkill,
         "sambanova": SambanovaListModelsSkill,
-        "perplexity": PerplexityListModelsSkill,
+        # Remove perplexity from this map as we'll handle it separately
+        # "perplexity": PerplexityListModelsSkill,
     }
 
     @classmethod
@@ -194,8 +195,17 @@ class ListModelsSkillFactory:
         """
         provider = provider.lower()
 
+        # Special case for perplexity to avoid circular import
+        if provider == "perplexity":
+            # Import here to avoid circular import
+            from airtrain.integrations.perplexity.list_models import (
+                PerplexityListModelsSkill,
+            )
+
+            return PerplexityListModelsSkill(credentials=credentials)
+
         if provider not in cls._PROVIDER_MAP:
-            supported = ", ".join(cls.get_supported_providers())
+            supported = ", ".join(cls.get_supported_providers() + ["perplexity"])
             raise ValueError(
                 f"Unsupported provider: {provider}. "
                 f"Supported providers are: {supported}"
@@ -207,4 +217,5 @@ class ListModelsSkillFactory:
     @classmethod
     def get_supported_providers(cls):
         """Return a list of supported provider names"""
-        return list(cls._PROVIDER_MAP.keys())
+        # Add perplexity to the list of supported providers
+        return list(cls._PROVIDER_MAP.keys()) + ["perplexity"]

airtrain/integrations/search/exa/schemas.py

@@ -4,7 +4,7 @@ Schemas for Exa Search API.
 This module defines the input and output schemas for the Exa search API.
 """
 
-from typing import Dict, List, Optional, Any, Union, bool
+from typing import Dict, List, Optional, Any, Union
 from pydantic import BaseModel, Field, HttpUrl
 
 from airtrain.core.schemas import InputSchema, OutputSchema

airtrain/integrations/search/exa/skills.py

@@ -11,8 +11,7 @@ from typing import Optional, Dict, Any, List, cast
 
 from pydantic import ValidationError
 
-from airtrain.core.skills import Skill
-from airtrain.core.errors import ProcessingError
+from airtrain.core.skills import Skill, ProcessingError
 from .credentials import ExaCredentials
 from .schemas import ExaSearchInputSchema, ExaSearchOutputSchema, ExaSearchResult
 

airtrain/tools/command.py

@@ -218,7 +218,7 @@ class FindFilesTool(StatelessTool):
         }
 
 
-@register_tool("terminal_navigation")
+@register_tool("terminal_navigation", tool_type="stateful")
 class TerminalNavigationTool(StatefulTool):
     """Tool for navigating through the terminal with state memory."""
 

{airtrain-0.1.61.dist-info → airtrain-0.1.62.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: airtrain
-Version: 0.1.61
+Version: 0.1.62
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai

{airtrain-0.1.61.dist-info → airtrain-0.1.62.dist-info}/RECORD

@@ -1,6 +1,6 @@
-airtrain/__init__.py,sha256=E3yMskC0rkqxxNnnn2IPtzBRsL2qT9UQMa2r9FT7jH4,3409
+airtrain/__init__.py,sha256=gTO8GEuN1gIKjo5HlBOURfVnRtEuv61zF-HMxTXQYQk,3357
 airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
-airtrain/__pycache__/__init__.cpython-313.pyc,sha256=4F7gemzpt0TGbv_Y-BLiSLd3HL8hCsmEw82IzG2gdUk,2816
+airtrain/__pycache__/__init__.cpython-313.pyc,sha256=gNzuKeI25qKWqhRq6MRzf8Otl1qJEekuX_QU0HAx7bE,2759
 airtrain/agents/__init__.py,sha256=r6v5_bblxamRgiaCT8CVhyzaDdWohGM7sSjLgIUpA5s,795
 airtrain/agents/example_agent.py,sha256=0dCS8QXIvUYYkxwyOEMLMdlQ4KgMAerQ56r7NcYGqTw,11681
 airtrain/agents/groq_agent.py,sha256=s-f-cGSrgsR4Jlrv6xzeg2Z0LAEHByhFMYVbTvu-Bkg,10558
@@ -19,10 +19,10 @@ airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
 airtrain/core/credentials.py,sha256=J1jd8vLrOfet0GhLI1J44d35o7skjriBsMgpODmXwfo,5592
 airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
 airtrain/core/skills.py,sha256=kIkI1MwIzuAwyoxdnZ5MGOb70BOB-njbVb_csJEdVvc,9244
-airtrain/core/__pycache__/__init__.cpython-313.pyc,sha256=E5o3fMQeU-E5G9UQoaEPyHXRN6uvDrcSdR88GvPb0cM,534
-airtrain/core/__pycache__/schemas.cpython-313.pyc,sha256=buRnJ0WO8QtpvaviXi7l7Qby6uOZIwmpsTJ2NMUd1o8,8327
-airtrain/core/__pycache__/skills.cpython-313.pyc,sha256=1-Q3pC2JXLvW1GarSKtqS38ImEztLTdWcsIwerJPBkk,11839
-airtrain/integrations/__init__.py,sha256=Fvr8keOQBe4PrxK7Is2B2jGodxY0J4zkcO6wT1T4e6A,2373
+airtrain/core/__pycache__/__init__.cpython-313.pyc,sha256=aHwrmX5Hf3s9pSa8V0PH3ytqldRjB_X9AWwA6IDpzyk,534
+airtrain/core/__pycache__/schemas.cpython-313.pyc,sha256=WpKRuARzTVGt_NDMApl8GNIQgF_jZcHV34z-KYs3g5g,8327
+airtrain/core/__pycache__/skills.cpython-313.pyc,sha256=4EppdsB6rISIJNF8CbyCPHXlMCX1JKETeKxOS8hENLw,11839
+airtrain/integrations/__init__.py,sha256=nGG3gZJ_sEaMzi_utsQCzExgnpHUjua8L9TyzofLQnw,2842
 airtrain/integrations/anthropic/__init__.py,sha256=K741w3v7fWsCknTo38ARqDL0D3HPlwDIvDuuBao9Tto,800
 airtrain/integrations/anthropic/credentials.py,sha256=hlTSw9HX66kYNaeQUtn0JjdZQBMNkzzFOJOoLOOzvcY,1246
 airtrain/integrations/anthropic/list_models.py,sha256=o7FABp0Cq3gs76zOF-CM9ohmYslWT6vK9qtQabV9XzI,3973
@@ -36,7 +36,7 @@ airtrain/integrations/cerebras/credentials.py,sha256=KDEH4r8FGT68L9p34MLZWK65wq_
 airtrain/integrations/cerebras/skills.py,sha256=BJEb_7TglCYAukD3kcx37R8ibnJWdxVrBrwf3ZTYP-4,4924
 airtrain/integrations/combined/__init__.py,sha256=EL_uZs8436abeZA6NbM-gLdur7kJr4YfGGOYybKOhjc,467
 airtrain/integrations/combined/groq_fireworks_skills.py,sha256=Kz8UDU4-Rl71znz3ml9qVMRz669iWx4sUl37iafW0NU,4612
-airtrain/integrations/combined/list_models_factory.py,sha256=3sQWQw3UwyzOzlNKDpRNzZiA-qN7zOIgyom7CDI0OiM,7724
+airtrain/integrations/combined/list_models_factory.py,sha256=pFvIMeAZSRC7e51qknD9LzLkNhNgZL4pItzGNGzZdNY,8263
 airtrain/integrations/fireworks/__init__.py,sha256=GstUg0rYC-7Pg0DVbDXwL5eO1hp3WCSfroWazbGpfi0,545
 airtrain/integrations/fireworks/completion_skills.py,sha256=zxx7aNlum9scQMri5Ek0qN8VfAomhyUp3u8JJo_AFWM,5615
 airtrain/integrations/fireworks/conversation_manager.py,sha256=ifscKHYKWM_NDElin-oTzpRhyoh6pzBnklmMuH5geOY,3706
@@ -75,8 +75,8 @@ airtrain/integrations/sambanova/skills.py,sha256=SZ_GAimMiOCILiNkzyhNflyRR6bdC5r
 airtrain/integrations/search/__init__.py,sha256=r-hCYff5IdgI4IgQyMUzKzc_CVryl1oiweaGC3TwZ-8,405
 airtrain/integrations/search/exa/__init__.py,sha256=YdINIFcuT4pC72obDYx7DHVdMzlEg03BRQJi57SKc3g,486
 airtrain/integrations/search/exa/credentials.py,sha256=yOJIQdWs6RXX2Msk453VHH7FYQffpF946afaEWgVg4o,900
-airtrain/integrations/search/exa/schemas.py,sha256=hCdg-J__3p666h_ep0GbKngbYqQG74XlhYrcFvJBet4,3797
-airtrain/integrations/search/exa/skills.py,sha256=z2k3f-t_WZiBKj82pkmAxtaqm1-P9yU0H2J44vOJbb0,3786
+airtrain/integrations/search/exa/schemas.py,sha256=QsTh-UfCAJ-YCy0TgjC7iJDADBCdY3Eu4NBi9BSGiuk,3791
+airtrain/integrations/search/exa/skills.py,sha256=zCoSOxeJk27umLHdYVZbW5sDz-kcbuBzH0SE0C7lLZg,3754
 airtrain/integrations/together/__init__.py,sha256=5p2v5kvryl-Z8eSMxdrMQtgzynPo6zQ6vZqpxBADA1I,878
 airtrain/integrations/together/audio_models_config.py,sha256=GtqfmKR1vJ5x4B3kScvEO3x4exvzwNP78vcGVTk_fBE,1004
 airtrain/integrations/together/credentials.py,sha256=cYNhyIwgsxm8LfiFfT-omBvgV3mUP6SZeRSukyzzDlI,747
@@ -95,14 +95,14 @@ airtrain/telemetry/__init__.py,sha256=_xDHzSmQvRCqihT0QTmF7bS9yKEl2LaZ3Zq05hXaI3
 airtrain/telemetry/service.py,sha256=6IB2-FL3fGsYYgMIVQ9MNCp9UlSE5cVhLlB3VBYkAnY,5592
 airtrain/telemetry/views.py,sha256=qob1swyLNEk_UIpDi5VTwMDsEsGZhJheFQrGbP8T5hw,8115
 airtrain/tools/__init__.py,sha256=AauO_EEBcK09mkyCuvvVK_Oz0L5DOrnuySViWXCOt6Y,1021
-airtrain/tools/command.py,sha256=-2axK36InyWSHf0F2N9WafAsi6746akUs8er9TWSy6o,13333
+airtrain/tools/command.py,sha256=dxvs6RzppjWmkUe1oMtxOc7w2mFOGFFZ9Gylwnm37Sw,13355
 airtrain/tools/filesystem.py,sha256=-YYdHj_KeSWPYXeRhWhIX9s_KujVA1R5tF3r93zRVTU,6324
 airtrain/tools/network.py,sha256=YR0AtMXDXkhCsXcx7_t2d12ItnKY8XXTmyP1kdj2M4c,3883
 airtrain/tools/registry.py,sha256=K-1H5EipYcDNDx2jdpsEY9gjfV4aNCGI1pY2UsgSpC0,10246
 airtrain/tools/search.py,sha256=MJNi17g6aBPSqbF0ChV8ZgMlzz_PoKSPAIpe_dazdt8,15081
 airtrain/tools/testing.py,sha256=q4ALEPRzukiadY6wFSPY7vA-T1o3XInLhXt18dsf6yY,4397
-airtrain-0.1.61.dist-info/METADATA,sha256=2igreLpEoXOlTprrUkh7KpgSCMy0CZSBlKJneL2WmoQ,6503
-airtrain-0.1.61.dist-info/WHEEL,sha256=DK49LOLCYiurdXXOXwGJm6U4DkHkg4lcxjhqwRa0CP4,91
-airtrain-0.1.61.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
-airtrain-0.1.61.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.61.dist-info/RECORD,,
+airtrain-0.1.62.dist-info/METADATA,sha256=fAOQQht25N594JsKGN6sFuiqjVMPjKMtly_RP6JW9co,6503
+airtrain-0.1.62.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+airtrain-0.1.62.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
+airtrain-0.1.62.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.62.dist-info/RECORD,,

{airtrain-0.1.61.dist-info → airtrain-0.1.62.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.0.2)
+Generator: setuptools (78.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any