Update app.py

app.py

@@ -70,7 +70,7 @@ Action Input: the input to the action
 Observation: the result of the action
 ... (this Thought/Action/Action Input/Observation can repeat N times)
 Thought: I now know the final answer
-Final Answer: {final_answer}  # Ensure this is defined as the output key
+Final Answer: the final response you want to return. This is your output.
 Begin!
 
 Question: {input}
@@ -79,19 +79,17 @@ Thought: Let's approach this step-by-step:
 )
 
 agent = create_react_agent(
-    local_llm, 
-    tools, 
-    prompt,
-    output_key="final_answer"  # Specify the output key here
+    llm=local_llm, 
+    tools=tools, 
+    prompt=prompt
 )
 
-# Create the agent executor
+# Create the agent executor without specifying output_keys
 agent_executor = AgentExecutor.from_agent_and_tools(
     agent=agent, 
     tools=tools, 
     verbose=True, 
-    memory=ConversationBufferMemory(),
-    output_keys=["final_answer"]  # Explicitly define the output key(s) here
+    memory=ConversationBufferMemory()
 )
 
 # Streamlit interface
@@ -115,7 +113,7 @@ if lead_name:
             with st.spinner("AI is generating a response..."):
                 response = agent_executor.run(initial_message)
                 st.write("AI Response:")
-                st.write(response)  # Now 'response' will directly be the final answer
+                st.write(response)  # The response should be the final output
 
 
 st.sidebar.title("About")