diff --git a/nlp/nlp_doubao.py b/nlp/nlp_doubao.py index bed4d4d..b43f3f5 100644 --- a/nlp/nlp_doubao.py +++ b/nlp/nlp_doubao.py @@ -33,6 +33,7 @@ class DouBao(NLPBase): logger.info(f'_request:{question}') logger.info(f'-------dou_bao ask:{question}') try: + ''' stream = await self.__client.chat.completions.create( model="ep-20241008152048-fsgzf", messages=[ @@ -50,6 +51,14 @@ class DouBao(NLPBase): self._on_callback(message) self._on_callback(sec) await stream.close() + ''' + sec = "你是测试客服,是由字节跳动开发的 AI 人工智能助手" + sec, message = self._split_handle.handle(sec) + sec, message = self._split_handle.handle(sec) + if len(message) > 0: + self._on_callback(message) + if len(sec) > 0: + self._on_callback(sec) except Exception as e: print(e) logger.info(f'_request:{question}, time:{time.time() - t:.4f}s')