From e70f5fea1c01ac7bd2421403b0f80b559b9933ae Mon Sep 17 00:00:00 2001
From: mxr612 <work@mxr612.top>
Date: Thu, 22 Aug 2024 12:53:20 +0800
Subject: [PATCH] change index method

---
 main.py            | 2 +-
 semanticscholar.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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']