V2LearningIntegration: Embedding/Rerank provider 因加载时序问题无法解析

[LucNocturne]

问题描述

V2LearningIntegration.__init__() 在 AstrBot 的 Provider 注册表（provider_manager.inst_map）填充之前就调用 _create_embedding_provider() 和 _create_rerank_provider()，导致 LightRAG 知识引擎和 Reranker 始终无法初始化。

复现步骤

1. 在 V2_Architecture_Settings 中配置 embedding_provider_id 和 rerank_provider_id
2. 将 knowledge_engine 设为 lightrag
3. 重启 AstrBot

实际行为

[V2Integration] Initialised — knowledge=lightrag, memory=legacy, embedding=no, reranker=no
[V2Integration] LightRAG requires an embedding provider but none is available; knowledge engine disabled

即使配置正确，embedding 和 rerank provider 始终解析为 None。

根因

AstrBot 启动时序：

1. 插件加载阶段（star_manager 载入插件）→ 此时 provider_manager.inst_map 为空
2. V2LearningIntegration.__init__() → 调用 EmbeddingProviderFactory.create() → context.get_provider_by_id() → 返回 None
3. V2LearningIntegration.start() → 同样时机过早，inst_map 仍为空
4. Provider 加载阶段（provider.manager 实例化各 provider）→ inst_map 才被填充
5. AstrBot 启动完成

FrameworkLLMAdapter 有类似的时序问题，但它有 _needs_lazy_init 机制在实际使用时重试。V2LearningIntegration 缺少这个机制。

临时 Workaround

在 start() 中用 asyncio.create_task() 启动后台 retry 任务：

async def start(self) -> None:
    # Schedule background retry for providers not yet available
    if (not self._embedding_provider or not self._rerank_provider) and self._context:
        asyncio.create_task(self._lazy_resolve_providers())
    # ... existing start logic ...

async def _lazy_resolve_providers(self) -> None:
    for attempt in range(30):
        await asyncio.sleep(3)
        if not self._embedding_provider:
            self._embedding_provider = self._create_embedding_provider()
        if not self._rerank_provider:
            self._rerank_provider = self._create_rerank_provider()
        if self._embedding_provider and self._rerank_provider:
            break
    if self._embedding_provider:
        if not self._knowledge_manager:
            self._knowledge_manager = self._create_knowledge_manager()
            if self._knowledge_manager and hasattr(self._knowledge_manager, 'start'):
                await self._knowledge_manager.start()
        if not self._exemplar_library:
            self._exemplar_library = self._create_exemplar_library()

建议修复

1. 在 V2LearningIntegration 中加入类似 FrameworkLLMAdapter._needs_lazy_init 的延迟初始化机制
2. 或者利用 AstrBot 的 on_astrbot_loaded hook，在所有 provider 就绪后再初始化 V2 模块

环境

• AstrBot: v4.19.4
• self-learning: Next-2.0.6
• Python: 3.12

[github-actions]

🤖 Issue 审查报告

优先级: 🟠 high

📋 用户处理意见

问题摘要: 即使正确配置了 embedding_provider_id / rerank_provider_id，在 AstrBot 启动阶段 provider 实例尚未注册到 provider_manager.inst_map 之前，V2LearningIntegration 就尝试解析 provider，导致 LightRAG/Reranker 初始化恒为 None。

问题分类: 代码问题

该问题需要开发者修复，请等待后续版本更新。

---

🔧 开发者修复建议（点击展开）

技术摘要: V2LearningIntegration 在插件加载/启动阶段过早调用 EmbeddingProviderFactory/RerankProviderFactory 解析 provider，因 AstrBot provider 注册时序导致 context.get_provider_by_id() 返回 None，最终禁用 LightRAG 知识引擎。

相关模块: core_learning, embedding, reranker, integration, plugin_lifecycle

相关文件: services/core_learning/v2_learning_integration.py, services/embedding/factory.py, services/embedding/framework_adapter.py, services/reranker/factory.py, services/reranker/framework_adapter.py, core/framework_llm_adapter.py, core/plugin_lifecycle.py, services/integration/lightrag_knowledge_manager.py, services/integration/maibot_enhanced_learning_manager.py, config.py

根因分析: services/core_learning/v2_learning_integration.py 中 V2LearningIntegration.init()（以及 start()）调用 _create_embedding_provider()/_create_rerank_provider() 时，底层 services/embedding/factory.py 与 services/reranker/factory.py 通过 context.get_provider_by_id(provider_id) 取 provider；但 AstrBot 在插件加载阶段 provider_manager.inst_map 尚未填充（provider 实例化发生在更后阶段），因此始终解析为 None，后续 _create_knowledge_manager() 会因缺 embedding 而禁用 lightrag（日志与 core/plugin_lifecycle.py 里 V2 模式启动路径一致）。

修复建议: 在 services/core_learning/v2_learning_integration.py 引入“延迟解析 + 在 provider 就绪后再创建依赖对象”的机制，避免在 init/start 阶段做硬解析：

1. 参照 core/framework_llm_adapter.py 的 lazy init 思路，为 V2LearningIntegration 增加类似 self._needs_lazy_init 标志与 _try_resolve_providers() 方法；当 provider 未就绪时仅记录 provider_id，不创建 _knowledge_manager/_exemplar_library。
2. 在实际首次使用 V2 能力的入口处（例如 get_enhanced_context / 任何会触发知识检索/重排的路径；或在 start() 内启动短周期重试任务）调用 _try_resolve_providers()：
   • 若 embedding/rerank provider 解析成功，再调用 _create_knowledge_manager() 并 await knowledge_manager.start()。
   • 若仍失败，保持降级到 legacy，但保留重试机会。
3. 更推荐的架构修复：在 core/plugin_lifecycle.py 或 services/plugin_lifecycle.py（当前为 core/plugin_lifecycle.py）增加/使用 AstrBot “on_astrbot_loaded / providers_ready” 类 hook（如果 AstrBot 框架有对应事件）来触发 V2LearningIntegration 的 provider 解析与 knowledge_manager 启动；确保发生在 provider.manager 完成实例化之后。
4. 同时补充单测：模拟 context.get_provider_by_id 初始返回 None，随后可返回 provider 实例，断言 V2LearningIntegration 能在后续调用时完成初始化并启用 lightrag。

---

_{🤖 此报告由 AI 自动生成，已分析仓库源码，仅供参考。开发者会尽快审阅您的 issue。}

[NickCharlie]

收到，后面有空会计划修复