paygent-sdk 3.0.0__py3-none-any.whl → 4.0.0__py3-none-any.whl

paygent_sdk/voice_client.py

@@ -21,251 +21,279 @@ from .constants import (
     DeepgramTTSModels,
     ElevenLabsTTSModels,
 )
-from .models import SttModelPricing, TtsModelPricing
+from .models import RawUsageData
 
 
-# STT model pricing (cost per hour in USD)
-STT_MODEL_PRICING: Dict[str, SttModelPricing] = {
-    # Deepgram Models
-    DeepgramSTTModels.FLUX: SttModelPricing(cost_per_hour=0.462),  # $0.462 per hour
-    DeepgramSTTModels.NOVA_3_MONOLINGUAL: SttModelPricing(cost_per_hour=0.462),  # $0.462 per hour
-    DeepgramSTTModels.NOVA_3_MULTILINGUAL: SttModelPricing(cost_per_hour=0.552),  # $0.552 per hour
-    DeepgramSTTModels.NOVA_1: SttModelPricing(cost_per_hour=0.348),  # $0.348 per hour
-    DeepgramSTTModels.NOVA_2: SttModelPricing(cost_per_hour=0.348),  # $0.348 per hour
-    DeepgramSTTModels.ENHANCED: SttModelPricing(cost_per_hour=0.99),  # $0.99 per hour
-    DeepgramSTTModels.BASE: SttModelPricing(cost_per_hour=0.87),  # $0.87 per hour
-    DeepgramSTTModels.REDACTION: SttModelPricing(cost_per_hour=0.12),  # $0.12 per hour (add-on)
-    DeepgramSTTModels.KEYTERM_PROMPTING: SttModelPricing(cost_per_hour=0.072),  # $0.072 per hour (add-on)
-    DeepgramSTTModels.SPEAKER_DIARIZATION: SttModelPricing(cost_per_hour=0.12),  # $0.12 per hour (add-on)
-    # Growth tier models
-    DeepgramSTTModels.GROWTH_NOVA_3_MONOLINGUAL: SttModelPricing(cost_per_hour=0.39),  # $0.39 per hour
-    DeepgramSTTModels.GROWTH_NOVA_3_MULTILINGUAL: SttModelPricing(cost_per_hour=0.468),  # $0.468 per hour
-    DeepgramSTTModels.GROWTH_NOVA_1: SttModelPricing(cost_per_hour=0.282),  # $0.282 per hour
-    DeepgramSTTModels.GROWTH_NOVA_2: SttModelPricing(cost_per_hour=0.282),  # $0.282 per hour
-
-    # Microsoft Azure Speech Service Models
-    MicrosoftAzureSpeechSTTModels.STANDARD: SttModelPricing(cost_per_hour=1.0),  # $1.0 per hour
-    MicrosoftAzureSpeechSTTModels.CUSTOM: SttModelPricing(cost_per_hour=1.2),  # $1.2 per hour
-
-    # Google Cloud Speech-to-Text Models
-    GoogleCloudSpeechSTTModels.STANDARD: SttModelPricing(cost_per_hour=0.96),  # $0.96 per hour
-
-    # AssemblyAI Models
-    AssemblyAISTTModels.UNIVERSAL_STREAMING: SttModelPricing(cost_per_hour=0.15),  # $0.15 per hour
-    AssemblyAISTTModels.UNIVERSAL_STREAMING_MULTILANG: SttModelPricing(cost_per_hour=0.15),  # $0.15 per hour
-    AssemblyAISTTModels.KEYTERMS_PROMPTING: SttModelPricing(cost_per_hour=0.04),  # $0.04 per hour
-
-    # Eleven Labs STT Models
-    ElevenLabsSTTModels.BUSINESS_SCRIBE_V1_V2: SttModelPricing(cost_per_hour=0.22),  # $0.22 per hour
-
-    # Soniox STT Models
-    SonioxSTTModels.REAL_TIME: SttModelPricing(cost_per_hour=0.12),  # $0.12 per hour
-}
-
-
-# TTS model pricing (cost per 1 million characters in USD)
-TTS_MODEL_PRICING: Dict[str, TtsModelPricing] = {
-    # Amazon Polly Models
-    AmazonPollyTTSModels.STANDARD: TtsModelPricing(cost_per_million_characters=0.4),  # $0.4 per 1 million characters
-    AmazonPollyTTSModels.NEURAL: TtsModelPricing(cost_per_million_characters=16.0),  # $16 per 1 million characters
-    AmazonPollyTTSModels.LONG_FORM: TtsModelPricing(cost_per_million_characters=100.0),  # $100 per 1 million characters
-    AmazonPollyTTSModels.GENERATIVE: TtsModelPricing(cost_per_million_characters=30.0),  # $30 per 1 million characters
-
-    # Microsoft Azure Speech Service TTS Models
-    MicrosoftAzureSpeechTTSModels.STANDARD_NEURAL: TtsModelPricing(cost_per_million_characters=15.0),  # $15 per 1 million characters
-    MicrosoftAzureSpeechTTSModels.CUSTOM_SYNTHESIS: TtsModelPricing(cost_per_million_characters=24.0),  # $24 per 1 million characters
-    MicrosoftAzureSpeechTTSModels.CUSTOM_SYNTHESIS_NEURAL_HD: TtsModelPricing(cost_per_million_characters=48.0),  # $48 per 1 million characters
-
-    # Google Cloud Text-to-Speech TTS Models
-    GoogleCloudTextToSpeechTTSModels.CHIRP_3_HD: TtsModelPricing(cost_per_million_characters=30.0),  # $30 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.INSTANT_CUSTOM: TtsModelPricing(cost_per_million_characters=60.0),  # $60 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.WAVENET: TtsModelPricing(cost_per_million_characters=4.0),  # $4 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.STUDIO: TtsModelPricing(cost_per_million_characters=160.0),  # $160 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.STANDARD: TtsModelPricing(cost_per_million_characters=4.0),  # $4 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.NEURAL2: TtsModelPricing(cost_per_million_characters=16.0),  # $16 per 1 million characters
-    GoogleCloudTextToSpeechTTSModels.POLYGLOT_PREVIEW: TtsModelPricing(cost_per_million_characters=16.0),  # $16 per 1 million characters
-
-    # Deepgram TTS Models
-    DeepgramTTSModels.AURA_2: TtsModelPricing(cost_per_million_characters=30.0),  # $30 per 1 million characters
-    DeepgramTTSModels.AURA_1: TtsModelPricing(cost_per_million_characters=15.0),  # $15 per 1 million characters
-    # Growth tier models
-    DeepgramTTSModels.GROWTH_AURA_2: TtsModelPricing(cost_per_million_characters=27.0),  # $27 per 1 million characters
-    DeepgramTTSModels.GROWTH_AURA_1: TtsModelPricing(cost_per_million_characters=13.5),  # $13.5 per 1 million characters
-
-    # Eleven Labs TTS Models
-    ElevenLabsTTSModels.BUSINESS_MULTILINGUAL_V2_V3: TtsModelPricing(cost_per_million_characters=120.0),  # $120 per 1 million characters
-}
-
-
-def _calculate_stt_cost(client_instance, model: str, audio_duration_seconds: int) -> float:
+def send_stt_usage(client_instance, agent_id: str, customer_id: str, stt_usage_data) -> None:
     """
-    Calculate the cost based on STT model and audio duration.
+    Send STT usage data to the Paygent API using V2 endpoint.
     
     Args:
         client_instance: The Client instance
-        model: The STT model name
-        audio_duration_seconds: Audio duration in seconds
+        agent_id: Unique identifier for the agent
+        customer_id: Unique identifier for the customer
+        stt_usage_data: SttUsageData containing model and audio duration information
         
-    Returns:
-        Calculated cost in USD
+    Raises:
+        requests.RequestException: If the request fails
     """
