Update query.py

query.py

@@ -73,24 +73,24 @@ class VectaraQuery():
             return "Sorry, something went wrong in my brain. Please try again later."
 
         res = response.json()
-        print(f"DEBUG request body = {body}")
-        print(f"DEBUG response summary = {res['responseSet'][0]['summary']}")
-
-        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.'
-
+#        print(f"DEBUG request body = {body}")
+#        print(f"DEBUG response summary = {res['responseSet'][0]['summary']}")
 
         top_k = 10
         summary = res['responseSet'][0]['summary'][0]['text']
         responses = res['responseSet'][0]['response'][:top_k]
         docs = res['responseSet'][0]['document']
+        chat = res['responseSet'][0]['summary'][0]['chat']
+
+        print(f"Chat = {chat}")
+        if chat['status'] != None:
+            st_code = chat['status']
+            print(f"Chat query failed with code {st_code}")
+            if st_code == '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.'
+        
         self.conv_id = res['responseSet'][0]['summary'][0]['chat']['conversationId']
         
         pattern = r'\[\d{1,2}\]'