feat: Add NaliiaBot API
This commit is contained in:
16
src/naliiabotapi/api/dependencies.py
Normal file
16
src/naliiabotapi/api/dependencies.py
Normal file
@@ -0,0 +1,16 @@
|
||||
|
||||
import os
|
||||
from src.naliiabot.bot.agent.agent import Agent
|
||||
from src.naliiabot.bot.factories.llm_factory import LLMFactory
|
||||
|
||||
MODEL_NAME = os.getenv("LLM_MODEL", "anthropic")
|
||||
|
||||
def get_agent():
|
||||
"""
|
||||
Dependency function to get an instance of the Agent class.
|
||||
|
||||
Returns:
|
||||
Agent: An instance of the Agent class.
|
||||
"""
|
||||
model = LLMFactory(MODEL_NAME).get_model()
|
||||
return Agent(model=model)
|
||||
Reference in New Issue
Block a user