airtrain 0.1.10__py3-none-any.whl → 0.1.12__py3-none-any.whl

airtrain/__init__.py

@@ -1,9 +1,60 @@
 """Airtrain - A platform for building and deploying AI agents with structured skills"""
 
-__version__ = "0.1.10"
+__version__ = "0.1.12"
 
-from .core.skills import Skill
+# Core imports
+from .core.skills import Skill, ProcessingError
 from .core.schemas import InputSchema, OutputSchema
 from .core.credentials import BaseCredentials
 
-__all__ = ["Skill", "InputSchema", "OutputSchema", "BaseCredentials"]
+# Integration imports - Credentials
+from .integrations.openai.credentials import OpenAICredentials
+from .integrations.aws.credentials import AWSCredentials
+from .integrations.google.credentials import GoogleCloudCredentials
+from .integrations.anthropic.credentials import AnthropicCredentials
+from .integrations.groq.credentials import GroqCredentials
+from .integrations.together.credentials import TogetherAICredentials
+from .integrations.ollama.credentials import OllamaCredentials
+from .integrations.sambanova.credentials import SambanovaCredentials
+from .integrations.cerebras.credentials import CerebrasCredentials
+
+# Integration imports - Skills
+from .integrations.openai.skills import OpenAIChatSkill, OpenAIParserSkill
+from .integrations.anthropic.skills import AnthropicChatSkill
+from .integrations.aws.skills import AWSBedrockSkill
+from .integrations.google.skills import VertexAISkill
+from .integrations.groq.skills import GroqChatSkill
+from .integrations.together.skills import TogetherAIChatSkill
+from .integrations.ollama.skills import OllamaChatSkill
+from .integrations.sambanova.skills import SambanovaChatSkill
+from .integrations.cerebras.skills import CerebrasChatSkill
+
+__all__ = [
+    # Core
+    "Skill",
+    "ProcessingError",
+    "InputSchema",
+    "OutputSchema",
+    "BaseCredentials",
+    # Credentials
+    "OpenAICredentials",
+    "AWSCredentials",
+    "GoogleCloudCredentials",
+    "AnthropicCredentials",
+    "GroqCredentials",
+    "TogetherAICredentials",
+    "OllamaCredentials",
+    "SambanovaCredentials",
+    "CerebrasCredentials",
+    # Skills
+    "OpenAIChatSkill",
+    "OpenAIParserSkill",
+    "AnthropicChatSkill",
+    "AWSBedrockSkill",
+    "VertexAISkill",
+    "GroqChatSkill",
+    "TogetherAIChatSkill",
+    "OllamaChatSkill",
+    "SambanovaChatSkill",
+    "CerebrasChatSkill",
+]

airtrain/contrib/__init__.py

@@ -0,0 +1,29 @@
+"""Airtrain contrib package for community contributions"""
+
+from .travel.agents import (
+    TravelAgentBase,
+    ClothingAgent,
+    HikingAgent,
+    InternetConnectivityAgent,
+    FoodRecommendationAgent,
+    PersonalizedRecommendationAgent,
+)
+from .travel.models import (
+    ClothingRecommendation,
+    HikingOption,
+    InternetAvailability,
+    FoodOption,
+)
+
+__all__ = [
+    "TravelAgentBase",
+    "ClothingAgent",
+    "HikingAgent",
+    "InternetConnectivityAgent",
+    "FoodRecommendationAgent",
+    "PersonalizedRecommendationAgent",
+    "ClothingRecommendation",
+    "HikingOption",
+    "InternetAvailability",
+    "FoodOption",
+]

airtrain/contrib/travel/__init__.py

