fix:issue 2323

eosphoros-ai · Jan 24, 2025 · 776fd85 · 776fd85
1 parent 1a6ad50
commit 776fd85
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/dbgpt/app/scene/base_chat.py b/dbgpt/app/scene/base_chat.py
@@ -231,10 +231,14 @@ async def _build_model_request(self) -> ModelRequest:
             span_id=root_tracer.get_current_span_id(),
         )
         temperature = float(
-            self._chat_param.get("temperature", self.prompt_template.temperature)
+            self._chat_param.get("temperature")
+            if self._chat_param.get("temperature")
+            else self.prompt_template.temperature
         )
         max_new_tokens = int(
-            self._chat_param.get("max_new_tokens", self.prompt_template.max_new_tokens)
+            self._chat_param.get("max_new_tokens")
+            if self._chat_param.get("max_new_tokens")
+            else self.prompt_template.max_new_tokens
         )
         node = AppChatComposerOperator(
             model=self.llm_model,