From 92b9f6def41024198900698e8e94384144396d43 Mon Sep 17 00:00:00 2001 From: brige Date: Mon, 28 Oct 2024 20:05:39 +0800 Subject: [PATCH] add test code --- nlp/nlp_doubao.py | 9 +++++++++ 1 file changed, 9 insertions(+) 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')