upload_file_to_kb>response.get('data') 防御性处理

Author: Samge0

ragflows/api.py

@@ -36,7 +36,8 @@ def check_api_url() -> tuple[bool, str]:
 
     response = r.json()
     if is_succeed(response):
-        return True, "API地址配置正确"
+        timeutils.print_log(f"ragflow version：{response.get('data')}")
+        return True, f"API地址配置正确"
 
     code = response.get("code")
     message = response.get("message")
@@ -139,8 +140,8 @@ def parse_chunks_with_check(filename, doc_id=None):
 
     if not doc_id:
         timeutils.print_log(f'根据文件名[{filename}]从数据库获取文档id')
-        doc_item = ragflowdb.get_doc_item_by_name(filename, max_retries=configs.SQL_RETRIES)
-        if not doc_item:
+        doc_item = ragflowdb.get_doc_item_by_name(filename, max_retries=configs.SQL_RETRIES) or {}
+        if not doc_item.get('id'):
             timeutils.print_log(f'找不到{filename}对应的数据库记录，跳过')
             return False
 
@@ -197,8 +198,14 @@ def set_document_metadata(doc_id, filepath) -> bool:
         return False
 
     if not doc_id:
-        timeutils.print_log(F'设置文档元数据失败: doc_id为空，跳过')
-        return False
+        filename = os.path.basename(filepath)
+        timeutils.print_log(f'根据文件名[{filename}]从数据库获取文档id')
+        doc_item = ragflowdb.get_doc_item_by_name(filename, max_retries=configs.SQL_RETRIES) or {}
+        if not doc_item.get('id'):
+            timeutils.print_log(F'设置文档元数据失败: doc_id为空，跳过')
+            return False
+        
+        doc_id = doc_item.get('id')
 
     # 构建元数据文件路径-移除原文件后缀再拼接元数据后缀
     filepath_without_ext = os.path.splitext(filepath)[0]

ragflows/main.py

@@ -159,9 +159,16 @@ def main():
             timeutils.print_log(f'{file_path} 上传失败：{response.get("text")}')
             continue
 
-        # 解析doc_id
-        doc_id = response.get('data')[0].get('id') if response.get('data') else None
-        
+        # 尝试从响应内容中解析doc_id，防御性处理，有的版本api返回格式为：{'code': 0, 'data': True, 'message': 'success'}
+        data = response.get('data')
+        doc_id = None
+        if isinstance(data, list) and data and isinstance(data[0], dict):
+            doc_id = data[0].get('id')
+        elif isinstance(data, dict):
+            doc_id = data.get('id')
+        else:
+            timeutils.print_log(f"上传文件后返回数据不包含文档id: {data}")
+
         # 检查配置并更新元数据
         api.set_document_metadata(doc_id, file_path)
 

scripts/launcher.py

@@ -81,7 +81,7 @@ def __init__(self):
         self.geometry("800x750")
 
         # 版本和仓库信息
-        self.version = "v1.0.5"  # 版本号
+        self.version = "v1.0.6-alpha"  # 版本号
         self.github_repo = "https://github.com/Samge0/ragflow-upload"  # GitHub仓库地址
 
         # 自定义图标