airtrain 0.1.3__tar.gz → 0.1.6__tar.gz

airtrain-0.1.6/.gitignore

@@ -0,0 +1,183 @@
+package
+.env
+.mypy_cache
+firebaseadmin.json
+**pyc=
+50mb_test3.bin
+*bin
+**bin
+token.pickle
+temp_workspace
+temp*
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+# PyPI configuration file
+.pypirc

airtrain-0.1.6/EXPERIMENTS/integrations_examples/anthropic_with_image.py

@@ -0,0 +1,43 @@
+import anthropic
+
+client = anthropic.Anthropic(
+    # defaults to os.environ.get("ANTHROPIC_API_KEY")
+    api_key="my_api_key",
+)
+
+# Replace placeholders like {{PR_DESCRIPTION}} with real values,
+# because the SDK does not support variables.
+message = client.messages.create(
+    model="claude-3-5-sonnet-20241022",
+    max_tokens=8192,
+    temperature=0,
+    system="You are an experienced software engineer tasked with reviewing a GitHub Pull Request (PR). Your goal is to analyze the code quality and suggest improvements. Follow these steps carefully:\n\n1. Review the PR description:\n<PR_DESCRIPTION>\n{{PR_DESCRIPTION}}\n</PR_DESCRIPTION>\n\n2. Examine the code changes:\n<CODE_CHANGES>\n{{CODE_CHANGES}}\n</CODE_CHANGES>\n\n3. Consider any existing comments:\n<EXISTING_COMMENTS>\n{{EXISTING_COMMENTS}}\n</EXISTING_COMMENTS>\n\n4. Analyze the code quality:\n   a. Check for adherence to coding standards and best practices\n   b. Evaluate code readability and maintainability\n   c. Assess performance implications\n   d. Look for potential bugs or edge cases\n   e. Consider security implications\n\n5. Suggest improvements:\n   a. Identify areas where the code can be optimized or simplified\n   b. Propose alternative approaches if applicable\n   c. Recommend additional tests or error handling if needed\n\n6. Format your response as follows:\n   <code_review>\n   <quality_analysis>\n   Provide a detailed analysis of the code quality, addressing points 4a-4e.\n   </quality_analysis>\n\n   <improvement_suggestions>\n   List your suggestions for improvement, addressing points 5a-5c. Number each suggestion.\n   </improvement_suggestions>\n\n   <summary>\n   Provide a brief summary of your overall assessment and key recommendations.\n   </summary>\n   </code_review>\n\nRemember to be constructive and specific in your feedback. Use code snippets or pseudocode to illustrate your suggestions when appropriate. If you need clarification on any part of the code or PR description, state your assumptions clearly.\n\nDo not comment on aspects unrelated to code quality or potential improvements. Focus solely on the technical aspects of the code changes presented.",
+    messages=[
+        {
+            "role": "user",
+            "content": [
+                {
+                    "type": "text",
+                    "text": "\nAnalyze the above examples and give me some updates. Analyze this image as well.\n\nOne more image is this. Can you image this as well.",
+                },
+                {
+                    "type": "image",
+                    "source": {
+                        "type": "base64",
+                        "media_type": "image/jpeg",
+                        "data": "<base64_encoded_image>",
+                    },
+                },
+                {
+                    "type": "image",
+                    "source": {
+                        "type": "base64",
+                        "media_type": "image/jpeg",
+                        "data": "<base64_encoded_image>",
+                    },
+                },
+            ],
+        }
+    ],
+)
+print(message.content)

airtrain-0.1.6/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,131 @@
+# 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.3 → airtrain-0.1.6}/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-0.1.3 → airtrain-0.1.6}/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-0.1.6/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/airtrain/integrations/anthropic/credentials.py

@@ -0,0 +1,32 @@
+from pydantic import Field, SecretStr, validator
+from airtrain.core.credentials import BaseCredentials, CredentialValidationError
+from anthropic import Anthropic
+
+
+class AnthropicCredentials(BaseCredentials):
+    """Anthropic API credentials"""
+
+    anthropic_api_key: SecretStr = Field(..., description="Anthropic API key")
+    version: str = Field(default="2023-06-01", description="API Version")
+
+    _required_credentials = {"anthropic_api_key"}
+
+    @validator("anthropic_api_key")
+    def validate_api_key_format(cls, v: SecretStr) -> SecretStr:
+        key = v.get_secret_value()
+        if not key.startswith("sk-ant-"):
+            raise ValueError("Anthropic API key must start with 'sk-ant-'")
+        return v
+
+    async def validate_credentials(self) -> bool:
+        """Validate Anthropic credentials"""
+        try:
+            client = Anthropic(api_key=self.anthropic_api_key.get_secret_value())
+            client.messages.create(
+                model="claude-3-opus-20240229",
+                max_tokens=1,
+                messages=[{"role": "user", "content": "Hi"}],
+            )
+            return True
+        except Exception as e:
+            raise CredentialValidationError(f"Invalid Anthropic credentials: {str(e)}")