youngtsai commited on
Commit
877f289
·
1 Parent(s): dad95ce

"response_format": response_format

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -73,15 +73,18 @@ def generate_topic_sentences(model, max_tokens, sys_content, scenario, eng_level
73
  {"role": "system", "content": sys_content},
74
  {"role": "user", "content": user_content}
75
  ]
 
 
76
 
77
  request_payload = {
78
  "model": model,
79
  "messages": messages,
80
  "max_tokens": max_tokens,
 
81
  }
82
 
83
  response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
84
- content = response.choices[0].message.content.strip()
85
 
86
  return content
87
 
 
73
  {"role": "system", "content": sys_content},
74
  {"role": "user", "content": user_content}
75
  ]
76
+ response_format = { "type": "json_object" }
77
+
78
 
79
  request_payload = {
80
  "model": model,
81
  "messages": messages,
82
  "max_tokens": max_tokens,
83
+ "response_format": response_format
84
  }
85
 
86
  response = OPEN_AI_CLIENT.chat.completions.create(**request_payload)
87
+ content = response.choices[0].message.content
88
 
89
  return content
90