Update query.py

query.py

@@ -81,6 +81,7 @@ class VectaraQuery():
         chat = res['responseSet'][0]['summary'][0]['chat']
 
 #        if chat['status'] != None:
+#  #          chat['status'] =  # I have no idea what to put here hahaha
  #           st_code = chat['status']
   #          print(f"Chat query failed with code {st_code}")
    #         if st_code == 'RESOURCE_EXHAUSTED':
@@ -88,11 +89,39 @@ class VectaraQuery():
      #              return 'Sorry, chat turns exceeds plan limit.'
       #      return 'Sorry, something went wrong in my brain. Please try again later.'
         
+#Another Bard patch for diagnostics
+
         try:
-            self.conv_id = res['responseSet'][0]['summary'][0]['chat']['conversationId']
+        # Step 1a: Check for 'responseSet'
+            if res is not None and 'responseSet' in res:
+                response_set = res['responseSet']
+        
+                # Step 1b: Check for 'summary' within the first item
+                if len(response_set) > 0 and 'summary' in response_set[0]:
+                    summary = response_set[0]['summary']
+        
+                    # Step 1c: Check for 'chat' within the first summary
+                    if len(summary) > 0 and 'chat' in summary[0]:
+                        chat = summary[0]['chat']
+        
+                        # Step 1d: Finally, access 'conversationId'
+                        self.conv_id = chat['conversationId']
+                    else:
+                        # Handle missing 'chat' in summary
+                        print("Chat data not found in response summary")
+                else:
+                    # Handle missing or empty 'summary'
+                    print("Response summary missing or empty")
+            else:
+                # Handle missing or empty 'responseSet'
+                print("Response data unavailable")
         except (TypeError):
-            return 'Sorry, something went wrong. Please try again later.'
+            # Handle TypeError if any of the access attempts fail
+            print('Sorry, something went wrong. Please try again later.')
 
+        
+#End diagnostics
+        
         pattern = r'\[\d{1,2}\]'
         matches = [match.span() for match in re.finditer(pattern, summary)]