Hugging Face's logo

Spaces:
hiasgnpsadgd
/
sage-api
Sleeping

App Files Community
sage-api
File size: 2,586 Bytes
de0f30b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""Service helpers for the Stage 01 rewrite."""

from .artifact_store import ArtifactStore, ArtifactStoreError, make_run_id, serialize_json_value
from .prompt_registry import PromptRegistry, PromptRegistryError, PromptRenderError, RenderedPrompt

# LLM providers live canonically under src/llm; re-export the names the rest of the
# codebase imports from `services` (works under both `services` and `src.services`).
try:  # pragma: no cover - import-root duality
    from llm import (  # type: ignore
        DEFAULT_QWEN_VLLM_PROFILE,
        DEFAULT_VLLM_HUB_DIR,
        DEFAULT_OPENAI_BASE_URL,
        DEFAULT_OPENAI_MODEL,
        DEFAULT_OPENAI_RESPONSE_FORMAT,
        AzureOpenAIError,
        AzureOpenAIJsonProvider,
        AzureOpenAISettings,
        JsonLlmProvider,
        LlmProviderError,
        OpenAICompatibleError,
        OpenAICompatibleJsonProvider,
        OpenAICompatibleSettings,
        build_azure_openai_provider,
        build_json_llm_provider,
        build_local_vllm_provider,
        build_openai_provider,
        build_qwen_vllm_provider,
        parse_json_content,
    )
except ImportError:  # pragma: no cover
    from src.llm import (  # type: ignore
        DEFAULT_QWEN_VLLM_PROFILE,
        DEFAULT_VLLM_HUB_DIR,
        DEFAULT_OPENAI_BASE_URL,
        DEFAULT_OPENAI_MODEL,
        DEFAULT_OPENAI_RESPONSE_FORMAT,
        AzureOpenAIError,
        AzureOpenAIJsonProvider,
        AzureOpenAISettings,
        JsonLlmProvider,
        LlmProviderError,
        OpenAICompatibleError,
        OpenAICompatibleJsonProvider,
        OpenAICompatibleSettings,
        build_azure_openai_provider,
        build_json_llm_provider,
        build_local_vllm_provider,
        build_openai_provider,
        build_qwen_vllm_provider,
        parse_json_content,
    )

__all__ = [
    "ArtifactStore",
    "ArtifactStoreError",
    "AzureOpenAIError",
    "AzureOpenAIJsonProvider",
    "AzureOpenAISettings",
    "DEFAULT_VLLM_HUB_DIR",
    "DEFAULT_QWEN_VLLM_PROFILE",
    "DEFAULT_OPENAI_BASE_URL",
    "DEFAULT_OPENAI_MODEL",
    "DEFAULT_OPENAI_RESPONSE_FORMAT",
    "JsonLlmProvider",
    "LlmProviderError",
    "OpenAICompatibleError",
    "OpenAICompatibleJsonProvider",
    "OpenAICompatibleSettings",
    "PromptRegistry",
    "PromptRegistryError",
    "PromptRenderError",
    "RenderedPrompt",
    "build_azure_openai_provider",
    "build_json_llm_provider",
    "build_local_vllm_provider",
    "build_openai_provider",
    "build_qwen_vllm_provider",
    "make_run_id",
    "parse_json_content",
    "serialize_json_value",
]