airtrain 0.1.3__py3-none-any.whl → 0.1.5__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.5"
 
 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.3.dist-info → airtrain-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.3
+Version: 0.1.5
 Summary: A platform for building and deploying AI agents with structured skills
 Home-page: https://github.com/rosaboyle/airtrain.dev
 Author: Dheeraj Pai

airtrain-0.1.5.dist-info/RECORD

@@ -0,0 +1,10 @@
+airtrain/__init__.py,sha256=eiw0x9Hie0dA0nrYtKSmD3XbW1KKOLr640Lu13DZFBs,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.5.dist-info/METADATA,sha256=YJiHLxGUOwqQvLlqis-JQrtmLcpeIP_11T_hn8pIFxg,2786
+airtrain-0.1.5.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+airtrain-0.1.5.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.5.dist-info/RECORD,,

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,,