feat(core): allow overriding ls_model_name from kwargs (#32541)

This commit is contained in:
Matthew Lapointe
2025-09-11 16:18:06 -04:00
committed by GitHub
parent 2903e08311
commit b1f08467cd
11 changed files with 129 additions and 7 deletions

View File

@@ -720,7 +720,9 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
ls_params["ls_stop"] = stop
# model
if hasattr(self, "model") and isinstance(self.model, str):
if "model" in kwargs and isinstance(kwargs["model"], str):
ls_params["ls_model_name"] = kwargs["model"]
elif hasattr(self, "model") and isinstance(self.model, str):
ls_params["ls_model_name"] = self.model
elif hasattr(self, "model_name") and isinstance(self.model_name, str):
ls_params["ls_model_name"] = self.model_name

View File

@@ -357,7 +357,9 @@ class BaseLLM(BaseLanguageModel[str], ABC):
ls_params["ls_stop"] = stop
# model
if hasattr(self, "model") and isinstance(self.model, str):
if "model" in kwargs and isinstance(kwargs["model"], str):
ls_params["ls_model_name"] = kwargs["model"]
elif hasattr(self, "model") and isinstance(self.model, str):
ls_params["ls_model_name"] = self.model
elif hasattr(self, "model_name") and isinstance(self.model_name, str):
ls_params["ls_model_name"] = self.model_name