diff --git a/gpt_engineer/ai.py b/gpt_engineer/ai.py index 0714869..0ccb59d 100644 --- a/gpt_engineer/ai.py +++ b/gpt_engineer/ai.py @@ -1,5 +1,9 @@ +import logging + import openai +logger = logging.getLogger(__name__) + class AI: def __init__(self, **kwargs): @@ -36,6 +40,7 @@ class AI: if prompt: messages = messages + [{"role": "user", "content": prompt}] + logger.debug(f"Creating a new chat completion: {messages}") response = openai.ChatCompletion.create( messages=messages, stream=True, **self.kwargs ) @@ -46,4 +51,6 @@ class AI: msg = delta.get("content", "") print(msg, end="") chat.append(msg) - return messages + [{"role": "assistant", "content": "".join(chat)}] + messages = messages + [{"role": "assistant", "content": "".join(chat)}] + logger.debug(f"Chat completion finished: {messages}") + return messages diff --git a/gpt_engineer/main.py b/gpt_engineer/main.py index 059f754..5695e7f 100644 --- a/gpt_engineer/main.py +++ b/gpt_engineer/main.py @@ -1,4 +1,5 @@ import json +import logging import os import pathlib import shutil @@ -26,7 +27,10 @@ def chat( model: str = "gpt-4", temperature: float = 0.1, steps_config: str = "default", + verbose: bool = typer.Option(False, "--verbose", "-v"), ): + logging.basicConfig(level=logging.DEBUG if verbose else logging.INFO) + app_dir = pathlib.Path(os.path.curdir) input_path = pathlib.Path(app_dir / "projects" / project_path) memory_path = input_path / (run_prefix + "memory")