Merge pull request #2550 from xinnan-tech/py_openai_fix

参数补充
This commit is contained in:
hrz
2025-11-18 22:07:01 +08:00
committed by GitHub
@@ -24,6 +24,7 @@ class LLMProvider(LLMProviderBase):
"max_tokens": int,
"temperature": lambda x: round(float(x), 1),
"top_p": lambda x: round(float(x), 1),
"top_k": int,
"frequency_penalty": lambda x: round(float(x), 1),
}
@@ -39,7 +40,7 @@ class LLMProvider(LLMProviderBase):
setattr(self, param, None)
logger.debug(
f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.frequency_penalty}"
f"意图识别参数初始化: {self.temperature}, {self.max_tokens}, {self.top_p}, {self.top_k}, {self.frequency_penalty}"
)
model_key_msg = check_model_key("LLM", self.api_key)
@@ -70,6 +71,7 @@ class LLMProvider(LLMProviderBase):
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
"temperature": kwargs.get("temperature", self.temperature),
"top_p": kwargs.get("top_p", self.top_p),
"top_k": kwargs.get("top_k", self.top_k),
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
}
@@ -114,6 +116,7 @@ class LLMProvider(LLMProviderBase):
"max_tokens": kwargs.get("max_tokens", self.max_tokens),
"temperature": kwargs.get("temperature", self.temperature),
"top_p": kwargs.get("top_p", self.top_p),
"top_k": kwargs.get("top_k", self.top_k),
"frequency_penalty": kwargs.get("frequency_penalty", self.frequency_penalty),
}