From 6b016481f1050c8557b1630f6199a86bd2ebcb68 Mon Sep 17 00:00:00 2001
From: mxr612 <work@mxr612.top>
Date: Wed, 21 Aug 2024 12:39:30 +0800
Subject: [PATCH] paper query for main and api supported

---
 fastgpt_uploader.py |  4 ++--
 main.py             | 29 +++++++++++++++++++++++++----
 2 files changed, 27 insertions(+), 6 deletions(-)

diff --git a/fastgpt_uploader.py b/fastgpt_uploader.py
index 2c7a43d..759a81b 100644
--- a/fastgpt_uploader.py
+++ b/fastgpt_uploader.py
@@ -26,13 +26,13 @@ def __new_set():
 
 colId = __new_set()
 
-def submit_data(q,a):
+def upload2fastgpt(data):
     try:
         headers = {"Authorization": f"Bearer {key}", "Content-Type": "application/json"}
         payload = {
             "collectionId": colId,
             "trainingMode": "chunk",
-            "data":  [{"q":q,"a":a}]
+            "data":  data
         }
         response = requests.post(
             url + "/core/dataset/data/pushData", headers=headers, data=json.dumps(payload)
diff --git a/main.py b/main.py
index 0ac1f68..759c4d9 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,28 @@
 import config
-import mysql_connector
-import fastgpt_uploader
+import json
+from fastapi import FastAPI
 
-config.end_mysql()
+# import mysql_connector
+from fastgpt_uploader import upload2fastgpt
+from semanticscholar import search_paper
 
-mysql_connector.end_mysql()
\ No newline at end of file
+app = FastAPI()
+
+def query(query:str):
+    res = []
+    list = search_paper(json.loads(query))
+    for i in list:
+        res.append({'q':i['title'], 'a':i['abstract']})
+        print(i['title'])
+    return res
+
+@app.get("/fastdoi")
+async def get_reference(questions):
+    res = query(questions)
+    upload2fastgpt(res)
+    return res
+
+if __name__ == '__main__':
+    import uvicorn
+    uvicorn.run(app, host="127.0.0.1", port=2501)
+    # mysql_connector.end_mysql()
\ No newline at end of file