fix: general structure of the project

.gitignore

@@ -2,5 +2,6 @@
 .idea/
 data/
 vector_stores/
-*/__pycache__/*
+*/__pycache__/
 .env
+__pycache__/

app.py

@@ -1,6 +1,4 @@
 import os
-import time
-
 import gradio as gr
 from langchain.agents import create_agent
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace

tools/RetriveBooksDataTool.py

@@ -1,17 +1,14 @@
 from langchain.tools import tool
+from rag import get_books_vector_store
 
 
 @tool
-def retrieve_book_context(query: str, vector_store):
-    """Search the customer database for records matching the query.
-
-    Args:
-        query: Search terms to look for
-        limit: Maximum number of results to return
-    """
-    retrieved_docs = vector_store.similarity_search(query, k=2)
+def retrieve_book_context(query: str):
+    """Search for relevant book context for a given query."""
+    vector_store = get_books_vector_store()
+    retrieved_docs = vector_store.similarity_search(query, k=3)
     serialized = "\n\n".join(
         (f"Source: {doc.metadata}\nContent: {doc.page_content}")
         for doc in retrieved_docs
     )
-    return serialized, retrieved_docs
+    return serialized