try rrf

2024-08-21 15:10:51 +08:00 · 2024-08-21 15:10:51 +08:00 · bb3e175968
commit bb3e175968
parent 1345f253a0
1 changed files with 14 additions and 1 deletions
--- a/main.py
+++ b/main.py
@ -11,7 +11,20 @@ def query(query:str):
    res = []
    list = search_paper(query)
    for i in list:
-        res.append({'id':'','datasetId':__CONFIG__['fastgpt_setId'], 'collectionId':__CONFIG__['fastgpt_colId'], 'sourceName':'Semantic Scholar','sourceId?':'', 'q':str(i['title']), 'a':str(i['abstract'])})
+        res.append({
+            'id':'','datasetId':__CONFIG__['fastgpt_setId'], 
+            'collectionId':__CONFIG__['fastgpt_colId'], 
+            'sourceName':'Semantic Scholar',
+            'sourceId?':'',
+             'q':str(i['title']), 
+             'a':str(i['abstract']),
+             'score':[
+                 {
+                     'type': 'rrf'
+                     'value':0.8
+                 }
+             ]
+             })
        print(i['title'])
    return res