Spaces:
Running
Running
create collection before inserting docs
Browse files
app.py
CHANGED
@@ -32,6 +32,7 @@ class RetrievalAugmentedQAPipeline:
|
|
32 |
self.llm = llm
|
33 |
self.vector_db_retriever = vector_db_retriever
|
34 |
self.vector_db_retriever.delete_collection()
|
|
|
35 |
|
36 |
async def arun_pipeline(self, user_query: str):
|
37 |
context_data = self.vector_db_retriever.search_similar(user_query)
|
|
|
32 |
self.llm = llm
|
33 |
self.vector_db_retriever = vector_db_retriever
|
34 |
self.vector_db_retriever.delete_collection()
|
35 |
+
self.vector_db_retriever.create_collection()
|
36 |
|
37 |
async def arun_pipeline(self, user_query: str):
|
38 |
context_data = self.vector_db_retriever.search_similar(user_query)
|