airtrain 0.1.3__tar.gz → 0.1.4__tar.gz

airtrain-0.1.4/MANIFEST.in

@@ -0,0 +1,3 @@
+include changelog.md
+include README.md
+include requirements.txt 

airtrain-0.1.4/PKG-INFO

@@ -0,0 +1,222 @@
+Metadata-Version: 2.4
+Name: airtrain
+Version: 0.1.4
+Summary: A platform for building and deploying AI agents with structured skills
+Home-page: https://github.com/rosaboyle/airtrain.dev
+Author: Dheeraj Pai
+Author-email: Dheeraj Pai <helloworldcmu@gmail.com>
+Project-URL: Homepage, https://github.com/rosaboyle/airtrain.dev
+Project-URL: Documentation, https://docs.airtrain.dev/
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: pydantic>=2.10.6
+Requires-Dist: openai>=1.60.1
+Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: PyYAML>=6.0.2
+Requires-Dist: firebase-admin>=6.6.0
+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
+Requires-Dist: cerebras-cloud-sdk>=1.19.0
+Requires-Dist: google-genai>=1.0.0
+Requires-Dist: fireworks-ai>=0.15.12
+Requires-Dist: google-generativeai>=0.8.4
+Requires-Dist: click>=8.0.0
+Requires-Dist: rich>=13.3.1
+Requires-Dist: prompt-toolkit>=3.0.36
+Requires-Dist: colorama>=0.4.6
+Requires-Dist: typer>=0.9.0
+Requires-Dist: posthog>=3.7.0
+Provides-Extra: dev
+Requires-Dist: black>=24.10.0; extra == "dev"
+Requires-Dist: flake8>=7.1.1; extra == "dev"
+Requires-Dist: isort>=5.13.0; extra == "dev"
+Requires-Dist: mypy>=1.9.0; extra == "dev"
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: twine>=4.0.0; extra == "dev"
+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
+
+A powerful platform for building and deploying AI agents with structured skills and capabilities.
+
+## Features
+
+- **Structured Skills**: Build modular AI skills with defined input/output schemas
+- **Multiple LLM Integrations**: Built-in support for OpenAI and Anthropic models
+- **Structured Outputs**: Parse LLM responses into structured Pydantic models
+- **Credential Management**: Secure handling of API keys and credentials
+- **Type Safety**: Full type hints and Pydantic model support
+- **Image Support**: Handle image inputs for multimodal models
+- **Error Handling**: Robust error handling and logging
+
+## Installation
+
+```bash
+pip install airtrain
+```
+
+## Quick Start
+
+### 1. Basic OpenAI Chat
+
+```python
+from airtrain.integrations.openai.skills import OpenAIChatSkill, OpenAIInput
+
+# Initialize the skill
+skill = OpenAIChatSkill()
+
+# Create input
+input_data = OpenAIInput(
+    user_input="Explain quantum computing in simple terms.",
+    system_prompt="You are a helpful teacher.",
+    max_tokens=500,
+    temperature=0.7
+)
+
+# Get response
+result = skill.process(input_data)
+print(result.response)
+print(f"Tokens Used: {result.usage['total_tokens']}")
+```
+
+### 2. Anthropic Claude Integration
+
+```python
+from airtrain.integrations.anthropic.skills import AnthropicChatSkill, AnthropicInput
+
+# Initialize the skill
+skill = AnthropicChatSkill()
+
+# Create input
+input_data = AnthropicInput(
+    user_input="Explain the theory of relativity.",
+    system_prompt="You are a physics expert.",
+    model="claude-3-opus-20240229",
+    temperature=0.3
+)
+
+# Get response
+result = skill.process(input_data)
+print(result.response)
+print(f"Usage: {result.usage}")
+```
+
+### 3. Structured Output with OpenAI
+
+```python
+from pydantic import BaseModel
+from typing import List
+from airtrain.integrations.openai.skills import OpenAIParserSkill, OpenAIParserInput
+
+# Define your response model
+class PersonInfo(BaseModel):
+    name: str
+    age: int
+    occupation: str
+    skills: List[str]
+
+# Initialize the parser skill
+parser_skill = OpenAIParserSkill()
+
+# Create input with response model
+input_data = OpenAIParserInput(
+    user_input="Tell me about John Doe, a 30-year-old software engineer who specializes in Python and AI",
+    system_prompt="Extract structured information about the person.",
+    response_model=PersonInfo
+)
+
+# Get structured response
+result = parser_skill.process(input_data)
+person_info = result.parsed_response
+print(f"Name: {person_info.name}")
+print(f"Skills: {', '.join(person_info.skills)}")
+```
+
+## Error Handling
+
+All skills include built-in error handling:
+
+```python
+from airtrain.core.skills import ProcessingError
+
+try:
+    result = skill.process(input_data)
+except ProcessingError as e:
+    print(f"Processing failed: {e}")
+```
+
+## Advanced Features
+
+- Image Analysis Support
+- Function Calling
+- Custom Validators
+- Async Processing
+- Token Usage Tracking
+
+For more examples and detailed documentation, visit our [documentation](https://airtrain.readthedocs.io/).
+
+## Documentation
+
+For detailed documentation, visit [our documentation site](https://docs.airtrain.dev/).
+
+## Telemetry
+
+Airtrain collects telemetry data to help improve the library. The data collected includes:
+
+- Agent run information (model used, task description, environment settings)
+- Agent steps and actions (full action details and reasoning)
+- Performance metrics (token usage, execution time, CPU and memory usage)
+- System information (OS, Python version, machine details)
+- Error information (complete stack traces and context)
+- Model usage details (prompts, responses, parameters)
+
+The telemetry helps us identify usage patterns, troubleshoot issues, and improve the library based on real-world usage. The user ID is stored at `~/.cache/airtrain/telemetry_user_id`.
+
+### Disabling Telemetry
+
+Telemetry is enabled by default, but you can disable it if needed:
+
+1. Set an environment variable:
+```bash
+export AIRTRAIN_TELEMETRY_ENABLED=false
+```
+
+2. In your Python code:
+```python
+import os
+os.environ["AIRTRAIN_TELEMETRY_ENABLED"] = "false"
+```
+
+### Viewing Telemetry Debug Information
+
+To see what telemetry data is being sent:
+```python
+os.environ["AIRTRAIN_LOGGING_LEVEL"] = "debug"
+```
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

airtrain-0.1.4/README.md

@@ -0,0 +1,166 @@
+# Airtrain
+
+A powerful platform for building and deploying AI agents with structured skills and capabilities.
+
+## Features
+
+- **Structured Skills**: Build modular AI skills with defined input/output schemas
+- **Multiple LLM Integrations**: Built-in support for OpenAI and Anthropic models
+- **Structured Outputs**: Parse LLM responses into structured Pydantic models
+- **Credential Management**: Secure handling of API keys and credentials
+- **Type Safety**: Full type hints and Pydantic model support
+- **Image Support**: Handle image inputs for multimodal models
+- **Error Handling**: Robust error handling and logging
+
+## Installation
+
+```bash
+pip install airtrain
+```
+
+## Quick Start
+
+### 1. Basic OpenAI Chat
+
+```python
+from airtrain.integrations.openai.skills import OpenAIChatSkill, OpenAIInput
+
+# Initialize the skill
+skill = OpenAIChatSkill()
+
+# Create input
+input_data = OpenAIInput(
+    user_input="Explain quantum computing in simple terms.",
+    system_prompt="You are a helpful teacher.",
+    max_tokens=500,
+    temperature=0.7
+)
+
+# Get response
+result = skill.process(input_data)
+print(result.response)
+print(f"Tokens Used: {result.usage['total_tokens']}")
+```
+
+### 2. Anthropic Claude Integration
+
+```python
+from airtrain.integrations.anthropic.skills import AnthropicChatSkill, AnthropicInput
+
+# Initialize the skill
+skill = AnthropicChatSkill()
+
+# Create input
+input_data = AnthropicInput(
+    user_input="Explain the theory of relativity.",
+    system_prompt="You are a physics expert.",
+    model="claude-3-opus-20240229",
+    temperature=0.3
+)
+
+# Get response
+result = skill.process(input_data)
+print(result.response)
+print(f"Usage: {result.usage}")
+```
+
+### 3. Structured Output with OpenAI
+
+```python
+from pydantic import BaseModel
+from typing import List
+from airtrain.integrations.openai.skills import OpenAIParserSkill, OpenAIParserInput
+
+# Define your response model
+class PersonInfo(BaseModel):
+    name: str
+    age: int
+    occupation: str
+    skills: List[str]
+
+# Initialize the parser skill
+parser_skill = OpenAIParserSkill()
+
+# Create input with response model
+input_data = OpenAIParserInput(
+    user_input="Tell me about John Doe, a 30-year-old software engineer who specializes in Python and AI",
+    system_prompt="Extract structured information about the person.",
+    response_model=PersonInfo
+)
+
+# Get structured response
+result = parser_skill.process(input_data)
+person_info = result.parsed_response
+print(f"Name: {person_info.name}")
+print(f"Skills: {', '.join(person_info.skills)}")
+```
+
+## Error Handling
+
+All skills include built-in error handling:
+
+```python
+from airtrain.core.skills import ProcessingError
+
+try:
+    result = skill.process(input_data)
+except ProcessingError as e:
+    print(f"Processing failed: {e}")
+```
+
+## Advanced Features
+
+- Image Analysis Support
+- Function Calling
+- Custom Validators
+- Async Processing
+- Token Usage Tracking
+
+For more examples and detailed documentation, visit our [documentation](https://airtrain.readthedocs.io/).
+
+## Documentation
+
+For detailed documentation, visit [our documentation site](https://docs.airtrain.dev/).
+
+## Telemetry
+
+Airtrain collects telemetry data to help improve the library. The data collected includes:
+
+- Agent run information (model used, task description, environment settings)
+- Agent steps and actions (full action details and reasoning)
+- Performance metrics (token usage, execution time, CPU and memory usage)
+- System information (OS, Python version, machine details)
+- Error information (complete stack traces and context)
+- Model usage details (prompts, responses, parameters)
+
+The telemetry helps us identify usage patterns, troubleshoot issues, and improve the library based on real-world usage. The user ID is stored at `~/.cache/airtrain/telemetry_user_id`.
+
+### Disabling Telemetry
+
+Telemetry is enabled by default, but you can disable it if needed:
+
+1. Set an environment variable:
+```bash
+export AIRTRAIN_TELEMETRY_ENABLED=false
+```
+
+2. In your Python code:
+```python
+import os
+os.environ["AIRTRAIN_TELEMETRY_ENABLED"] = "false"
+```
+
+### Viewing Telemetry Debug Information
+
+To see what telemetry data is being sent:
+```python
+os.environ["AIRTRAIN_LOGGING_LEVEL"] = "debug"
+```
+
+## Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details. 

airtrain-0.1.4/airtrain/__init__.py

@@ -0,0 +1,149 @@
+"""
+Airtrain: AI Agent Framework
+
+This library provides a flexible framework for building AI agents
+that can complete complex tasks using AI models, skills, and tools.
+"""
+
+__version__ = "0.1.4"
+
+import sys
+
+# Core imports
+from .core import Skill, ProcessingError, InputSchema, OutputSchema, BaseCredentials
+
+# Integration imports - Credentials
+from .integrations import (
+    # OpenAI
+    OpenAICredentials,
+    OpenAIChatSkill,
+    OpenAICompletionSkill,
+    # Anthropic
+    AnthropicCredentials,
+    AnthropicChatSkill,
+    # Together.ai
+    TogetherAICredentials,
+    TogetherChatSkill,
+    # Fireworks
+    FireworksCredentials,
+    FireworksChatSkill,
+    # Google
+    GeminiCredentials,
+    GeminiChatSkill,
+    # Search
+    ExaCredentials,
+    ExaSearchSkill,
+    ExaSearchInputSchema,
+    ExaSearchOutputSchema,
+)
+
+# Integration imports - Skills
+from .integrations.aws.skills import AWSBedrockSkill
+from .integrations.google.skills import GoogleChatSkill
+from .integrations.groq.skills import GroqChatSkill
+from .integrations.ollama.skills import OllamaChatSkill
+from .integrations.sambanova.skills import SambanovaChatSkill
+from .integrations.cerebras.skills import CerebrasChatSkill
+
+# Tool imports
+from .tools import (
+    ToolFactory,
+    register_tool,
+    StatelessTool,
+    StatefulTool,
+    BaseTool,
+    ListDirectoryTool,
+    DirectoryTreeTool,
+    ApiCallTool,
+    ExecuteCommandTool,
+    FindFilesTool,
+    TerminalNavigationTool,
+    SearchTermTool,
+    RunPytestTool,
+)
+
+# Agent imports
+from .agents import (
+    BaseAgent,
+    AgentFactory,
+    register_agent,
+    BaseMemory,
+    ShortTermMemory,
+    LongTermMemory,
+    SharedMemory,
+)
+
+# Telemetry import - must be imported after version is defined
+from .telemetry import telemetry
+from .telemetry import PackageImportTelemetryEvent
+
+
+__all__ = [
+    # Core
+    "Skill",
+    "ProcessingError",
+    "InputSchema",
+    "OutputSchema",
+    "BaseCredentials",
+    # OpenAI Integration
+    "OpenAICredentials",
+    "OpenAIChatSkill",
+    "OpenAICompletionSkill",
+    # Anthropic Integration
+    "AnthropicCredentials",
+    "AnthropicChatSkill",
+    # Together Integration
+    "TogetherAICredentials",
+    "TogetherChatSkill",
+    # Fireworks Integration
+    "FireworksCredentials",
+    "FireworksChatSkill",
+    # Google Integration
+    "GeminiCredentials",
+    "GeminiChatSkill",
+    # Search Integration
+    "ExaCredentials",
+    "ExaSearchSkill",
+    "ExaSearchInputSchema",
+    "ExaSearchOutputSchema",
+    # Tools
+    "ToolFactory",
+    "register_tool",
+    "StatelessTool",
+    "StatefulTool",
+    "BaseTool",
+    "ListDirectoryTool",
+    "DirectoryTreeTool",
+    "ApiCallTool",
+    "ExecuteCommandTool",
+    "FindFilesTool",
+    "TerminalNavigationTool",
+    "SearchTermTool",
+    "RunPytestTool",
+    # Agents
+    "BaseAgent",
+    "AgentFactory",
+    "register_agent",
+    "BaseMemory",
+    "ShortTermMemory",
+    "LongTermMemory",
+    "SharedMemory",
+    # Telemetry - not directly exposed to users
+    # but initialized at import time
+]
+
+# Capture import telemetry
+try:
+    telemetry.capture(
+        PackageImportTelemetryEvent(
+            version=__version__,
+            python_version=(
+                f"{sys.version_info.major}."
+                f"{sys.version_info.minor}."
+                f"{sys.version_info.micro}"
+            ),
+        )
+    )
+except Exception:
+    # Silently continue if telemetry fails
+    pass

airtrain-0.1.4/airtrain/__main__.py

@@ -0,0 +1,4 @@
+from airtrain.cli.main import cli
+
+if __name__ == "__main__":
+    cli()

airtrain-0.1.4/airtrain/agents/__init__.py

@@ -0,0 +1,45 @@
+"""
+Agents package for AirTrain.
+
+This package provides a registry of agents that can be used to build AI systems.
+"""
+
+# Import registry components
+from .registry import (
+    BaseAgent,
+    AgentFactory,
+    register_agent,
+    AgentRegistry
+)
+
+# Import memory components
+from .memory import (
+    BaseMemory,
+    ShortTermMemory,
+    LongTermMemory,
+    SharedMemory,
+    AgentMemoryManager
+)
+
+# Import agent implementations
+from .groq_agent import GroqAgent
+
+__all__ = [
+    # Base classes
+    "BaseAgent",
+    
+    # Registry components
+    "AgentFactory",
+    "register_agent",
+    "AgentRegistry",
+    
+    # Memory components
+    "BaseMemory",
+    "ShortTermMemory",
+    "LongTermMemory",
+    "SharedMemory",
+    "AgentMemoryManager",
+    
+    # Agent implementations
+    "GroqAgent",
+]