brige/human
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 2 Wiki Activity

add test code

Browse Source
This commit is contained in:
brige 2024-10-28 20:05:39 +08:00
parent 3f833a3c42
commit 92b9f6def4
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nlp/nlp_doubao.py
View File

@ -33,6 +33,7 @@ class DouBao(NLPBase):
logger.info(f'_request:{question}') logger.info(f'_request:{question}')
logger.info(f'-------dou_bao ask:{question}') logger.info(f'-------dou_bao ask:{question}')
try: try:
'''
stream = await self.__client.chat.completions.create( stream = await self.__client.chat.completions.create(
model="ep-20241008152048-fsgzf", model="ep-20241008152048-fsgzf",
messages=[ messages=[
@ -50,6 +51,14 @@ class DouBao(NLPBase):
self._on_callback(message) self._on_callback(message)
self._on_callback(sec) self._on_callback(sec)
await stream.close() 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: except Exception as e:
print(e) print(e)
logger.info(f'_request:{question}, time:{time.time() - t:.4f}s') logger.info(f'_request:{question}, time:{time.time() - t:.4f}s')