anubhav77 commited on
Commit
c738eed
·
1 Parent(s): 74ee141
Files changed (2) hide show
  1. src/chromaIntf.py +3 -2
  2. src/main.py +3 -1
src/chromaIntf.py CHANGED
@@ -128,11 +128,12 @@ class ChromaIntf():
128
  Document(page_content=inStr, metadata=metadata)]
129
  with open("./docs/"+metadata['ID'],"w") as fd:
130
  fd.write(inStr)
 
131
  try:
132
- return await self.vectorstore.aadd_documents(docs,ids=[metadata.ID])
133
  except:
134
  print("inside expect of addText")
135
- return await self.vectorstore.aadd_documents(docs,ids=[metadata['ID']])
136
 
137
  async def listDocs(self):
138
  collection=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
 
128
  Document(page_content=inStr, metadata=metadata)]
129
  with open("./docs/"+metadata['ID'],"w") as fd:
130
  fd.write(inStr)
131
+ print("written to file", inStr)
132
  try:
133
+ return await self.vectorstore.aadd_documents(docs,ids=[metadata['ID']])
134
  except:
135
  print("inside expect of addText")
136
+ return await self.vectorstore.aadd_documents(docs,ids=[metadata.ID])
137
 
138
  async def listDocs(self):
139
  collection=self.vectorstore._client.get_collection(self.vectorstore._LANGCHAIN_DEFAULT_COLLECTION_NAME,embedding_function=self.embedding)
src/main.py CHANGED
@@ -88,7 +88,7 @@ async def reset_db():
88
 
89
  @app.post(api_base+"/walk")
90
  def walk(path: PathRequest):
91
- print("Received walk request")
92
  dirs=[]
93
  fileList=[]
94
  try:
@@ -99,6 +99,8 @@ def walk(path: PathRequest):
99
  fileList.append(filea)
100
  except Exception:
101
  print("got exception",Exception)
 
 
102
  response= JSONResponse(content= {"dirs":dirs,"files":fileList})
103
  return response
104
 
 
88
 
89
  @app.post(api_base+"/walk")
90
  def walk(path: PathRequest):
91
+ print("Received walk request for",path)
92
  dirs=[]
93
  fileList=[]
94
  try:
 
99
  fileList.append(filea)
100
  except Exception:
101
  print("got exception",Exception)
102
+ print("dirs ",dirs)
103
+ print("files ",fileList)
104
  response= JSONResponse(content= {"dirs":dirs,"files":fileList})
105
  return response
106