fix: fix the chat function case

dbpunk-labs · Oct 20, 2023 · 1ece762 · 1ece762
1 parent dfabb5f
commit 1ece762
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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()