paygent-sdk 1.0.0__py3-none-any.whl

examples/__init__.py

@@ -0,0 +1 @@
+# Examples package

examples/advanced_usage.py

@@ -0,0 +1,101 @@
+"""
+Advanced usage example for the Paygent SDK with token string functionality.
+"""
+
+import sys
+import os
+
+# Add parent directory to path to allow imports
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from paygent_sdk import (
+    Client, 
+    UsageData, 
+    UsageDataWithStrings,
+    ServiceProvider,
+    OpenAIModels,
+    AnthropicModels,
+    GoogleDeepMindModels,
+    MetaModels,
+    MistralAIModels
+)
+
+def main():
+    """Advanced usage example with both token counting methods."""
+    # Create a new client with your API key
+    client = Client.new_client_with_url(
+        "pk_e0ea0d11bb7f0d174caf578d665454acff97bdb1f85c235af547ccd9a733ef35",
+        "http://localhost:8080"
+    )
+    
+    print("Example 1: Basic Usage with Pre-calculated Tokens")
+    
+    # Define usage data with pre-calculated tokens using constants
+    usage_data = UsageData(
+        service_provider=ServiceProvider.OPENAI,
+        model=OpenAIModels.GPT_4_0613,
+        prompt_tokens=1000,
+        completion_tokens=500,
+        total_tokens=1500
+    )
+    
+    # Send usage data
+    try:
+        client.send_usage("agent-123", "customer-456", "chat-completion", usage_data)
+        print("✓ Usage data sent successfully!")
+    except Exception as e:
+        print(f"✗ Failed to send usage: {e}")
+    
+    print("\nExample 2: Advanced Usage with Token String Counting")
+    
+    # Define usage data with prompt and output strings using constants
+    usage_data_with_strings = UsageDataWithStrings(
+        service_provider=ServiceProvider.OPENAI,
+        model=OpenAIModels.GPT_3_5_TURBO,
+        prompt_string="What is the capital of France? Please provide a detailed explanation.",
+        output_string="The capital of France is Paris. Paris is located in the north-central part of France and is the country's largest city and economic center. It serves as the political, economic, and cultural hub of the nation."
+    )
+    
+    # Send usage data (tokens will be automatically counted)
+    try:
+        client.send_usage_with_token_string(
+            "agent-789", 
+            "customer-101", 
+            "question-answer", 
+            usage_data_with_strings
+        )
+        print("✓ Usage data with token strings sent successfully!")
+    except Exception as e:
+        print(f"✗ Failed to send usage with token strings: {e}")
+    
+    print("\nExample 3: Different AI Models")
+    
+    # Test different AI models using constants
+    models_to_test = [
+        (AnthropicModels.SONNET_3_7, ServiceProvider.ANTHROPIC, "Anthropic"),
+        (GoogleDeepMindModels.GEMINI_2_5_PRO, ServiceProvider.GOOGLE_DEEPMIND, "Google DeepMind"),
+        (MetaModels.LLAMA_3_1_8B_INSTRUCT_TURBO, ServiceProvider.META, "Meta"),
+        (MistralAIModels.MISTRAL_LARGE, ServiceProvider.MISTRAL_AI, "Mistral AI")
+    ]
+    
+    for model, provider, provider_name in models_to_test:
+        usage_data_test = UsageDataWithStrings(
+            service_provider=provider,
+            model=model,
+            prompt_string="Hello, how are you?",
+            output_string="I'm doing well, thank you for asking! How can I help you today?"
+        )
+        
+        try:
+            client.send_usage_with_token_string(
+                f"agent-{model.replace('-', '_')}", 
+                "customer-test", 
+                "greeting", 
+                usage_data_test
+            )
+            print(f"✓ {provider_name} {model} usage sent successfully!")
+        except Exception as e:
+            print(f"✗ Failed to send usage for {provider_name} {model}: {e}")
+
+if __name__ == "__main__":
+    main()

