Add --verbose option for logging request/response

This commit is contained in:
Carl Thomé
2023-06-18 16:21:16 +02:00
parent 16a3278ee3
commit 38401b2f35
2 changed files with 12 additions and 1 deletions

View File

@@ -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

View File

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