|
import json |
|
import logging |
|
import shutil |
|
|
|
from pathlib import Path |
|
|
|
import typer |
|
|
|
from gpt_engineer import steps |
|
from gpt_engineer.ai import AI |
|
from gpt_engineer.db import DB, DBs |
|
from gpt_engineer.steps import STEPS |
|
|
|
app = typer.Typer() |
|
|
|
|
|
@app.command() |
|
def main( |
|
project_path: str = typer.Argument("example", help="path"), |
|
delete_existing: bool = typer.Argument(False, help="delete existing files"), |
|
model: str = "gpt-4", |
|
temperature: float = 0.1, |
|
steps_config: steps.Config = typer.Option( |
|
steps.Config.DEFAULT, "--steps", "-s", help="decide which steps to run" |
|
), |
|
verbose: bool = typer.Option(False, "--verbose", "-v"), |
|
run_prefix: str = typer.Option( |
|
"", |
|
help=( |
|
"run prefix, if you want to run multiple variants of the same project and " |
|
"later compare them" |
|
), |
|
), |
|
): |
|
logging.basicConfig(level=logging.DEBUG if verbose else logging.INFO) |
|
|
|
input_path = Path(project_path).absolute() |
|
memory_path = input_path / f"{run_prefix}memory" |
|
workspace_path = input_path / f"{run_prefix}workspace" |
|
|
|
if delete_existing: |
|
|
|
shutil.rmtree(memory_path, ignore_errors=True) |
|
shutil.rmtree(workspace_path, ignore_errors=True) |
|
|
|
ai = AI( |
|
model=model, |
|
temperature=temperature, |
|
) |
|
|
|
dbs = DBs( |
|
memory=DB(memory_path), |
|
logs=DB(memory_path / "logs"), |
|
input=DB(input_path), |
|
workspace=DB(workspace_path), |
|
preprompts=DB(Path(__file__).parent / "preprompts"), |
|
) |
|
|
|
for step in STEPS[steps_config]: |
|
messages = step(ai, dbs) |
|
dbs.logs[step.__name__] = json.dumps(messages) |
|
|
|
|
|
if __name__ == "__main__": |
|
app() |
|
|
