Update tools.py

tools.py

@@ -25,10 +25,13 @@ retriever = None
 
 
 def load_retriever():
+    """Load FAISS retriever from disk if available."""
     global retriever
     try:
         embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
-        if os.path.exists(os.path.join(VECTORSTORE_DIR, "index.faiss")):
+        index_path = os.path.join(VECTORSTORE_DIR, "index.faiss")
+
+        if os.path.exists(index_path):
             vectorstore = FAISS.load_local(
                 VECTORSTORE_DIR,
                 embeddings,
@@ -41,6 +44,7 @@ def load_retriever():
 
 
 def build_vectorstore(path: str):
+    """Build FAISS vector store from uploaded PDF."""
     loader = PyPDFLoader(path)
     docs = loader.load()
 
@@ -50,22 +54,23 @@ def build_vectorstore(path: str):
     )
 
     split_docs = splitter.split_documents(docs)
-
     embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
-    vectorstore = FAISS.from_documents(split_docs, embeddings)
 
+    vectorstore = FAISS.from_documents(split_docs, embeddings)
     vectorstore.save_local(VECTORSTORE_DIR)
+
     return vectorstore
 
 
 def update_retriever(pdf_path: str):
+    """Update retriever after document upload."""
     global retriever
     vectorstore = build_vectorstore(pdf_path)
     retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
 
 
 # ==============================
-# RAG TOOL (FIXED)
+# RAG TOOL
 # ==============================
 def create_rag_tool():
 
@@ -73,6 +78,7 @@ def create_rag_tool():
     def rag_search(query: str) -> str:
         """
         Retrieve relevant information from uploaded documents.
+        Uses FAISS-based semantic search.
         """
         global retriever
 
@@ -92,10 +98,13 @@ def create_rag_tool():
     return rag_search
 
 
-# ---------------- OTHER TOOLS ---------------- #
+# ==============================
+# EXTERNAL TOOLS
+# ==============================
 
 @tool
 def arxiv_search(query: str) -> dict:
+    """Search academic papers from arXiv."""
     try:
         arxiv = ArxivQueryRun(api_wrapper=ArxivAPIWrapper())
         return {"results": arxiv.run(query)}
@@ -105,6 +114,7 @@ def arxiv_search(query: str) -> dict:
 
 @tool
 def wikipedia_search(query: str) -> dict:
+    """Search Wikipedia for relevant information."""
     try:
         wiki = WikipediaQueryRun(api_wrapper=WikipediaAPIWrapper())
         return {"results": wiki.run(query)}
@@ -114,9 +124,9 @@ def wikipedia_search(query: str) -> dict:
 
 @tool
 def tavily_search(query: str) -> dict:
+    """Search the web using Tavily."""
     try:
         search = TavilySearchResults(max_results=5)
         return {"results": search.run(query)}
     except Exception as e:
         return {"error": str(e)}
-