-    pricing = STT_MODEL_PRICING.get(model)
-    
-    if not pricing:
-        client_instance.logger.warning(f"Unknown STT model '{model}', using default pricing")
-        # Use default pricing for unknown models (per hour)
-        pricing = SttModelPricing(cost_per_hour=0.5)  # $0.50 per hour default
+    client_instance.logger.debug(f"Sending STT usage data for model: {stt_usage_data.model}")
     
-    # Calculate cost: (duration in seconds / 3600) * cost per hour
-    # Convert seconds to hours and multiply by cost per hour
-    duration_hours = audio_duration_seconds / 3600.0
-    total_cost = duration_hours * pricing.cost_per_hour
-    
-    client_instance.logger.debug(
-        f"STT cost calculation for model '{model}': "
-        f"duration={audio_duration_seconds} seconds ({duration_hours:.6f} hours), "
-        f"cost_per_hour={pricing.cost_per_hour:.6f}, total={total_cost:.6f}"
-    )
-    
-    return total_cost
+    try:
+        from .models import RawUsageData
+        
+        # Use V2 usage tracking for server-side pricing
+        raw_usage = RawUsageData(
+            provider=stt_usage_data.service_provider,
+            model=stt_usage_data.model,
+            audio_duration=stt_usage_data.audio_duration
+        )
+        
+        client_instance.send_usage_v2(agent_id, customer_id, 'stt-usage', raw_usage)
+        client_instance.logger.info('STT usage data sent successfully')
+        
+    except Exception as e:
+        client_instance.logger.error(f"Failed to send STT usage data: {str(e)}")
+        raise
 
 
-def _calculate_tts_cost(client_instance, model: str, character_count: int) -> float:
+def send_tts_usage(client_instance, agent_id: str, customer_id: str, tts_usage_data) -> None:
     """
-    Calculate the cost based on TTS model and character count.
+    Send TTS usage data to the Paygent API using V2 endpoint.
     
     Args:
         client_instance: The Client instance
-        model: The TTS model name
-        character_count: Number of characters
+        agent_id: Unique identifier for the agent
+        customer_id: Unique identifier for the customer
+        tts_usage_data: TtsUsageData containing model and character count information
         
-    Returns:
-        Calculated cost in USD
+    Raises:
+        requests.RequestException: If the request fails
     """
