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'