From 38401b2f355cc7da18c34affe445d3bf65fe05cc Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Carl=20Thom=C3=A9?= <carlthome@gmail.com>
Date: Sun, 18 Jun 2023 16:21:16 +0200
Subject: [PATCH] Add --verbose option for logging request/response

---
 gpt_engineer/ai.py   | 9 ++++++++-
 gpt_engineer/main.py | 4 ++++
 2 files changed, 12 insertions(+), 1 deletion(-)

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")