airtrain 0.1.61__tar.gz → 0.1.62__tar.gz

{airtrain-0.1.61 → airtrain-0.1.62}/PKG-INFO

@@ -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 → airtrain-0.1.62}/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-0.1.61 → airtrain-0.1.62}/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-0.1.61 → airtrain-0.1.62}/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-0.1.61 → airtrain-0.1.62}/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-0.1.61 → airtrain-0.1.62}/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-0.1.61 → airtrain-0.1.62}/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 → airtrain-0.1.62}/airtrain.egg-info/PKG-INFO

@@ -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 → airtrain-0.1.62}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "airtrain"
-version = "0.1.61"
+version = "0.1.62"
 description = "A platform for building and deploying AI agents with structured skills"
 readme = "README.md"
 requires-python = ">=3.8"