diff --git a/api/core/llm_generator/llm_generator.py b/api/core/llm_generator/llm_generator.py index 88479a8502..e9e3f7c5d8 100644 --- a/api/core/llm_generator/llm_generator.py +++ b/api/core/llm_generator/llm_generator.py @@ -605,7 +605,7 @@ class LLMGenerator: LLMResult, model_instance.invoke_llm( prompt_messages=[UserPromptMessage(content=formatted_prompt)], - model_parameters={"temperature": 0.01, "max_tokens": 2000}, + model_parameters=memory_spec.model.completion_params, stream=False, ) )