Update app.py

app.py

@@ -3,7 +3,7 @@ from transformers import pipeline
 from langchain_huggingface import HuggingFacePipeline, ChatHuggingFace
 from langchain.tools import tool
 from langchain_core.output_parsers import JsonOutputParser
-from langchain.agents import AgentExecutor
+from langchain.agents import AgentExecutor, create_react_agent
 from youtube_transcript_api import YouTubeTranscriptApi
 import chess, chess.engine
 from bs4 import BeautifulSoup
@@ -152,11 +152,12 @@ class BasicAgent:
         prompt = f"{self.system_prompt}Q: {question}\nA:"
         #out = self.generator(prompt, max_new_tokens=16, return_full_text=False)
         #return out[0]["generated_text"].strip()
-        agent = AgentExecutor.from_llm_and_tools(
-            llm=self.llm,
+        agent = create_react_agent(llm=self.llm, tools=TOOLS, prompt=prompt)
+        agent = AgentExecutor(
+            agent=agent,
             tools=TOOLS,
-            system_message=prompt,
             verbose=True,
+            return_intermediate_steps=False
         )
         #agent = AgentExecutor(agent=self.llm, tools=TOOLS, prompt=prompt, verbose=False, return_intermediate_steps=False)
         result = agent.invoke({"input": question})