From 7e329a478afcedfe6a353cf5fb5cec1863fb9f83 Mon Sep 17 00:00:00 2001 From: shenchucheng Date: Thu, 17 Aug 2023 17:14:43 +0800 Subject: [PATCH] Make the report output directory if it doesn't exist. --- metagpt/roles/researcher.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/metagpt/roles/researcher.py b/metagpt/roles/researcher.py index 815cfa172..acb46c718 100644 --- a/metagpt/roles/researcher.py +++ b/metagpt/roles/researcher.py @@ -84,10 +84,17 @@ class Researcher(Role): return msg def write_report(self, topic: str, content: str): + if not RESEARCH_PATH.exists(): + RESEARCH_PATH.mkdir(parents=True) filepath = RESEARCH_PATH / f"{topic}.md" filepath.write_text(content) if __name__ == "__main__": - role = Researcher(language="en-us") - asyncio.run(role.run("dataiku vs. datarobot")) + import fire + + async def main(topic: str, language="en-us"): + role = Researcher(topic, language=language) + await role.run(topic) + + fire.Fire(main)