Update query.py

query.py

@@ -76,6 +76,15 @@ class VectaraQuery():
         print(f"DEBUG request body = {body}")
         print(f"DEBUG response summary = {res['responseSet'][0]['summary']}")
 
+        ecode = res['responseSet'][0]['status'].get('code','')
+        if  ecode != '':
+            print(f"Chat query failed with code {ecode}")
+            if ecode == 'RESOURCE_EXHAUSTED':
+                self.conv_id = None
+                return 'Sorry, Vectara chat turns exceeds plan limit.'
+            return 'Sorry, something went wrong in my brain. Please try again later.'
+
+
         top_k = 10
         summary = res['responseSet'][0]['summary'][0]['text']
         responses = res['responseSet'][0]['response'][:top_k]