refactor: rename 中间件

backend/package/yuxi/agents/buildin/chatbot/graph.py

@@ -9,9 +9,9 @@ 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.skills_middleware import SkillsMiddleware
-from yuxi.agents.middlewares.subagent_task_middleware import create_subagent_task_middleware
+from yuxi.agents.middlewares.knowledge_base import KnowledgeBaseMiddleware
+from yuxi.agents.middlewares.skills import SkillsMiddleware
+from yuxi.agents.middlewares.subagent_task import create_subagent_task_middleware
 from yuxi.agents.toolkits.service import resolve_configured_runtime_tools
 
 from .context import ChatBotContext

backend/package/yuxi/agents/buildin/subagent/graph.py

@@ -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.context import prepare_agent_runtime_context
 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.skills_middleware import SkillsMiddleware
+from yuxi.agents.middlewares.knowledge_base import KnowledgeBaseMiddleware
+from yuxi.agents.middlewares.skills import SkillsMiddleware
 from yuxi.agents.toolkits.service import resolve_configured_runtime_tools
 
 

backend/package/yuxi/agents/context.py

@@ -401,7 +401,7 @@ async def prepare_agent_runtime_context(
         return 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.storage.postgres.manager import pg_manager
 

backend/package/yuxi/agents/middlewares/__init__.py

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

backend/test/unit/agents/test_context_auth.py

@@ -246,7 +246,7 @@ async def test_prepare_agent_runtime_context_filters_resources_and_derives_runti
     )
     monkeypatch.setitem(
         sys.modules,
-        "yuxi.agents.middlewares.skills_middleware",
+        "yuxi.agents.middlewares.skills",
         types.SimpleNamespace(resolve_runtime_skills_for_context=fake_resolve_runtime_skills_for_context),
     )
     monkeypatch.setitem(
@@ -331,7 +331,7 @@ async def test_prepare_agent_runtime_context_clears_resources_for_missing_user(m
     )
     monkeypatch.setitem(
         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),
     )
     monkeypatch.setitem(

backend/test/unit/backends/test_sandbox_backends.py

@@ -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.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
 
 

backend/test/unit/middlewares/test_skills_middleware.py

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

backend/test/unit/middlewares/test_subagent_task_middleware.py

@@ -3,12 +3,12 @@ from __future__ import annotations
 from types import SimpleNamespace
 
 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 langgraph.prebuilt.tool_node import ToolRuntime
 from langgraph.types import Command
 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.repositories.agent_repository import SUB_AGENT_BACKEND_ID
 

backend/test/unit/middlewares/test_summary_middleware.py

@@ -6,7 +6,7 @@ import pytest
 from deepagents.middleware.summarization import SummarizationMiddleware
 
 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