Merge pull request #429 from davidleon/improvement/lazy_external_load

fix extra kwargs error: keyword_extraction.
This commit is contained in:
zrguo
2024-12-09 18:07:30 +08:00
committed by GitHub
2 changed files with 26 additions and 8 deletions

View File

@@ -919,6 +919,8 @@ class MultiModel:
self, prompt, system_prompt=None, history_messages=[], **kwargs
) -> str:
kwargs.pop("model", None) # stop from overwriting the custom model name
kwargs.pop("keyword_extraction", None)
kwargs.pop("mode", None)
next_model = self._next_model()
args = dict(
prompt=prompt,