anubhav77 commited on
Commit
9467b57
·
1 Parent(s): 8d7feb0
Files changed (2) hide show
  1. src/chromaIntf.py +1 -1
  2. src/main.py +1 -1
src/chromaIntf.py CHANGED
@@ -125,7 +125,7 @@ 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)[0]
129
  return self.vectorstore._client._get(collection_id=self._uuid(collectionInfo.id))
130
 
131
 
 
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
 
src/main.py CHANGED
@@ -102,7 +102,7 @@ def walk(path: PathRequest):
102
  return response
103
 
104
  @app.get(api_base+"/list")
105
- async def list_collection():
106
  return await chromaIntf.listDocs()
107
 
108
  print(__name__)
 
102
  return response
103
 
104
  @app.get(api_base+"/list")
105
+ async def list_docs():
106
  return await chromaIntf.listDocs()
107
 
108
  print(__name__)