orbitalsai 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

orbitalsai/__init__.py

@@ -20,7 +20,7 @@ Example:
 from .client import Client
 from .async_client import AsyncClient
 from .models import (
-    TranscriptTask, Transcript, Balance, UsageHistory, DailyUsage, User,
+    TranscriptTask, Transcript, Balance, UsageHistory, DailyUsage, User, Model,
     SUPPORTED_LANGUAGES, SUPPORTED_AUDIO_FORMATS, SUPPORTED_AUDIO_MIMETYPES
 )
 from .exceptions import (
@@ -45,6 +45,7 @@ __all__ = [
     "UsageHistory",
     "DailyUsage",
     "User",
+    "Model",
     
     # Constants
     "SUPPORTED_LANGUAGES",

orbitalsai/async_client.py

@@ -11,7 +11,7 @@ from datetime import datetime, date
 
 from .models import (
     TranscriptTask, Transcript, Balance, UsageHistory, DailyUsage, 
-    User, APIKey, UsageRecord, DailyUsageRecord
+    User, APIKey, UsageRecord, DailyUsageRecord, Model
 )
 from .exceptions import (
     OrbitalsAIError, AuthenticationError, InsufficientBalanceError,
@@ -101,11 +101,33 @@ class AsyncClient:
         except aiohttp.ClientError as e:
             raise OrbitalsAIError(f"Request failed: {str(e)}")
     
+    async def get_models(self) -> List[Model]:
+        """
+        Get all available AI models with their pricing information.
+        
+        Returns:
+            List of Model objects with pricing details
+        """
+        response = await self._make_request("GET", "/models")
+        
+        models = []
+        for model_data in response:
+            models.append(Model(
+                id=model_data["id"],
+                model_name=model_data["model_name"],
+                transcription_rate_per_second=model_data["transcription_rate_per_second"],
+                transcription_rate_per_hour=model_data["transcription_rate_per_hour"],
+                is_active=model_data["is_active"]
+            ))
+        
+        return models
+    
     async def transcribe(
         self, 
         file_path: str, 
         language: str = "english", 
         generate_srt: bool = False,
+        model_name: str = "Perigee-1",
         wait: bool = True,
         timeout: int = 300,
         poll_interval: int = 5
@@ -117,6 +139,7 @@ class AsyncClient:
             file_path: Path to the audio file
             language: Language of the audio (default: "english")
             generate_srt: Whether to generate SRT subtitles (default: False)
+            model_name: AI model to use for transcription (default: "Perigee-1")
             wait: Whether to wait for completion (default: True)
             timeout: Maximum time to wait in seconds (default: 300)
             poll_interval: Seconds to wait between status checks (default: 5)
@@ -141,6 +164,7 @@ class AsyncClient:
             data.add_field('file', f, filename=file_path.split('/')[-1], content_type='audio/mpeg')
             data.add_field('language', language)
             data.add_field('generate_srt', str(generate_srt).lower())
+            data.add_field('model_name', model_name)
             
             # Upload file
             response = await self._make_request("POST", "/audio/upload", data=data)

orbitalsai/client.py

@@ -12,7 +12,7 @@ from pathlib import Path
 
 from .models import (
     TranscriptTask, Transcript, Balance, UsageHistory, DailyUsage, 
-    User, APIKey, UsageRecord, DailyUsageRecord
+    User, APIKey, UsageRecord, DailyUsageRecord, Model
 )
 from .exceptions import (
     OrbitalsAIError, AuthenticationError, InsufficientBalanceError,
@@ -89,11 +89,33 @@ class Client:
         except requests.exceptions.RequestException as e:
             raise OrbitalsAIError(f"Request failed: {str(e)}")
     
+    def get_models(self) -> List[Model]:
+        """
+        Get all available AI models with their pricing information.
+        
+        Returns:
+            List of Model objects with pricing details
+        """
+        response = self._make_request("GET", "/models")
+        
+        models = []
+        for model_data in response:
+            models.append(Model(
+                id=model_data["id"],
+                model_name=model_data["model_name"],
+                transcription_rate_per_second=model_data["transcription_rate_per_second"],
+                transcription_rate_per_hour=model_data["transcription_rate_per_hour"],
+                is_active=model_data["is_active"]
+            ))
+        
+        return models
+    
     def transcribe(
         self, 
         file_path: str, 
         language: str = "english", 
         generate_srt: bool = False,
+        model_name: str = "Perigee-1",
         wait: bool = True,
         timeout: int = 300,
         poll_interval: int = 5
@@ -105,6 +127,7 @@ class Client:
             file_path: Path to the audio file
             language: Language of the audio (default: "english")
             generate_srt: Whether to generate SRT subtitles (default: False)
+            model_name: AI model to use for transcription (default: "Perigee-1")
             wait: Whether to wait for completion (default: True)
             timeout: Maximum time to wait in seconds (default: 300)
             poll_interval: Seconds to wait between status checks (default: 5)
@@ -128,7 +151,8 @@ class Client:
             files = {"file": (Path(file_path).name, f, "audio/mpeg")}
             data = {
                 "language": language,
-                "generate_srt": str(generate_srt).lower()
+                "generate_srt": str(generate_srt).lower(),
+                "model_name": model_name
             }
             
             # Upload file
@@ -249,54 +273,6 @@ class Client:
             last_updated=datetime.fromisoformat(response["last_updated"].replace('Z', '+00:00'))
         )
     
-    def get_usage_history(
-        self, 
-        start_date: Optional[datetime] = None, 
-        end_date: Optional[datetime] = None,
-        page: int = 1,
-        page_size: int = 50
-    ) -> UsageHistory:
-        """
-        Get usage history for the current user.
-        
-        Args:
-            start_date: Start date for the history (default: 30 days ago)
-            end_date: End date for the history (default: now)
-            page: Page number (default: 1)
-            page_size: Number of records per page (default: 50)
-            
-        Returns:
-            UsageHistory object with usage records
-        """
-        params = {"page": page, "page_size": page_size}
-        if start_date:
-            params["start_date"] = start_date.isoformat()
-        if end_date:
-            params["end_date"] = end_date.isoformat()
-        
-        response = self._make_request("GET", "/billing/usage-history", params=params)
-        
-        records = []
-        for record_data in response["records"]:
-            records.append(UsageRecord(
-                id=record_data["id"],
-                service_type=record_data["service_type"],
-                usage_amount=record_data["total_audio_usage"],
-                cost=record_data["cost"],
-                timestamp=datetime.fromisoformat(record_data["timestamp"].replace('Z', '+00:00')),
-                api_key_id=record_data.get("api_key_id")
-            ))
-        
-        return UsageHistory(
-            records=records,
-            total_records=response["total_records"],
-            total_pages=response["total_pages"],
-            current_page=response["current_page"],
-            start_date=datetime.fromisoformat(response["start_date"].replace('Z', '+00:00')),
-            end_date=datetime.fromisoformat(response["end_date"].replace('Z', '+00:00')),
-            period_summary=response["period_summary"]
-        )
-    
     def get_daily_usage(
         self, 
         start_date: Optional[date] = None, 

orbitalsai/models.py

@@ -114,6 +114,16 @@ class APIKey:
     last_used: Optional[datetime] = None
 
 
+@dataclass
+class Model:
+    """Represents an AI model with pricing information."""
+    id: int
+    model_name: str
+    transcription_rate_per_second: float
+    transcription_rate_per_hour: float
+    is_active: bool
+
+
 # Supported languages and file formats
 SUPPORTED_LANGUAGES = [
     "english", "hausa", "igbo", "yoruba", "swahili", "pidgin", "kinyarwanda"

{orbitalsai-1.0.0.dist-info → orbitalsai-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orbitalsai
-Version: 1.0.0
+Version: 1.1.0
 Summary: A simple and powerful Python SDK for the OrbitalsAI API
 Home-page: https://github.com/orbitalsai/orbitalsai-python-sdk
 Author: OrbitalsAI
@@ -138,27 +138,61 @@ print(transcript.text)
 print(transcript.srt_content)  # SRT subtitle content
 ```
 
-### Don't Wait (Advanced)
+## 🤖 AI Model Selection
+
+Choose which AI model to use for transcription. Different models may have different pricing and capabilities.
+
+### List Available Models
 
 ```python
 import orbitalsai
 
 client = orbitalsai.Client(api_key="your_api_key_here")
 
-# Start transcription without waiting
-task = client.transcribe("audio.mp3", wait=False)
-print(f"Task started: {task.task_id}")
+# Get all available models
+models = client.get_models()
+
+for model in models:
+    print(f"{model.model_name}: ${model.transcription_rate_per_hour:.2f}/hour")
+```
+
+### Transcribe with Specific Model
 
-# Check status later
-status = client.get_task(task.task_id)
-if status.status == "completed":
-    print(status.result_text)
+```python
+import orbitalsai
+
+client = orbitalsai.Client(api_key="your_api_key_here")
+
+# Transcribe with Perigee-1 model
+transcript = client.transcribe(
+    "audio.mp3",
+    language="hausa",
+    model_name="Perigee-1"  # Specify the model
+)
 
-# Or wait for completion
-transcript = client.wait_for_task(task.task_id)
 print(transcript.text)
 ```
 
+### Choose Model Based on Budget
+
+```python
+import orbitalsai
+
+client = orbitalsai.Client(api_key="your_api_key_here")
+
+# Get the cheapest available model
+models = client.get_models()
+cheapest_model = min(models, key=lambda m: m.transcription_rate_per_hour)
+
+print(f"Using {cheapest_model.model_name} at ${cheapest_model.transcription_rate_per_hour:.2f}/hour")
+
+transcript = client.transcribe(
+    "audio.mp3",
+    language="english",
+    model_name=cheapest_model.model_name
+)
+```
+
 ## 🔄 Async Usage
 
 Perfect for processing multiple files or building web applications.
@@ -169,11 +203,15 @@ import orbitalsai
 
 async def main():
     async with orbitalsai.AsyncClient(api_key="your_api_key_here") as client:
+        # List available models
+        models = await client.get_models()
+        print(f"Available models: {[m.model_name for m in models]}")
+        
         # Transcribe multiple files concurrently
         tasks = await asyncio.gather(
-            client.transcribe("audio1.mp3"),
-            client.transcribe("audio2.wav"),
-            client.transcribe("audio3.m4a")
+            client.transcribe("audio1.mp3", model_name="Perigee-1"),
+            client.transcribe("audio2.wav", model_name="Perigee-1"),
+            client.transcribe("audio3.m4a", model_name="Perigee-1")
         )
         
         for transcript in tasks:
@@ -202,7 +240,7 @@ print(f"Last updated: {balance.last_updated}")
 import orbitalsai
 from datetime import date, timedelta
 
-client = orbitalsai.Client(api_key="QDOBQ2PJ.4ThV4fwk-27hpBvh8pYVvAdOOQlA4Lk1fJQdI6EL9Yk")
+client = orbitalsai.Client(api_key="your_api_key_here")
 
 # Get last 7 days of usage
 end_date = date.today()
@@ -280,7 +318,13 @@ except Exception as e:
 
 ### Client Methods
 
-#### `transcribe(file_path, language="english", generate_srt=False, wait=True, timeout=300, poll_interval=5)`
+#### `get_models()`
+
+Get all available AI models with their pricing information.
+
+**Returns:** List of `Model` objects
+
+#### `transcribe(file_path, language="english", generate_srt=False, model_name="Perigee-1", wait=True, timeout=300, poll_interval=5)`
 
 Transcribe an audio file.
 
@@ -288,6 +332,7 @@ Transcribe an audio file.
 - `file_path` (str): Path to the audio file
 - `language` (str): Language code (default: "english")
 - `generate_srt` (bool): Generate SRT subtitles (default: False)
+- `model_name` (str): AI model to use (default: "Perigee-1")
 - `wait` (bool): Wait for completion (default: True)
 - `timeout` (int): Maximum wait time in seconds (default: 300)
 - `poll_interval` (int): Seconds between status checks (default: 5)
@@ -367,6 +412,13 @@ Get current user details.
 - `balance` (float): Current balance in credits
 - `last_updated` (datetime): Last update timestamp
 
+#### `Model`
+- `id` (int): Model ID
+- `model_name` (str): Name of the model (e.g., "Perigee-1")
+- `transcription_rate_per_second` (float): Cost per second of audio
+- `transcription_rate_per_hour` (float): Cost per hour of audio
+- `is_active` (bool): Whether the model is currently available
+
 ## 🌍 Supported Languages
 
 - **English** (`english`)

orbitalsai-1.1.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+orbitalsai/__init__.py,sha256=l1PSfVe3tVHU53VWOMc7c6qx3iBP_SPyRFN98-taCo4,1634
+orbitalsai/async_client.py,sha256=zW8eJrGTnu7Z0mDKo6y8b732s_oGpM9zXWWukj_QQhQ,14983
+orbitalsai/client.py,sha256=5pUFDVWXfRbrMVUx7Pw7lV7fvEJfaCqexmNk8aTOHjo,12352
+orbitalsai/exceptions.py,sha256=5jdLYiWUXsbpifFV4-rY4FRZtmRqnqvqXTlunh5DxgE,1390
+orbitalsai/models.py,sha256=ktmZJoYRnEuCZ1bGhR8vy6eBXkXC-wNMdA11CalZ458,3373
+orbitalsai/utils.py,sha256=HtRCWfg7xmk3c_KPqm2-P5kx178QZYxy2hDGv1IS9hM,2791
+orbitalsai-1.1.0.dist-info/licenses/LICENSE,sha256=C7Eee8rAV6XiweKYlvcabxiMvFX75B_0arFlrLnfnf0,1067
+orbitalsai-1.1.0.dist-info/METADATA,sha256=wcuxtbOZRljBf1Y4DRtJO_GetzmRJILwiOzHNQDmSoo,13278
+orbitalsai-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+orbitalsai-1.1.0.dist-info/top_level.txt,sha256=KZ4ueNRg-PyNeu3RKTOpe5PRCSidh3ieL70-YGf62ro,11
+orbitalsai-1.1.0.dist-info/RECORD,,

orbitalsai-1.0.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-orbitalsai/__init__.py,sha256=cpl690jw9L93AYPqOFV_1kKVdfZI6UUiVl2av5KLtPg,1614
-orbitalsai/async_client.py,sha256=4-_AtYLNtSZ97xgLfiVebxTRLY9I9xglmFWDT9rVteA,14054
-orbitalsai/client.py,sha256=G7zuiyeQatWKqfIbiSMJ2wtCjQdf-aingeQ2UE9vDk4,13344
-orbitalsai/exceptions.py,sha256=5jdLYiWUXsbpifFV4-rY4FRZtmRqnqvqXTlunh5DxgE,1390
-orbitalsai/models.py,sha256=8SEo8zfxTdO4Rlhrhey21bILv2Vq4qwFQDIzhwPp9aA,3156
-orbitalsai/utils.py,sha256=HtRCWfg7xmk3c_KPqm2-P5kx178QZYxy2hDGv1IS9hM,2791
-orbitalsai-1.0.0.dist-info/licenses/LICENSE,sha256=C7Eee8rAV6XiweKYlvcabxiMvFX75B_0arFlrLnfnf0,1067
-orbitalsai-1.0.0.dist-info/METADATA,sha256=ysRwoHUQEzZe8X67cTtj9Jkn-ZQAjX4hA9T2E0l2Bxo,11853
-orbitalsai-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-orbitalsai-1.0.0.dist-info/top_level.txt,sha256=KZ4ueNRg-PyNeu3RKTOpe5PRCSidh3ieL70-YGf62ro,11
-orbitalsai-1.0.0.dist-info/RECORD,,