Upload Index.py

Index.py

@@ -187,9 +187,11 @@ def _load_embeddings_from_db(
     embeddings = Embeddings({"path": path, "content": True})
     # if Vector DB is not present
     if not db_present:
+        print("db not present")
         return embeddings
     else:
         if domain == "":
+            print("domain empty")
             embeddings.load("index")  # change this later
         else:
             print(3)
@@ -228,11 +230,13 @@ def rag(domain: str, question: str):
     # Create extractor instance
     #extractor = Extractor(embeddings, "google/flan-t5-base")
     #extractor = Extractor(embeddings, "TheBloke/Llama-2-7B-GGUF")
+    print("before calling extractor")
     extractor = Extractor(embeddings, "google/flan-t5-base")
     # llm = HuggingFaceHub(
     #     repo_id="google/flan-t5-xxl",
     #     model_kwargs={"temperature": 1, "max_length": 1000000},
     # )
     # else:
+    print("before doing Q&A")
     answer = _search(question, extractor)
     return {"question": question, "answer": answer}