Update query.py

query.py

@@ -48,19 +48,7 @@ class VectaraQuery():
                         'sentences_after': 2,
                         'start_tag': "%START_SNIPPET%",
                         'end_tag': "%END_SNIPPET%",
-                    },
-                    'summary': [
-                        {
-                            'responseLang': 'eng',
-                            'maxSummarizedResults': 1,
-                            'summarizerPromptName': self.prompt_name,
-                            'promptText': prompt,
-                            'chat': {
-                                'store': True,
-                                'conversationId': self.conv_id
-                            },
-                        }
-                    ]
+                    }
                 } 
             ]
         }
@@ -106,60 +94,51 @@ class VectaraQuery():
                             continue
                         text = result['text']
                         print(f"Processing text: {text}")  # Debugging line
-                        accumulated_text += text
+                        accumulated_text += text + " "
 
         if accumulated_text:
-            return self.summarize_text(accumulated_text)
+            return self.format_response_using_vectara(accumulated_text)
 
         return "No relevant information found."
 
-    def summarize_text(self, text):
+    def format_response_using_vectara(self, text):
         endpoint = f"https://api.vectara.io/v1/stream-summary"
-        prompt = f'''
-        [
-            {{
-                "role": "system",
-                "content": "You are an assistant that provides information about drink names based on a given corpus. \
-                Format the response in the following way:\n\
-                Reason: <reason why the name cannot be used>\n\
-                Alternative: <alternative name>\n\
-                Notes: <additional notes>\n\n\
-                Example:\n\
-                Reason: The name 'Vodka Sunrise' cannot be used because it is trademarked.\n\
-                Alternative: Use 'Morning Delight' instead.\n\
-                Notes: Ensure the drink contains vodka to match the alternative name."
-            }},
-            {{
-                "role": "user",
-                "content": "{text}"
-            }}
-        ]
-        '''
         body = {
             'text': text,
             'summary': {
                 'responseLang': 'eng',
                 'maxSummarizedResults': 1,
                 'summarizerPromptName': self.prompt_name,
-                'promptText': prompt,
-                'chat': {
-                    'store': True,
-                    'conversationId': self.conv_id
-                },
+                'promptText': f'''
+                [
+                    {{
+                        "role": "system",
+                        "content": "You are an assistant that provides information about drink names based on a given corpus. \
+                        Format the response in the following way:\n\
+                        Reason: <reason why the name cannot be used>\n\
+                        Alternative: <alternative name>\n\
+                        Notes: <additional notes>\n\n\
+                        Example:\n\
+                        Reason: The name 'Vodka Sunrise' cannot be used because it is trademarked.\n\
+                        Alternative: Use 'Morning Delight' instead.\n\
+                        Notes: Ensure the drink contains vodka to match the alternative name."
+                    }},
+                    {{
+                        "role": "user",
+                        "content": "{text}"
+                    }}
+                ]
+                '''
             }
         }
-        response = requests.post(endpoint, data=json.dumps(body), verify=True, headers=self.get_headers(), stream=True)
+        headers = self.get_headers()
+        response = requests.post(endpoint, data=json.dumps(body), headers=headers)
         if response.status_code != 200:
             print(f"Summary query failed with code {response.status_code}, reason {response.reason}, text {response.text}")
             return "Sorry, something went wrong. Please try again later."
 
-        for line in response.iter_lines():
-            if line:  # filter out keep-alive new lines
-                data = json.loads(line.decode('utf-8'))
-                print(f"Received summary data chunk: {json.dumps(data, indent=2)}")  # Debugging line
-
-                if 'summary' in data:
-                    return data['summary']['text']
-
+        data = response.json()
+        if 'summary' in data:
+            return data['summary']['text']
+        
         return "No relevant information found."
-