refactor: rename 中间件

This commit is contained in:
Wenjie Zhang 2026-06-02 19:20:15 +08:00
parent 01d21e2759
commit d4bc51714c
16 changed files with 18 additions and 18 deletions

View File

@ -9,9 +9,9 @@ from yuxi.agents.middlewares import (
create_summary_middleware, create_summary_middleware,
save_attachments_to_fs, save_attachments_to_fs,
) )
from yuxi.agents.middlewares.knowledge_base_middleware import KnowledgeBaseMiddleware from yuxi.agents.middlewares.knowledge_base import KnowledgeBaseMiddleware
from yuxi.agents.middlewares.skills_middleware import SkillsMiddleware from yuxi.agents.middlewares.skills import SkillsMiddleware
from yuxi.agents.middlewares.subagent_task_middleware import create_subagent_task_middleware from yuxi.agents.middlewares.subagent_task import create_subagent_task_middleware
from yuxi.agents.toolkits.service import resolve_configured_runtime_tools from yuxi.agents.toolkits.service import resolve_configured_runtime_tools
from .context import ChatBotContext from .context import ChatBotContext

View File

@ -11,8 +11,8 @@ from yuxi.agents.buildin.chatbot.prompt import TODO_MID_PROMPT, build_prompt_wit
from yuxi.agents.buildin.subagent.context import SubAgentContext from yuxi.agents.buildin.subagent.context import SubAgentContext
from yuxi.agents.context import prepare_agent_runtime_context from yuxi.agents.context import prepare_agent_runtime_context
from yuxi.agents.middlewares import create_summary_middleware, save_attachments_to_fs from yuxi.agents.middlewares import create_summary_middleware, save_attachments_to_fs
from yuxi.agents.middlewares.knowledge_base_middleware import KnowledgeBaseMiddleware from yuxi.agents.middlewares.knowledge_base import KnowledgeBaseMiddleware
from yuxi.agents.middlewares.skills_middleware import SkillsMiddleware from yuxi.agents.middlewares.skills import SkillsMiddleware
from yuxi.agents.toolkits.service import resolve_configured_runtime_tools from yuxi.agents.toolkits.service import resolve_configured_runtime_tools

View File

