Guiyom commited on
Commit
96211e0
·
verified ·
1 Parent(s): ad15485

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -68,7 +68,8 @@ def openai_call(prompt: str, messages: list = None, model: str = "o3-mini",
68
  General LLM call to the OpenAI API using the specified model.
69
  If the message content exceeds MAX_MESSAGE_LENGTH, it is trimmed.
70
  """
71
- if messages is None:
 
72
  messages = [{"role": "user", "content": prompt}]
73
  if len(messages[0]['content']) > MAX_MESSAGE_LENGTH:
74
  messages[0]['content'] = messages[0]['content'][:MAX_MESSAGE_LENGTH]
 
68
  General LLM call to the OpenAI API using the specified model.
69
  If the message content exceeds MAX_MESSAGE_LENGTH, it is trimmed.
70
  """
71
+ # If messages is None or empty, create a default message using the prompt.
72
+ if messages is None or len(messages) == 0:
73
  messages = [{"role": "user", "content": prompt}]
74
  if len(messages[0]['content']) > MAX_MESSAGE_LENGTH:
75
  messages[0]['content'] = messages[0]['content'][:MAX_MESSAGE_LENGTH]