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

refactor: write prd & system design

Browse source
This commit is contained in:
莘权 马 2023-11-22 21:59:14 +08:00
parent e8131652de
commit 62d93517b4
2 changed files with 17 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

21
metagpt/actions/design_api.py
View file

@ -290,10 +290,9 @@ class WriteDesign(Action):
data_api_design = m.get("Data structures and interface definitions")
if not data_api_design:
return
path = CONFIG.git_repo.workdir / DATA_API_DESIGN_FILE_REPO
if not path.exists():
path.mkdir(parents=True, exists_ok=True)
await mermaid_to_file(data_api_design, path / Path(design_doc).with_suffix(".mmd"))
pathname = CONFIG.git_repo.workdir / DATA_API_DESIGN_FILE_REPO / Path(design_doc).with_suffix(".mmd")
await WriteDesign._save_mermaid_file(data_api_design, pathname)
logger.info(f"Save class view to {str(pathname)}")
@staticmethod
async def _save_seq_flow(design_doc):
@ -301,13 +300,19 @@ class WriteDesign(Action):
seq_flow = m.get("Program call flow")
if not seq_flow:
return
path = CONFIG.git_repo.workdir / SEQ_FLOW_FILE_REPO
if not path.exists():
path.mkdir(parents=True, exists_ok=True)
await mermaid_to_file(seq_flow, path / Path(design_doc).with_suffix(".mmd"))
pathname = CONFIG.git_repo.workdir / SEQ_FLOW_FILE_REPO / Path(design_doc).with_suffix(".mmd")
await WriteDesign._save_mermaid_file(seq_flow, pathname)
logger.info(f"Saving sequence flow to {str(pathname)}")
@staticmethod
async def _save_pdf(design_doc):
m = json.loads(design_doc.content)
file_repo = CONFIG.git_repo.new_file_repository(SYSTEM_DESIGN_PDF_FILE_REPO)
await file_repo.save(filename=design_doc.filename, content=json_to_markdown(m))
logger.info(f"Saving system design pdf to {design_doc.root_relative_path}")
@staticmethod
async def _save_mermaid_file(data: str, pathname: Path):
if not pathname.parent.exists():
pathname.parent.mkdir(parents=True, exists_ok=True)
await mermaid_to_file(data, pathname)

8
metagpt/actions/write_prd.py
View file

@ -305,10 +305,10 @@ class WritePRD(Action):
quadrant_chart = m.get("Competitive Quadrant Chart")
if not quadrant_chart:
return
path = CONFIG.git_repo.workdir / COMPETITIVE_ANALYSIS_FILE_REPO
if not path.exists():
path.mkdir(parents=True, exists_ok=True)
await mermaid_to_file(quadrant_chart, path / Path(prd_doc).with_suffix(".mmd"))
pathname = CONFIG.git_repo.workdir / Path(COMPETITIVE_ANALYSIS_FILE_REPO) / Path(prd_doc).with_suffix(".mmd")
if not pathname.parent.exists():
pathname.parent.mkdir(parents=True, exists_ok=True)
await mermaid_to_file(quadrant_chart, pathname)
@staticmethod
async def _save_pdf(prd_doc):