Update app.py

app.py

@@ -518,17 +518,15 @@ def safety_node(state: AgentState):
     information = state["information"]
 
     prompt = f"""
-You are a response agent.
+You are a reasoning agent who takes into account the provided information, if available, and answers to the user request.
 
-You must reason over the user request and the provided information and output the answer to the user's request. Reason well over the information provided, if any, and output the answer that satisfies the user's question exactly.
+You must reason over the user request and the provided information and output the answer to the user's request. Reason well over the information provided, if any, and output the answer to the user's question exactly.
 
 You MUST ONLY return EXACTLY the answer to the user's question in the following format:
 Response: <answer>
 
 DO NOT add anything additional and return ONLY what is asked and in the format asked.
 
-ONLY return a response if you are confident about the answer, otherwise return empty string.
-
 If you output anything else, it is incorrect.
 
 If there is no information provided or the information is not relevant then answer as best based on your own knowledge.
@@ -549,8 +547,6 @@ Information:
     raw_output = reasoning_generate(prompt)
     # raw_output = generate(prompt)
 
-    state["raw_output"] = raw_output
-
     logger.info(f"Raw Output: {raw_output}")
 
     # output = raw_output.split("Response:")[-1].strip()
@@ -1027,12 +1023,12 @@ class BasicAgent:
 
             try:
                 response = self.safe_app.invoke(state)
-                agent_answer = response["output"]
 
                 if "answer: " in response["information"]:
                     response["output"] = response["information"].split("answer: ")[-1].strip()
 
-                agent_answer = str(response)
+                agent_answer = response["output"]
+                # agent_answer = str(response)
 
                 # agent_answer = str(df)
                 # agent_answer = str(response.status_code) + " - " + task_id