From 2b7bb0201c8b29a640a84680f0d495f468abd9c9 Mon Sep 17 00:00:00 2001
From: mxr612 <work@mxr612.top>
Date: Thu, 22 Aug 2024 16:22:45 +0800
Subject: [PATCH] updated output

---
 main.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 48dc46a..55c1ee0 100644
--- a/main.py
+++ b/main.py
@@ -13,12 +13,14 @@ def query(query:str):
     list = search_paper(query)
     for i in list:
         if not mysql_connector.is_loaded(i['paperId']):
+            print('New: '+i['q'])
             res.append({
                 'id':i['paperId'],
                 'q':str(i['citationStyles']['bibtex']), 
                 'a':str(i['abstract']),
                 'score':[]
                 })
+    print('New: '+len(res)+'/'+len(list))
     return res
 
 @app.get("/fastdoi")
@@ -28,7 +30,7 @@ async def get_reference(questions):
     if(upload2fastgpt(res)):
         for i in res:
             mysql_connector.new_load(i['id'])
-            print('New: '+i['q'])
+
     return res
 
 if __name__ == '__main__':