OneTeam/NaliiaBot
16
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Change main to so print graph

Browse Source
This commit is contained in:
aserrador
2026-03-04 19:00:44 -05:00
parent 75adb658e8
commit e440c97743
1 changed files with 5 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.py
View File

@@ -1,24 +1,18 @@
import os import os
from naliiabot.bot.agent.agent import Agent, AgentConfig from naliiabot.bot.agent.agent import Agent, AgentConfig
from naliiabot.bot.factories.llm_factory import LLMFactory
from naliiabot.bot.tools.naliia_tools import NaliiaTools from naliiabot.bot.tools.naliia_tools import NaliiaTools
from naliiabot.bot.prompts.load_prompt import get_prompt_template from naliiabot.bot.prompts.load_prompt import get_prompt_template
from langchain_anthropic import ChatAnthropic
model_name = "claude-sonnet-4-5-20250929"
model = ChatAnthropic(
model=model_name
)
MODEL_NAME = os.getenv("LLM_MODEL", "anthropic") model_name = os.getenv("LLM_MODEL", "anthropic")
NALIIA_PROMPT = get_prompt_template( model = LLMFactory(model_name).get_model()
"NALIIA_PROMPT") naliia_prompt = get_prompt_template("NALIIA_PROMPT")
naliia_tools = NaliiaTools().get_tools() naliia_tools = NaliiaTools().get_tools()
config = AgentConfig(system_prompt=NALIIA_PROMPT) config = AgentConfig(system_prompt=naliia_prompt)
agent = Agent(model=model, config=config, tools=naliia_tools) agent = Agent(model=model, config=config, tools=naliia_tools)
graph_image = agent._compiled_agent.get_graph().draw_mermaid_png() graph_image = agent._compiled_agent.get_graph().draw_mermaid_png()
with open("graph.png", "wb") as f: with open("graph.png", "wb") as f: