diff --git a/main.py b/main.py index 55c1ee0..0070d67 100644 --- a/main.py +++ b/main.py @@ -13,14 +13,14 @@ def query(query:str): list = search_paper(query) for i in list: if not mysql_connector.is_loaded(i['paperId']): - print('New: '+i['q']) + print(i['citationStyles']['bibtex']) res.append({ 'id':i['paperId'], 'q':str(i['citationStyles']['bibtex']), 'a':str(i['abstract']), 'score':[] }) - print('New: '+len(res)+'/'+len(list)) + print('New load: '+str(len(res))+'/'+str(len(list))) return res @app.get("/fastdoi") @@ -30,7 +30,6 @@ async def get_reference(questions): if(upload2fastgpt(res)): for i in res: mysql_connector.new_load(i['id']) - return res if __name__ == '__main__':