Stop generating unit test for non python files

When trying to create a simple HelloWorld with test, metagpt creates test for README.md
geekan · Jan 22, 2024 · b4e0934 · b4e0934
1 parent b5922d4
commit b4e0934
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/metagpt/roles/qa_engineer.py b/metagpt/roles/qa_engineer.py
@@ -65,6 +65,8 @@ async def _write_test(self, message: Message) -> None:
             code_doc = await src_file_repo.get(filename)
             if not code_doc:
                 continue
+            if not code_doc.filename.endswith(".py"):
+                continue
             test_doc = await tests_file_repo.get("test_" + code_doc.filename)
             if not test_doc:
                 test_doc = Document(