fix(credential): correct typo in access key secret parameter name

fix(agui_protocol): optimize import statements and event handling logic

- Move imports to local scope to avoid unnecessary dependencies
- Simplify conditional expressions and method calls
- Improve type hinting for better code clarity

修复凭据模型中的访问密钥参数拼写错误

优化 AGUI 协议处理器的导入语句和事件处理逻辑：
- 将部分导入移至函数内部以减少不必要的依赖
- 简化条件表达式与方法调用
- 改进类型提示以增强代码可读性

Change-Id: I1f4b89fa0a0ceeed09bcf1e6bb03370dfd1547de
Signed-off-by: OhYee <oyohyee@oyohyee.com>

Author: OhYee

agentrun/credential/model.py

@@ -134,7 +134,7 @@ def outbound_tool_ak_sk(
         cls,
         provider: str,
         access_key_id: str,
-        access_key_secred: str,
+        access_key_secret: str,
         account_id: str,
     ):
         """配置访问第三方工具的 ak/sk 凭证"""
@@ -148,7 +148,7 @@ def outbound_tool_ak_sk(
                     "accountId": account_id,
                 },
             },
-            credential_secret=access_key_secred,
+            credential_secret=access_key_secret,
         )
 
     @classmethod

agentrun/server/agui_protocol.py

@@ -19,36 +19,12 @@
 )
 import uuid
 
-from ag_ui.core import AssistantMessage
-from ag_ui.core import CustomEvent as AguiCustomEvent
-from ag_ui.core import EventType as AguiEventType
-from ag_ui.core import Message as AguiMessage
-from ag_ui.core import MessagesSnapshotEvent
-from ag_ui.core import RawEvent as AguiRawEvent
-from ag_ui.core import (
-    RunErrorEvent,
-    RunFinishedEvent,
-    RunStartedEvent,
-    StateDeltaEvent,
-    StateSnapshotEvent,
-    StepFinishedEvent,
-    StepStartedEvent,
-    SystemMessage,
-    TextMessageContentEvent,
-    TextMessageEndEvent,
-    TextMessageStartEvent,
-)
-from ag_ui.core import Tool as AguiTool
-from ag_ui.core import ToolCall as AguiToolCall
-from ag_ui.core import (
-    ToolCallArgsEvent,
-    ToolCallEndEvent,
-    ToolCallResultEvent,
-    ToolCallStartEvent,
-)
-from ag_ui.core import ToolMessage as AguiToolMessage
-from ag_ui.core import UserMessage
-from ag_ui.encoder import EventEncoder
+if TYPE_CHECKING:
+    from ag_ui.core import (
+        Message as AguiMessage,
+    )
+    from ag_ui.encoder import EventEncoder
+
 from fastapi import APIRouter, Request
 from fastapi.responses import StreamingResponse
 import pydash
@@ -101,16 +77,20 @@ class StreamStateMachine:
     run_errored: bool = False
 
     def end_all_tools(
-        self, encoder: EventEncoder, exclude: Optional[str] = None
+        self, encoder: "EventEncoder", exclude: Optional[str] = None
     ) -> Iterator[str]:
+        from ag_ui.core import ToolCallEndEvent
+
         for tool_id, state in self.tool_call_states.items():
             if exclude and tool_id == exclude:
                 continue
             if state.started and not state.ended:
                 yield encoder.encode(ToolCallEndEvent(tool_call_id=tool_id))
                 state.ended = True
 
-    def ensure_text_started(self, encoder: EventEncoder) -> Iterator[str]:
+    def ensure_text_started(self, encoder: "EventEncoder") -> Iterator[str]:
+        from ag_ui.core import TextMessageStartEvent
+
         if not self.text.started or self.text.ended:
             if self.text.ended:
                 self.text = TextState()
@@ -123,7 +103,9 @@ def ensure_text_started(self, encoder: EventEncoder) -> Iterator[str]:
             self.text.started = True
             self.text.ended = False
 
-    def end_text_if_open(self, encoder: EventEncoder) -> Iterator[str]:
+    def end_text_if_open(self, encoder: "EventEncoder") -> Iterator[str]:
+        from ag_ui.core import TextMessageEndEvent
+
         if self.text.started and not self.text.ended:
             yield encoder.encode(
                 TextMessageEndEvent(message_id=self.text.message_id)
@@ -168,6 +150,8 @@ class AGUIProtocolHandler(BaseProtocolHandler):
     name = "ag-ui"
 
     def __init__(self, config: Optional[ServerConfig] = None):
+        from ag_ui.encoder import EventEncoder
+
         self._config = config.agui if config else None
         self._encoder = EventEncoder()
 
@@ -420,6 +404,18 @@ def _process_event_with_boundaries(
         """处理事件并注入边界事件"""
         import json
 
+        from ag_ui.core import CustomEvent as AguiCustomEvent
+        from ag_ui.core import (
+            RunErrorEvent,
+            StateDeltaEvent,
+            StateSnapshotEvent,
+            TextMessageContentEvent,
+            ToolCallArgsEvent,
+            ToolCallEndEvent,
+            ToolCallResultEvent,
+            ToolCallStartEvent,
+        )
+
         # RAW 事件直接透传
         if event.event == EventType.RAW:
             raw_data = event.data.get("raw", "")
@@ -703,7 +699,7 @@ def _process_event_with_boundaries(
 
     def _convert_messages_for_snapshot(
         self, messages: List[Dict[str, Any]]
-    ) -> List[AguiMessage]:
+    ) -> List["AguiMessage"]:
         """将消息列表转换为 ag-ui-protocol 格式
 
         Args:
@@ -712,6 +708,10 @@ def _convert_messages_for_snapshot(
         Returns:
             ag-ui-protocol 消息列表
         """
+        from ag_ui.core import AssistantMessage, SystemMessage
+        from ag_ui.core import ToolMessage as AguiToolMessage
+        from ag_ui.core import UserMessage
+
         result = []
         for msg in messages:
             if not isinstance(msg, dict):
@@ -779,6 +779,8 @@ async def _error_stream(self, message: str) -> AsyncIterator[str]:
         Yields:
             SSE 格式的错误事件
         """
+        from ag_ui.core import RunErrorEvent, RunStartedEvent
+
         thread_id = str(uuid.uuid4())
         run_id = str(uuid.uuid4())