mitulagr2 commited on
Commit
09e28b7
·
1 Parent(s): f4f9ced
Files changed (1) hide show
  1. app/rag.py +2 -2
app/rag.py CHANGED
@@ -78,7 +78,7 @@ class ChatPDF:
78
  logger.info("enumerating text_chunks")
79
  for idx, text_chunk in enumerate(self.text_chunks):
80
  node = TextNode(text=text_chunk)
81
- self.src_doc = docs[doc_ids[idx]]
82
  node.metadata = src_doc.metadata
83
  self.nodes.append(node)
84
 
@@ -109,7 +109,7 @@ class ChatPDF:
109
  return streaming_response
110
 
111
  def clear(self):
112
- self.self.client.delete(collection_name="rag_documents", ids=doc_ids)
113
  self.pdf_count = 0
114
  self.text_chunks = []
115
  self.doc_ids = []
 
78
  logger.info("enumerating text_chunks")
79
  for idx, text_chunk in enumerate(self.text_chunks):
80
  node = TextNode(text=text_chunk)
81
+ self.src_doc = docs[self.doc_ids[idx]]
82
  node.metadata = src_doc.metadata
83
  self.nodes.append(node)
84
 
 
109
  return streaming_response
110
 
111
  def clear(self):
112
+ self.self.client.delete(collection_name="rag_documents", ids=self.doc_ids)
113
  self.pdf_count = 0
114
  self.text_chunks = []
115
  self.doc_ids = []