firts agent concept

This commit is contained in:
2026-02-03 21:20:00 -05:00
parent daa9893d2e
commit aff0a588c5
12 changed files with 2030 additions and 1 deletions

View 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