graph test

app.py

@@ -141,15 +141,15 @@ class BasicAgent:
 
     def wiki_search(self, state: AnswerState) -> Dict[str, Any]:
         print("[wiki_search] Searching for: " + str(state["search_request"]))
-        # wiki_tool = WikipediaAPIWrapper()
-        # results = wiki_tool.run(str(state["search_request"]))
-        # print(f"Wiki results: {results}")
-        # state["messages"].append(
-        #     ChatMessage(
-        #         role="tool",
-        #         content=results,
-        #     )
-        # )
+        wiki_tool = WikipediaAPIWrapper()
+        results = wiki_tool.run(str(state["search_request"]))
+        print(f"Wiki results: {results}")
+        state["messages"].append(
+            ChatMessage(
+                role="user",
+                content="WikipediaAgent tool results are: " + results,
+            )
+        )
         return {"messages": state["messages"]}
 
     def web_search(self, state: AnswerState) -> Dict[str, Any]:

requirements.txt

@@ -8,3 +8,4 @@ langchain_huggingface
 langchain-community
 ddgs
 duckduckgo-search
+wikipedia