From 472e4963f5d0fbf852851586e850e5397739b8ea Mon Sep 17 00:00:00 2001 From: Chenwei Xiao Date: Sun, 3 Aug 2025 20:50:12 +0800 Subject: [PATCH] Convert dict to string for chat response --- .gitignore | 1 + local_mode/models/codegeex.py | 2 ++ 2 files changed, 3 insertions(+) create mode 100644 .gitignore diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..bee8a64 --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +__pycache__ diff --git a/local_mode/models/codegeex.py b/local_mode/models/codegeex.py index a46c942..3365e07 100644 --- a/local_mode/models/codegeex.py +++ b/local_mode/models/codegeex.py @@ -76,6 +76,8 @@ def chat(self, request: ChatCompletionRequest): top_p=request.top_p, repetition_penalty=request.presence_penalty ) + if isinstance(response, dict): + response = '\n'.join([response.get('name'), response.get('content') or str(response.get('parameters'))]) resp = ChatCompletionResponse() resp.choices[0].message.content = response resp.choices[0].finish_reason = 'stop'