Update app.py

app.py

@@ -865,7 +865,7 @@ def tool_executor(state: AgentState):
                      "extremely",]
                     webpage_information_complete = " \n ".join(texts)
                 else:
-                    webpage_results = visit_webpage_wiki(action["args"]["url"])
+                    webpage_results = visit_webpage_wiki(action.args["url"])
                     webpage_result = " \n ".join(webpage_results)
 
                     # for webpage_result in webpage_results:
@@ -909,6 +909,7 @@ def tool_executor(state: AgentState):
                 pass
         elif "answer" in state["proposed_action"]:
             webpage_information_complete = f"""answer: {state["proposed_action"]["answer"]}"""
+            state["best_query_webpage_information_similarity_score"] = 1.0
         else:
             webpage_information_complete = ""
     
@@ -918,8 +919,12 @@ def tool_executor(state: AgentState):
         state["information"] = webpage_information_complete[:3000]
         state["best_query_webpage_information_similarity_score"] = best_query_webpage_information_similarity_score
     except:
-        state["information"] = ""
-        state["best_query_webpage_information_similarity_score"] = -1.0
+        if "answer" in state["proposed_action"]:
+            webpage_information_complete = f"""answer: {state["proposed_action"]["answer"]}"""
+            state["best_query_webpage_information_similarity_score"] = 1.0
+        else:
+            state["information"] = ""
+            state["best_query_webpage_information_similarity_score"] = -1.0
     
     # logger.info(f"Information: {state['information']}")
     # logger.info(f"Information: {state['best_query_webpage_information_similarity_score']}")