book_rag_server/utils/ragflow_sdk/base_util.py

from ragflow_sdk import RAGFlow
from conf.settings import ragflow_settings


class RAGFlowBaseUtil:
    """
    RAGFlow 工具类基础类，封装了 RAGFlow 客户端初始化逻辑
    """
    
    def __init__(self):
        """
        初始化 RAGFlow 客户端
        """
        self.ragflow_client = RAGFlow(
            api_key=ragflow_settings.ragflow_api_key,
            base_url=ragflow_settings.ragflow_api_url
        )