Update generate_answer.py
Browse files- generate_answer.py +2 -2
generate_answer.py
CHANGED
@@ -23,7 +23,7 @@ def is_response_complete(response: str) -> bool:
|
|
23 |
# Retry mechanism for incomplete responses
|
24 |
def retry_response(messages):
|
25 |
response = openai.ChatCompletion.create(
|
26 |
-
model="gpt-
|
27 |
messages=messages
|
28 |
).choices[0].message['content']
|
29 |
if not is_response_complete(response):
|
@@ -36,7 +36,7 @@ def base_model_chatbot(messages):
|
|
36 |
]
|
37 |
messages = system_message + messages
|
38 |
response = openai.ChatCompletion.create(
|
39 |
-
model="gpt-
|
40 |
messages=messages
|
41 |
).choices[0].message['content']
|
42 |
# Validate response completeness
|
|
|
23 |
# Retry mechanism for incomplete responses
|
24 |
def retry_response(messages):
|
25 |
response = openai.ChatCompletion.create(
|
26 |
+
model="gpt-4o-mini",
|
27 |
messages=messages
|
28 |
).choices[0].message['content']
|
29 |
if not is_response_complete(response):
|
|
|
36 |
]
|
37 |
messages = system_message + messages
|
38 |
response = openai.ChatCompletion.create(
|
39 |
+
model="gpt-4o-mini",
|
40 |
messages=messages
|
41 |
).choices[0].message['content']
|
42 |
# Validate response completeness
|