RagFlow_Plugs/test_upload_document.py

import sys
import os

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from services.utils.http_client import HTTPClient
from services.ragflow.document_service import DocumentService

# 配置信息
API_URL = "http://localhost:8000"  # 替换为实际的RAGFlow API URL
API_KEY = "your_api_key"  # 替换为实际的API密钥
DATASET_ID = "your_dataset_id"  # 替换为实际的数据集ID
PDF_PATH = r"D:\project\work\ragflow_plugs\book\不一样的卡梅拉1-我想去看海.pdf"  # 使用已有的测试PDF文件

def test_upload_document():
    """测试上传文档功能"""
    try:
        # 创建HTTP客户端实例
        http_client = HTTPClient(base_url=API_URL, api_key=API_KEY)
        
        # 创建文档服务实例
        document_service = DocumentService(http_client)
        
        # 调用上传文档方法
        print(f"开始上传文档: {PDF_PATH}")
        result = document_service.upload_document(DATASET_ID, PDF_PATH)
        
        # 打印结果
        print(f"文档上传成功: {result}")
        return True
    except Exception as e:
        print(f"文档上传失败: {str(e)}")
        import traceback
        traceback.print_exc()
        return False

if __name__ == "__main__":
    test_upload_document()