@@ -0,0 +1,35 @@
+"""Travel related agents and models"""
+
+from .agents import (
+    TravelAgentBase,
+    ClothingAgent,
+    HikingAgent,
+    InternetConnectivityAgent,
+    FoodRecommendationAgent,
+    PersonalizedRecommendationAgent,
+)
+from .models import (
+    ClothingRecommendation,
+    HikingOption,
+    InternetAvailability,
+    FoodOption,
+)
+from .agents.verification_agent import UserVerificationAgent
+from .models.verification import UserTravelInfo, TravelCompanion, HealthCondition
+
+__all__ = [
+    "TravelAgentBase",
+    "ClothingAgent",
+    "HikingAgent",
+    "InternetConnectivityAgent",
+    "FoodRecommendationAgent",
+    "PersonalizedRecommendationAgent",
+    "ClothingRecommendation",
+    "HikingOption",
+    "InternetAvailability",
+    "FoodOption",
+    "UserVerificationAgent",
+    "UserTravelInfo",
+    "TravelCompanion",
+    "HealthCondition",
+]

airtrain/contrib/travel/agents.py

@@ -0,0 +1,243 @@
+from typing import Optional, List, Any
+from airtrain.core.skills import Skill, ProcessingError
+from airtrain.integrations.openai.skills import OpenAIParserSkill, OpenAIParserInput
+from .models import (
+    ClothingRecommendation,
+    HikingOption,
+    InternetAvailability,
+    FoodOption,
+    PersonalizedRecommendation,
+)
+
+
+class TravelAgentBase(OpenAIParserSkill):
+    def __init__(self, credentials=None):
+        super().__init__(credentials)
+        self.model = "gpt-4o"
+        self.temperature = 0.0
+
+
+class ClothingAgent(TravelAgentBase):
+    """Agent for clothing recommendations"""
+
+    def get_recommendations(
+        self, location: str, duration: int, activities: List[str], season: str
+    ) -> ClothingRecommendation:
+        prompt = f"""
+        Provide clothing recommendations for a {duration}-day trip to {location} during {season}.
+        Activities planned: {', '.join(activities)}.
+        Include essential items, weather-specific clothing, and cultural considerations.
+        """
+
+        input_data = OpenAIParserInput(
+            user_input=prompt,
+            system_prompt="You are a travel clothing expert. Provide detailed packing recommendations.",
+            response_model=ClothingRecommendation,
+            model=self.model,
+            temperature=self.temperature,
+        )
+
+        result = self.process(input_data)
+        return result.parsed_response
+
+
+class HikingAgent(TravelAgentBase):
+    """Agent for hiking recommendations"""
+
+    def get_hiking_options(
+        self, location: str, difficulty: str, duration_preference: float
+    ) -> List[HikingOption]:
+        prompt = f"""
+        Find hiking trails in {location} that match:
+        - Difficulty level: {difficulty}
+        - Preferred duration: around {duration_preference} hours
+        Provide detailed trail information and safety tips.
+        """
+
+        input_data = OpenAIParserInput(
+            user_input=prompt,
+            system_prompt="You are a hiking expert. Recommend suitable trails with safety considerations.",
+            response_model=List[HikingOption],
+            model=self.model,
+        )
+
+        result = self.process(input_data)
+        return result.parsed_response
+
+
+class InternetAgent(TravelAgentBase):
+    """Agent for internet availability information"""
+
+    def get_connectivity_info(
+        self,
+        location: str,
+        duration: int,
+        work_requirements: Optional[List[str]] = None,
+    ) -> InternetAvailability:
+        prompt = f"""
+        Provide detailed internet connectivity information for {location} for a {duration}-day stay.
+        {f'Work requirements: {", ".join(work_requirements)}' if work_requirements else ''}
+        Include public WiFi spots, recommended providers, and connectivity tips.
+        """
+
+        input_data = OpenAIParserInput(
+            user_input=prompt,
+            system_prompt="You are a connectivity expert. Provide detailed internet availability information.",
+            response_model=InternetAvailability,
+            model=self.model,
+            temperature=self.temperature,
+        )
+
+        result = self.process(input_data)
+        return result.parsed_response
+
+
+class FoodAgent(TravelAgentBase):
+    """Agent for food recommendations"""
+
+    def get_food_recommendations(
+        self,
+        location: str,
+        dietary_restrictions: Optional[List[str]] = None,
+        preferences: Optional[List[str]] = None,
+        budget_level: str = "medium",
+    ) -> FoodOption:
+        prompt = f"""
+        Provide food recommendations for {location}.
+        {f'Dietary restrictions: {", ".join(dietary_restrictions)}' if dietary_restrictions else ''}
+        {f'Food preferences: {", ".join(preferences)}' if preferences else ''}
+        Budget level: {budget_level}
+        Include local specialties, restaurant recommendations, and food safety tips.
+        """
+
+        input_data = OpenAIParserInput(
+            user_input=prompt,
+            system_prompt="You are a culinary expert. Provide detailed food recommendations.",
+            response_model=FoodOption,
+            model=self.model,
+            temperature=self.temperature,
+        )
+
+        result = self.process(input_data)
+        return result.parsed_response
+
+
+class PersonalizedAgent(TravelAgentBase):
+    """Agent for personalized travel recommendations"""
+
+    def get_personalized_recommendations(
+        self,
+        location: str,
+        duration: int,
+        interests: List[str],
+        budget_level: str,
+        travel_style: str,
+        previous_destinations: Optional[List[str]] = None,
+    ) -> PersonalizedRecommendation:
+        prompt = f"""
+        Create personalized travel recommendations for {location} for {duration} days.
+        Interests: {', '.join(interests)}
+        Travel style: {travel_style}
+        Budget level: {budget_level}
+        {f'Previous destinations: {", ".join(previous_destinations)}' if previous_destinations else ''}
+        Include hidden gems, local events, and a custom itinerary.
+        """
+
+        input_data = OpenAIParserInput(
+            user_input=prompt,
+            system_prompt="You are a personal travel consultant. Provide tailored recommendations.",
+            response_model=PersonalizedRecommendation,
+            model=self.model,
+            temperature=self.temperature,
+        )
+
+        result = self.process(input_data)
+        return result.parsed_response
+
+
+# Similar pattern for other agents...
+
+if __name__ == "__main__":
+    # Initialize all agents
+    clothing_agent = ClothingAgent()
+    hiking_agent = HikingAgent()
+    internet_agent = InternetAgent()
+    food_agent = FoodAgent()
+    personalized_agent = PersonalizedAgent()
+
+    # Example location and common parameters
+    location = "Kyoto, Japan"
+    duration = 7
+    season = "spring"
+
+    try:
+        # Get clothing recommendations
+        clothing_result = clothing_agent.get_recommendations(
+            location=location,
+            duration=duration,
+            activities=["hiking", "temple visits", "city walking", "photography"],
+            season=season,
+        )
+        print("\n=== Clothing Recommendations ===")
+        print(f"Essential items: {', '.join(clothing_result.essential_items)}")
+        print(f"Weather specific: {', '.join(clothing_result.weather_specific)}")
+        print(
+            f"Cultural considerations: {', '.join(clothing_result.cultural_considerations)}"
+        )
+
+        # Get hiking options
+        hiking_result = hiking_agent.get_hiking_options(
+            location=location, difficulty="moderate", duration_preference=4.0
+        )
+        print("\n=== Hiking Options ===")
+        for trail in hiking_result:
+            print(f"\nTrail: {trail.trail_name}")
+            print(f"Difficulty: {trail.difficulty}")
+            print(f"Duration: {trail.duration_hours} hours")
+            print(f"Distance: {trail.distance_km} km")
+
+        # Get internet availability
+        internet_result = internet_agent.get_connectivity_info(
+            location=location,
+            duration=duration,
+            work_requirements=["video calls", "cloud storage access"],
+        )
+        print("\n=== Internet Availability ===")
+        print(f"General availability: {internet_result.general_availability}")
+        print(f"Average speed: {internet_result.average_speed_mbps} Mbps")
+        print(
+            f"Recommended providers: {', '.join(internet_result.recommended_providers)}"
+        )
+
+        # Get food recommendations
+        food_result = food_agent.get_food_recommendations(
+            location=location,
+            dietary_restrictions=["vegetarian"],
+            preferences=["traditional", "local specialties"],
+            budget_level="medium",
+        )
+        print("\n=== Food Recommendations ===")
+        print("Local specialties:", ", ".join(food_result.local_specialties))
+        print("Must-try dishes:", ", ".join(food_result.must_try_dishes))
+        for restaurant in food_result.recommended_restaurants[:3]:  # Show top 3
+            print(f"Restaurant: {restaurant['name']} - {restaurant['type']}")
+
+        # Get personalized recommendations
+        personal_result = personalized_agent.get_personalized_recommendations(
+            location=location,
+            duration=duration,
+            interests=["photography", "culture", "nature", "food"],
+            budget_level="medium",
+            travel_style="balanced",
+            previous_destinations=["Tokyo", "Seoul"],
+        )
+        print("\n=== Personalized Recommendations ===")
+        print("Hidden gems:", ", ".join(personal_result.hidden_gems))
+        print("\nCustom Itinerary:")
+        for day in personal_result.custom_itinerary:
+            print(f"Day {day['day']}: {day['activities']}")
+
+    except ProcessingError as e:
+        print(f"Error processing travel recommendations: {str(e)}")
+    except Exception as e:
+        print(f"Unexpected error: {str(e)}")

