chore(core.agent): ruff rewrite

Author: Dt8333

astrbot/core/agent/mcp_client.py

@@ -1,7 +1,6 @@
 import asyncio
 import logging
 from datetime import timedelta
-from typing import Optional
 from contextlib import AsyncExitStack
 from astrbot import logger
 from astrbot.core.utils.log_pipe import LogPipe
@@ -96,7 +95,7 @@ async def _quick_test_mcp_connection(config: dict) -> tuple[bool, str]:
 class MCPClient:
     def __init__(self):
         # Initialize session and client objects
-        self.session: Optional[mcp.ClientSession] = None
+        self.session: mcp.ClientSession | None = None
         self.exit_stack = AsyncExitStack()
 
         self.name: str | None = None

astrbot/core/agent/tool.py

@@ -1,6 +1,8 @@
 from dataclasses import dataclass
 from deprecated import deprecated
-from typing import Awaitable, Callable, Literal, Any, Optional
+from typing import Literal, Any
+
+from collections.abc import Awaitable, Callable
 from .mcp_client import MCPClient
 
 
@@ -71,7 +73,7 @@ def remove_tool(self, name: str):
         """Remove a tool by its name."""
         self.tools = [tool for tool in self.tools if tool.name != name]
 
-    def get_tool(self, name: str) -> Optional[FunctionTool]:
+    def get_tool(self, name: str) -> FunctionTool | None:
         """Get a tool by its name."""
         for tool in self.tools:
             if tool.name == name:

astrbot/core/agent/tool_executor.py

@@ -1,5 +1,7 @@
 import mcp
-from typing import Any, Generic, AsyncGenerator
+from typing import Any, Generic
+
+from collections.abc import AsyncGenerator
 from .run_context import TContext, ContextWrapper
 from .tool import FunctionTool