Spaces:
Sleeping
Sleeping
| """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", | |
| ] | |