anubhav77 commited on
Commit
4caf01e
·
1 Parent(s): d8648b8
Files changed (1) hide show
  1. src/chromaIntf.py +3 -2
src/chromaIntf.py CHANGED
@@ -125,8 +125,9 @@ class ChromaIntf():
125
  return await self.vectorstore.add_documents(docs,ids=[metadata['ID']])
126
 
127
  async def listDocs(self):
128
- collectionInfo=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
129
- return self.vectorstore._client._get(collection_id=self._uuid(collectionInfo.id))
 
130
 
131
 
132
  async def persist(self):
 
125
  return await self.vectorstore.add_documents(docs,ids=[metadata['ID']])
126
 
127
  async def listDocs(self):
128
+ collection=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
129
+ return collection.get()
130
+ #return self.vectorstore._client._get(collection_id=self._uuid(collectionInfo.id))
131
 
132
 
133
  async def persist(self):