Update query.py

query.py

@@ -123,9 +123,9 @@ class VectaraQuery():
         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']
+        chat = res['responseSet'][0]['summary'][0].get('chat', None)
 
-        if chat['status'] != None:
+        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':
@@ -133,8 +133,7 @@ class VectaraQuery():
                 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']
-
+        self.conv_id = chat['conversationId'] if chat else None
         summary = CitationNormalizer().normalize_citations(summary, responses, docs)        
         return summary