From 1ece762a230593c092ad99dbd9edfb2bb63181e5 Mon Sep 17 00:00:00 2001 From: imotai Date: Fri, 20 Oct 2023 16:24:33 +0800 Subject: [PATCH] fix: fix the chat function case --- chat/tests/test_chat_function.py | 2 +- kernel/tests/kernel_client_tests.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/chat/tests/test_chat_function.py b/chat/tests/test_chat_function.py index fddd6f9..82de0b0 100644 --- a/chat/tests/test_chat_function.py +++ b/chat/tests/test_chat_function.py @@ -35,7 +35,7 @@ def test_handle_final_answer_smoke_test(): respond_content = agent_server_pb2.TaskResponse( state=task_state, response_type=agent_server_pb2.TaskResponse.OnModelTypeText, - typing_content="hello world!", + typing_content=agent_server_pb2.TypingContent(content="hello world!", language="text") ) respond_final = agent_server_pb2.TaskResponse( state=task_state, diff --git a/kernel/tests/kernel_client_tests.py b/kernel/tests/kernel_client_tests.py index c2f9c11..08d812d 100644 --- a/kernel/tests/kernel_client_tests.py +++ b/kernel/tests/kernel_client_tests.py @@ -192,6 +192,8 @@ async def test_generate_pie_chart(kernel_manager): if msg: logger.debug(f"{msg}") messages.append(msg) + + logger.info(f"{messages}") assert len(list(filter(lambda x: x["msg_type"] == "display_data", messages))) > 0 await asyncio.sleep(2) await kernel_client.stop_watch()