mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
72 lines
1.9 KiB
Python
72 lines
1.9 KiB
Python
import json
|
|
import logging
|
|
import shutil
|
|
|
|
from pathlib import Path
|
|
|
|
import typer
|
|
|
|
from gpt_engineer import steps
|
|
from gpt_engineer.ai import AI, fallback_model
|
|
from gpt_engineer.collect import collect_learnings
|
|
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:
|
|
# 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,
|
|
)
|
|
|
|
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"),
|
|
)
|
|
|
|
steps = STEPS[steps_config]
|
|
for step in steps:
|
|
messages = step(ai, dbs)
|
|
dbs.logs[step.__name__] = json.dumps(messages)
|
|
|
|
collect_learnings(model, temperature, steps, dbs)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|