anubhav77 commited on
Commit
01de2b3
·
1 Parent(s): 80d6857
Files changed (2) hide show
  1. src/chromaIntf.py +8 -0
  2. src/main.py +2 -2
src/chromaIntf.py CHANGED
@@ -83,9 +83,13 @@ class ChromaIntf():
83
  self.retriever.search_kwargs[key]=kwargs[key]
84
  except:
85
  print("setting search args failed")
 
86
  retVal=await self.retriever.aget_relevant_documents(query)
87
  value=[]
88
  excludeMeta=True
 
 
 
89
  try:
90
  for item in retVal:
91
  if excludeMeta:
@@ -96,8 +100,10 @@ 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
  except:
 
101
  for item in retVal:
102
  if excludeMeta:
103
  v=item.page_content+" \n"
@@ -107,7 +113,9 @@ class ChromaIntf():
107
  if key != "ID":
108
  v+=key+":"+str(item.metadata[key])+" "
109
  value.append(v)
 
110
  self.db_interface.add_to_cache(input=query,value=value)
 
111
  return retVal
112
 
113
 
 
83
  self.retriever.search_kwargs[key]=kwargs[key]
84
  except:
85
  print("setting search args failed")
86
+ print("reaching step2")
87
  retVal=await self.retriever.aget_relevant_documents(query)
88
  value=[]
89
  excludeMeta=True
90
+ print("reaching step3")
91
+ print(",".join(retVal))
92
+ print("reaching step4")
93
  try:
94
  for item in retVal:
95
  if excludeMeta:
 
100
  if key != "ID":
101
  v+=key+":"+str(item.metadata[key])+" "
102
  value.append(v)
103
+ print("reaching step5")
104
  self.db_interface.add_to_cache(input=query,value=value)
105
  except:
106
+ print("reaching step6")
107
  for item in retVal:
108
  if excludeMeta:
109
  v=item.page_content+" \n"
 
113
  if key != "ID":
114
  v+=key+":"+str(item.metadata[key])+" "
115
  value.append(v)
116
+ print("reaching step7")
117
  self.db_interface.add_to_cache(input=query,value=value)
118
+ print("reaching step8")
119
  return retVal
120
 
121
 
src/main.py CHANGED
@@ -86,11 +86,11 @@ async def get_matching_docs_sync(inStr: str, kwargs: Dict [Any, Any]) -> Any:
86
  TODO: Add parameter to return the source information as well
87
  """
88
  logger.info("================================================")
89
- logger.info("Received new request at /getMatchingDocs:"+inStr)
90
  retVal=await chromaIntf.getRelevantDocs(inStr,kwargs)
91
  #background_tasks.add_task(chromaIntf.getRelevantDocs,inStr,kwargs)
92
  #retVal="added to queue"
93
- logger.info("Returning:"+retVal)
94
  logger.info("================================================")
95
  return retVal
96
 
 
86
  TODO: Add parameter to return the source information as well
87
  """
88
  logger.info("================================================")
89
+ logger.info("Received new request at /getMatchingDocsSync:"+inStr)
90
  retVal=await chromaIntf.getRelevantDocs(inStr,kwargs)
91
  #background_tasks.add_task(chromaIntf.getRelevantDocs,inStr,kwargs)
92
  #retVal="added to queue"
93
+ logger.info("Returning:"+",,".join(retVal))
94
  logger.info("================================================")
95
  return retVal
96