fix all openai_client.chat calls to have n=1 for google api

modified:   app.py
This commit is contained in:
Russell Ballestrini 2024-11-21 10:33:32 -05:00
parent b0aee48352
commit c8ffe15e66

12
app.py
View file

@ -1625,6 +1625,7 @@ def gpt_generate_room_title(messages):
messages=chat_history, messages=chat_history,
model=model_name, # or any appropriate model model=model_name, # or any appropriate model
max_tokens=20, max_tokens=20,
n=1,
) )
title = response.choices[0].message.content title = response.choices[0].message.content
@ -2822,7 +2823,11 @@ def generate_grading(chat_history, rubric):
try: try:
completion = openai_client.chat.completions.create( completion = openai_client.chat.completions.create(
model=model_name, messages=messages, max_tokens=1000, temperature=0.7 model=model_name,
messages=messages,
max_tokens=1000,
temperature=0.7,
n=1,
) )
grading = completion.choices[0].message.content.strip() grading = completion.choices[0].message.content.strip()
return grading return grading
@ -2869,6 +2874,7 @@ def categorize_response(question, response, buckets, tokens_for_ai):
completion = openai_client.chat.completions.create( completion = openai_client.chat.completions.create(
model=model_name, model=model_name,
messages=messages, messages=messages,
n=1,
max_tokens=10, max_tokens=10,
temperature=0, temperature=0,
) )
@ -2904,7 +2910,7 @@ def generate_ai_feedback(
try: try:
completion = openai_client.chat.completions.create( completion = openai_client.chat.completions.create(
model=model_name, messages=messages, max_tokens=1000, temperature=0.7 model=model_name, messages=messages, max_tokens=1000, temperature=0.7, n=1
) )
feedback = completion.choices[0].message.content.strip() feedback = completion.choices[0].message.content.strip()
return feedback return feedback
@ -2961,7 +2967,7 @@ def translate_text(text, target_language):
try: try:
completion = openai_client.chat.completions.create( completion = openai_client.chat.completions.create(
model=model_name, messages=messages, max_tokens=2000, temperature=0.7 model=model_name, messages=messages, max_tokens=2000, temperature=0.7, n=1
) )
translation = completion.choices[0].message.content.strip() translation = completion.choices[0].message.content.strip()
return translation return translation