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

updated output

Browse Source
This commit is contained in:
mxr612 2024-08-22 16:22:45 +08:00
parent e70f5fea1c
commit 2b7bb0201c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -13,12 +13,14 @@ def query(query:str):
list = search_paper(query) list = search_paper(query)
for i in list: for i in list:
if not mysql_connector.is_loaded(i['paperId']): if not mysql_connector.is_loaded(i['paperId']):
print('New: '+i['q'])
res.append({ res.append({
'id':i['paperId'], 'id':i['paperId'],
'q':str(i['citationStyles']['bibtex']), 'q':str(i['citationStyles']['bibtex']),
'a':str(i['abstract']), 'a':str(i['abstract']),
'score':[] 'score':[]
}) })
print('New: '+len(res)+'/'+len(list))
return res return res
@app.get("/fastdoi") @app.get("/fastdoi")
@ -28,7 +30,7 @@ async def get_reference(questions):
if(upload2fastgpt(res)): if(upload2fastgpt(res)):
for i in res: for i in res:
mysql_connector.new_load(i['id']) mysql_connector.new_load(i['id'])
print('New: '+i['q'])
return res return res
if __name__ == '__main__': if __name__ == '__main__':