From 85a4106e18daf9fe5a99512232439bcb07b15994 Mon Sep 17 00:00:00 2001
From: mxr612 <work@mxr612.top>
Date: Tue, 20 Aug 2024 19:07:07 +0800
Subject: [PATCH] semanticscholar api used

---
 fastgpt_uploader.py |  8 ++++----
 semanticscholar.py  | 24 ++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 4 deletions(-)
 create mode 100644 semanticscholar.py

diff --git a/fastgpt_uploader.py b/fastgpt_uploader.py
index 75a3732..2c7a43d 100644
--- a/fastgpt_uploader.py
+++ b/fastgpt_uploader.py
@@ -2,11 +2,11 @@ import json
 import requests
 from markdownify import markdownify as md
 
-import config
+from config import __CONFIG__
 
-key = config.__CONFIG__["fastgpt_api"]
-url = config.__CONFIG__["fastgpt_host"]
-setId =  config.__CONFIG__["fastgpt_setId"]
+key = __CONFIG__["fastgpt_api"]
+url = __CONFIG__["fastgpt_host"]
+setId =  __CONFIG__["fastgpt_setId"]
 
 # Config FastGPT
 
diff --git a/semanticscholar.py b/semanticscholar.py
new file mode 100644
index 0000000..20d976a
--- /dev/null
+++ b/semanticscholar.py
@@ -0,0 +1,24 @@
+import requests
+import json
+
+from config import __CONFIG__
+
+def latest_release():
+    r2 = requests.get('https://api.semanticscholar.org/datasets/v1/release/latest').json()
+    return r2['release_id']
+
+def first_pull():
+    res = requests.get('https://api.semanticscholar.org/datasets/v1/release/latest/dataset/abstract')
+    return res.json()
+
+def abstract():
+    r3 = requests.get(headers={'x-api-key':__CONFIG__['semAPI']},url=
+        'https://api.semanticscholar.org/datasets/v1/release/latest/dataset/abstracts').json()
+
+def search_paper(question):
+    query=""
+    for i in question:
+        query += "+" + i
+    res = requests.get(headers={'x-api-key':__CONFIG__['semAPI']},
+                       url= 'https://api.semanticscholar.org/graph/v1/paper/search?fields=title,abstract&query='+query[+1:]).json()
+    return res['data']