koffiwind commited on
Commit
fcc34f7
·
1 Parent(s): d9c3884

create collection before inserting docs

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)