소스 검색

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()