cd45212acb
* feat: add zai provider support * add zai_api_key secret to deploy-core * add to justfile * add testing, provider integration skill * enable zai key * fix zai test * clean up skill a little * small changes
50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# Provider base classes and utilities
|
|
# Provider implementations
|
|
from .anthropic import AnthropicProvider
|
|
from .azure import AzureProvider
|
|
from .base import Provider, ProviderBase, ProviderCheck, ProviderCreate, ProviderUpdate
|
|
from .bedrock import BedrockProvider
|
|
from .cerebras import CerebrasProvider
|
|
from .deepseek import DeepSeekProvider
|
|
from .google_gemini import GoogleAIProvider
|
|
from .google_vertex import GoogleVertexProvider
|
|
from .groq import GroqProvider
|
|
from .letta import LettaProvider
|
|
from .lmstudio import LMStudioOpenAIProvider
|
|
from .mistral import MistralProvider
|
|
from .ollama import OllamaProvider
|
|
from .openai import OpenAIProvider
|
|
from .openrouter import OpenRouterProvider
|
|
from .together import TogetherProvider
|
|
from .vllm import VLLMProvider
|
|
from .xai import XAIProvider
|
|
from .zai import ZAIProvider
|
|
|
|
__all__ = [
|
|
# Base classes
|
|
"Provider",
|
|
"ProviderBase",
|
|
"ProviderCreate",
|
|
"ProviderUpdate",
|
|
"ProviderCheck",
|
|
# Provider implementations
|
|
"AnthropicProvider",
|
|
"AzureProvider",
|
|
"BedrockProvider",
|
|
"CerebrasProvider", # NEW
|
|
"DeepSeekProvider",
|
|
"GoogleAIProvider",
|
|
"GoogleVertexProvider",
|
|
"GroqProvider",
|
|
"LettaProvider",
|
|
"LMStudioOpenAIProvider",
|
|
"MistralProvider",
|
|
"OllamaProvider",
|
|
"OpenAIProvider",
|
|
"TogetherProvider",
|
|
"VLLMProvider", # Replaces ChatCompletions and Completions
|
|
"XAIProvider",
|
|
"ZAIProvider",
|
|
"OpenRouterProvider",
|
|
]
|