venice-sdk 0.3.3__py3-none-any.whl

venice_sdk/__init__.py

@@ -0,0 +1,297 @@
+"""
+Venice SDK for Python.
+
+This module provides a comprehensive Python interface to the Venice AI API.
+"""
+
+import logging
+
+from .client import HTTPClient
+from .venice_client import VeniceClient, create_client
+from .config import Config, load_config
+from .errors import (
+    VeniceError,
+    VeniceAPIError,
+    VeniceConnectionError,
+    RateLimitError,
+    UnauthorizedError,
+    InsufficientBalanceError,
+    InvalidRequestError,
+    ModelNotFoundError,
+    CharacterNotFoundError,
+    ImageGenerationError,
+    AudioGenerationError,
+    VideoGenerationError,
+    BillingError,
+    APIKeyError,
+    EmbeddingError,
+)
+from .models import (
+    Model,
+    ModelCapabilities,
+    ModelsAPI,
+    get_models,
+    get_model_by_id,
+    get_text_models,
+)
+from .chat import (
+    Message,
+    Choice,
+    Usage,
+    ChatCompletion,
+    ChatAPI,
+    chat_complete,
+)
+from .images import (
+    ImageGeneration,
+    ImageEditResult,
+    ImageUpscaleResult,
+    ImageStyle,
+    ImageAPI,
+    ImageEditAPI,
+    ImageUpscaleAPI,
+    ImageStylesAPI,
+    generate_image,
+    get_image_model_constraints,
+    edit_image,
+    upscale_image,
+    multi_edit_image,
+    remove_background,
+)
+from .audio import (
+    Voice,
+    AudioResult,
+    TranscriptionResult,
+    AudioAPI,
+    MusicJob,
+    MusicAPI,
+    AudioBatchProcessor,
+    text_to_speech,
+    text_to_speech_file,
+)
+from .video import (
+    VideoMetadata,
+    VideoJob,
+    VideoQuote,
+    VideoAPI,
+)
+from .characters import (
+    Character,
+    CharactersAPI,
+    CharacterManager,
+    get_character,
+    list_characters,
+    search_characters,
+)
+from .account import (
+    APIKey,
+    Web3APIKey,
+    RateLimits,
+    RateLimitLog,
+    UsageInfo,
+    ModelUsage,
+    APIKeysAPI,
+    BillingAPI,
+    AccountManager,
+    get_account_usage,
+    get_rate_limits,
+    list_api_keys,
+)
+from .models_advanced import (
+    ModelTraits,
+    CompatibilityMapping,
+    ModelsTraitsAPI,
+    ModelsCompatibilityAPI,
+    ModelRecommendationEngine,
+    get_model_traits,
+    get_compatibility_mapping,
+    find_models_by_capability,
+)
+from .embeddings import (
+    Embedding,
+    EmbeddingResult,
+    EmbeddingsAPI,
+    EmbeddingSimilarity,
+    SemanticSearch,
+    EmbeddingClustering,
+    generate_embedding,
+    calculate_similarity,
+    generate_embeddings,
+)
+from .tee import TEEAPI
+from .augment import (
+    AugmentAPI,
+    SearchHit,
+    SearchResponse,
+    ScrapeResponse,
+    ParsedDocument,
+)
+from .responses import ResponsesAPI, Response, ResponseOutputItem
+from .crypto import CryptoAPI
+from .x402 import (
+    X402API,
+    X402Balance,
+    X402TopUpResult,
+    X402Transactions,
+    PaymentSigner,
+    build_payment_payload,
+    encode_payment_signature,
+    decode_payment_signature,
+)
+from .decisions import DecisionsAPI, DecisionAnswer, DecisionResult
+from .voice_changer import VoiceChangerAPI, VoiceChangerJob
+from .logging_config import setup_logging
+
+_root_logger = logging.getLogger("venice_sdk")
+if not any(isinstance(handler, logging.NullHandler) for handler in _root_logger.handlers):
+    _root_logger.addHandler(logging.NullHandler())
+
+__version__ = "0.3.3"
+
+__all__ = [
+    # Core client
+    "HTTPClient",
+    "VeniceClient",
+    "create_client",
+    "Config",
+    "load_config",
+    
+    # Errors
+    "VeniceError",
+    "VeniceAPIError",
+    "VeniceConnectionError",
+    "RateLimitError",
+    "UnauthorizedError",
+    "InsufficientBalanceError",
+    "InvalidRequestError",
+    "ModelNotFoundError",
+    "CharacterNotFoundError",
+    "ImageGenerationError",
+    "AudioGenerationError",
+    "VideoGenerationError",
+    "BillingError",
+    "APIKeyError",
+    "EmbeddingError",
+    
+    # Models
+    "Model",
+    "ModelCapabilities",
+    "ModelsAPI",
+    "get_models",
+    "get_model_by_id",
+    "get_text_models",
+    
+    # Chat
+    "Message",
+    "Choice",
+    "Usage",
+    "ChatCompletion",
+    "ChatAPI",
+    "chat_complete",
+    
+    # Images
+    "ImageGeneration",
+    "ImageEditResult",
+    "ImageUpscaleResult",
+    "ImageStyle",
+    "ImageAPI",
+    "ImageEditAPI",
+    "ImageUpscaleAPI",
+    "ImageStylesAPI",
+    "generate_image",
+    "get_image_model_constraints",
+    "edit_image",
+    "upscale_image",
+    "multi_edit_image",
+    "remove_background",
+    
+    # Audio
+    "Voice",
+    "AudioResult",
+    "TranscriptionResult",
+    "AudioAPI",
+    "MusicJob",
+    "MusicAPI",
+    "AudioBatchProcessor",
+    "text_to_speech",
+    "text_to_speech_file",
+    
+    # Video
+    "VideoMetadata",
+    "VideoJob",
+    "VideoQuote",
+    "VideoAPI",
+    
+    # Characters
+    "Character",
+    "CharactersAPI",
+    "CharacterManager",
+    "get_character",
+    "list_characters",
+    "search_characters",
+    
+    # Account
+    "APIKey",
+    "Web3APIKey",
+    "RateLimits",
+    "RateLimitLog",
+    "UsageInfo",
+    "ModelUsage",
+    "APIKeysAPI",
+    "BillingAPI",
+    "AccountManager",
+    "get_account_usage",
+    "get_rate_limits",
+    "list_api_keys",
+    
+    # Advanced Models
+    "ModelTraits",
+    "CompatibilityMapping",
+    "ModelsTraitsAPI",
+    "ModelsCompatibilityAPI",
+    "ModelRecommendationEngine",
+    "get_model_traits",
+    "get_compatibility_mapping",
+    "find_models_by_capability",
+    
+    # Embeddings
+    "Embedding",
+    "EmbeddingResult",
+    "EmbeddingsAPI",
+    "EmbeddingSimilarity",
+    "SemanticSearch",
+    "EmbeddingClustering",
+    "generate_embedding",
+    "calculate_similarity",
+    "generate_embeddings",
+
+    # TEE
+    "TEEAPI",
+
+    # Augment / Responses / Crypto / x402
+    "AugmentAPI",
+    "SearchHit",
+    "SearchResponse",
+    "ScrapeResponse",
+    "ParsedDocument",
+    "ResponsesAPI",
+    "Response",
+    "ResponseOutputItem",
+    "CryptoAPI",
+    "X402API",
+    "X402Balance",
+    "X402TopUpResult",
+    "X402Transactions",
+    "PaymentSigner",
+    "build_payment_payload",
+    "encode_payment_signature",
+    "decode_payment_signature",
+    "DecisionsAPI",
+    "DecisionAnswer",
+    "DecisionResult",
+    "VoiceChangerAPI",
+    "VoiceChangerJob",
+
+    # Logging
+    "setup_logging",
+] 

