shukdevdatta123 commited on
Commit
37fdc36
·
verified ·
1 Parent(s): 6648da4

Update generate_answer.py

Browse files
Files changed (1) hide show
  1. 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-3.5-turbo",
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-3.5-turbo",
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