Fix: tools addition

utils/helpers.py

@@ -8,6 +8,42 @@ import os
 import random
 from datetime import datetime
 from zoneinfo import ZoneInfo
+from langchain_community.document_loaders import PyPDFDirectoryLoader
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_google_genai import GoogleGenerativeAIEmbeddings
+from langchain_community.vectorstores import FAISS
+
+def create_vector_store():
+    """
+    Checks if a vector store index exists. If not, it creates one from
+    the PDFs in the knowledge_base folder.
+    """
+    persist_directory = 'faiss_index'
+    if os.path.exists(persist_directory):
+        print("--- Knowledge base (FAISS index) already exists. Loading... ---")
+        return
+
+    # Check if there are files to process
+    if not os.path.exists("./knowledge_base") or not os.listdir("./knowledge_base"):
+         print("--- 'knowledge_base' folder is empty or does not exist. Skipping index creation. ---")
+         return
+
+    print("--- Creating new knowledge base... ---")
+    loader = PyPDFDirectoryLoader("./knowledge_base/")
+    documents = loader.load()
+    if not documents:
+        print("--- No documents could be loaded. Skipping index creation. ---")
+        return
+
+    print(f"--- Loaded {len(documents)} document(s). Splitting text... ---")
+    text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
+    docs = text_splitter.split_documents(documents)
+    
+    print(f"--- Creating embeddings and vector store. This may take a moment... ---")
+    embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
+    db = FAISS.from_documents(docs, embeddings)
+    db.save_local(persist_directory)
+    print("--- Knowledge base created successfully. ---")
 
 def load_quotes():
     """Load inspirational quotes from Gita/Vedas"""