mxr612/fastdoi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix variable name problem, upd stringize

Browse Source
This commit is contained in:
mxr612 2024-08-21 14:46:12 +08:00
parent b4aa5b1147
commit 1345f253a0
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
fastgpt_uploader.py
View File

@ -24,13 +24,13 @@ def __new_set():
).json()
return result["data"]
__CONFIG__['colId'] = __new_set()
__CONFIG__['fastgpt_colId'] = __new_set()
def upload2fastgpt(data):
try:
headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
payload = {
"collectionId": __CONFIG__['colId'],
"collectionId": __CONFIG__['fastgpt_colId'],
"trainingMode": "chunk",
"data": data
}

2
main.py
View File

@ -11,7 +11,7 @@ def query(query:str):
res = []
list = search_paper(query)
for i in list:
res.append({'id':'','datasetId':__CONFIG__['setId'], 'collectionId':__CONFIG__['colId'], 'sourceName':'Semantic Scholar','sourceId?':'', 'q':i['title'], 'a':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'])})
print(i['title'])
return res