decart-python/decart/__init__.py at 0f63c9e588ff5e96d78ca0aad1079ec7c8777710 · DecartAI/decart-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
from .client import DecartClient
from .errors import (
    DecartSDKError,
    InvalidAPIKeyError,
    InvalidBaseURLError,
    InvalidInputError,
    ModelNotFoundError,
    ProcessingError,
    WebRTCError,
    QueueSubmitError,
    QueueStatusError,
    QueueResultError,
    TokenCreateError,
    FileTooLargeError,
    MAX_FILE_SIZE,
)
from .models import models, ModelDefinition, VideoRestyleInput
from .types import FileInput, ModelState, Prompt
from .queue import (
    QueueClient,
    JobStatus,
    JobSubmitResponse,
    JobStatusResponse,
    QueueJobResult,
)
from .tokens import (
    TokensClient,
    CreateTokenResponse,
)

try:
    from .realtime import (
        RealtimeClient,
        SetInput,
        SubscribeClient,
        SubscribeOptions,
        encode_subscribe_token,
        decode_subscribe_token,
        RealtimeConnectOptions,
        ConnectionState,
        AvatarOptions,
    )

    REALTIME_AVAILABLE = True
except ImportError:
    REALTIME_AVAILABLE = False
    RealtimeClient = None  # type: ignore
    SetInput = None  # type: ignore
    SubscribeClient = None  # type: ignore
    SubscribeOptions = None  # type: ignore
    encode_subscribe_token = None  # type: ignore
    decode_subscribe_token = None  # type: ignore
    RealtimeConnectOptions = None  # type: ignore
    ConnectionState = None  # type: ignore
    AvatarOptions = None  # type: ignore

__version__ = "0.0.1"

__all__ = [
    "DecartClient",
    "DecartSDKError",
    "InvalidAPIKeyError",
    "InvalidBaseURLError",
    "InvalidInputError",
    "ModelNotFoundError",
    "ProcessingError",
    "WebRTCError",
    "QueueSubmitError",
    "QueueStatusError",
    "QueueResultError",
    "FileTooLargeError",
    "MAX_FILE_SIZE",
    "models",
    "ModelDefinition",
    "VideoRestyleInput",
    "FileInput",
    "ModelState",
    "Prompt",
    "QueueClient",
    "JobStatus",
    "JobSubmitResponse",
    "JobStatusResponse",
    "QueueJobResult",
    "TokensClient",
    "CreateTokenResponse",
    "TokenCreateError",
]

if REALTIME_AVAILABLE:
    __all__.extend(
        [
            "RealtimeClient",
            "SetInput",
            "SubscribeClient",
            "SubscribeOptions",
            "encode_subscribe_token",
            "decode_subscribe_token",
            "RealtimeConnectOptions",
            "ConnectionState",
            "AvatarOptions",
        ]
    )