anubhav77 commited on
Commit
34de5cc
·
1 Parent(s): 89eca8d
Files changed (1) hide show
  1. src/chromaIntf.py +4 -4
src/chromaIntf.py CHANGED
@@ -70,12 +70,12 @@ class ChromaIntf():
70
  )
71
 
72
 
73
- def getRelevantDocs(self,query:str,count:int=8):
74
  """This should also post the result to firebase"""
75
  print("retriver state",self.retriever.search_kwargs)
76
  print("retriver state",self.retriever.search_type)
77
  self.retriever.search_kwargs["k"]=count
78
- retVal=self.retriever.get_relevant_documents(query)
79
  value=[]
80
  excludeMeta=True
81
  try:
@@ -88,7 +88,7 @@ class ChromaIntf():
88
  if key != "ID":
89
  v+=key+":"+str(item.metadata[key])+" "
90
  value.append(v)
91
- self.db_interface.add_to_cache(input=query,value=value)
92
  except:
93
  for item in retVal:
94
  v="Info:"+item.page_content+" "
@@ -96,7 +96,7 @@ class ChromaIntf():
96
  if key != "ID":
97
  v+=key+":"+str(item.metadata[key])+" "
98
  value.append(v)
99
- self.db_interface.add_to_cache(input=query,value=value)
100
  return retVal
101
 
102
 
 
70
  )
71
 
72
 
73
+ async def getRelevantDocs(self,query:str,count:int=8):
74
  """This should also post the result to firebase"""
75
  print("retriver state",self.retriever.search_kwargs)
76
  print("retriver state",self.retriever.search_type)
77
  self.retriever.search_kwargs["k"]=count
78
+ retVal=await self.retriever.get_relevant_documents(query)
79
  value=[]
80
  excludeMeta=True
81
  try:
 
88
  if key != "ID":
89
  v+=key+":"+str(item.metadata[key])+" "
90
  value.append(v)
91
+ await self.db_interface.add_to_cache(input=query,value=value)
92
  except:
93
  for item in retVal:
94
  v="Info:"+item.page_content+" "
 
96
  if key != "ID":
97
  v+=key+":"+str(item.metadata[key])+" "
98
  value.append(v)
99
+ await self.db_interface.add_to_cache(input=query,value=value)
100
  return retVal
101
 
102