75 lines
2.6 KiB
Python
75 lines
2.6 KiB
Python
from app.contracts import ModelCapability, ProviderConfig, ProviderPreset, ProviderType
|
|
from app.providers.base import ModelProvider
|
|
from app.providers.credentials import CredentialResolver
|
|
from app.providers.ollama import OllamaProvider
|
|
from app.providers.openai_compatible import OpenAICompatibleProvider
|
|
|
|
|
|
class UnsupportedProviderError(ValueError):
|
|
pass
|
|
|
|
|
|
class ProviderFactory:
|
|
def __init__(self, credentials: CredentialResolver) -> None:
|
|
self.credentials = credentials
|
|
|
|
def build(self, config: ProviderConfig) -> ModelProvider:
|
|
if config.provider_type in {
|
|
ProviderType.openai_chat,
|
|
ProviderType.openai_compatible,
|
|
}:
|
|
return OpenAICompatibleProvider(
|
|
base_url=config.base_url or "https://api.openai.com/v1",
|
|
credential_id=config.credential_id,
|
|
credentials=self.credentials,
|
|
)
|
|
if config.provider_type == ProviderType.ollama:
|
|
return OllamaProvider(config.base_url or "http://127.0.0.1:11434")
|
|
raise UnsupportedProviderError(config.provider_type.value)
|
|
|
|
@staticmethod
|
|
def presets() -> list[ProviderPreset]:
|
|
return [
|
|
ProviderPreset(
|
|
preset_id="openai",
|
|
name="OpenAI",
|
|
provider_type=ProviderType.openai_chat,
|
|
base_url="https://api.openai.com/v1",
|
|
default_credential_id="openai",
|
|
),
|
|
ProviderPreset(
|
|
preset_id="deepseek",
|
|
name="DeepSeek",
|
|
provider_type=ProviderType.openai_compatible,
|
|
base_url="https://api.deepseek.com",
|
|
default_credential_id="deepseek",
|
|
),
|
|
ProviderPreset(
|
|
preset_id="ollama",
|
|
name="Ollama",
|
|
provider_type=ProviderType.ollama,
|
|
base_url="http://127.0.0.1:11434",
|
|
requires_credential=False,
|
|
),
|
|
]
|
|
|
|
@staticmethod
|
|
def capabilities(provider_type: ProviderType) -> list[ModelCapability]:
|
|
if provider_type in {
|
|
ProviderType.openai_chat,
|
|
ProviderType.openai_compatible,
|
|
}:
|
|
return [
|
|
ModelCapability.chat,
|
|
ModelCapability.tool_calling,
|
|
ModelCapability.streaming,
|
|
ModelCapability.structured_output,
|
|
]
|
|
if provider_type == ProviderType.ollama:
|
|
return [
|
|
ModelCapability.chat,
|
|
ModelCapability.tool_calling,
|
|
ModelCapability.streaming,
|
|
]
|
|
return []
|