annikwag commited on
Commit
4bb94c0
·
verified ·
1 Parent(s): 38d8c2b

Update appStore/rag_utils.py

Browse files
Files changed (1) hide show
  1. appStore/rag_utils.py +8 -0
appStore/rag_utils.py CHANGED
@@ -124,5 +124,13 @@ def get_rag_answer(query, top_results, endpoint, token):
124
  if "Answer:" in answer:
125
  answer = answer.split("Answer:")[-1].strip()
126
  return answer
 
 
 
 
 
 
 
 
127
  else:
128
  return f"Error in generating answer: {response.text}"
 
124
  if "Answer:" in answer:
125
  answer = answer.split("Answer:")[-1].strip()
126
  return answer
127
+ elif response.status_code == 503:
128
+ # Custom message with a larger llama icon and red highlighted text
129
+ return (
130
+ "<span style='color: red;'>"
131
+ "<span style='font-size: 2em;'>🦙</span> tzzz tzzz I'm currently sleeping. "
132
+ "Please come back in 10 minutes, and I'll be fully awake to answer your question."
133
+ "</span>"
134
+ )
135
  else:
136
  return f"Error in generating answer: {response.text}"