anubhav77 commited on
Commit
c5c98c5
·
1 Parent(s): ed9eaab
Files changed (1) hide show
  1. src/chromaIntf.py +4 -1
src/chromaIntf.py CHANGED
@@ -76,7 +76,10 @@ class ChromaIntf():
76
  print("retriver state",self.retriever.search_type)
77
  try:
78
  for key in kwargs.keys():
79
- self.retriever.search_kwargs[key]=kwargs[key]
 
 
 
80
  except:
81
  print("setting search args failed")
82
  retVal=await self.retriever.aget_relevant_documents(query)
 
76
  print("retriver state",self.retriever.search_type)
77
  try:
78
  for key in kwargs.keys():
79
+ if "search_type" in key:
80
+ self.retriever.search_type=kwargs[key]
81
+ else:
82
+ self.retriever.search_kwargs[key]=kwargs[key]
83
  except:
84
  print("setting search args failed")
85
  retVal=await self.retriever.aget_relevant_documents(query)