airtrain 0.1.3__py3-none-any.whl → 0.1.6__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.3"
+__version__ = "0.1.6"
 
 from .core.skills import Skill
 from .core.schemas import InputSchema, OutputSchema

airtrain/core/credentials.py

@@ -5,7 +5,7 @@ from pathlib import Path
 from abc import ABC, abstractmethod
 import dotenv
 from pydantic import BaseModel, Field, SecretStr
-import yaml  # type: ignore
+import yaml
 
 
 class CredentialError(Exception):
@@ -100,7 +100,7 @@ class BaseCredentials(BaseModel):
         else:
             raise ValueError(f"Unsupported file format: {file_path.suffix}")
 
-    def validate_credentials(self) -> None:
+    async def validate_credentials(self) -> bool:
         """Validate that all required credentials are present"""
         missing = []
         for field_name in self._required_credentials:
@@ -114,6 +114,7 @@ class BaseCredentials(BaseModel):
             raise CredentialValidationError(
                 f"Missing required credentials: {', '.join(missing)}"
             )
+        return True
 
     def clear_from_env(self) -> None:
         """Remove credentials from environment variables"""
@@ -122,32 +123,3 @@ class BaseCredentials(BaseModel):
             if env_key in os.environ:
                 del os.environ[env_key]
         self._loaded = False
-
-
-class OpenAICredentials(BaseCredentials):
-    """OpenAI API credentials"""
-
-    api_key: SecretStr = Field(..., description="OpenAI API key")
-    organization_id: Optional[str] = Field(None, description="OpenAI organization ID")
-
-    _required_credentials = {"api_key"}
-
-
-class AWSCredentials(BaseCredentials):
-    """AWS credentials"""
-
-    aws_access_key_id: SecretStr
-    aws_secret_access_key: SecretStr
-    aws_region: str = "us-east-1"
-    aws_session_token: Optional[SecretStr] = None
-
-    _required_credentials = {"aws_access_key_id", "aws_secret_access_key"}
-
-
-class GoogleCloudCredentials(BaseCredentials):
-    """Google Cloud credentials"""
-
-    project_id: str
-    service_account_key: SecretStr
-
-    _required_credentials = {"project_id", "service_account_key"}

airtrain/integrations/__init__.py

@@ -0,0 +1,26 @@
+"""Airtrain integrations package"""
+
+from .openai.credentials import OpenAICredentials
+from .aws.credentials import AWSCredentials
+from .google.credentials import GoogleCloudCredentials
+from .anthropic.credentials import AnthropicCredentials
+from .groq.credentials import GroqCredentials
+from .together.credentials import TogetherAICredentials
+from .ollama.credentials import OllamaCredentials
+from .sambanova.credentials import SambanovaCredentials
+from .cerebras.credentials import CerebrasCredentials
+
+from .anthropic.skills import AnthropicChatSkill
+
+__all__ = [
+    "OpenAICredentials",
+    "AWSCredentials",
+    "GoogleCloudCredentials",
+    "AnthropicCredentials",
+    "AnthropicChatSkill",
+    "GroqCredentials",
+    "TogetherAICredentials",
+    "OllamaCredentials",
+    "SambanovaCredentials",
+    "CerebrasCredentials",
+]

airtrain-0.1.6.dist-info/METADATA

@@ -0,0 +1,164 @@
+Metadata-Version: 2.2
+Name: airtrain
+Version: 0.1.6
+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: helloworldcmu@gmail.com
+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.0.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: python-dotenv>=0.19.0
+Requires-Dist: PyYAML>=5.4.1
+Requires-Dist: firebase-admin>=5.0.0
+Requires-Dist: loguru>=0.5.3
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# 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/).
+
+## 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.6.dist-info/RECORD

@@ -0,0 +1,10 @@
+airtrain/__init__.py,sha256=t0n2IItXdHzaLZf_1zSUmSf3buF-8Y4FJol1LAzSclk,312
+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
+airtrain/core/skills.py,sha256=LljalzeSHK5eQPTAOEAYc5D8Qn1kVSfiz9WgziTD5UM,4688
+airtrain/integrations/__init__.py,sha256=PRKI_A-KE307C4lpXgFAsZA2oFtTl1kt_4CrRUF2rpU,832
+airtrain-0.1.6.dist-info/METADATA,sha256=C0hcwg_Am0cUqrv8vKfNJ_AeMdBfGhxrSEmzDbjJA7o,4380
+airtrain-0.1.6.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.6.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.6.dist-info/RECORD,,

airtrain-0.1.3.dist-info/METADATA

@@ -1,106 +0,0 @@
-Metadata-Version: 2.2
-Name: airtrain
-Version: 0.1.3
-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: helloworldcmu@gmail.com
-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.0.0
-Requires-Dist: openai>=1.0.0
-Requires-Dist: python-dotenv>=0.19.0
-Requires-Dist: PyYAML>=5.4.1
-Requires-Dist: firebase-admin>=5.0.0
-Requires-Dist: loguru>=0.5.3
-Dynamic: author
-Dynamic: author-email
-Dynamic: classifier
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# 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
-- **OpenAI Integration**: Built-in support for OpenAI's GPT models with structured outputs
-- **Credential Management**: Secure handling of API keys and credentials
-- **Type Safety**: Full type hints and Pydantic model support
-- **Async Support**: Both synchronous and asynchronous API implementations
-
-## Installation
-
-```bash
-pip install airtrain
-```
-
-## Quick Start
-
-### Creating a Structured OpenAI Skill
-
-```python
-from airtrain.core.skills import Skill
-from airtrain.core.schemas import InputSchema, OutputSchema
-from pydantic import BaseModel
-from typing import List
-
-# Define your response model
-class PersonInfo(BaseModel):
-    name: str
-    age: int
-    occupation: str
-    skills: List[str]
-
-# Create a skill
-class OpenAIParserSkill(Skill):
-    def process(self, input_data):
-        # Implementation
-        return parsed_response
-
-# Use the skill
-skill = OpenAIParserSkill()
-result = skill.process(input_data)
-```
-
-### Managing Credentials
-
-```python
-from airtrain.core.credentials import OpenAICredentials
-from pathlib import Path
-
-# Load credentials
-creds = OpenAICredentials(
-    api_key="your-api-key",
-    organization_id="optional-org-id"
-)
-
-# Save to environment
-creds.load_to_env()
-```
-
-## Documentation
-
-For detailed documentation, visit [our documentation site](https://docs.airtrain.dev/).
-
-## 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.3.dist-info/RECORD

@@ -1,9 +0,0 @@
-airtrain/__init__.py,sha256=tgJhVEwGGFzwlIrtXHGfUULQOZm8Tb4NOrRCXUTau9g,312
-airtrain/core/__init__.py,sha256=9h7iKwTzZocCPc9bU6j8bA02BokteWIOcO1uaqGMcrk,254
-airtrain/core/credentials.py,sha256=CzUZkAFxrSMC0nq70zybkkJmeIZDYiNBuzfivOTEgH0,4773
-airtrain/core/schemas.py,sha256=MMXrDviC4gRea_QaPpbjgO--B_UKxnD7YrxqZOLJZZU,7003
-airtrain/core/skills.py,sha256=LljalzeSHK5eQPTAOEAYc5D8Qn1kVSfiz9WgziTD5UM,4688
-airtrain-0.1.3.dist-info/METADATA,sha256=mbq-9NwqtCnevHmiKtC4r1jneyDrFyQJE0uvfSOir6s,2786
-airtrain-0.1.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-airtrain-0.1.3.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.3.dist-info/RECORD,,