ailine/backend/rag_core/client.py

# rag_core/client.py
import os
from .config import QDRANT_URL, QDRANT_API_KEY
from qdrant_client import QdrantClient


def create_qdrant_client(timeout: int = 300) -> QdrantClient:
    """
    创建并返回一个配置好的 Qdrant 客户端。

    Args:
        timeout: 请求超时时间（秒），默认 300 秒（索引构建需要较长超时）。

    Returns:
        配置好的 QdrantClient 实例。

    Raises:
        ValueError: 如果 QDRANT_URL 未配置。
    """
    if not QDRANT_URL:
        raise ValueError("Qdrant URL 未配置")

    client_kwargs = {
        "url": QDRANT_URL,
        "timeout": timeout,
    }
    if QDRANT_API_KEY:
        client_kwargs["api_key"] = QDRANT_API_KEY

    return QdrantClient(**client_kwargs)