venice_sdk/_http.py

@@ -0,0 +1,67 @@
+from __future__ import annotations
+
+from typing import Callable, Dict, Optional
+
+from .client import HTTPClient, get_http_client_manager
+
+_manual_override_client: Optional[HTTPClient] = None
+_factory_http_clients: Dict[int, HTTPClient] = {}
+_http_client_manager = get_http_client_manager()
+
+
+def set_shared_http_client(client: Optional[HTTPClient]) -> None:
+    """Override the cached shared HTTP client (primarily for testing)."""
+    global _manual_override_client
+    _manual_override_client = client
+    if client is None:
+        _http_client_manager.clear()
+        _factory_http_clients.clear()
+
+
+def reset_shared_http_client() -> None:
+    """Clear the cached shared HTTP client so it will be recreated when needed."""
+    set_shared_http_client(None)
+
+
+def _default_http_client_factory(config: Optional["Config"] = None) -> HTTPClient:
+    """Build an HTTPClient via the unified VeniceClient for compatibility."""
+    from .venice_client import VeniceClient
+    from .config import Config as _Config  # Local import to avoid cycles
+
+    cfg: Optional[_Config] = config
+    venice_client = VeniceClient(cfg)
+    http_client = getattr(venice_client, "http_client", None)
+    if isinstance(http_client, HTTPClient):
+        return http_client
+    if isinstance(venice_client, HTTPClient):
+        return venice_client
+    # Tests commonly patch VeniceClient to return a mock; fall back to that mock.
+    return venice_client
+
+
+_http_client_manager.set_builder(_default_http_client_factory)
+
+
+def get_shared_http_client(
+    factory: Optional[Callable[[], HTTPClient]] = None
+) -> HTTPClient:
+    """Return the cached shared HTTP client, creating it on first use."""
+    if factory is None:
+        if _manual_override_client is not None:
+            return _manual_override_client
+        return _http_client_manager.get_client()
+
+    key = id(factory)
+    if key not in _factory_http_clients:
+        _factory_http_clients[key] = factory()
+    return _factory_http_clients[key]
+
+
+def ensure_http_client(
+    client: Optional[HTTPClient],
+    factory: Optional[Callable[[], HTTPClient]] = None,
+) -> HTTPClient:
+    """Return a concrete HTTPClient, instantiating one if necessary."""
+    if client is not None:
+        return client
+    return get_shared_http_client(factory=factory)