diff --git a/main.py b/main.py index 5f61837..48dc46a 100644 --- a/main.py +++ b/main.py @@ -15,7 +15,7 @@ def query(query:str): if not mysql_connector.is_loaded(i['paperId']): res.append({ 'id':i['paperId'], - 'q':str(i['title']), + 'q':str(i['citationStyles']['bibtex']), 'a':str(i['abstract']), 'score':[] }) diff --git a/semanticscholar.py b/semanticscholar.py index 4692e61..96ccc35 100644 --- a/semanticscholar.py +++ b/semanticscholar.py @@ -20,5 +20,5 @@ def search_paper(question): for i in question: query += "+" + i res = requests.get(headers={'x-api-key':__CONFIG__['semAPI']}, - url= 'https://api.semanticscholar.org/graph/v1/paper/search?fieldsOfStudy=Psychology&fields=title,abstract&query='+query[+1:]).json() + url= 'https://api.semanticscholar.org/graph/v1/paper/search?fieldsOfStudy=Psychology&fields=abstract,citationStyles&query='+query[+1:]).json() return res['data']