Update query.py

query.py

@@ -81,6 +81,8 @@ class VectaraQuery():
         response = requests.post(endpoint, data=json.dumps(body), verify=True, headers=self.get_headers())
         if response.status_code != 200:
             print(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
+            if response.status_code == 429:
+                return "Sorry, Vectara chat turns exceeds plan limit."
             return "Sorry, something went wrong in my brain. Please try again later."
 
         res = response.json()
@@ -89,15 +91,6 @@ class VectaraQuery():
             self.conv_id = res['chat_id']
 
         summary = res['answer']
-
-        # FIGURE OUT HOW TO IMPLEMENT THIS IN APIV2
-        # if chat and chat['status'] is not 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.'
         
         return summary