graph test

app.py

@@ -8,6 +8,8 @@ import gradio as gr
 import pandas as pd
 import requests
 from langchain_community.tools import DuckDuckGoSearchRun
+from langchain_community.utilities.wikipedia import WikipediaAPIWrapper
+
 from langchain_core import messages
 from langchain_core.messages import (
     BaseMessage,
@@ -139,12 +141,12 @@ 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"]))
+        # wiki_tool = WikipediaAPIWrapper()
+        # results = wiki_tool.run(str(state["search_request"]))
         # print(f"Wiki results: {results}")
         # state["messages"].append(
         #     ChatMessage(
-        #         role="WikipediaAgent",
+        #         role="tool",
         #         content=results,
         #     )
         # )
@@ -157,8 +159,8 @@ class BasicAgent:
         print(f"Search results: {results}")
         state["messages"].append(
             ChatMessage(
-                role="tool",
-                content=results,
+                role="user",
+                content="DuckDuckGoSearchAgent tool results are: " + results,
             )
         )
         return {"messages": state["messages"]}