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

update code due to failed unittests

Browse source
This commit is contained in:
better629 2024-02-03 13:33:02 +08:00
parent 2eeb9556f5
commit f2dbb51094
4 changed files with 14 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/metagpt/actions/test_action_node.py
View file

@ -244,13 +244,19 @@ def test_create_model_class_with_mapping():
@pytest.mark.asyncio
async def test_action_node_with_image():
async def test_action_node_with_image(mocker):
# add a mock to update model in unittest, due to the gloabl MockLLM
def _cons_kwargs(self, messages: list[dict], timeout=3, **extra_kwargs) -> dict:
kwargs = {"messages": messages, "temperature": 0.3, "model": "gpt-4-vision-preview"}
return kwargs
invoice = ActionNode(
key="invoice", expected_type=bool, instruction="if it's a invoice file, return True else False", example="False"
)
invoice_path = Path(__file__).parent.joinpath("..", "..", "data", "invoices", "invoice-2.png")
img_base64 = encode_image(invoice_path)
mocker.patch("metagpt.provider.openai_api.OpenAILLM._cons_kwargs", _cons_kwargs)
node = await invoice.fill(context="", llm=LLM(), images=[img_base64])
assert node.instruct_content.invoice

6
tests/metagpt/utils/test_repair_llm_raw_output.py
View file

@ -135,7 +135,7 @@ def test_repair_json_format():
}
"""
target_output = """{
"Language": "en_us",
"Language": "en_us",
"Programming Language": "Python"
}"""
output = repair_llm_raw_output(output=raw_output, req_keys=[None], repair_type=RepairType.JSON)
@ -148,7 +148,7 @@ def test_repair_json_format():
}
"""
target_output = """{
"Language": "en_us",
"Language": "en_us",
"Programming Language": "Python"
}"""
output = repair_llm_raw_output(output=raw_output, req_keys=[None], repair_type=RepairType.JSON)
@ -161,7 +161,7 @@ def test_repair_json_format():
}
"""
target_output = """{
"Language": "#en_us#",
"Language": "#en_us#",
"Programming Language": "//Python # Code // Language//"
}"""
output = repair_llm_raw_output(output=raw_output, req_keys=[None], repair_type=RepairType.JSON)