updates

agent.py

@@ -1,11 +1,10 @@
+import json
 import os
 from typing import Annotated
 
+import pandas as pd
 from langchain_community.document_loaders import WikipediaLoader, YoutubeLoader
 from langchain_community.document_loaders.youtube import TranscriptFormat
-
-# --- Langchain / Langraph ---
-from langchain_community.tools.tavily_search import TavilySearchResults
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
 from langchain_core.runnables import RunnableLambda
 from langchain_core.tools import tool
@@ -16,6 +15,9 @@ from langchain_huggingface import (
     HuggingFaceEmbeddings,
     HuggingFaceEndpoint,
 )
+
+# --- Langchain / Langraph ---
+from langchain_tavily import TavilySearch
 from langgraph.graph import END, START, MessagesState, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
@@ -27,7 +29,7 @@ def search_web_sources(query: Annotated[str, "Search query string"]) -> dict:
 
     if not os.environ.get("TAVILY_API_KEY"):
         raise EnvironmentError("TAVILY_API_KEY is not set in environment variables.")
-    search_docs = TavilySearchResults(max_results=3).invoke({"query": query})
+    search_docs = TavilySearch(max_results=3).invoke({"query": query})
     formatted = "\n\n---\n\n".join(
         [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}">\n{doc.page_content}\n</Document>'

requirements.txt

@@ -10,4 +10,5 @@ langchain-tavily
 langgraph
 tavily-python
 wikipedia
-youtube-transcript-api==0.6.3
+youtube-transcript-api==0.6.3
+pytube