feat: Implemented load PROMPTS strategy
This commit is contained in:
@@ -1,7 +1,8 @@
|
||||
|
||||
import os
|
||||
from src.naliiabot.bot.agent.agent import Agent
|
||||
from src.naliiabot.bot.agent.agent import Agent, AgentConfig
|
||||
from src.naliiabot.bot.factories.llm_factory import LLMFactory
|
||||
from src.naliiabot.bot.prompts.load_prompt import get_prompt_template
|
||||
|
||||
MODEL_NAME = os.getenv("LLM_MODEL", "anthropic")
|
||||
|
||||
@@ -12,5 +13,9 @@ def get_agent():
|
||||
Returns:
|
||||
Agent: An instance of the Agent class.
|
||||
"""
|
||||
|
||||
NALIIA_PROMPT = get_prompt_template(
|
||||
"NALIIA_PROMPT")
|
||||
config = AgentConfig(system_prompt=NALIIA_PROMPT)
|
||||
model = LLMFactory(MODEL_NAME).get_model()
|
||||
return Agent(model=model)
|
||||
Reference in New Issue
Block a user