TejAndrewsACC commited on
Commit
bd89c17
·
verified ·
1 Parent(s): 2964355

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -6
app.py CHANGED
@@ -3636,12 +3636,26 @@ def respond(message, history, max_tokens, temperature, top_p):
3636
  messages.append(["assistant", val["content"]])
3637
  messages.append(["user", message])
3638
 
3639
- response = ""
3640
- for message in client.chat_completion(
3641
- messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
3642
- ):
3643
- token = message.choices[0].delta.content
3644
- response += token
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3645
  yield response
3646
 
3647
 
 
3636
  messages.append(["assistant", val["content"]])
3637
  messages.append(["user", message])
3638
 
3639
+ max_retries = 25
3640
+ retry_count = 0
3641
+
3642
+ while retry_count < max_retries:
3643
+ try:
3644
+ response = ""
3645
+ for message in client.chat_completion(
3646
+ messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p
3647
+ ):
3648
+ token = message.choices[0].delta.content
3649
+ response += token
3650
+ yield response
3651
+ break
3652
+ except Exception as e:
3653
+ print(f"Attempt {retry_count + 1} failed: {e}")
3654
+ retry_count += 1
3655
+ time.sleep(1)
3656
+
3657
+ if retry_count == max_retries:
3658
+ response = random.choice(errored_responses)
3659
  yield response
3660
 
3661