Update query.py

query.py

@@ -76,13 +76,15 @@ class VectaraQuery():
         print(f"DEBUG request body = {body}")
         print(f"DEBUG response summary = {res['responseSet'][0]['summary']}")
 
-        ecode = res['responseSet'][0]['status'][0].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.'
+        st = res['responseSet'][0]['status']
+        if len(st)>0:
+            ecode = st[0].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