diff --git a/rag2_0/dify/dify_tool.py b/rag2_0/dify/dify_tool.py
index 2d1cbaa..50d2e01 100755
--- a/rag2_0/dify/dify_tool.py
+++ b/rag2_0/dify/dify_tool.py
@@ -341,7 +341,7 @@ content: "{content}"
         except Exception as e:
             return -1
     
-    def get_retrieve_info(self, query: str, outputs: dict, reranker_sorce_info:list) -> tuple:
+    def get_retrieve_info(self, query: str, outputs: list[dict], reranker_sorce_info:list) -> tuple:
         """
         获取检索信息并计算分数
         
@@ -357,6 +357,7 @@ content: "{content}"
         total_score = 0
         valid_scores = 0
         retrieve_title = []
+        segmentid_to_title = { result["segment_id"]:result["title"].split("/")[-1] for result in outputs}
         
         # 使用线程池并发计算分数
         with ThreadPoolExecutor() as executor:
@@ -372,7 +373,7 @@ content: "{content}"
             for future in as_completed(future_to_content):
                 content, segment_id = future_to_content[future]
                 score = future.result()
-                content_title = content.split("\n")[0]
+                content_title = segmentid_to_title[segment_id]
                 
                 if score != -1:
                     max_score = max(max_score, score)