aljaz/gpt-engineer
1
0
Fork 0
mirror of https://github.com/aljazceru/gpt-engineer.git synced 2025-12-17 12:45:26 +01:00
Code Issues Packages Projects Releases Wiki Activity

Remove delete_existing option; Introduce archive (#409)

Browse Source 
* Remove `delete_existing` option; Introduce archive

* Update gpt_engineer/db.py

* Update gpt_engineer/main.py

* Update gpt_engineer/main.py

* Update gpt_engineer/steps.py

* Update gpt_engineer/steps.py

---------

Co-authored-by: Anton Osika <anton.osika@gmail.com>
This commit is contained in:
Anton Azarov
2023-07-02 16:56:31 +03:00
committed by GitHub
parent 14ae0e7d0e
commit 60e0a7e1dd
7 changed files with 79 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
gpt_engineer/main.py
View File

@@ -1,6 +1,6 @@
import json
import logging
import shutil
import os
from pathlib import Path
@@ -18,7 +18,6 @@ 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 = typer.Argument("gpt-4", help="model id string"),
temperature: float = 0.1,
steps_config: steps.Config = typer.Option(
@@ -35,28 +34,25 @@ def main(
):
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:
# Delete files and subdirectories in paths
shutil.rmtree(memory_path, ignore_errors=True)
shutil.rmtree(workspace_path, ignore_errors=True)
model = fallback_model(model)
ai = AI(
model=model,
temperature=temperature,
)
input_path = Path(project_path).absolute()
memory_path = input_path / f"{run_prefix}memory"
workspace_path = input_path / f"{run_prefix}workspace"
archive_path = input_path / f"{run_prefix}archive"
initial_run = not os.path.exists(memory_path) and not os.path.exists(workspace_path)
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"),
archive=None if initial_run else DB(archive_path),
)
steps = STEPS[steps_config]