Added web search 4 out of 20

agent.py

@@ -13,6 +13,10 @@ from langchain_community.utilities import WikipediaAPIWrapper
 from langchain_community.document_loaders import WikipediaLoader
 from langchain_community.retrievers import WikipediaRetriever
 
+# Tavily web search
+from langchain_community.tools.tavily_search import TavilySearchResults
+
+
 load_dotenv()
 
 
@@ -43,20 +47,49 @@ def wikipedia_search(query: str) -> str:
         load_all_available_meta=False,
     )
     documents = loader.load()
+    formatted_search_docs = "\n\n---\n\n"
 
-    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
-        ]
-    )
+    for next_doc in documents:
+        formatted_doc = f'<Document source="{next_doc.metadata["source"]}" title="{next_doc.metadata.get("title", "")}"/>\n{next_doc.page_content}\n</Document>'
+        formatted_search_docs.join(formatted_doc)
 
     result = f"{{wiki_results: {formatted_search_docs}}}"
 
     return result
 
 
-tools = [meaning_of_life, wikipedia_search]
+def web_search(query: str) -> str:
+    """Search Web with Tavily for a query and return results.
+
+    Args:
+        query: The search query."""
+    tool = TavilySearchResults(
+        max_results=3,
+        include_answer=True,
+        include_raw_content=True,
+        include_images=True,
+        # search_depth="advanced",
+        # include_domains = []
+        # exclude_domains = []
+    )
+
+    documents = tool.invoke(input=query)
+    formatted_search_docs = "\n\n---\n\n"
+    for next_doc in documents:
+        url = next_doc["url"]
+        title = next_doc["title"]
+        content = next_doc["content"]
+        formatted_doc = (
+            f'<Document source="{url}" title="{title}"/>\n{content}\n</Document>'
+        )
+        formatted_search_docs = formatted_search_docs + formatted_doc
+
+    result = f"{{web_results: {formatted_search_docs}}}"
+
+    return result
+
+
+tools = [meaning_of_life, wikipedia_search, web_search]
 
 
 # --- GRAPH ---
@@ -121,6 +154,7 @@ def create_graph():
 if __name__ == "__main__":
     # Build the graph
     graph = create_graph()
+
     # Run the graph
     """ 
     print(f"******** TEST NORMAL LLM CALL ********")
@@ -137,7 +171,7 @@ if __name__ == "__main__":
     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"
@@ -145,3 +179,13 @@ if __name__ == "__main__":
     messages = graph.invoke({"messages": messages})
     for m in messages["messages"]:
         m.pretty_print()
+
+"""
+
+    print("******** TESTING WEB SEARCH TOOL ********")
+    # expected answer is "Samuel"
+    question = "Search web for information about mozart"
+    messages = [HumanMessage(content=question)]
+    messages = graph.invoke({"messages": messages})
+    for m in messages["messages"]:
+        m.pretty_print()

requirements.txt

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