Added wiki search

.gitignore

@@ -171,4 +171,6 @@ cython_debug/
 .ruff_cache/
 
 # PyPI configuration file
-.pypirc
+.pypirc
+
+.langgraph_api/

agent.py

@@ -7,32 +7,56 @@ from langgraph.graph import START, StateGraph, MessagesState
 from langgraph.prebuilt import tools_condition, ToolNode
 from langchain_core.runnables import RunnableConfig
 
+# Wikepedia tool
+from langchain_community.tools import WikipediaQueryRun
+from langchain_community.utilities import WikipediaAPIWrapper
+from langchain_community.document_loaders import WikipediaLoader
+from langchain_community.retrievers import WikipediaRetriever
 
 load_dotenv()
 
 
 # --- TOOLS ---
-def add(a: int, b: int) -> int:
-    """Adds a and b.
+# Simple tool to test a tool call
+def meaning_of_life(a: int, b: int) -> int:
+    """Returns meaning of life
 
     Args:
         a: first int
         b: second int
     """
-    return a + b
+    return 42
 
 
-def multiply(a: int, b: int) -> int:
-    """Multiplies a and b.
+# https://www.restack.io/docs/langchain-knowledge-wikipedia-loader-cat-ai
+# https://api.python.langchain.com/en/latest/community/document_loaders/langchain_community.document_loaders.wikipedia.WikipediaLoader.html#
+def wikipedia_search(query: str) -> str:
+    """Searches Wikipedia for a given query and fetches full document
 
     Args:
-        a: first int
-        b: second int
+        query: the query to search for
     """
-    return a * b
+    loader = loader = WikipediaLoader(
+        query=query,
+        load_max_docs=1,
+        doc_content_chars_max=4000,
+        load_all_available_meta=False,
+    )
+    documents = loader.load()
+
+    formatted_search_docs = "\n\n---\n\n".join(
+        [
+            f'<Document source="{next_doc.metadata["source"]}" title="{next_doc.metadata.get("title", "")}"/>\n{next_doc.page_content}\n</Document>'
+            for next_doc in documents
+        ]
+    )
+
+    result = f"{{wiki_results: {formatted_search_docs}}}"
 
+    return result
 
-tools = [add, multiply]
+
+tools = [meaning_of_life, wikipedia_search]
 
 
 # --- GRAPH ---
@@ -98,13 +122,25 @@ if __name__ == "__main__":
     # Build the graph
     graph = create_graph()
     # Run the graph
+    """ 
+    print(f"******** TEST NORMAL LLM CALL ********")
     question = "What is an elephant? "
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:
         m.pretty_print()
 
-    question = "What is 10+10?"
+    print(f"******** TESTING MEANING OF LIFE TOOL ********")
+    question = "What is meaning of life 10+10?"
+    messages = [HumanMessage(content=question)]
+    messages = graph.invoke({"messages": messages})
+    for m in messages["messages"]:
+        m.pretty_print()
+
+ """
+    print("******** TESTING WIKEPEDIA TOOL ********")
+    # expected answer is "Samuel"
+    question = "Search Wikipedia and find out who is the recipient of the malko competition in 2024"
     messages = [HumanMessage(content=question)]
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:

requirements.txt

@@ -7,3 +7,4 @@ langchain-core
 langchain-community
 langchain-openai
 langgraph-cli[inmem]
+wikipedia