Update query.py

query.py

@@ -13,6 +13,7 @@ class VectaraQuery():
         self.customer_id = customer_id
         self.corpus_ids = corpus_ids
         self.api_key = api_key
+        self.conv_id = None
 
     def submit_query(self, query_str: str):
         corpora_key_list = [{
@@ -38,15 +39,19 @@ class VectaraQuery():
                     'numResults': 7,
                     'corpusKey': corpora_key_list,
                     'context_config': {
-                        'sentences_before': 3,
-                        'sentences_after': 3,
+                        'sentences_before': 2,
+                        'sentences_after': 2,
                         'start_tag': start_tag,
                         'end_tag': end_tag,
                     },
                     'summary': [
                         {
                             'responseLang': 'eng',
-                            'maxSummarizedResults': 7,
+                            'maxSummarizedResults': 5,
+                            'chat': {
+                                'store': True,
+                                'conversationId': self.conv_id
+                            }
                         }
                     ]
                 }
@@ -63,6 +68,7 @@ class VectaraQuery():
         summary = res['responseSet'][0]['summary'][0]['text']
         responses = res['responseSet'][0]['response']
         docs = res['responseSet'][0]['document']
+        self.conv_id = summary[0]['chat']['conversationId']
         pattern = r'\[\d{1,2}\]'
         matches = [match.span() for match in re.finditer(pattern, summary)]
 
@@ -89,4 +95,5 @@ class VectaraQuery():
             url = f"{metadata['url']}#:~:text={quote(text)}"
             citation_inx = refs_dict[url]
             summary = summary[:start] + f'[\[{citation_inx}\]]({url})' + summary[end:]
+
         return summary