Update query.py

query.py

@@ -56,7 +56,6 @@ class VectaraQuery():
         }
 
     def submit_query(self, query_str: str):
-
         endpoint = f"https://api.vectara.io/v1/stream-query"
         body = self.get_body(query_str)
         response = requests.post(endpoint, data=json.dumps(body), verify=True, headers=self.get_headers(), stream=True) 
@@ -64,14 +63,12 @@ class VectaraQuery():
             print(f"Query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             return "Sorry, something went wrong. Please try again later."
 
-        chunks = []
-        accumulated_text = ""  # Initialize text accumulation
-        pattern_max_length = 50  # Example heuristic
+        accumulated_text = ""
         for line in response.iter_lines():
             if line:  # filter out keep-alive new lines
                 data = json.loads(line.decode('utf-8'))
                 print(f"Received data chunk: {json.dumps(data, indent=2)}")  # Debugging line
-                
+
                 if 'result' not in data:
                     continue
 
@@ -87,14 +84,14 @@ class VectaraQuery():
                             continue
                         text = result['text']
                         print(f"Processing text: {text}")  # Debugging line
-                        # Extract relevant information from the text
-                        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_match = re.search(r"Reason Why it Can't be Used: (.*?)\n", text, re.DOTALL)
+                        alternative_match = re.search(r"Alternative: (.*?)\n", text, re.DOTALL)
+                        notes_match = re.search(r"Notes: (.*?)\n", text, re.DOTALL)
                         
-                        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"
+                        reason = reason_match.group(1).strip() if reason_match else "Not available"
+                        alternative = alternative_match.group(1).strip() if alternative_match else "Not available"
+                        notes = notes_match.group(1).strip() if notes_match else "Not available"
                         
                         response = f"Reason: {reason}\nAlternative: {alternative}\nNotes: {notes}"
                         return response