Update query.py

query.py

@@ -36,7 +36,7 @@ class VectaraQuery():
                 { 
                     'query': query_str,
                     'start': 0,
-                    'numResults': 10,
+                    'numResults': 50,
                     'corpusKey': corpora_key_list,
                     'context_config': {
                         'sentences_before': 2,
@@ -44,6 +44,13 @@ class VectaraQuery():
                         'start_tag': start_tag,
                         'end_tag': end_tag,
                     },
+                    'rerankingConfig':
+                    {
+                        'rerankerId': '272725718',
+                        'mmrConfig': {
+                            'diversityBias': 0.2
+                        }
+                    }
                     'summary': [
                         {
                             'responseLang': 'eng',
@@ -68,13 +75,12 @@ class VectaraQuery():
         res = response.json()
         print(f"DEBUG request body = {body}")
         print(f"DEBUG response summary = {res['responseSet'][0]['summary']}")
-        
+
+        top_k = 10
         summary = res['responseSet'][0]['summary'][0]['text']
-        responses = res['responseSet'][0]['response']
+        responses = res['responseSet'][0]['response'][:top_k]
         docs = res['responseSet'][0]['document']
         self.conv_id = res['responseSet'][0]['summary'][0]['chat']['conversationId']
-
-        
         
         pattern = r'\[\d{1,2}\]'
         matches = [match.span() for match in re.finditer(pattern, summary)]