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

feat: ask_code message support type list[str].

Browse source
This commit is contained in:
刘棒棒 2023-11-21 12:01:28 +08:00
parent 4418fe8283
commit 657571a192
2 changed files with 12 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -68,3 +68,13 @@ def test_ask_code_list_Message():
assert "language" in rsp
assert "code" in rsp
assert len(rsp["code"]) > 0
def test_ask_code_list_str():
llm = OpenAIGPTAPI()
msg = ["a=[1,2,5,10,-10]", "写出求a中最大值的代码python"]
rsp = llm.ask_code(msg) # -> {'language': 'python', 'code': 'max_value = max(a)\nmax_value'}
print(rsp)
assert "language" in rsp
assert "code" in rsp
assert len(rsp["code"]) > 0