From 38aa99382f5e8c6282df1152872033cc50cb0d82 Mon Sep 17 00:00:00 2001 From: Akio Nishimura Date: Wed, 27 Sep 2023 12:56:33 +0900 Subject: [PATCH] Fix stop key of TextGen. --- libs/langchain/langchain/llms/textgen.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libs/langchain/langchain/llms/textgen.py b/libs/langchain/langchain/llms/textgen.py index 6b409ecb12f69..e962c7be278bb 100644 --- a/libs/langchain/langchain/llms/textgen.py +++ b/libs/langchain/langchain/llms/textgen.py @@ -175,7 +175,7 @@ def _get_parameters(self, stop: Optional[List[str]] = None) -> Dict[str, Any]: params = {"preset": self.preset} # then sets it as configured, or default to an empty list: - params["stop"] = self.stopping_strings or stop or [] + params["stopping_strings"] = self.stopping_strings or stop or [] return params