apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-30 03:16:23 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: save files

Browse source
This commit is contained in:
莘权 马 2023-11-23 11:29:09 +08:00
parent 369047e558
commit 438fbe28c0
5 changed files with 67 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

18
metagpt/utils/file_repository.py
View file

@ -8,8 +8,8 @@
"""
from __future__ import annotations
import json
import os
import uuid
from datetime import datetime
from pathlib import Path
from typing import Dict, List, Set
@ -18,6 +18,7 @@ import aiofiles
from metagpt.logs import logger
from metagpt.schema import Document
from metagpt.utils.json_to_markdown import json_to_markdown
class FileRepository:
@ -165,5 +166,16 @@ class FileRepository:
:return: A new filename string.
"""
current_time = datetime.now().strftime("%Y%m%d%H%M%S")
guid_suffix = str(uuid.uuid4())[:8]
return f"{current_time}x{guid_suffix}"
return current_time
# guid_suffix = str(uuid.uuid4())[:8]
# return f"{current_time}x{guid_suffix}"
async def save_pdf(self, doc: Document):
"""Save a Document as a PDF file.
:param doc: The Document instance to be saved.
"""
m = json.loads(doc.content)
filename = Path(doc.filename).with_suffix(".md")
await self.save(filename=str(filename), content=json_to_markdown(m))
logger.info(f"File Saved: {str(filename)}")