examples/basic_usage.py

@@ -0,0 +1,38 @@
+"""
+Basic usage example for the Paygent SDK.
+"""
+
+import sys
+import os
+
+# Add parent directory to path to allow imports
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from paygent_sdk import Client, UsageData, ServiceProvider, MetaModels
+
+def main():
+    """Basic usage example."""
+    # Create a new client with your API key
+    client = Client.new_client_with_url(
+        "pk_e0ea0d11bb7f0d174caf578d665454acff97bdb1f85c235af547ccd9a733ef35",
+        "http://localhost:8080"
+    )
+    
+    # Define usage data using constants
+    usage_data = UsageData(
+        service_provider=ServiceProvider.META,
+        model=MetaModels.LLAMA_2,
+        prompt_tokens=756,
+        completion_tokens=244,
+        total_tokens=1000
+    )
+    
+    # Send usage data
+    try:
+        client.send_usage("agent-123", "customer-456", "email-sent", usage_data)
+        print("Usage data sent successfully!")
+    except Exception as e:
+        print(f"Failed to send usage: {e}")
+
+if __name__ == "__main__":
+    main()

examples/constants_usage.py

@@ -0,0 +1,136 @@
+"""
+Example demonstrating the use of model constants in the Paygent SDK.
+"""
+
+from paygent_sdk import (
+    Client,
+    UsageData,
+    UsageDataWithStrings,
+    OpenAIModels,
+    AnthropicModels,
+    GoogleDeepMindModels,
+    MetaModels,
+    DeepSeekModels,
+    ServiceProvider,
+    is_model_supported,
+)
+
+
+def main():
+    """Main function demonstrating model constants usage."""
+    # Create a client
+    client = Client.new_client("your-api-key-here")
+
+    print("=== Model Constants Example ===\n")
+
+    # Example 1: Using OpenAI model constants
+    print("1. Using OpenAI Model Constants:")
+    openai_usage = UsageData(
+        service_provider=ServiceProvider.OPENAI,
+        model=OpenAIModels.GPT_4O,
+        prompt_tokens=100,
+        completion_tokens=50,
+        total_tokens=150,
+    )
+
+    print(f"   Model: {openai_usage.model}")
+    print(f"   Provider: {openai_usage.service_provider}")
+    print(f"   Tokens: {openai_usage.total_tokens}\n")
+
+    # Example 2: Using Anthropic model constants
+    print("2. Using Anthropic Model Constants:")
+    anthropic_usage = UsageData(
+        service_provider=ServiceProvider.ANTHROPIC,
+        model=AnthropicModels.SONNET_4_5,
+        prompt_tokens=200,
+        completion_tokens=100,
+        total_tokens=300,
+    )
+
+    print(f"   Model: {anthropic_usage.model}")
+    print(f"   Provider: {anthropic_usage.service_provider}\n")
+
+    # Example 3: Using Google DeepMind model constants
+    print("3. Using Google DeepMind Model Constants:")
+    gemini_usage = UsageData(
+        service_provider=ServiceProvider.GOOGLE_DEEPMIND,
+        model=GoogleDeepMindModels.GEMINI_2_5_PRO,
+        prompt_tokens=150,
+        completion_tokens=75,
+        total_tokens=225,
+    )
+
+    print(f"   Model: {gemini_usage.model}")
+    print(f"   Provider: {gemini_usage.service_provider}\n")
+
+    # Example 4: Using Meta model constants
+    print("4. Using Meta Model Constants:")
+    llama_usage = UsageData(
+        service_provider=ServiceProvider.META,
+        model=MetaModels.LLAMA_4_MAVERICK,
+        prompt_tokens=300,
+        completion_tokens=150,
+        total_tokens=450,
+    )
+
+    print(f"   Model: {llama_usage.model}")
+    print(f"   Provider: {llama_usage.service_provider}\n")
+
+    # Example 5: Using DeepSeek model constants
+    print("5. Using DeepSeek Model Constants:")
+    deepseek_usage = UsageData(
+        service_provider=ServiceProvider.DEEPSEEK,
+        model=DeepSeekModels.DEEPSEEK_R1_GLOBAL,
+        prompt_tokens=250,
+        completion_tokens=125,
+        total_tokens=375,
+    )
+
+    print(f"   Model: {deepseek_usage.model}")
+    print(f"   Provider: {deepseek_usage.service_provider}\n")
+
+    # Example 6: Check if a model is supported
+    print("6. Checking Model Support:")
+    models_to_check = [
+        OpenAIModels.GPT_5,
+        OpenAIModels.O3,
+        AnthropicModels.SONNET_4_5,
+        "unknown-model",
+    ]
+
+    for model in models_to_check:
+        supported = is_model_supported(model)
+        status = "✓ Supported" if supported else "✗ Not Supported"
+        print(f"   {model}: {status}")
+    print()
+
+    # Example 7: Service providers are explicit (user provides them)
+    print("7. Service Providers (User-Provided):")
+    print(f"   OpenAI models use: {ServiceProvider.OPENAI}")
+    print(f"   Anthropic models use: {ServiceProvider.ANTHROPIC}")
+    print(f"   Google DeepMind models use: {ServiceProvider.GOOGLE_DEEPMIND}")
+    print(f"   Meta models use: {ServiceProvider.META}")
+    print()
+
+    # Example 8: Using constants with send_usage_with_token_string
+    print("8. Using Constants with send_usage_with_token_string:")
+    string_usage = UsageDataWithStrings(
+        service_provider=ServiceProvider.OPENAI,
+        model=OpenAIModels.GPT_4O_MINI,
+        prompt_string="What is the capital of France?",
+        output_string="The capital of France is Paris.",
+    )
+
+    print(f"   Model: {string_usage.model}")
+    print(f"   Provider: {string_usage.service_provider}")
+    print(f'   Prompt: "{string_usage.prompt_string}"')
+    print(f'   Output: "{string_usage.output_string}"\n')
+
+    print("=== All examples completed successfully! ===")
+    print("\nNote: To actually send data to the API, uncomment the client.send_usage() calls below:\n")
+    print('# client.send_usage("agent-123", "customer-456", "test", openai_usage)')
+    print('# client.send_usage_with_token_string("agent-123", "customer-456", "test", string_usage)')
+
+
+if __name__ == "__main__":
+    main()

