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

修改函数名称

Browse source
This commit is contained in:
黄伟韬 2024-08-16 11:22:02 +08:00
parent 0a1c965933
commit b27ba8476f
2 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
metagpt/exp_pool/context_builders/role_zero.py
View file

@ -30,10 +30,10 @@ class RoleZeroContextBuilder(BaseContextBuilder):
return req_copy
def replace_example_content(self, text: str, new_example_content: str) -> str:
return self.replace_content_of_example_tag(text, new_example_content)
return self.fill_experience(text, new_example_content)
@staticmethod
def replace_content_of_example_tag(text: str, new_example_content: str) -> str:
pattern = "# Past Experience\n<experience>"
replaced_text = text.replace(pattern, "# Past Experience\n" + new_example_content)
def fill_experience(text: str, new_example_content: str) -> str:
pattern = "<experience>"
replaced_text = text.replace(pattern, new_example_content)
return replaced_text

12
tests/metagpt/exp_pool/test_context_builders/test_rolezero_context_builder.py
View file

@ -30,20 +30,20 @@ class TestRoleZeroContextBuilder:
assert result == [{"content": "Updated content"}]
def test_replace_example_content(self, context_builder, mocker):
mocker.patch.object(RoleZeroContextBuilder, "replace_content_of_example_tag", return_value="Replaced content")
mocker.patch.object(RoleZeroContextBuilder, "fill_experience", return_value="Replaced content")
result = context_builder.replace_example_content("Original text", "New example content")
assert result == "Replaced content"
context_builder.replace_content_of_example_tag.assert_called_once_with("Original text", "New example content")
context_builder.fill_experience.assert_called_once_with("Original text", "New example content")
def test_replace_content_of_example_tag(self):
def test_fill_experience(self):
text = "Start\n# Past Experience\n<experience>\n\n# Instruction\nEnd"
new_content = "New content"
result = RoleZeroContextBuilder.replace_content_of_example_tag(text, new_content)
result = RoleZeroContextBuilder.fill_experience(text, new_content)
expected = "Start\n# Past Experience\nNew content\n\n# Instruction\nEnd"
assert result == expected
def test_replace_content_of_example_tag_no_match(self):
def test_fill_experience_no_match(self):
text = "Start\nNo markers\nEnd"
new_content = "New content"
result = RoleZeroContextBuilder.replace_content_of_example_tag(text, new_content)
result = RoleZeroContextBuilder.fill_experience(text, new_content)
assert result == text