apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add files via upload

Browse source
This commit is contained in:
YangQianli92 2024-02-29 11:13:40 +08:00 • committed by GitHub
parent b5be7a28bf
commit 0776dc7c45
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
tests/metagpt/provider/test_openai.py
View file

@ -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)