From 222694c329d5bddc412317d4e20c774d391776b3 Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Fri, 15 Dec 2023 00:37:10 +0800
Subject: [PATCH] fix bugs

---
 metagpt/actions/write_code.py        | 13 +++++++++----
 metagpt/actions/write_code_review.py |  2 +-
 metagpt/schema.py                    |  6 ++++++
 3 files changed, 16 insertions(+), 5 deletions(-)

diff --git a/metagpt/actions/write_code.py b/metagpt/actions/write_code.py
index a91e4ee1e..5960e2621 100644
--- a/metagpt/actions/write_code.py
+++ b/metagpt/actions/write_code.py
@@ -71,7 +71,7 @@ ATTENTION: Use '##' to SPLIT SECTIONS, not '#'. Output format carefully referenc
 
 # Instruction: Based on the context, follow "Format example", write code.
 
-## Code: {filename} Write code with triple quoto, based on the following attentions and context.
+## Code: {filename}. Write code with triple quoto, based on the following attentions and context.
 1. Only One file: do your best to implement THIS ONLY ONE FILE.
 2. COMPLETE CODE: Your code will be part of the entire project, so please implement complete, reliable, reusable code snippets.
 3. Set default value: If there is any setting, ALWAYS SET A DEFAULT VALUE, ALWAYS USE STRONG TYPE AND EXPLICIT VARIABLE. AVOID circular import.
@@ -100,7 +100,7 @@ class WriteCode(Action):
             filename="test_" + coding_context.filename + ".json", relative_path=TEST_OUTPUTS_FILE_REPO
         )
         summary_doc = None
-        if coding_context.design_doc.filename:
+        if coding_context.design_doc and coding_context.design_doc.filename:
             summary_doc = await FileRepository.get_file(
                 filename=coding_context.design_doc.filename, relative_path=CODE_SUMMARIES_FILE_REPO
             )
@@ -108,9 +108,14 @@ class WriteCode(Action):
         if test_doc:
             test_detail = RunCodeResult.loads(test_doc.content)
             logs = test_detail.stderr
-        code_context = await self.get_codes(coding_context.task_doc, exclude=self.context.filename)
+
+        if bug_feedback:
+            code_context = coding_context.code_doc.content
+        else:
+            code_context = await self.get_codes(coding_context.task_doc, exclude=self.context.filename)
+
         prompt = PROMPT_TEMPLATE.format(
-            design=coding_context.design_doc.content,
+            design=coding_context.design_doc.content if coding_context.design_doc else "",
             tasks=coding_context.task_doc.content if coding_context.task_doc else "",
             code=code_context,
             logs=logs,
diff --git a/metagpt/actions/write_code_review.py b/metagpt/actions/write_code_review.py
index f63a399a9..62e96acd8 100644
--- a/metagpt/actions/write_code_review.py
+++ b/metagpt/actions/write_code_review.py
@@ -124,7 +124,7 @@ class WriteCodeReview(Action):
             code_context = await WriteCode.get_codes(self.context.task_doc, exclude=self.context.filename)
             context = "\n".join(
                 [
-                    "## System Design\n" + self.context.design_doc.content + "\n",
+                    "## System Design\n" + str(self.context.design_doc) + "\n",
                     "## Tasks\n" + task_content + "\n",
                     "## Code Files\n" + code_context + "\n",
                 ]
diff --git a/metagpt/schema.py b/metagpt/schema.py
index 799bb9253..758149efa 100644
--- a/metagpt/schema.py
+++ b/metagpt/schema.py
@@ -74,6 +74,12 @@ class Document(BaseModel):
             return None
         return str(CONFIG.git_repo.workdir / self.root_path / self.filename)
 
+    def __str__(self):
+        return self.content
+
+    def __repr__(self):
+        return self.content
+
 
 class Documents(BaseModel):
     """A class representing a collection of documents.