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

airtrain/cli/main.py

@@ -1,13 +1,23 @@
 import click
+from typing import Optional
 from airtrain.integrations.openai.skills import OpenAIChatSkill, OpenAIInput
+from airtrain.integrations.anthropic.skills import AnthropicChatSkill, AnthropicInput
 import os
 from dotenv import load_dotenv
+import sys
 
+# Load environment variables
 load_dotenv()
 
 
-def initialize_chat():
-    return OpenAIChatSkill()
+def initialize_chat(provider: str = "openai"):
+    """Initialize chat skill based on provider"""
+    if provider == "openai":
+        return OpenAIChatSkill()
+    elif provider == "anthropic":
+        return AnthropicChatSkill()
+    else:
+        raise ValueError(f"Unsupported provider: {provider}")
 
 
 @click.group()
@@ -17,37 +27,84 @@ def cli():
 
 
 @cli.command()
-def chat():
+@click.option(
+    "--provider",
+    type=click.Choice(["openai", "anthropic"]),
+    default="openai",
+    help="The AI provider to use",
+)
+@click.option(
+    "--temperature",
+    type=float,
+    default=0.7,
+    help="Temperature for response generation (0.0-1.0)",
+)
+@click.option(
+    "--system-prompt",
+    type=str,
+    default="You are a helpful AI assistant that helps users build their own AI agents. Be helpful and provide clear explanations.",
+    help="System prompt to guide the model",
+)
+def chat(provider: str, temperature: float, system_prompt: str):
     """Start an interactive chat session with Airtrain"""
-    skill = initialize_chat()
-    click.echo("Welcome to Airtrain! I'm here to help you build your AI Agent.")
-    click.echo("Type 'exit' to end the conversation.\n")
-
-    while True:
-        user_input = click.prompt("You", type=str)
-
-        if user_input.lower() == "exit":
-            click.echo("\nGoodbye! Have a great day!")
-            break
-
-        try:
-            input_data = OpenAIInput(
-                user_input=user_input,
-                system_prompt="You are an AI assistant that helps users build their own AI agents. Be helpful and provide clear explanations.",
-                model="gpt-4o",
-                temperature=0.7,
-            )
-
-            result = skill.process(input_data)
-            click.echo(f"\nAirtrain: {result.response}\n")
-
-        except Exception as e:
-            click.echo(f"\nError: {str(e)}\n")
-
-
-def main():
-    cli()
+    try:
+        skill = initialize_chat(provider)
+
+        click.echo(f"\nWelcome to Airtrain! Using {provider.upper()} as the provider.")
+        click.echo("Type 'exit' to end the conversation.")
+        click.echo("Type 'clear' to clear the conversation history.\n")
+
+        conversation_history = []
+
+        while True:
+            user_input = click.prompt("You", type=str)
+
+            if user_input.lower() == "exit":
+                click.echo("\nGoodbye! Have a great day!")
+                break
+
+            if user_input.lower() == "clear":
+                conversation_history = []
+                click.echo("\nConversation history cleared!")
+                continue
+
+            try:
+                if provider == "openai":
+                    input_data = OpenAIInput(
+                        user_input=user_input,
+                        system_prompt=system_prompt,
+                        conversation_history=conversation_history,
+                        model="gpt-4o",
+                        temperature=temperature,
+                    )
+                else:
+                    input_data = AnthropicInput(
+                        user_input=user_input,
+                        system_prompt=system_prompt,
+                        conversation_history=conversation_history,
+                        model="claude-3-opus-20240229",
+                        temperature=temperature,
+                    )
+
+                result = skill.process(input_data)
+
+                # Update conversation history
+                conversation_history.extend(
+                    [
+                        {"role": "user", "content": user_input},
+                        {"role": "assistant", "content": result.response},
+                    ]
+                )
+
+                click.echo(f"\nAirtrain: {result.response}\n")
+
+            except Exception as e:
+                click.echo(f"\nError: {str(e)}\n")
+
+    except Exception as e:
+        click.echo(f"Failed to initialize chat: {str(e)}")
+        sys.exit(1)
 
 
 if __name__ == "__main__":
-    main()
+    cli()

{airtrain-0.1.32.dist-info → airtrain-0.1.35.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.32
+Version: 0.1.35
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai
@@ -48,6 +48,7 @@ Requires-Dist: build>=0.10.0; extra == "dev"
 Requires-Dist: types-PyYAML>=6.0; extra == "dev"
 Requires-Dist: types-requests>=2.31.0; extra == "dev"
 Requires-Dist: types-Markdown>=3.5.0; extra == "dev"
+Requires-Dist: toml>=0.10.2; extra == "dev"
 Dynamic: author
 Dynamic: home-page
 Dynamic: requires-python

{airtrain-0.1.32.dist-info → airtrain-0.1.35.dist-info}/RECORD

@@ -1,6 +1,6 @@
-airtrain/__init__.py,sha256=pF1FDnJhZuGJNN6mpZWRCaLVjfwV1uXcp5e8m55u-e0,2099
+airtrain/__init__.py,sha256=HxQoDsFRceRrgV-8HMhQ9NQYkwqFREBJoxpqHBVcE40,2099
 airtrain/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-airtrain/cli/main.py,sha256=5tOBEEJILcIjIBB8JL3djgKpVqImaniJBuJyh0DG0Hg,1297
+airtrain/cli/main.py,sha256=Y6QxIcoqctJNn7_HLRvOb3WhF9MhBeK3KacpEV-Xwto,3415
 airtrain/contrib/__init__.py,sha256=pG-7mJ0pBMqp3Q86mIF9bo1PqoBOVSGlnEK1yY1U1ok,641
 airtrain/contrib/travel/__init__.py,sha256=clmBodw4nkTA-DsgjVGcXfJGPaWxIpCZDtdO-8RzL0M,811
 airtrain/contrib/travel/agents.py,sha256=tpQtZ0WUiXBuhvZtc2JlEam5TuR5l-Tndi14YyImDBM,8975
@@ -59,8 +59,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.32.dist-info/METADATA,sha256=Ouqg9FfKefq9PpncN-RIR2ahMp4AvGaJEIJFni8tcZQ,5331
-airtrain-0.1.32.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-airtrain-0.1.32.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
-airtrain-0.1.32.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.32.dist-info/RECORD,,
+airtrain-0.1.35.dist-info/METADATA,sha256=puapNvVJkJ9KZ5BBsnrkQ5aFEetk52M79ePuZd6EOCo,5375
+airtrain-0.1.35.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.35.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
+airtrain-0.1.35.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.35.dist-info/RECORD,,