Update app.py

app.py

@@ -865,16 +865,16 @@ def tool_executor(state: AgentState):
                      "extremely",]
                     webpage_information_complete = " \n ".join(texts)
                 else:
-                    webpage_results = visit_webpage_wiki(result)
+                    webpage_results = visit_webpage_wiki(action["args"]["url"])
                     webpage_result = " \n ".join(webpage_results)
-    
+
                     # for webpage_result in webpage_results:
                     query_embeddings = sentence_transformer_model.encode_query(state["messages"][-1].content).reshape(1, -1)
                     webpage_information_embeddings = sentence_transformer_model.encode_query(webpage_result).reshape(1, -1)
                     query_webpage_information_similarity_score = float(cosine_similarity(query_embeddings, webpage_information_embeddings)[0][0])
         
                     # logger.info(f"Webpage Information and Similarity Score: {result} - {webpage_result} - {query_webpage_information_similarity_score}")
-        
+
                     if query_webpage_information_similarity_score > 0.65:
                         webpage_information_complete += webpage_result
                         webpage_information_complete += " \n "
@@ -907,8 +907,10 @@ def tool_executor(state: AgentState):
             except Exception as e:
                 webpage_information_complete = str(e)
                 pass
+        elif "answer" in state["proposed_action"]:
+            webpage_information_complete = f"""answer: {state["proposed_action"]["answer"]}"""
         else:
-            result = "Unknown tool"
+            webpage_information_complete = ""
     
         if webpage_information_complete == "" and best_query_webpage_information_similarity_score > 0.30:
             webpage_information_complete = best_webpage_information
@@ -1016,6 +1018,9 @@ class BasicAgent:
                 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 = str(df)