浏览代码

fix: return message error in blocking mode (#657)

John Wang 1 年之前
父节点
当前提交
538e3fc256
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      api/services/completion_service.py

+ 2 - 2
api/services/completion_service.py

@@ -390,8 +390,8 @@ class CompletionService:
                         result = json.loads(result)
                         if result.get('error'):
                             cls.handle_error(result)
-
-                        return cls.get_message_response_data(result.get('data'))
+                        if 'data' in result:
+                            return cls.get_message_response_data(result.get('data'))
             except ValueError as e:
                 if e.args[0] != "I/O operation on closed file.":  # ignore this error
                     raise CompletionStoppedError()