airtrain/contrib/travel/models.py

@@ -0,0 +1,59 @@
+from typing import List, Optional, Dict, Any
+from pydantic import BaseModel, Field, validator
+
+
+class ClothingRecommendation(BaseModel):
+    """Model for clothing recommendations"""
+
+    essential_items: List[str]
+    weather_specific: List[str]
+    activity_specific: List[str]
+    cultural_considerations: List[str]
+    packing_tips: List[str]
+
+
+class HikingOption(BaseModel):
+    """Model for hiking recommendations"""
+
+    trail_name: str
+    difficulty: str
+    duration_hours: float
+    distance_km: float
+    elevation_gain_m: float
+    best_season: List[str]
+    required_gear: List[str]
+    safety_tips: List[str]
+    highlights: List[str]
+
+
+class InternetAvailability(BaseModel):
+    """Model for internet availability information"""
+
+    general_availability: str
+    average_speed_mbps: float
+    public_wifi_spots: List[str]
+    recommended_providers: List[str]
+    connectivity_tips: List[str]
+    offline_alternatives: List[str]
+
+
+class FoodOption(BaseModel):
+    """Model for food recommendations"""
+
+    local_specialties: List[str]
+    recommended_restaurants: List[Dict[str, str]]
+    dietary_considerations: List[str]
+    food_safety_tips: List[str]
+    price_ranges: Dict[str, str]
+    must_try_dishes: List[str]
+
+
+class PersonalizedRecommendation(BaseModel):
+    """Model for personalized recommendations"""
+
+    activities: List[Dict[str, str]]
+    hidden_gems: List[str]
+    local_events: List[Dict[str, str]]
+    custom_itinerary: List[Dict[str, Any]]
+    safety_tips: List[str]
+    budget_recommendations: Dict[str, str]

