mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
f159dc45f2
* First step in collecting learnings * Rename prompts * remove requirements, use pip install -e . instead * Add requirements * Fix tests
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
from dataclasses import dataclass
|
|
from pathlib import Path
|
|
|
|
|
|
# This class represents a simple database that stores its data as files in a directory.
|
|
class DB:
|
|
"""A simple key-value store, where keys are filenames and values are file contents."""
|
|
|
|
def __init__(self, path):
|
|
self.path = Path(path).absolute()
|
|
|
|
self.path.mkdir(parents=True, exist_ok=True)
|
|
|
|
def __contains__(self, key):
|
|
return (self.path / key).is_file()
|
|
|
|
def __getitem__(self, key):
|
|
full_path = self.path / key
|
|
|
|
if not full_path.is_file():
|
|
raise KeyError(key)
|
|
with full_path.open("r", encoding="utf-8") as f:
|
|
return f.read()
|
|
|
|
def get(self, key, default=None):
|
|
try:
|
|
return self[key]
|
|
except KeyError:
|
|
return default
|
|
|
|
def __setitem__(self, key, val):
|
|
full_path = self.path / key
|
|
full_path.parent.mkdir(parents=True, exist_ok=True)
|
|
|
|
if isinstance(val, str):
|
|
full_path.write_text(val, encoding="utf-8")
|
|
else:
|
|
# If val is neither a string nor bytes, raise an error.
|
|
raise TypeError("val must be either a str or bytes")
|
|
|
|
|
|
# dataclass for all dbs:
|
|
@dataclass
|
|
class DBs:
|
|
memory: DB
|
|
logs: DB
|
|
preprompts: DB
|
|
input: DB
|
|
workspace: DB
|