Spaces:
Paused
Paused
File size: 4,722 Bytes
1bc3f18 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 | from .LLMEnums import LLMEnums
from stores.llm.providers.OpenAIProvider import OpenAIProvider
from stores.llm.providers.OllamaProvider import OllamaProvider
from stores.llm.providers.CohereProvider import CohereProvider
from stores.llm.providers.MistralProvider import MistralProvider
from stores.llm.providers.GroqProvider import GroqProvider
from stores.llm.providers.OpenRouterProvider import OpenRouterProvider
from stores.llm.providers.HuggingFaceProvider import HuggingFaceProvider
from stores.llm.providers.DeepSeekProvider import DeepSeekProvider
from stores.llm.providers.GeminiProvider import GeminiProvider
class LLMProviderFactory:
def __init__(self, config: dict):
self.config = config
def create(self, provider: str):
if provider == LLMEnums.OPENAI.value:
return OpenAIProvider(
api_key=self.config.OPENAI_API_KEY,
api_url=self.config.OPENAI_API_URL,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.OLLAMA.value:
return OllamaProvider(
url=self.config.OLLAMA_URL,
api_key=self.config.OLLAMA_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.COHERE.value:
return CohereProvider(
api_key=self.config.COHERE_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.MISTRAL.value:
return MistralProvider(
api_key=self.config.MISTRAL_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.GROQ.value:
return GroqProvider(
api_key=self.config.GROQ_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.OPENROUTER.value:
return OpenRouterProvider(
api_key=self.config.OPENROUTER_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.HUGGINGFACE.value:
return HuggingFaceProvider(
api_key=self.config.HF_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.DEEPSEEK.value:
return DeepSeekProvider(
api_key=self.config.DEEPSEEK_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
if provider == LLMEnums.GEMINI.value:
return GeminiProvider(
api_key=self.config.GEMINI_API_KEY,
default_input_max_characters=self.config.INPUT_DAFAULT_MAX_CHARACTERS,
default_generation_max_output_tokens=self.config.GENERATION_DAFAULT_MAX_TOKENS,
default_generation_temperature=self.config.GENERATION_DAFAULT_TEMPERATURE,
)
return None
|