-    pricing = TTS_MODEL_PRICING.get(model)
-    
-    if not pricing:
-        client_instance.logger.warning(f"Unknown TTS model '{model}', using default pricing")
-        # Use default pricing for unknown models (per 1 million characters)
-        pricing = TtsModelPricing(cost_per_million_characters=10.0)  # $10 per 1 million characters default
+    client_instance.logger.debug(f"Sending TTS usage data for model: {tts_usage_data.model}")
     
-    # Calculate cost: (character count / 1,000,000) * cost per 1 million characters
-    # Convert character count to millions and multiply by cost per million
-    characters_in_millions = character_count / 1000000.0
-    total_cost = characters_in_millions * pricing.cost_per_million_characters
-    
-    client_instance.logger.debug(
-        f"TTS cost calculation for model '{model}': "
-        f"characters={character_count} ({characters_in_millions:.6f} millions), "
-        f"cost_per_million={pricing.cost_per_million_characters:.6f}, total={total_cost:.6f}"
-    )
-    
-    return total_cost
+    try:
+        from .models import RawUsageData
+        
+        # Use V2 usage tracking for server-side pricing
+        raw_usage = RawUsageData(
+            provider=tts_usage_data.service_provider,
+            model=tts_usage_data.model,
+            character_count=tts_usage_data.character_count
+        )
+        
+        client_instance.send_usage_v2(agent_id, customer_id, 'tts-usage', raw_usage)
+        client_instance.logger.info('TTS usage data sent successfully')
+        
+    except Exception as e:
+        client_instance.logger.error(f"Failed to send TTS usage data: {str(e)}")
+        raise
 
 
-def send_stt_usage(client_instance, agent_id: str, customer_id: str, stt_usage_data) -> None:
+def initialize_voice_session(client_instance, session_id: str, agent_id: str, customer_id: str) -> None:
     """
-    Send STT usage data to the Paygent API.
+    Initialize a new voice session.
     
     Args:
         client_instance: The Client instance
+        session_id: Unique identifier for the session
         agent_id: Unique identifier for the agent
         customer_id: Unique identifier for the customer
-        stt_usage_data: SttUsageData containing model and audio duration information
         
     Raises:
         requests.RequestException: If the request fails
     """
