klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rename method _get_gpt4all_answer to _get_answer (#727)

Browse Source
This commit is contained in:
Prikshit
2023-09-30 09:45:31 +05:30
committed by GitHub
parent 6ae33d04b8
commit 16b0d5b829
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
embedchain/llm/gpt4all.py
View File

@@ -14,7 +14,7 @@ class GPT4ALLLlm(BaseLlm):
self.instance = GPT4ALLLlm._get_instance(self.config.model) self.instance = GPT4ALLLlm._get_instance(self.config.model)
def get_llm_model_answer(self, prompt): def get_llm_model_answer(self, prompt):
return self._get_gpt4all_answer(prompt=prompt, config=self.config) return self._get_answer(prompt=prompt, config=self.config)
@staticmethod @staticmethod
def _get_instance(model): def _get_instance(model):
@@ -27,7 +27,7 @@ class GPT4ALLLlm(BaseLlm):
return GPT4All(model_name=model) return GPT4All(model_name=model)
def _get_gpt4all_answer(self, prompt: str, config: BaseLlmConfig) -> Union[str, Iterable]: def _get_answer(self, prompt: str, config: BaseLlmConfig) -> Union[str, Iterable]:
if config.model and config.model != self.config.model: if config.model and config.model != self.config.model:
raise RuntimeError( raise RuntimeError(
"OpenSourceApp does not support switching models at runtime. Please create a new app instance." "OpenSourceApp does not support switching models at runtime. Please create a new app instance."