airtrain 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

airtrain/__init__.py

@@ -1,6 +1,6 @@
 """Airtrain - A platform for building and deploying AI agents with structured skills"""
 
-__version__ = "0.1.9"
+__version__ = "0.1.11"
 
 from .core.skills import Skill
 from .core.schemas import InputSchema, OutputSchema

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,29 @@
+"""Travel related agents and models"""
+
+from .agents import (
+    TravelAgentBase,
+    ClothingAgent,
+    HikingAgent,
+    InternetConnectivityAgent,
+    FoodRecommendationAgent,
+    PersonalizedRecommendationAgent,
+)
+from .models import (
+    ClothingRecommendation,
+    HikingOption,
+    InternetAvailability,
+    FoodOption,
+)
+
+__all__ = [
+    "TravelAgentBase",
+    "ClothingAgent",
+    "HikingAgent",
+    "InternetConnectivityAgent",
+    "FoodRecommendationAgent",
+    "PersonalizedRecommendationAgent",
+    "ClothingRecommendation",
+    "HikingOption",
+    "InternetAvailability",
+    "FoodOption",
+]

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.9.dist-info → airtrain-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.9
+Version: 0.1.11
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai
@@ -24,6 +24,8 @@ Requires-Dist: loguru>=0.7.3
 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.9.dist-info → airtrain-0.1.11.dist-info}/RECORD

@@ -1,4 +1,8 @@
-airtrain/__init__.py,sha256=tG5y_cTnqzViCKyybcr-KpH6IDLcBOSaZXsqHN6-EmQ,312
+airtrain/__init__.py,sha256=9Iz01w_MQoT9YAt5RVbxwQHBt_TR-kr1KI-DdxaYfk8,313
+airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
+airtrain/contrib/travel/__init__.py,sha256=2xWgnyQdHOvfSHcrjeb66Md8P7Jl4DTKOvC6osubQQE,607
+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.9.dist-info/METADATA,sha256=xO55FRnnSnJoVa0GvQ03HA2GJw5IBR4rhTV__ralmRg,4474
-airtrain-0.1.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-airtrain-0.1.9.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.9.dist-info/RECORD,,
+airtrain-0.1.11.dist-info/METADATA,sha256=8qZMiOyqKPZWWYPGuSgXyeAfOzHXuuHGODdnUERYwKA,4536
+airtrain-0.1.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.11.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.11.dist-info/RECORD,,