airtrain 0.1.40__py3-none-any.whl → 0.1.41__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.40"
+__version__ = "0.1.41"
 
 # Core imports
 from .core.skills import Skill, ProcessingError

airtrain/integrations/together/__init__.py

@@ -2,5 +2,18 @@
 
 from .credentials import TogetherAICredentials
 from .skills import TogetherAIChatSkill
+from .list_models import (
+    TogetherListModelsSkill,
+    TogetherListModelsInput,
+    TogetherListModelsOutput,
+)
+from .models import TogetherModel
 
-__all__ = ["TogetherAICredentials", "TogetherAIChatSkill"]
+__all__ = [
+    "TogetherAICredentials", 
+    "TogetherAIChatSkill",
+    "TogetherListModelsSkill",
+    "TogetherListModelsInput",
+    "TogetherListModelsOutput",
+    "TogetherModel",
+]

airtrain/integrations/together/list_models.py

@@ -0,0 +1,77 @@
+from typing import Optional
+import requests
+from pydantic import Field
+
+from airtrain.core.skills import Skill, ProcessingError
+from airtrain.core.schemas import InputSchema, OutputSchema
+from .credentials import TogetherAICredentials
+from .models import TogetherModel
+
+
+class TogetherListModelsInput(InputSchema):
+    """Schema for Together AI list models input"""
+    pass
+
+
+class TogetherListModelsOutput(OutputSchema):
+    """Schema for Together AI list models output"""
+    
+    data: list[TogetherModel] = Field(
+        default_factory=list,
+        description="List of Together AI models"
+    )
+    object: Optional[str] = Field(
+        default=None, 
+        description="Object type"
+    )
+
+
+class TogetherListModelsSkill(Skill[TogetherListModelsInput, TogetherListModelsOutput]):
+    """Skill for listing Together AI models"""
+
+    input_schema = TogetherListModelsInput
+    output_schema = TogetherListModelsOutput
+    
+    def __init__(self, credentials: Optional[TogetherAICredentials] = None):
+        """Initialize the skill with optional credentials"""
+        super().__init__()
+        self.credentials = credentials or TogetherAICredentials.from_env()
+        self.base_url = "https://api.together.xyz/v1"
+    
+    def process(
+        self, input_data: TogetherListModelsInput
+    ) -> TogetherListModelsOutput:
+        """Process the input and return a list of models."""
+        try:
+            # Build the URL
+            url = f"{self.base_url}/models"
+            
+            # Make the request
+            headers = {
+                "Authorization": (
+                    f"Bearer {self.credentials.together_api_key.get_secret_value()}"
+                ),
+                "accept": "application/json"
+            }
+            
+            response = requests.get(url, headers=headers)
+            response.raise_for_status()
+            
+            # Parse the response
+            result = response.json()
+            
+            # Convert the models to TogetherModel objects
+            models = []
+            for model_data in result.get("data", []):
+                models.append(TogetherModel(**model_data))
+            
+            # Return the output
+            return TogetherListModelsOutput(
+                data=models,
+                object=result.get("object")
+            )
+            
+        except requests.RequestException as e:
+            raise ProcessingError(f"Failed to list Together AI models: {str(e)}")
+        except Exception as e:
+            raise ProcessingError(f"Error listing Together AI models: {str(e)}") 

airtrain/integrations/together/models.py

@@ -1,6 +1,5 @@
-from typing import List, Optional
+from typing import List, Optional, Dict, Any
 from pydantic import BaseModel, Field, validator
-from pathlib import Path
 
 
 class TogetherAIImageInput(BaseModel):
@@ -30,7 +29,7 @@ class TogetherAIImageInput(BaseModel):
             if width <= 0 or height <= 0:
                 raise ValueError
             return v
-        except:
+        except ValueError:
             raise ValueError("Size must be in format WIDTHxHEIGHT (e.g., 1024x1024)")
 
 
@@ -54,3 +53,43 @@ class TogetherAIImageOutput(BaseModel):
     usage: dict = Field(
         default_factory=dict, description="Usage statistics and billing information"
     )
+
+
+class TogetherModel(BaseModel):
+    """Schema for Together AI model"""
+
+    id: str = Field(..., description="Model ID")
+    name: Optional[str] = Field(None, description="Model name")
+    object: Optional[str] = Field(None, description="Object type")
+    created: Optional[int] = Field(None, description="Creation timestamp")
+    owned_by: Optional[str] = Field(None, description="Model owner")
+    root: Optional[str] = Field(None, description="Root model identifier")
+    parent: Optional[str] = Field(None, description="Parent model identifier")
+    permission: Optional[List[Dict[str, Any]]] = Field(
+        None, description="Permission details"
+    )
+    metadata: Optional[Dict[str, Any]] = Field(
+        None, description="Additional metadata for the model"
+    )
+    description: Optional[str] = Field(None, description="Model description")
+    pricing: Optional[Dict[str, Any]] = Field(None, description="Pricing information")
+    context_length: Optional[int] = Field(
+        None, description="Maximum context length supported by the model"
+    )
+    capabilities: Optional[List[str]] = Field(
+        None, description="Model capabilities"
+    )
+
+
+class TogetherListModelsInput(BaseModel):
+    """Schema for listing Together AI models input"""
+    pass
+
+
+class TogetherListModelsOutput(BaseModel):
+    """Schema for listing Together AI models output"""
+    
+    data: List[TogetherModel] = Field(
+        ..., description="List of Together AI models"
+    )
+    object: Optional[str] = Field(None, description="Object type")

