format json stream log

software_company.py

@@ -1,4 +1,5 @@
 import asyncio
+from functools import partial
 import json
 from pydantic import BaseModel, Field
 import datetime
@@ -28,6 +29,9 @@ from metagpt.utils.common import any_to_str, read_json_file, write_json_file
 from metagpt.utils.git_repository import GitRepository
 
 
+_default_llm_stream_log = partial(print, end="")
+
+
 class RoleRun(Action):
     role: Role
 
@@ -131,10 +135,29 @@ class SoftwareCompany(Role):
             self.finish = True
             return Message(output.content, role=self.profile, cause_by=type(self._rc.todo))
 
+        default_log_stream = CONFIG.get("LLM_STREAM_LOG", _default_llm_stream_log)
+
+        start = False
+        insert_code = False
+
+        def log_stream(msg):
+            nonlocal start, insert_code
+            if not start:
+                if msg.startswith("["):
+                    msg = "```json\n" + msg
+                    insert_code = True
+                start = True
+            return default_log_stream(msg)
+
+        CONFIG.LLM_STREAM_LOG = log_stream
+
         output = await self.active_role._act()
         self.active_role._set_state(state=-1)
         self.active_role.publish_message(output)
 
+        if insert_code:
+            default_log_stream("```")
+
         cause_by = output.cause_by
 
         if cause_by == any_to_str(WritePRD):