Update app.py

app.py

@@ -10,6 +10,16 @@ from langchain.tools.base import StructuredTool
 from langchain.agents import initialize_agent
 from qa_txt import llm
 import gradio as gr
+from langchain.agents import (
+    create_react_agent,
+    AgentExecutor,
+    tool,
+)
+from langchain import hub
+
+
+
+prompt = hub.pull("hwchase17/react")
 
 def faq(query: str) -> str:
     reponse = conversation_chain({"question": query, "chat_history": []})
@@ -29,7 +39,7 @@ qa_faq = StructuredTool.from_function(
 )
 
 def request_data(query: str) -> str:
-    request = chain.run(query)
+    request = chain.invoke({"input": query})['text']
     mot_cle = nettoyer_string(request)
     mots = mot_cle.split()
     ui = mots[0]
@@ -58,17 +68,18 @@ tools_add = [
     fetch_data,
 ]
 
-agent = initialize_agent(
-    tools = tools_add,
-    llm = llm,
-    agent = "zero-shot-react-description",
-    verbose = True
+agent = create_react_agent(llm=llm, tools=tools_add, prompt=prompt)
+
+agent_executor = AgentExecutor(
+    agent=agent,
+    tools=tools_add,
+    verbose=True,
 )
 
 def data_gov_ma(message, history):
   try:
-    response = agent.run(message)
-    return response
+    response = agent_executor.invoke({"input": message})
+    return response['output']
   except ValueError as e:
     return "Je suis désolé, je n'ai pas compris votre question. Pourriez-vous la reformuler s'il vous plaît ?"