@ -401,7 +401,7 @@ async def prepare_agent_runtime_context(
return context return context
from yuxi.agents.backends.knowledge_base_backend import resolve_visible_knowledge_bases_for_context from yuxi.agents.backends.knowledge_base_backend import resolve_visible_knowledge_bases_for_context
from yuxi.agents.middlewares.skills_middleware import resolve_runtime_skills_for_context from yuxi.agents.middlewares.skills import resolve_runtime_skills_for_context
from yuxi.repositories.user_repository import UserRepository from yuxi.repositories.user_repository import UserRepository
from yuxi.storage.postgres.manager import pg_manager from yuxi.storage.postgres.manager import pg_manager

View File

@ -1,7 +1,7 @@
from .attachment_middleware import inject_attachment_context, save_attachments_to_fs from .attachment import inject_attachment_context, save_attachments_to_fs
from .context_middlewares import context_aware_prompt, context_based_model from .context import context_aware_prompt, context_based_model
from .dynamic_tool_middleware import DynamicToolMiddleware from .dynamic_tool import DynamicToolMiddleware
from .summary_middleware import create_summary_middleware from .summary import create_summary_middleware
__all__ = [ __all__ = [
"DynamicToolMiddleware", "DynamicToolMiddleware",

View File

@ -246,7 +246,7 @@ async def test_prepare_agent_runtime_context_filters_resources_and_derives_runti
) )
monkeypatch.setitem( monkeypatch.setitem(
sys.modules, sys.modules,
"yuxi.agents.middlewares.skills_middleware", "yuxi.agents.middlewares.skills",
types.SimpleNamespace(resolve_runtime_skills_for_context=fake_resolve_runtime_skills_for_context), types.SimpleNamespace(resolve_runtime_skills_for_context=fake_resolve_runtime_skills_for_context),
) )
monkeypatch.setitem( monkeypatch.setitem(
@ -331,7 +331,7 @@ async def test_prepare_agent_runtime_context_clears_resources_for_missing_user(m
) )
monkeypatch.setitem( monkeypatch.setitem(
sys.modules, sys.modules,
"yuxi.agents.middlewares.skills_middleware", "yuxi.agents.middlewares.skills",
types.SimpleNamespace(resolve_runtime_skills_for_context=lambda _context, db=None, user=None: None), types.SimpleNamespace(resolve_runtime_skills_for_context=lambda _context, db=None, user=None: None),
) )
monkeypatch.setitem( monkeypatch.setitem(

View File

@ -14,7 +14,7 @@ from yuxi.agents.backends.composite import (
) )
from yuxi.agents.backends.sandbox import resolve_virtual_path, sandbox_id_for_thread from yuxi.agents.backends.sandbox import resolve_virtual_path, sandbox_id_for_thread
from yuxi.agents.backends.sandbox.backend import ProvisionerSandboxBackend from yuxi.agents.backends.sandbox.backend import ProvisionerSandboxBackend
from yuxi.agents.middlewares.skills_middleware import SkillsMiddleware from yuxi.agents.middlewares.skills import SkillsMiddleware
from yuxi.utils.paths import VIRTUAL_PATH_CONVERSATION_HISTORY, VIRTUAL_PATH_LARGE_TOOL_RESULTS from yuxi.utils.paths import VIRTUAL_PATH_CONVERSATION_HISTORY, VIRTUAL_PATH_LARGE_TOOL_RESULTS

View File

@ -6,8 +6,8 @@ import pytest
from langchain_core.messages import SystemMessage, ToolMessage from langchain_core.messages import SystemMessage, ToolMessage
from langgraph.types import Command from langgraph.types import Command
import yuxi.agents.middlewares.skills_middleware as skills_middleware import yuxi.agents.middlewares.skills as skills_middleware
from yuxi.agents.middlewares.skills_middleware import SkillsMiddleware, resolve_runtime_skills_for_context from yuxi.agents.middlewares.skills import SkillsMiddleware, resolve_runtime_skills_for_context
def _system_message_text(message: SystemMessage) -> str: def _system_message_text(message: SystemMessage) -> str:

View File

@ -3,12 +3,12 @@ from __future__ import annotations
from types import SimpleNamespace from types import SimpleNamespace
import pytest import pytest
import yuxi.agents.middlewares.subagent_task_middleware as subagent_task_middleware import yuxi.agents.middlewares.subagent_task as subagent_task_middleware
from langchain_core.messages import AIMessage, HumanMessage from langchain_core.messages import AIMessage, HumanMessage
from langgraph.prebuilt.tool_node import ToolRuntime from langgraph.prebuilt.tool_node import ToolRuntime
from langgraph.types import Command from langgraph.types import Command
from yuxi.agents.buildin.chatbot.state import merge_subagent_runs from yuxi.agents.buildin.chatbot.state import merge_subagent_runs
from yuxi.agents.middlewares.subagent_task_middleware import YuxiSubAgentMiddleware from yuxi.agents.middlewares.subagent_task import YuxiSubAgentMiddleware
from yuxi.utils.subagent_thread_utils import make_child_thread_id from yuxi.utils.subagent_thread_utils import make_child_thread_id
from yuxi.repositories.agent_repository import SUB_AGENT_BACKEND_ID from yuxi.repositories.agent_repository import SUB_AGENT_BACKEND_ID

View File

@ -6,7 +6,7 @@ import pytest
from deepagents.middleware.summarization import SummarizationMiddleware from deepagents.middleware.summarization import SummarizationMiddleware
from yuxi.agents.backends.composite import create_agent_composite_backend from yuxi.agents.backends.composite import create_agent_composite_backend
from yuxi.agents.middlewares.summary_middleware import create_summary_middleware from yuxi.agents.middlewares.summary import create_summary_middleware
from yuxi.utils.paths import VIRTUAL_PATH_CONVERSATION_HISTORY, VIRTUAL_PATH_LARGE_TOOL_RESULTS from yuxi.utils.paths import VIRTUAL_PATH_CONVERSATION_HISTORY, VIRTUAL_PATH_LARGE_TOOL_RESULTS