[Fix] Fix lightllmapi list bug (#635)

522241a8 · Yang Yong · GitHub · 18849126 · 522241a8
Unverified Commit 522241a8 authored Nov 24, 2023 by Yang Yong Committed by GitHub Nov 24, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

opencompass/models/lightllm_api.py opencompass/models/lightllm_api.py +4 -1

No files found.
--- a/opencompass/models/lightllm_api.py
+++ b/opencompass/models/lightllm_api.py
@@ -73,7 +73,10 @@ class LightllmAPI(BaseAPIModel):
                continue
            try:
                response = raw_response.json()
-                return response['generated_text']
+                generated_text = response['generated_text']
+                if isinstance(generated_text, list):
+                    generated_text = generated_text[0]
+                return generated_text
            except requests.JSONDecodeError:
                self.logger.error('JsonDecode error, got',
                                  str(raw_response.content))