Update app/policy_vector_db.py

app/policy_vector_db.py

@@ -8,10 +8,10 @@ from sentence_transformers import SentenceTransformer
 
 class PolicyVectorDB:
     """Manages the creation and searching of a persistent vector database."""
-    def __init__(self, persist_directory: str = "chroma_db"):
+    def __init__(self, persist_directory: str = "/tmp/policy_vector_db"):
         self.client = chromadb.PersistentClient(path=persist_directory)
         self.collection_name = "neepco_dop_policies"
-        self.embedding_model = SentenceTransformer('BAAI/bge-large-en-v1.5', device = 'cpu')
+        self.embedding_model = SentenceTransformer('BAAI/bge-large-en-v1.5', device='cpu')
         self.collection = self.client.get_or_create_collection(
             name=self.collection_name,
             metadata={"description": "NEEPCO Delegation of Powers Policy"}
@@ -75,7 +75,8 @@ class PolicyVectorDB:
 
 def main():
     """Main function to build and verify the vector database."""
-    INPUT_CHUNKS_PATH = os.path.join(os.path.dirname(__file__), "../processed_chunks.json")
+    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
+    INPUT_CHUNKS_PATH = os.path.join(BASE_DIR, "../processed_chunks.json")
     PERSIST_DIRECTORY = "/tmp/policy_vector_db"
 
     if not os.path.exists(INPUT_CHUNKS_PATH):
@@ -125,4 +126,4 @@ def main():
             print("  No results found.")
 
 if __name__ == "__main__":
-    main()
+    main()