参数配置统一

This commit is contained in:
2026-04-21 19:06:34 +08:00
parent e2eaac9498
commit 37e86f3bb1
10 changed files with 120 additions and 166 deletions

View File

@@ -1,27 +1,30 @@
# rag_core/client.py
import os
from .config import QDRANT_URL, QDRANT_API_KEY
from typing import Optional
from qdrant_client import QdrantClient
def create_qdrant_client(timeout: int = 300) -> QdrantClient:
"""
创建并返回一个配置好的 Qdrant 客户端。
def create_qdrant_client(
url: Optional[str] = None,
api_key: Optional[str] = None,
timeout: int = 300, # 索引构建需要较长超时
) -> QdrantClient:
effective_url = url or QDRANT_URL
effective_api_key = api_key or QDRANT_API_KEY
Args:
timeout: 请求超时时间(秒),默认 300 秒(索引构建需要较长超时)。
if not effective_url:
Returns:
配置好的 QdrantClient 实例。
Raises:
ValueError: 如果 QDRANT_URL 未配置。
"""
if not QDRANT_URL:
raise ValueError("Qdrant URL 未配置")
client_kwargs = {
"url": effective_url,
"url": QDRANT_URL,
"timeout": timeout,
}
if effective_api_key:
client_kwargs["api_key"] = effective_api_key
if QDRANT_API_KEY:
client_kwargs["api_key"] = QDRANT_API_KEY
return QdrantClient(**client_kwargs)
return QdrantClient(**client_kwargs)