Update query.py

query.py

@@ -9,10 +9,11 @@ def extract_between_tags(text, start_tag, end_tag):
     return text[start_index+len(start_tag):end_index-len(end_tag)]
 
 class VectaraQuery():
-    def __init__(self, api_key: str, customer_id: str, corpus_ids: list[str]):
+    def __init__(self, api_key: str, customer_id: str, corpus_ids: list[str], prompt_name: str = None):
         self.customer_id = customer_id
         self.corpus_ids = corpus_ids
         self.api_key = api_key
+        self.prompt_name = prompt_name if prompt_name else "vectara-summary-ext-v1.2.0"
         self.conv_id = None
 
     def submit_query(self, query_str: str):
@@ -55,12 +56,12 @@ class VectaraQuery():
                         {
                             'responseLang': 'eng',
                             'maxSummarizedResults': 5,
-                            'summarizerPromptName': 'vectara-experimental-summary-ext-2023-12-11-large',
+                            'summarizerPromptName': self.prompt_name,
                             'chat': {
                                 'store': True,
                                 'conversationId': self.conv_id
                             },
-                            'debug': True,
+#                            'debug': True,
                         }
                     ]
                 }