Add extra documentation

This commit is contained in:
Andres Caicedo
2023-04-03 14:10:02 +02:00
parent 8e2074f483
commit 765210f0cd
7 changed files with 18 additions and 3 deletions

View File

@@ -6,6 +6,7 @@ openai.api_key = cfg.openai_api_key
# Overly simple abstraction until we create something better
def create_chat_completion(messages, model=None, temperature=None, max_tokens=None)->str:
"""Create a chat completion using the OpenAI API."""
response = openai.ChatCompletion.create(
model=model,
messages=messages,