{airtrain-0.1.40.dist-info → airtrain-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: airtrain
-Version: 0.1.40
+Version: 0.1.41
 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.40.dist-info → airtrain-0.1.41.dist-info}/RECORD

@@ -1,4 +1,4 @@
-airtrain/__init__.py,sha256=fRBP9B8-SN5didcE8h1LLAPTuQTxiODu_-bbKGWLDR4,2099
+airtrain/__init__.py,sha256=KrDT-Fn1iruBhdnav_vK-wkcO6-ZfwcOCDukwElOxEU,2099
 airtrain/__main__.py,sha256=EU8ffFmCdC1G-UcHHt0Oo3lB1PGqfC6kwzH39CnYSwU,72
 airtrain/builder/__init__.py,sha256=D33sr0k_WAe6FAJkk8rUaivEzFaeVqLXkQgyFWEhfPU,110
 airtrain/builder/agent_builder.py,sha256=3XnGUAcK_6lWoUDtL0TanliQZuh7u0unhNbnrz1z2-I,5018
@@ -52,21 +52,22 @@ airtrain/integrations/openai/skills.py,sha256=1dvRJYrnU2hOmGRlkHBtyR6P8D7aIwHZfU
 airtrain/integrations/sambanova/__init__.py,sha256=dp_263iOckM_J9pOEvyqpf3FrejD6-_x33r0edMCTe0,179
 airtrain/integrations/sambanova/credentials.py,sha256=JyN8sbMCoXuXAjim46aI3LTicBijoemS7Ao0rn4yBJU,824
 airtrain/integrations/sambanova/skills.py,sha256=SZ_GAimMiOCILiNkzyhNflyRR6bdC5r0Tnog19K8geU,4997
-airtrain/integrations/together/__init__.py,sha256=we4KXn_pUs6Dxo3QcB-t40BSRraQFdKg2nXw7yi2FjM,185
+airtrain/integrations/together/__init__.py,sha256=P6AH0kWgYmXGOupbLqBXCsY-KV1tcCV9UtvmJ6BGqFE,463
 airtrain/integrations/together/audio_models_config.py,sha256=GtqfmKR1vJ5x4B3kScvEO3x4exvzwNP78vcGVTk_fBE,1004
 airtrain/integrations/together/credentials.py,sha256=cYNhyIwgsxm8LfiFfT-omBvgV3mUP6SZeRSukyzzDlI,747
 airtrain/integrations/together/embedding_models_config.py,sha256=F0ISAXCG_Pcnf-ojkvZwIXacXD8LaU8hQmGHCFzmlds,2927
 airtrain/integrations/together/image_models_config.py,sha256=JlCozrphI9zE4uYpGfj4DCWSN6GZGyr84Tb1HmjNQ28,2455
 airtrain/integrations/together/image_skill.py,sha256=wQ8wSzfL-QHpM_esYGLNXf8ciOPPsz-QJw6zSrxZT68,5214
-airtrain/integrations/together/models.py,sha256=ZW5xfEN9fU18aeltb-sB2O-Bnu5sLkDPZqvUtxgoH-U,2112
+airtrain/integrations/together/list_models.py,sha256=BFq_w3Rz9WP2gKIaQNNIyUJUaYkz-FCSEbNMClccrsY,2580
+airtrain/integrations/together/models.py,sha256=q5KsouOK7IvyzGZ7nhSjTpZw-CcLfPghJr6o_UU9uMo,3652
 airtrain/integrations/together/models_config.py,sha256=XMKp0Oq1nWWnMMdNAZxkFXmJaURwWrwLE18kFXsMsRw,8829
 airtrain/integrations/together/rerank_models_config.py,sha256=coCg0IOG2tU4L2uc2uPtPdoBwGjSc_zQwxENwdDuwHE,1188
 airtrain/integrations/together/rerank_skill.py,sha256=gjH24hLWCweWKPyyfKZMG3K_g9gWzm80WgiJNjkA9eg,1894
 airtrain/integrations/together/schemas.py,sha256=pBMrbX67oxPCr-sg4K8_Xqu1DWbaC4uLCloVSascROg,1210
 airtrain/integrations/together/skills.py,sha256=8DwkexMJu1Gm6QmNDfNasYStQ31QsXBbFP99zR-YCf0,7598
 airtrain/integrations/together/vision_models_config.py,sha256=m28HwYDk2Kup_J-a1FtynIa2ZVcbl37kltfoHnK8zxs,1544
-airtrain-0.1.40.dist-info/METADATA,sha256=3-KnsKsMdriztxTjdEWoJF599N3oicxpw2x17-mrQaw,5375
-airtrain-0.1.40.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
-airtrain-0.1.40.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
-airtrain-0.1.40.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
-airtrain-0.1.40.dist-info/RECORD,,
+airtrain-0.1.41.dist-info/METADATA,sha256=mOYF47bkfI4rQJFBkcmnHZ47u1Pqh6e_S8-4Ps3KmGg,5375
+airtrain-0.1.41.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+airtrain-0.1.41.dist-info/entry_points.txt,sha256=rrJ36IUsyq6n1dSfTWXqVAgpQLPRWDfCqwd6_3B-G0U,52
+airtrain-0.1.41.dist-info/top_level.txt,sha256=cFWW1vY6VMCb3AGVdz6jBDpZ65xxBRSqlsPyySxTkxY,9
+airtrain-0.1.41.dist-info/RECORD,,