Function generate_tsne_visualization

astrbot/dashboard/utils.py:14–167

生成 t-SNE 可视化图片 Args: query: 查询文本 kb_names: 知识库名称列表 kb_manager: 知识库管理器 Returns: 图片路径或 None

(
    query: str,
    kb_names: list[str],
    kb_manager: KnowledgeBaseManager,
)

async def generate_tsne_visualization(
    query: str,
    kb_names: list[str],
    kb_manager: KnowledgeBaseManager,
) -> str | None:
    """生成 t-SNE 可视化图片

    Args:
        query: 查询文本
        kb_names: 知识库名称列表
        kb_manager: 知识库管理器

    Returns:
        图片路径或 None

    """
    try:
        import faiss
        import matplotlib  # type: ignore[reportMissingImports]
        import numpy as np

        matplotlib.use("Agg")  # 使用非交互式后端
        import matplotlib.pyplot as plt  # type: ignore[reportMissingImports]
        from sklearn.manifold import TSNE  # type: ignore[reportMissingImports]
    except ImportError as e:
        raise Exception(
            "缺少必要的库以生成 t-SNE 可视化。请安装 matplotlib 和 scikit-learn: {e}",
        ) from e

    try:
        # 获取第一个知识库的向量数据
        kb_helper: KBHelper | None = None
        for kb_name in kb_names:
            kb_helper = await kb_manager.get_kb_by_name(kb_name)
            if kb_helper:
                break

        if not kb_helper:
            logger.warning("未找到知识库")
            return None

        kb = kb_helper.kb
        index_path = kb_helper.kb_dir / "index.faiss"

        # 读取 FAISS 索引
        if not index_path.exists():
            logger.warning(f"FAISS 索引不存在: {index_path!s}")
            return None

        index = faiss.read_index(str(index_path))

        if index.ntotal == 0:
            logger.warning("索引为空")
            return None

        # 提取所有向量
        logger.info(f"提取 {index.ntotal} 个向量用于可视化...")
        if isinstance(index, faiss.IndexIDMap):