black formatted

This commit is contained in:
PatillaCode
2023-06-15 18:33:52 +02:00
parent 12b280fbbb
commit f4eb1e109f
6 changed files with 38 additions and 41 deletions

18
main.py
View File

@@ -1,14 +1,11 @@
import json
import os
import pathlib
from typing import Optional
import openai
from chat_to_files import to_files
from ai import AI
from steps import STEPS
from db import DB, DBs
import typer
from ai import AI
from db import DB, DBs
from steps import STEPS
app = typer.Typer()
@@ -16,11 +13,13 @@ app = typer.Typer()
@app.command()
def chat(
project_path: str = typer.Argument(None, help="path"),
run_prefix: str = typer.Option("", help="run prefix, if you want to run multiple variants of the same project and later compare them"),
run_prefix: str = typer.Option(
"",
help="run prefix, if you want to run multiple variants of the same project and later compare them",
),
model: str = "gpt-4",
temperature: float = 0.1,
):
if project_path is None:
project_path = str(pathlib.Path(__file__).parent / "example")
@@ -41,7 +40,6 @@ def chat(
identity=DB(pathlib.Path(__file__).parent / "identity"),
)
for step in STEPS:
messages = step(ai, dbs)
dbs.logs[step.__name__] = json.dumps(messages)