Bugfixes

pantel2 · Jun 11, 2023 · 35c75d2 · 35c75d2
1 parent 86426e9
commit 35c75d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -19,7 +19,6 @@ def chat(
     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,
-    max_tokens: int = 8192,
 ):
 
     if project_path is None:
@@ -32,7 +31,6 @@ def chat(
     ai = AI(
         model=model,
         temperature=temperature,
-        max_tokens=max_tokens,
     )
 
     dbs = DBs(
@@ -50,4 +48,4 @@ def chat(
 
 
 if __name__ == "__main__":
-    app()
+    app()
diff --git a/steps.py b/steps.py
@@ -1,6 +1,7 @@
 from ai import AI
 from chat_to_files import to_files
 from db import DBs
+import json
 
 
 def setup_sys_prompt(dbs):