firts agent concept
This commit is contained in:
48
src/naliiabot/bot/factories/llm_factory.py
Normal file
48
src/naliiabot/bot/factories/llm_factory.py
Normal file
@@ -0,0 +1,48 @@
|
||||
from enum import Enum
|
||||
from langchain_anthropic import ChatAnthropic
|
||||
from langchain_deepseek import ChatDeepSeek
|
||||
from src.naliiabot.bot.exceptions import NotFoundProviderError
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
|
||||
class LLMProvider(str, Enum):
|
||||
"""Supported Providers."""
|
||||
|
||||
ANTHROPIC = "anthropic"
|
||||
DEEPSEEK = "deepseek"
|
||||
|
||||
|
||||
class LLMFactory:
|
||||
|
||||
def __init__(self, provider):
|
||||
self.provider = provider
|
||||
|
||||
def get_model(self):
|
||||
model = None
|
||||
|
||||
if self.provider == LLMProvider.ANTHROPIC:
|
||||
model = self._create_anthropic()
|
||||
elif self.provider == LLMProvider.DEEPSEEK:
|
||||
model = self._create_deepseek()
|
||||
else:
|
||||
raise NotFoundProviderError(
|
||||
f"{self.provider.upper()} Not Provider Found"
|
||||
)
|
||||
|
||||
return model
|
||||
|
||||
def _create_anthropic(self):
|
||||
model_name = "claude-sonnet-4-5-20250929"
|
||||
model = ChatAnthropic(model=model_name)
|
||||
|
||||
return model
|
||||
|
||||
def _create_deepseek(self):
|
||||
model_name = "deepseek-chat"
|
||||
model = ChatDeepSeek(
|
||||
model=model_name
|
||||
)
|
||||
|
||||
return model
|
||||
Reference in New Issue
Block a user