From 0e2e5f1e37c22628ba98bf43d38c65870068414a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=A0=D0=B8=D0=BD=D0=B0=D1=80=20=D0=9C=D0=B8=D1=80=D0=B7?= =?UTF-8?q?=D0=B0=D0=B3=D0=B8=D1=82=D0=BE=D0=B2?= Date: Thu, 18 Jun 2026 09:48:03 +0000 Subject: [PATCH] ready: update src/agent.py --- src/agent.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/agent.py b/src/agent.py index e9e0b8e..bf56ade 100644 --- a/src/agent.py +++ b/src/agent.py @@ -1,7 +1,7 @@ import os from typing import Dict, List -from langchain.agents import AgentExecutor, create_agent +from langchain.agents import create_agent from langchain_ollama import Ollama from langchain_core.prompts import ChatPromptTemplate from langchain_core.messages import HumanMessage @@ -49,7 +49,6 @@ PROMPT = ChatPromptTemplate.from_messages([ # Create agent llm = Ollama(model="llama2") agent = create_agent(llm, TOOL_DEFINITIONS, PROMPT) -executor = AgentExecutor(agent=agent, tools=TOOL_DEFINITIONS, verbose=True) # CLI entry point if __name__ == "__main__": @@ -58,14 +57,14 @@ if __name__ == "__main__": parser.add_argument("--question", type=str, help="Question to ask the bot") args = parser.parse_args() if args.question: - result = executor.invoke({"input": args.question}) + result = agent.invoke({"input": args.question}) print(result["output"]) else: print("Enter questions (Ctrl-D to exit):") try: while True: q = input("Q: ") - res = executor.invoke({"input": q}) + res = agent.invoke({"input": q}) print("A:", res["output"]) except EOFError: pass