chore: Optimize the log display.

qanything_kernel/connector/rerank/rerank_backend.py

@@ -86,10 +86,7 @@ def tokenize_preproc(self,
         return merge_inputs, merge_inputs_idxs
 
     @get_time
-    def predict(self,
-                query: str,
-                passages: List[str],
-                ):
+    def get_rerank(self, query: str, passages: List[str]):
         tot_batches, merge_inputs_idxs_sort = self.tokenize_preproc(query, passages)
 
         tot_scores = []

qanything_kernel/core/local_doc_qa.py

@@ -224,7 +224,7 @@ def rerank_documents(self, query, source_documents):
         if num_tokens(query) > 300:  # tokens数量超过300时不使用local rerank
             return source_documents
 
-        scores = self.local_rerank_backend.predict(query, [doc.page_content for doc in source_documents])
+        scores = self.local_rerank_backend.get_rerank(query, [doc.page_content for doc in source_documents])
         debug_logger.info(f"rerank scores: {scores}")
         for idx, score in enumerate(scores):
                 source_documents[idx].metadata['score'] = score

qanything_kernel/qanything_server/sanic_api.py

@@ -203,6 +203,7 @@ async def init_local_doc_qa(app, loop):
 @app.after_server_start
 async def print_info(app, loop):
     print("已启动后端服务，请复制[  http://0.0.0.0:8777/qanything/  ]到浏览器进行测试。", flush=True)
+    print("详细调试日志请查看 logs/debug_logs/debug.log，问答日志请查看logs/qa_logs/qa.log", flush=True)
     # os.system("tail -f logs/debug_logs/debug.log")
 
 app.add_route(document, "/api/docs", methods=['GET'])

qanything_kernel/utils/general_utils.py

@@ -72,7 +72,7 @@ def inner(*arg, **kwargs):
         s_time = time.time()
         res = func(*arg, **kwargs)
         e_time = time.time()
-        print('函数 {} 执行耗时: {} 秒'.format(func.__name__, e_time - s_time))
+        debug_logger.info('函数 {} 执行耗时: {} 秒'.format(func.__name__, e_time - s_time))
         return res
 
     return inner