paygent_sdk/__init__.py

@@ -0,0 +1,47 @@
+"""
+Paygent SDK for Python
+
+A Python SDK for integrating with the Paygent API to track usage and costs for AI models.
+
+For the Go SDK equivalent, see: https://github.com/paygent/paygent-sdk-go
+"""
+
+from .client import Client
+from .models import UsageData, UsageDataWithStrings, APIRequest, ModelPricing, MODEL_PRICING
+from .constants import (
+    ServiceProvider,
+    OpenAIModels,
+    AnthropicModels,
+    GoogleDeepMindModels,
+    MetaModels,
+    AWSModels,
+    MistralAIModels,
+    CohereModels,
+    DeepSeekModels,
+    is_model_supported
+)
+
+__version__ = "1.0.0"
+__all__ = [
+    # Core classes
+    "Client",
+    "UsageData", 
+    "UsageDataWithStrings", 
+    "APIRequest", 
+    "ModelPricing",
+    "MODEL_PRICING",
+    
+    # Constants
+    "ServiceProvider",
+    "OpenAIModels",
+    "AnthropicModels", 
+    "GoogleDeepMindModels",
+    "MetaModels",
+    "AWSModels",
+    "MistralAIModels",
+    "CohereModels",
+    "DeepSeekModels",
+    
+    # Utility functions
+    "is_model_supported"
+]