fix: default proxyllm generator function (#971)

eosphoros-ai · Dec 25, 2023 · 048fb6c · 048fb6c
1 parent 0c46c33
commit 048fb6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dbgpt/model/llm_out/proxy_llm.py b/dbgpt/model/llm_out/proxy_llm.py
@@ -34,7 +34,7 @@ def proxyllm_generate_stream(
     model_name = model_params.model_name
     default_error_message = f"{model_name} LLM is not supported"
     generator_function = generator_mapping.get(
-        model_name, lambda: default_error_message
+        model_name, lambda *args: [default_error_message]
     )
 
     yield from generator_function(model, tokenizer, params, device, context_len)