From 8779907a685220a19d202d374002fabdf0b53f80 Mon Sep 17 00:00:00 2001 From: mxr612 Date: Thu, 22 Aug 2024 19:34:46 +0800 Subject: [PATCH] improve random limit for paper --- semanticscholar.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/semanticscholar.py b/semanticscholar.py index 96ccc35..720d2ee 100644 --- a/semanticscholar.py +++ b/semanticscholar.py @@ -1,5 +1,6 @@ import requests -import json +import random +import datetime from config import __CONFIG__ @@ -20,5 +21,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=abstract,citationStyles&query='+query[+1:]).json() + url= 'https://api.semanticscholar.org/graph/v1/paper/search?fieldsOfStudy=Psychology&fields=abstract,citationStyles&minCitationCount='+str(random.randint(10, 500))+'&year='+str(random.randint(1900,datetime.datetime.now().year))+'-&query='+query[+1:]).json() return res['data']