RagFlow_Plugs/test/test_infinity_sdk.py

import sys
import os

# 添加项目根目录到Python路径
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

def test_infinity_import():
    """测试Infinity SDK导入"""
    try:
        from services.utils.infinity_util import InfinityVectorDB
        print("✓ InfinityVectorDB导入成功")
        return True
    except Exception as e:
        print(f"✗ InfinityVectorDB导入失败: {str(e)}")
        return False

def test_infinity_sdk_version():
    """测试Infinity SDK版本"""
    try:
        import infinity
        print(f"✓ Infinity SDK版本: {infinity.__version__}")
        return True
    except Exception as e:
        print(f"✗ 无法获取Infinity SDK版本: {str(e)}")
        return False

def test_infinity_api():
    """测试Infinity API可用性"""
    try:
        import infinity
        from infinity.common import ConflictType
        
        print(f"✓ infinity模块: {infinity}")
        print(f"✓ ConflictType: {ConflictType}")
        print(f"✓ NetworkAddress: {infinity.NetworkAddress}")
        print(f"✓ DataType: {infinity.DataType}")
        print(f"✓ IndexType: {infinity.IndexType}")
        print(f"✓ MetricType: {infinity.MetricType}")
        return True
    except Exception as e:
        print(f"✗ 无法访问Infinity API: {str(e)}")
        return False

def main():
    """主测试函数"""
    print("=== 测试Infinity SDK实现 ===")
    
    # 测试导入
    test_infinity_import()
    
    # 测试SDK版本
    test_infinity_sdk_version()
    
    # 测试API可用性
    test_infinity_api()
    
    print("\n=== 测试完成 ===")

if __name__ == "__main__":
    main()