From fa400a0b0d43d59afdea037430cc7fac57e34634 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Tue, 21 Nov 2023 19:15:16 +0800
Subject: [PATCH] chore: rename WriteCode -> WriteCodeFunction.

---
 metagpt/actions/__init__.py                                 | 2 +-
 .../actions/{write_code_v2.py => write_code_function.py}    | 2 +-
 .../{test_write_code_v2.py => test_write_code_function.py}  | 6 +++---
 3 files changed, 5 insertions(+), 5 deletions(-)
 rename metagpt/actions/{write_code_v2.py => write_code_function.py} (97%)
 rename tests/metagpt/actions/{test_write_code_v2.py => test_write_code_function.py} (78%)

diff --git a/metagpt/actions/__init__.py b/metagpt/actions/__init__.py
index d7afae2fe..d0163c24e 100644
--- a/metagpt/actions/__init__.py
+++ b/metagpt/actions/__init__.py
@@ -24,7 +24,7 @@ from metagpt.actions.write_prd import WritePRD
 from metagpt.actions.write_prd_review import WritePRDReview
 from metagpt.actions.write_test import WriteTest
 from metagpt.actions.code_executor import PyCodeExecutor
-from metagpt.actions.write_code_v2 import WriteCode as WriteCodeFunction
+from metagpt.actions.write_code_function import WriteCodeFunction
 from metagpt.actions.plan import Plan
 
 
diff --git a/metagpt/actions/write_code_v2.py b/metagpt/actions/write_code_function.py
similarity index 97%
rename from metagpt/actions/write_code_v2.py
rename to metagpt/actions/write_code_function.py
index 335e70dc0..2d943176a 100644
--- a/metagpt/actions/write_code_v2.py
+++ b/metagpt/actions/write_code_function.py
@@ -10,7 +10,7 @@ from metagpt.actions import Action
 from metagpt.schema import Message
 
 
-class WriteCode(Action):
+class WriteCodeFunction(Action):
     """Use openai function to generate code."""
 
     def __init__(self, name: str = "", context=None, llm=None) -> str:
diff --git a/tests/metagpt/actions/test_write_code_v2.py b/tests/metagpt/actions/test_write_code_function.py
similarity index 78%
rename from tests/metagpt/actions/test_write_code_v2.py
rename to tests/metagpt/actions/test_write_code_function.py
index 929407051..0e57b4ced 100644
--- a/tests/metagpt/actions/test_write_code_v2.py
+++ b/tests/metagpt/actions/test_write_code_function.py
@@ -1,11 +1,11 @@
 import pytest
 
-from metagpt.actions.write_code_v2 import WriteCode
+from metagpt.actions.write_code_function import WriteCodeFunction
 
 
 @pytest.mark.asyncio
 async def test_write_code():
-    coder = WriteCode()
+    coder = WriteCodeFunction()
     code = await coder.run("Write a hello world code.")
     assert "language" in code.content
     assert "code" in code.content
@@ -14,7 +14,7 @@ async def test_write_code():
 
 @pytest.mark.asyncio
 async def test_write_code_by_list_prompt():
-    coder = WriteCode()
+    coder = WriteCodeFunction()
     msg = ["a=[1,2,5,10,-10]", "写出求a中最大值的代码python"]
     code = await coder.run(msg)
     assert "language" in code.content