revise qwen

2024-07-02 14:22:45 +00:00
parent 6cd2d80dfd
commit 0c6c357e9c
2 changed files with 8 additions and 3 deletions
--- a/request_llms/com_qwenapi.py
+++ b/request_llms/com_qwenapi.py
@@ -65,8 +65,12 @@ class QwenRequestInstance():
                self.result_buf += f"[Local Message] 请求错误：状态码：{response.status_code}，错误码:{response.code}，消息：{response.message}"
                yield self.result_buf
                break
-        logging.info(f'[raw_input] {inputs}')
-        logging.info(f'[response] {self.result_buf}')
+
+        # 耗尽generator避免报错
+        while True:
+            try: next(responses)
+            except: break
+
        return self.result_buf