airtrain 0.1.37__py3-none-any.whl → 0.1.38__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.37"
+__version__ = "0.1.38"
 
 # Core imports
 from .core.skills import Skill, ProcessingError

airtrain/builder/__init__.py

@@ -0,0 +1,3 @@
+from .agent_builder import AgentBuilder, AgentSpecification
+
+__all__ = ["AgentBuilder", "AgentSpecification"]

airtrain/builder/agent_builder.py

@@ -0,0 +1,122 @@
+from typing import Dict, List, Optional
+from pydantic import BaseModel, Field
+from airtrain.integrations.fireworks.skills import FireworksChatSkill, FireworksInput
+from airtrain.core.skills import ProcessingError
+import json
+
+
+class AgentSpecification(BaseModel):
+    """Model to capture agent specifications"""
+
+    name: str = Field(..., description="Name of the agent")
+    purpose: str = Field(..., description="Primary purpose of the agent")
+    input_type: str = Field(..., description="Type of input the agent accepts")
+    output_type: str = Field(..., description="Type of output the agent produces")
+    required_skills: List[str] = Field(
+        default_factory=list, description="Skills required by the agent"
+    )
+    conversation_style: str = Field(
+        ..., description="Style of conversation (formal, casual, technical, etc.)"
+    )
+    safety_constraints: List[str] = Field(
+        default_factory=list, description="Safety constraints for the agent"
+    )
+    reasoning: Optional[str] = Field(
+        None, description="Reasoning behind agent design decisions"
+    )
+
+
+class AgentBuilder:
+    """AI-powered agent builder"""
+
+    def __init__(self):
+        self.skill = FireworksChatSkill()
+        self.system_prompt = """You are an expert AI Agent architect. Your role is to help users build AI agents by:
+1. Understanding their requirements through targeted questions
+2. Designing appropriate agent architectures
+3. Selecting optimal skills and models
+4. Ensuring safety and ethical constraints
+5. Providing clear reasoning for all decisions
+
+Ask one question at a time. Wait for user response before proceeding.
+Start by asking about the primary purpose of the agent they want to build.
+
+Your responses must be in this format:
+QUESTION: [Your question here]
+CONTEXT: [Brief context about why this question is important]
+
+When creating the final specification, output valid JSON matching the AgentSpecification schema."""
+
+    def _get_next_question(self, conversation_history: List[Dict[str, str]]) -> str:
+        input_data = FireworksInput(
+            user_input="Based on the conversation so far, what's the next question to ask?",
+            system_prompt=self.system_prompt,
+            model="accounts/fireworks/models/deepseek-r1",
+            temperature=0.7,
+            conversation_history=conversation_history,
+        )
+
+        try:
+            result = self.skill.process(input_data)
+            return result.response
+        except Exception as e:
+            raise ProcessingError(f"Failed to generate next question: {str(e)}")
+
+    def _create_specification(
+        self, conversation_history: List[Dict[str, str]]
+    ) -> AgentSpecification:
+        input_data = FireworksInput(
+            user_input="Based on our conversation, create a complete agent specification in valid JSON format.",
+            system_prompt=self.system_prompt,
+            model="accounts/fireworks/models/deepseek-r1",
+            temperature=0.7,
+            conversation_history=conversation_history,
+        )
+
+        result = self.skill.process(input_data)
+
+        try:
+            # Extract JSON from the response (it might be wrapped in markdown or other text)
+            json_str = result.response
+            if "```json" in json_str:
+                json_str = json_str.split("```json")[1].split("```")[0].strip()
+            elif "```" in json_str:
+                json_str = json_str.split("```")[1].split("```")[0].strip()
+
+            return AgentSpecification.model_validate_json(json_str)
+        except Exception as e:
+            raise ProcessingError(f"Failed to parse agent specification: {str(e)}")
+
+    def build_agent(self) -> AgentSpecification:
+        conversation_history = []
+
+        print("\nWelcome to the AI Agent Builder!")
+        print("I'll help you create a custom AI agent through a series of questions.\n")
+
+        while True:
+            next_question = self._get_next_question(conversation_history)
+            print(f"\n{next_question}")
+
+            user_input = input("\nYour response (type 'done' when finished): ").strip()
+
+            if user_input.lower() == "done":
+                if len(conversation_history) < 6:  # Minimum questions needed
+                    print(
+                        "\nPlease answer a few more questions to create a complete specification."
+                    )
+                    continue
+                try:
+                    return self._create_specification(conversation_history)
+                except ProcessingError as e:
+                    print(f"\nError creating specification: {str(e)}")
+                    print(
+                        "Let's continue with a few more questions to gather complete information."
+                    )
+                    continue
+
+            conversation_history.extend(
+                [
+                    {"role": "assistant", "content": next_question},
+                    {"role": "user", "content": user_input},
+                ]
+            )

{airtrain-0.1.37.dist-info → airtrain-0.1.38.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.37
+Version: 0.1.38
 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.37.dist-info → airtrain-0.1.38.dist-info}/RECORD

@@ -1,5 +1,7 @@
-airtrain/__init__.py,sha256=WbNEBdO9_ZcI0yNOBlO_zo7g8oZ_piJcTBuG6doFpNo,2099
+airtrain/__init__.py,sha256=mw-sMkN9E8eDGEJ0qNfMzAnDgDrNzA6dt19aE0rFM1M,2099
 airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
+airtrain/builder/__init__.py,sha256=D33sr0k_WAe6FAJkk8rUaivEzFaeVqLXkQgyFWEhfPU,110
+airtrain/builder/agent_builder.py,sha256=3XnGUAcK_6lWoUDtL0TanliQZuh7u0unhNbnrz1z2-I,5018
 airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 airtrain/cli/builder.py,sha256=cI0FZCfRrgXPmjt8lOnHZwCWKrOB2doaOn49kmxVxHs,669
 airtrain/cli/main.py,sha256=Sf4-I4ENlZZ_AehPEaf1rdebZEMCv5K41sSZTnrH9y4,3456
@@ -61,8 +63,8 @@ airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9
 airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
 airtrain/integrations/together/skills.py,sha256=8DwkexMJu1Gm6QmNDfNasYStQ31QsXBbFP99zR-YCf0,7598
 airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
-airtrain-0.1.37.dist-info/METADATA,sha256=bnCDFNSAXHCwH7K8Gdnile1QREhrOB19MIdua51Ktr8,5375
-airtrain-0.1.37.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-airtrain-0.1.37.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
-airtrain-0.1.37.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.37.dist-info/RECORD,,
+airtrain-0.1.38.dist-info/METADATA,sha256=2PlIeaOOUFSrfhqEkQ5M9CM7hX8jSEd4tGt5ZvKwwas,5375
+airtrain-0.1.38.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.38.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
+airtrain-0.1.38.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.38.dist-info/RECORD,,