airtrain 0.1.32__tar.gz → 0.1.35__tar.gz

{airtrain-0.1.32 → airtrain-0.1.35}/PKG-INFO

@@ -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 → airtrain-0.1.35}/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-0.1.35/airtrain/cli/main.py

@@ -0,0 +1,110 @@
+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(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()
+def cli():
+    """Airtrain CLI - Your AI Agent Building Assistant"""
+    pass
+
+
+@cli.command()
+@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"""
+    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__":
+    cli()

{airtrain-0.1.32 → airtrain-0.1.35}/airtrain.egg-info/PKG-INFO

@@ -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 → airtrain-0.1.35}/airtrain.egg-info/requires.txt

@@ -30,3 +30,4 @@ build>=0.10.0
 types-PyYAML>=6.0
 types-requests>=2.31.0
 types-Markdown>=3.5.0
+toml>=0.10.2

airtrain-0.1.35/pyproject.toml

@@ -0,0 +1,45 @@
+[build-system]
+requires = [ "setuptools>=61.0",]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "airtrain"
+version = "0.1.35"
+description = "A platform for building and deploying AI agents with structured skills"
+readme = "README.md"
+requires-python = ">=3.8"
+classifiers = [ "Development Status :: 3 - Alpha", "Intended Audience :: Developers", "License :: OSI Approved :: MIT License", "Operating System :: OS Independent", "Programming Language :: Python :: 3", "Programming Language :: Python :: 3.8", "Programming Language :: Python :: 3.9", "Programming Language :: Python :: 3.10",]
+dependencies = [ "pydantic>=2.10.6", "openai>=1.60.1", "python-dotenv>=1.0.1", "PyYAML>=6.0.2", "firebase-admin>=6.6.0", "loguru>=0.7.3", "requests>=2.32.3", "boto3>=1.36.6", "together>=1.3.13", "anthropic>=0.45.0", "groq>=0.15.0", "cerebras-cloud-sdk>=1.19.0", "google-genai>=1.0.0", "fireworks-ai>=0.15.12", "google-generativeai>=0.8.4", "click>=8.0.0", "rich>=13.3.1", "prompt-toolkit>=3.0.36", "colorama>=0.4.6", "typer>=0.9.0",]
+[[project.authors]]
+name = "Dheeraj Pai"
+email = "helloworldcmu@gmail.com"
+
+[project.optional-dependencies]
+dev = [ "black>=24.10.0", "flake8>=7.1.1", "isort>=5.13.0", "mypy>=1.9.0", "pytest>=7.0.0", "twine>=4.0.0", "build>=0.10.0", "types-PyYAML>=6.0", "types-requests>=2.31.0", "types-Markdown>=3.5.0", "toml>=0.10.2",]
+
+[project.urls]
+Homepage = "https://github.com/rosaboyle/airtrain.dev"
+Documentation = "https://docs.airtrain.dev/"
+
+[project.scripts]
+airtrain = "airtrain.cli.main:main"
+
+[tool.black]
+line-length = 88
+target-version = [ "py38",]
+include = "\\.pyi?$"
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3
+include_trailing_comma = true
+force_grid_wrap = 0
+use_parentheses = true
+line_length = 88
+
+[tool.mypy]
+python_version = "3.8"
+warn_return_any = true
+warn_unused_configs = true
+disallow_untyped_defs = true
+check_untyped_defs = true

airtrain-0.1.32/airtrain/cli/main.py

@@ -1,53 +0,0 @@
-import click
-from airtrain.integrations.openai.skills import OpenAIChatSkill, OpenAIInput
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-
-
-def initialize_chat():
-    return OpenAIChatSkill()
-
-
-@click.group()
-def cli():
-    """Airtrain CLI - Your AI Agent Building Assistant"""
-    pass
-
-
-@cli.command()
-def chat():
-    """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()
-
-
-if __name__ == "__main__":
-    main()

airtrain-0.1.32/pyproject.toml

@@ -1,87 +0,0 @@
-[build-system]
-requires = ["setuptools>=61.0"]
-build-backend = "setuptools.build_meta"
-
-[project]
-name = "airtrain"
-version = "0.1.32"
-authors = [
-    { name="Dheeraj Pai", email="helloworldcmu@gmail.com" },
-]
-description = "A platform for building and deploying AI agents with structured skills"
-readme = "README.md"
-requires-python = ">=3.8"
-classifiers = [
-    "Development Status :: 3 - Alpha",
-    "Intended Audience :: Developers",
-    "License :: OSI Approved :: MIT License",
-    "Operating System :: OS Independent",
-    "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.8",
-    "Programming Language :: Python :: 3.9",
-    "Programming Language :: Python :: 3.10",
-]
-
-dependencies = [
-    "pydantic>=2.10.6",
-    "openai>=1.60.1",
-    "python-dotenv>=1.0.1",
-    "PyYAML>=6.0.2",
-    "firebase-admin>=6.6.0",
-    "loguru>=0.7.3",
-    "requests>=2.32.3",
-    "boto3>=1.36.6",
-    "together>=1.3.13",
-    "anthropic>=0.45.0",
-    "groq>=0.15.0",
-    "cerebras-cloud-sdk>=1.19.0",
-    "google-genai>=1.0.0",
-    "fireworks-ai>=0.15.12",
-    "google-generativeai>=0.8.4",
-    "click>=8.0.0",
-    "rich>=13.3.1",
-    "prompt-toolkit>=3.0.36",
-    "colorama>=0.4.6",
-    "typer>=0.9.0",
-]
-
-[project.optional-dependencies]
-dev = [
-    "black>=24.10.0",
-    "flake8>=7.1.1",
-    "isort>=5.13.0",
-    "mypy>=1.9.0",
-    "pytest>=7.0.0",
-    "twine>=4.0.0",
-    "build>=0.10.0",
-    "types-PyYAML>=6.0",
-    "types-requests>=2.31.0",
-    "types-Markdown>=3.5.0",
-]
-
-[project.urls]
-"Homepage" = "https://github.com/rosaboyle/airtrain.dev"
-"Documentation" = "https://docs.airtrain.dev/"
-
-[tool.black]
-line-length = 88
-target-version = ['py38']
-include = '\.pyi?$'
-
-[tool.isort]
-profile = "black"
-multi_line_output = 3
-include_trailing_comma = true
-force_grid_wrap = 0
-use_parentheses = true
-line_length = 88
-
-[tool.mypy]
-python_version = "3.8"
-warn_return_any = true
-warn_unused_configs = true
-disallow_untyped_defs = true
-check_untyped_defs = true
-
-[project.scripts]
-airtrain = "airtrain.cli.main:main"