graph test

app.py

@@ -139,15 +139,15 @@ class BasicAgent:
 
     def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
         print("[wiki_search] Searching for: " + str(state["search_request"]))
-        wiki_tool = WikipediaQueryRunTool()
-        results = wiki_tool.invoke(str(state["search_request"]))
-        print(f"Wiki results: {results}")
-        state["messages"].append(
-            ChatMessage(
-                role="WikipediaAgent",
-                content=results,
-            )
-        )
+        # wiki_tool = WikipediaQueryRunTool()
+        # results = wiki_tool.invoke(str(state["search_request"]))
+        # print(f"Wiki results: {results}")
+        # state["messages"].append(
+        #     ChatMessage(
+        #         role="WikipediaAgent",
+        #         content=results,
+        #     )
+        # )
         return {"messages": state["messages"]}
 
     def web_search(self, state: AnswerState) -> Dict[str, Any]:
@@ -157,7 +157,7 @@ class BasicAgent:
         print(f"Search results: {results}")
         state["messages"].append(
             ChatMessage(
-                role="DuckDuckGoSearchAgent",
+                role="tool",
                 content=results,
             )
         )