anubhav77 commited on
Commit
4ab9cb1
·
1 Parent(s): 37419af
Files changed (2) hide show
  1. src/chromaIntf.py +2 -1
  2. src/llm/geminiLLM.py +2 -1
src/chromaIntf.py CHANGED
@@ -59,7 +59,8 @@ class ChromaIntf():
59
  ]
60
  self.document_content_description = "Information to store for retrival from LLM based chatbot"
61
  lf=LLMFactory()
62
- self.llm=lf.get_llm("executor2")
 
63
 
64
  self.retriever = SelfQueryRetriever.from_llm(
65
  self.llm,
 
59
  ]
60
  self.document_content_description = "Information to store for retrival from LLM based chatbot"
61
  lf=LLMFactory()
62
+ #self.llm=lf.get_llm("executor2")
63
+ self.llm=lf.get_llm("executor3")
64
 
65
  self.retriever = SelfQueryRetriever.from_llm(
66
  self.llm,
src/llm/geminiLLM.py CHANGED
@@ -74,7 +74,8 @@ class GeminiLLM(LLM):
74
  logger.error("Will switch to fallback LLM as response from palm is None::")
75
  raise(Exception)
76
  if run_manager:
77
- run_manager.on_llm_end(val)
 
78
  return val
79
 
80
  @property
 
74
  logger.error("Will switch to fallback LLM as response from palm is None::")
75
  raise(Exception)
76
  if run_manager:
77
+ pass
78
+ #run_manager.on_llm_end(val)
79
  return val
80
 
81
  @property