Update query.py

query.py

@@ -69,17 +69,30 @@ class VectaraQuery():
         pattern_max_length = 50  # Example heuristic
         for line in response.iter_lines():
             if line:  # filter out keep-alive new lines
-                data = json.loads(line.decode('utf-8'))                
+                data = json.loads(line.decode('utf-8'))
+                
+                if 'result' not in data:
+                    continue
+
                 res = data['result']
+                if 'responseSet' not in res:
+                    continue
+                
                 response_set = res['responseSet']
                 
                 if response_set:
                     for result in response_set:
+                        if 'text' not in result:
+                            continue
                         text = result['text']
                         # Extract relevant information from the text
-                        reason = re.search(r"Reason Why it Can't be Used: (.*?)\n", text).group(1)
-                        alternative = re.search(r"Alternative: (.*?)\n", text).group(1)
-                        notes = re.search(r"Notes: (.*?)\n", text).group(1)
+                        reason_match = re.search(r"Reason Why it Can't be Used: (.*?)\n", text)
+                        alternative_match = re.search(r"Alternative: (.*?)\n", text)
+                        notes_match = re.search(r"Notes: (.*?)\n", text)
+                        
+                        reason = reason_match.group(1) if reason_match else "Not available"
+                        alternative = alternative_match.group(1) if alternative_match else "Not available"
+                        notes = notes_match.group(1) if notes_match else "Not available"
                         
                         response = f"Reason: {reason}\nAlternative: {alternative}\nNotes: {notes}"
                         return response