airtrain/integrations/google/credentials.py

@@ -1,6 +1,7 @@
 from pydantic import Field, SecretStr
 from airtrain.core.credentials import BaseCredentials, CredentialValidationError
-from google.cloud import storage
+
+# from google.cloud import storage
 
 
 class GoogleCloudCredentials(BaseCredentials):

{airtrain-0.1.10.dist-info → airtrain-0.1.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.10
+Version: 0.1.12
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai
@@ -25,6 +25,7 @@ Requires-Dist: requests>=2.32.3
 Requires-Dist: boto3>=1.36.6
 Requires-Dist: together>=1.3.13
 Requires-Dist: anthropic>=0.45.0
+Requires-Dist: groq>=0.15.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

{airtrain-0.1.10.dist-info → airtrain-0.1.12.dist-info}/RECORD

@@ -1,4 +1,8 @@
-airtrain/__init__.py,sha256=QlWDodHKi49nnPROPULRasxMFzKN_R6iz8tPk5xT4rM,313
+airtrain/__init__.py,sha256=rk7ANjWmvBbBrDNBi_3_UsNXya3ifz8Qa9RLDFcrdrk,2095
+airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
+airtrain/contrib/travel/__init__.py,sha256=rx_zYv8Qt5-DDiGT3H8IyGmEuBlNe3r71p_erYJxaLc,847
+airtrain/contrib/travel/agents.py,sha256=tpQtZ0WUiXBuhvZtc2JlEam5TuR5l-Tndi14YyImDBM,8975
+airtrain/contrib/travel/models.py,sha256=E4Mtds5TINmXLXu65aTYqv6wwOh2KclJHZ2eXRrBqiY,1547
 airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
 airtrain/core/credentials.py,sha256=PgQotrQc46J5djidKnkK1znUv3fyNkUFDO-m2Kn_Gzo,4006
 airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
@@ -14,7 +18,7 @@ airtrain/integrations/cerebras/__init__.py,sha256=zAD-qV38OzHhMCz1z-NvjjqcYEhURb
 airtrain/integrations/cerebras/credentials.py,sha256=IFkn8LxMAaOpvEWXDpb94VQGtqcDxQ7rZHKH-tX4Nuw,884
 airtrain/integrations/cerebras/skills.py,sha256=O9vwFzvv_tUOwFOVE8CszAQEac711eVYVUj_8dVMTpc,1596
 airtrain/integrations/google/__init__.py,sha256=INZFNOcNebz3m-Ggk07ZjmX0kNHIbTe_St9gBlZBki8,176
-airtrain/integrations/google/credentials.py,sha256=yyl-MWl06wr4SWvcvJGSpJ3hGTz21ByrRSr_3np5cbU,1030
+airtrain/integrations/google/credentials.py,sha256=Mm4jNWF02rIf0_GuHLcUUPyLHC4NMRdF_iTCoVTQ0Bs,1033
 airtrain/integrations/google/skills.py,sha256=uwmgetl5Ien7fLOA5HIZdqoL6AZnexFDyzfsrGuJ1RU,1606
 airtrain/integrations/groq/__init__.py,sha256=B_X2fXbsJfFD6GquKeVCsEJjwd9Ygbq1uEHlV4Jy7YE,154
 airtrain/integrations/groq/credentials.py,sha256=A8-VIyoZTkHFQb-O-lmu-UrgaLZ3hfWfzzigkYteESk,829
@@ -32,7 +36,7 @@ airtrain/integrations/sambanova/skills.py,sha256=Po1ur_QFwzVIugbkk2mt73WdXDz_Gr9
 airtrain/integrations/together/__init__.py,sha256=we4KXn_pUs6Dxo3QcB-t40BSRraQFdKg2nXw7yi2FjM,185
 airtrain/integrations/together/credentials.py,sha256=y5M6ZQrfYJLJbClxEasq4HaVyZM0l5lFshwVP6jq2E4,720
 airtrain/integrations/together/skills.py,sha256=YMOULyk2TX32rCjhxK29e4ehn8iIzMXpg3xmdYtuyQQ,1664
-airtrain-0.1.10.dist-info/METADATA,sha256=frRmuMVDKKHudrTIAIaIkIecEMYwkcb3BAbjGYxYZqA,4508
-airtrain-0.1.10.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-airtrain-0.1.10.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.10.dist-info/RECORD,,
+airtrain-0.1.12.dist-info/METADATA,sha256=NGt35x8Yel7vDKoe-AzD77E14JK5IG6fw7kR8tiNX4Y,4536
+airtrain-0.1.12.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.12.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.12.dist-info/RECORD,,