From 0776dc7c45a143636324dc28cf753b20c64bac1e Mon Sep 17 00:00:00 2001
From: YangQianli92 <108046369+YangQianli92@users.noreply.github.com>
Date: Thu, 29 Feb 2024 11:13:40 +0800
Subject: [PATCH] Add files via upload

---
 tests/metagpt/provider/test_openai.py | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/tests/metagpt/provider/test_openai.py b/tests/metagpt/provider/test_openai.py
index 1c28fbd7b..3ce38d2a5 100644
--- a/tests/metagpt/provider/test_openai.py
+++ b/tests/metagpt/provider/test_openai.py
@@ -26,13 +26,11 @@ from tests.metagpt.provider.req_resp_const import (
     resp_cont_tmpl,
 )
 
-prompt_msg = "who are you"
-messages = [{"role": "user", "content": prompt_msg}]
-name = 'AI'
+name = "AI assistant"
 resp_cont = resp_cont_tmpl.format(name=name)
 default_resp = get_openai_chat_completion(name)
 
-default_resp_chunk = get_openai_chat_completion_chunk(name)
+default_resp_chunk = get_openai_chat_completion_chunk(name, usage_as_dict=True)
 
 usage = CompletionUsage(completion_tokens=110, prompt_tokens=92, total_tokens=202)
 
@@ -166,6 +164,3 @@ async def test_openai_acompletion(mocker):
     assert resp.usage == usage
 
     await llm_general_chat_funcs_test(llm, prompt, messages, resp_cont)
-
-
-