RagFlow_Plugs/utils/infinity/__init__.py

# Infinity Python API客户端，带有连接池机制

# 从分离的模块中导入所有组件，确保向后兼容性
from .pool import InfinityConnectionPool
from .client import InfinityClient, get_client, close_client

# 重新导出所有组件，使它们可以从包级别访问
__all__ = [
    'InfinityConnectionPool',
    'InfinityClient',
    'get_client',
    'close_client'
]

# 使用示例
"""
# 示例1：基本使用
client = InfinityClient()
databases = client.get_databases()
print(f"Databases: {databases}")

# 示例2：使用连接池上下文
with client.get_connection() as conn:
    tables = conn.get_tables()
    print(f"Tables: {tables}")

# 示例3：使用全局客户端
from utils.infinity import get_client
client = get_client()
databases = client.get_databases()
print(f"Databases from global client: {databases}")

# 示例4：混合检索
result = client.hybrid_search(
    table_name="my_table",
    vector_field="vector",
    query_vector=[0.1, 0.2, 0.3],
    text_query="test",
    text_field="content",
    topn=5
)
print(f"Hybrid search result: {result}")
"""