-    client_instance.logger.info(
-        f"Starting send_stt_usage for agentID={agent_id}, customerID={customer_id}, "
-        f"model={stt_usage_data.model}, duration={stt_usage_data.audio_duration} seconds"
-    )
+    client_instance.logger.debug(f"Initializing voice session: {session_id}")
     
     try:
-        # Calculate cost
-        cost = _calculate_stt_cost(client_instance, stt_usage_data.model, stt_usage_data.audio_duration)
-        client_instance.logger.info(f"Calculated STT cost: {cost:.6f} for model {stt_usage_data.model}")
+        url = urljoin(client_instance.base_url, "/api/v1/voice/session")
+        headers = {
+            "Content-Type": "application/json",
+            "paygent-api-key": client_instance.api_key,
+        }
         
-        # Prepare API request
-        api_request = {
+        request_data = {
+            "sessionId": session_id,
             "agentId": agent_id,
             "customerId": customer_id,
-            "indicator": "stt-usage",  # Default indicator for STT usage
-            "amount": cost,
-            "audioDuration": stt_usage_data.audio_duration,
-            "model": stt_usage_data.model,
-            "serviceProvider": stt_usage_data.service_provider,
         }
         
-        # Make HTTP request
-        url = urljoin(client_instance.base_url, "/api/v1/usage")
+        response = client_instance.session.post(url, json=request_data, headers=headers)
+        response.raise_for_status()
+        
+        client_instance.logger.info("Voice session initialized successfully")
+    except requests.RequestException as e:
+        client_instance.logger.error(f"Failed to initialize voice session: {str(e)}")
+        raise
+
+
+def track_voice_stt(
+    client_instance,
+    session_id: str,
+    stt_usage_data: dict
+) -> None:
+    """
+    Track STT usage for a voice session.
+    
+    Args:
+        client_instance: The Client instance
+        session_id: Unique identifier for the session
+        stt_usage_data: Dict containing audioDuration, serviceProvider, model, and optional language
+        
+    Raises:
+        requests.RequestException: If the request fails
+    """
+    client_instance.logger.debug(f"Tracking voice STT usage for session: {session_id}")
+    
+    try:
+        url = urljoin(client_instance.base_url, "/api/v1/voice/stt")
         headers = {
             "Content-Type": "application/json",
             "paygent-api-key": client_instance.api_key,
         }
         
-        response = client_instance.session.post(url, json=api_request, headers=headers)
+        request_data = {
+            "sessionId": session_id,
+            "audioMinutes": stt_usage_data.get("audioDuration"),
+            "provider": stt_usage_data.get("serviceProvider"),
+            "model": stt_usage_data.get("model"),
+        }
+        
+        # Add optional language field
+        if "language" in stt_usage_data:
+            request_data["language"] = stt_usage_data["language"]
+        
+        response = client_instance.session.post(url, json=request_data, headers=headers)
         response.raise_for_status()
         
-        client_instance.logger.info(
-            f"Successfully sent STT usage data for agentID={agent_id}, "
-            f"customerID={customer_id}, cost={cost:.6f}"
-        )
+        client_instance.logger.info("Voice STT usage tracked successfully")
     except requests.RequestException as e:
-        client_instance.logger.error(f"Failed to send STT usage data: {str(e)}")
+        client_instance.logger.error(f"Failed to track voice STT: {str(e)}")
         raise
 
 
-def send_tts_usage(client_instance, agent_id: str, customer_id: str, tts_usage_data) -> None:
+def track_voice_llm(
+    client_instance,
+    session_id: str,
+    llm_usage_data: dict
+) -> None:
     """
-    Send TTS usage data to the Paygent API.
+    Track LLM usage for a voice session.
     
     Args:
         client_instance: The Client instance
-        agent_id: Unique identifier for the agent
-        customer_id: Unique identifier for the customer
-        tts_usage_data: TtsUsageData containing model and character count information
+        session_id: Unique identifier for the session
+        llm_usage_data: Dict containing serviceProvider, model, promptTokens, completionTokens
         
     Raises:
         requests.RequestException: If the request fails
     """
-    client_instance.logger.info(
-        f"Starting send_tts_usage for agentID={agent_id}, customerID={customer_id}, "
-        f"model={tts_usage_data.model}, characters={tts_usage_data.character_count}"
-    )
+    client_instance.logger.debug(f"Tracking voice LLM usage for session: {session_id}")
     
     try:
-        # Calculate cost
-        cost = _calculate_tts_cost(client_instance, tts_usage_data.model, tts_usage_data.character_count)
-        client_instance.logger.info(f"Calculated TTS cost: {cost:.6f} for model {tts_usage_data.model}")
+        url = urljoin(client_instance.base_url, "/api/v1/voice/llm")
+        headers = {
+            "Content-Type": "application/json",
+            "paygent-api-key": client_instance.api_key,
+        }
         
-        # Prepare API request
-        api_request = {
-            "agentId": agent_id,
-            "customerId": customer_id,
-            "indicator": "tts-usage",  # Default indicator for TTS usage
-            "amount": cost,
-            "characterCount": tts_usage_data.character_count,
-            "model": tts_usage_data.model,
-            "serviceProvider": tts_usage_data.service_provider,
+        request_data = {
+            "sessionId": session_id,
+            "provider": llm_usage_data.get("serviceProvider"),
+            "model": llm_usage_data.get("model"),
+            "promptTokens": llm_usage_data.get("promptTokens"),
+            "completionTokens": llm_usage_data.get("completionTokens"),
+        }
+        
+        response = client_instance.session.post(url, json=request_data, headers=headers)
+        response.raise_for_status()
+        
+        client_instance.logger.info("Voice LLM usage tracked successfully")
+    except requests.RequestException as e:
+        client_instance.logger.error(f"Failed to track voice LLM: {str(e)}")
+        raise
+
+
+def track_voice_tts(
+    client_instance,
+    session_id: str,
+    tts_usage_data: dict
+) -> None:
+    """
+    Track TTS usage for a voice session.
+    
+    Args:
+        client_instance: The Client instance
+        session_id: Unique identifier for the session
+        tts_usage_data: Dict containing serviceProvider, model, characterCount
+        
+    Raises:
+        requests.RequestException: If the request fails
+    """
+    client_instance.logger.debug(f"Tracking voice TTS usage for session: {session_id}")
+    
+    try:
+        url = urljoin(client_instance.base_url, "/api/v1/voice/tts")
+        headers = {
+            "Content-Type": "application/json",
+            "paygent-api-key": client_instance.api_key,
+        }
+        
+        request_data = {
+            "sessionId": session_id,
+            "provider": tts_usage_data.get("serviceProvider"),
+            "model": tts_usage_data.get("model"),
+            "characters": tts_usage_data.get("characterCount"),
         }
         
-        # Make HTTP request
-        url = urljoin(client_instance.base_url, "/api/v1/usage")
+        response = client_instance.session.post(url, json=request_data, headers=headers)
+        response.raise_for_status()
+        
+        client_instance.logger.info("Voice TTS usage tracked successfully")
+    except requests.RequestException as e:
+        client_instance.logger.error(f"Failed to track voice TTS: {str(e)}")
+        raise
+
+
+def set_voice_indicator(
+    client_instance,
+    session_id: str,
+    indicator: str,
+    total_duration: int
+) -> None:
+    """
+    Set an indicator for a voice session with final duration.
+    
+    Args:
+        client_instance: The Client instance
+        session_id: Unique identifier for the session
+        indicator: Indicator for the session
+        total_duration: Total duration of the session in seconds
+        
+    Raises:
+        requests.RequestException: If the request fails
+    """
+    client_instance.logger.debug(f"Setting voice indicator: {indicator} for session: {session_id}")
+    
+    try:
+        url = urljoin(client_instance.base_url, "/api/v1/voice/indicator")
         headers = {
             "Content-Type": "application/json",
             "paygent-api-key": client_instance.api_key,
         }
         
-        response = client_instance.session.post(url, json=api_request, headers=headers)
+        request_data = {
+            "sessionId": session_id,
+            "indicator": indicator,
+            "totalDuration": total_duration,
+        }
+        
+        response = client_instance.session.post(url, json=request_data, headers=headers)
         response.raise_for_status()
         
-        client_instance.logger.info(
-            f"Successfully sent TTS usage data for agentID={agent_id}, "
-            f"customerID={customer_id}, cost={cost:.6f}"
-        )
+        client_instance.logger.info("Voice indicator set successfully")
     except requests.RequestException as e:
-        client_instance.logger.error(f"Failed to send TTS usage data: {str(e)}")
+        client_instance.logger.error(f"Failed to set voice indicator: {str(e)}")
         raise
 
 
@@ -276,6 +304,11 @@ def _add_voice_methods_to_client():
     
     Client.send_stt_usage = send_stt_usage
     Client.send_tts_usage = send_tts_usage
+    Client.initialize_voice_session = initialize_voice_session
+    Client.track_voice_stt = track_voice_stt
+    Client.track_voice_llm = track_voice_llm
+    Client.track_voice_tts = track_voice_tts
+    Client.set_voice_indicator = set_voice_indicator
 
 
 # Call this function to attach methods when module is imported

paygent_sdk/wrappers/anthropic_wrapper.py

@@ -16,6 +16,7 @@ except ImportError:
 
 from ..client import Client
 from ..models import UsageData, UsageDataWithStrings
+from ..constants import ServiceProvider
 
 
 class PaygentAnthropic:
@@ -94,7 +95,7 @@ class MessagesWrapper:
         if has_valid_usage:
             # Primary path: Use usage data from API response
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.ANTHROPIC,
                 model=model,
                 prompt_tokens=response.usage.input_tokens,
                 completion_tokens=response.usage.output_tokens,
@@ -116,7 +117,7 @@ class MessagesWrapper:
                     output_string = response.content[0].text or ''
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.ANTHROPIC,
                 model=model,
                 prompt_string=prompt_string,
                 output_string=output_string

paygent_sdk/wrappers/gemini_wrapper.py

@@ -18,6 +18,7 @@ except ImportError:
 
 from ..client import Client
 from ..models import UsageData, UsageDataWithStrings
+from ..constants import ServiceProvider
 
 
 class PaygentGemini:
@@ -89,7 +90,7 @@ class ModelsWrapper:
         if has_valid_usage:
             # Primary path: Use usage metadata from API response
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.GOOGLE,
                 model=model,
                 prompt_tokens=response.usage_metadata.prompt_token_count or 0,
                 completion_tokens=response.usage_metadata.candidates_token_count or 0,
@@ -117,7 +118,7 @@ class ModelsWrapper:
                                 output_string = part.text or ''
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.GOOGLE,
                 model=model,
                 prompt_string=prompt_string,
                 output_string=output_string
@@ -208,7 +209,7 @@ class ModelsWrapper:
         if has_valid_usage:
             # Primary path: Use usage metadata from API response
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.GOOGLE,
                 model=model,
                 prompt_tokens=response.usage_metadata.prompt_token_count or 0,
                 completion_tokens=response.usage_metadata.candidates_token_count or 0,
@@ -224,7 +225,7 @@ class ModelsWrapper:
         else:
             # Fallback path: Calculate tokens from prompt string
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.GOOGLE,
                 model=model,
                 prompt_string=prompt,
                 output_string=''  # Image generation doesn't have text output
@@ -283,7 +284,7 @@ class ChatSessionWrapper:
         if has_valid_usage:
             # Primary path: Use usage metadata from API response
             usage_data = UsageData(
-                service_provider=self.model_name,
+                service_provider=ServiceProvider.GOOGLE,
                 model=self.model_name,
                 prompt_tokens=response.usage_metadata.prompt_token_count or 0,
                 completion_tokens=response.usage_metadata.candidates_token_count or 0,
@@ -309,7 +310,7 @@ class ChatSessionWrapper:
                                 output_string = part.text or ''
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=self.model_name,
+                service_provider=ServiceProvider.GOOGLE,
                 model=self.model_name,
                 prompt_string=message,
                 output_string=output_string

paygent_sdk/wrappers/mistral_wrapper.py

@@ -16,6 +16,7 @@ except ImportError:
 
 from ..client import Client
 from ..models import UsageData, UsageDataWithStrings
+from ..constants import ServiceProvider
 
 
 class PaygentMistral:
@@ -88,7 +89,7 @@ class ChatWrapper:
         if has_valid_usage:
             # Primary path: Use usage data from API response
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.MISTRAL_AI,
                 model=model,
                 prompt_tokens=response.usage.prompt_tokens,
                 completion_tokens=response.usage.completion_tokens,
@@ -112,7 +113,7 @@ class ChatWrapper:
                         output_string = choice.message.content or ''
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.MISTRAL_AI,
                 model=model,
                 prompt_string=prompt_string,
                 output_string=output_string

paygent_sdk/wrappers/openai_wrapper.py

@@ -16,6 +16,7 @@ except ImportError:
 
 from ..client import Client
 from ..models import UsageData, UsageDataWithStrings
+from ..constants import ServiceProvider
 
 
 class PaygentOpenAI:
@@ -121,7 +122,7 @@ class ChatCompletionsWrapper:
                     cached_tokens = prompt_details.cached_tokens
             
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.OPENAI,
                 model=model,
                 prompt_tokens=response.usage.prompt_tokens,
                 completion_tokens=response.usage.completion_tokens,
@@ -145,7 +146,7 @@ class ChatCompletionsWrapper:
                     output_string = response.choices[0].message.content or ''
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.OPENAI,
                 model=model,
                 prompt_string=prompt_string,
                 output_string=output_string
@@ -209,7 +210,7 @@ class EmbeddingsWrapper:
         if has_valid_usage:
             # Primary path: Use usage data from API response
             usage_data = UsageData(
-                service_provider=model,
+                service_provider=ServiceProvider.OPENAI,
                 model=model,
                 prompt_tokens=response.usage.prompt_tokens,
                 completion_tokens=response.usage.prompt_tokens,  # Embeddings don't have completion tokens
@@ -231,7 +232,7 @@ class EmbeddingsWrapper:
                 input_text = str(input)
 
             usage_data_with_strings = UsageDataWithStrings(
-                service_provider=model,
+                service_provider=ServiceProvider.OPENAI,
                 model=model,
                 prompt_string=input_text,
                 output_string=''  # Embeddings don't have output
@@ -297,8 +298,8 @@ class ImagesWrapper:
         # For image generation, we'll use a simplified usage tracking
         # since OpenAI doesn't provide token counts for images
         usage_data = UsageData(
-            service_provider=model,
-            model=model,
+            service_provider=ServiceProvider.OPENAI,
+            model=model if model else 'dall-e-2',
             prompt_tokens=0,  # Images don't use traditional tokens
             completion_tokens=0,
             total_tokens=0

{paygent_sdk-3.0.0.dist-info → paygent_sdk-4.0.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: paygent-sdk
-Version: 3.0.0
+Version: 4.0.0
 Summary: Official Python SDK for Paygent - Track AI usage and costs across multiple providers (OpenAI, Anthropic, Google, DeepSeek, etc.)
 Home-page: https://github.com/paygent/paygent-sdk-python
 Author: Paygent