18 lines
666 B
Python
18 lines
666 B
Python
|
from app.config import settings
|
||
|
|
from app.services.llm.base import LLMProvider
|
||
|
|
|
||
|
|
|
||
|
|
def get_llm_provider() -> LLMProvider:
|
||
|
|
provider = settings.llm_provider.lower()
|
||
|
|
if provider == "anthropic":
|
||
|
|
from app.services.llm.anthropic import AnthropicProvider
|
||
|
|
return AnthropicProvider()
|
||
|
|
elif provider == "openai":
|
||
|
|
from app.services.llm.openai_provider import OpenAIProvider
|
||
|
|
return OpenAIProvider()
|
||
|
|
elif provider == "ollama":
|
||
|
|
from app.services.llm.ollama import OllamaProvider
|
||
|
|
return OllamaProvider()
|
||
|
|
else:
|
||
|
|
raise ValueError(f"Unknown LLM provider: {provider}. Must be 'anthropic', 'openai', or 'ollama'.")
|