参数配置统一
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user