chore

agent.py

@@ -1,6 +1,6 @@
 import time
 from langchain.chains import RetrievalQA
-from langchain.vectorstores import Chroma
+from langchain_chroma import Chroma
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from langgraph.graph import StateGraph, START
@@ -34,8 +34,6 @@ sys_msg = SystemMessage(
     """
 )
 
-print("agent.py loaded")
-
 class CUSTOM_AGENT:
     """
         A simple deterministic agent that leverages our tools directly and avoids
@@ -76,7 +74,7 @@ class CUSTOM_AGENT:
     def _retriever_node(self, state: MessagesState):
         """Retriever node"""
         question = state["messages"][ -1 ].content
-        docs = self.retriever.get_relevant_documents(question)
+        docs = self.retriever.invoke(question)
 
         if docs:
             context = "\n\n".join([d.page_content for d in docs])

agent_tools.py

@@ -99,7 +99,8 @@ def visit_webpage(url: str) -> str:
     """)
 def duckduck_websearch(query: str) -> str:
     search_engine = DuckDuckGoSearchResults(output_format="list", num_results=2)
-    page_urls = [url["link"] for url in search_engine(query)]
+    results = search_engine.invoke({"query": query})
+    page_urls = [url["link"] for url in results]
 
     loader = WebBaseLoader(web_paths=page_urls)
     docs = loader.load()

requirements.txt

@@ -31,3 +31,4 @@ langchain-google-genai
 pytesseract
 matplotlib
 sentence_transformers
+langchain_chroma

test.py

@@ -107,7 +107,6 @@ questions = [
 # Test
 if __name__ == "__main__":
     agent = CUSTOM_AGENT()
-    q = questions[9]
-    print("Question:", q["question"])
+    q = questions[0]
     answer = agent.run(q)
     print("Answer:", answer)