add Custom provider
This commit is contained in:
@@ -3,8 +3,9 @@ from services.llm_providers.ionet_provider import IoNetProvider
|
||||
from services.llm_providers.gemini_provider import GeminiProvider
|
||||
from services.llm_providers.gpt4free_provider import Gpt4FreeProvider
|
||||
from services.llm_providers.base_provider import BaseLLMProvider
|
||||
from services.llm_providers.custom_provider import CustomProvider
|
||||
|
||||
def get_llm_provider(provider_name: str, api_key: str | None) -> BaseLLMProvider:
|
||||
def get_llm_provider(provider_name: str, api_key: str | None = None, base_url: str | None = None) -> BaseLLMProvider:
|
||||
"""
|
||||
Фабричная функция для получения экземпляра провайдера LLM.
|
||||
"""
|
||||
@@ -18,5 +19,9 @@ def get_llm_provider(provider_name: str, api_key: str | None) -> BaseLLMProvider
|
||||
return GeminiProvider(api_key)
|
||||
elif provider_name == 'gpt4free':
|
||||
return Gpt4FreeProvider()
|
||||
elif provider_name == 'Custom':
|
||||
if not base_url:
|
||||
raise ValueError("Base URL обязателен для Custom провайдера")
|
||||
return CustomProvider(api_key, base_url) # base_url будет установлен позже
|
||||
else:
|
||||
raise ValueError(f"Неизвестный провайдер: {provider_name}")
|
||||
Reference in New Issue
Block a user