Enhance chat UI and compatible anthropic thinking messages (#1018)

backend/src/agents/middlewares/title_middleware.py

@@ -43,7 +43,7 @@ class TitleMiddleware(AgentMiddleware[TitleMiddlewareState]):
         # Generate title after first complete exchange
         return len(user_messages) == 1 and len(assistant_messages) >= 1
 
-    def _generate_title(self, state: TitleMiddlewareState) -> str:
+    async def _generate_title(self, state: TitleMiddlewareState) -> str:
         """Generate a concise title based on the conversation."""
         config = get_title_config()
         messages = state.get("messages", [])
@@ -66,7 +66,7 @@ class TitleMiddleware(AgentMiddleware[TitleMiddlewareState]):
         )
 
         try:
-            response = model.invoke(prompt)
+            response = await model.ainvoke(prompt)
             # Ensure response content is string
             title_content = str(response.content) if response.content else ""
             title = title_content.strip().strip('"').strip("'")
@@ -81,10 +81,10 @@ class TitleMiddleware(AgentMiddleware[TitleMiddlewareState]):
             return user_msg if user_msg else "New Conversation"
 
     @override
-    def after_agent(self, state: TitleMiddlewareState, runtime: Runtime) -> dict | None:
+    async def aafter_model(self, state: TitleMiddlewareState, runtime: Runtime) -> dict | None:
         """Generate and set thread title after the first agent response."""
         if self._should_generate_title(state):
-            title = self._generate_title(state)
+            title = await self._generate_title(state)
             print(f"Generated thread title: {title}")
 
             # Store title in state (will be persisted by checkpointer if configured)

backend/src/sandbox/local/local_sandbox.py

@@ -179,22 +179,34 @@ class LocalSandbox(Sandbox):
 
     def read_file(self, path: str) -> str:
         resolved_path = self._resolve_path(path)
-        with open(resolved_path) as f:
-            return f.read()
+        try:
+            with open(resolved_path) as f:
+                return f.read()
+        except OSError as e:
+            # Re-raise with the original path for clearer error messages, hiding internal resolved paths
+            raise type(e)(e.errno, e.strerror, path) from None
 
     def write_file(self, path: str, content: str, append: bool = False) -> None:
         resolved_path = self._resolve_path(path)
-        dir_path = os.path.dirname(resolved_path)
-        if dir_path:
-            os.makedirs(dir_path, exist_ok=True)
-        mode = "a" if append else "w"
-        with open(resolved_path, mode) as f:
-            f.write(content)
+        try:
+            dir_path = os.path.dirname(resolved_path)
+            if dir_path:
+                os.makedirs(dir_path, exist_ok=True)
+            mode = "a" if append else "w"
+            with open(resolved_path, mode) as f:
+                f.write(content)
+        except OSError as e:
+            # Re-raise with the original path for clearer error messages, hiding internal resolved paths
+            raise type(e)(e.errno, e.strerror, path) from None
 
     def update_file(self, path: str, content: bytes) -> None:
         resolved_path = self._resolve_path(path)
-        dir_path = os.path.dirname(resolved_path)
-        if dir_path:
-            os.makedirs(dir_path, exist_ok=True)
-        with open(resolved_path, "wb") as f:
-            f.write(content)
+        try:
+            dir_path = os.path.dirname(resolved_path)
+            if dir_path:
+                os.makedirs(dir_path, exist_ok=True)
+            with open(resolved_path, "wb") as f:
+                f.write(content)
+        except OSError as e:
+            # Re-raise with the original path for clearer error messages, hiding internal resolved paths
+            raise type(e)(e.errno, e.strerror, path) from None

backend/tests/test_title_middleware_core_logic.py

@@ -1,6 +1,7 @@
 """Core behavior tests for TitleMiddleware."""
 
-from unittest.mock import MagicMock
+import asyncio
+from unittest.mock import AsyncMock, MagicMock
 
 from langchain_core.messages import AIMessage, HumanMessage
 
@@ -76,7 +77,7 @@ class TestTitleMiddlewareCoreLogic:
         _set_test_title_config(max_chars=12)
         middleware = TitleMiddleware()
         fake_model = MagicMock()
-        fake_model.invoke.return_value = MagicMock(content='"A very long generated title"')
+        fake_model.ainvoke = AsyncMock(return_value=MagicMock(content='"A very long generated title"'))
         monkeypatch.setattr("src.agents.middlewares.title_middleware.create_chat_model", lambda **kwargs: fake_model)
 
         state = {
@@ -85,7 +86,7 @@ class TestTitleMiddlewareCoreLogic:
                 AIMessage(content="好的，先确认需求"),
             ]
         }
-        title = middleware._generate_title(state)
+        title = asyncio.run(middleware._generate_title(state))
 
         assert '"' not in title
         assert "'" not in title
@@ -95,7 +96,7 @@ class TestTitleMiddlewareCoreLogic:
         _set_test_title_config(max_chars=20)
         middleware = TitleMiddleware()
         fake_model = MagicMock()
-        fake_model.invoke.side_effect = RuntimeError("LLM unavailable")
+        fake_model.ainvoke = AsyncMock(side_effect=RuntimeError("LLM unavailable"))
         monkeypatch.setattr("src.agents.middlewares.title_middleware.create_chat_model", lambda **kwargs: fake_model)
 
         state = {
@@ -104,7 +105,7 @@ class TestTitleMiddlewareCoreLogic:
                 AIMessage(content="收到"),
             ]
         }
-        title = middleware._generate_title(state)
+        title = asyncio.run(middleware._generate_title(state))
 
         # Assert behavior (truncated fallback + ellipsis) without overfitting exact text.
         assert title.endswith("...")
@@ -113,11 +114,11 @@ class TestTitleMiddlewareCoreLogic:
     def test_after_agent_returns_title_only_when_needed(self, monkeypatch):
         middleware = TitleMiddleware()
         monkeypatch.setattr(middleware, "_should_generate_title", lambda state: True)
-        monkeypatch.setattr(middleware, "_generate_title", lambda state: "核心逻辑回归")
+        monkeypatch.setattr(middleware, "_generate_title", AsyncMock(return_value="核心逻辑回归"))
 
-        result = middleware.after_agent({"messages": []}, runtime=MagicMock())
+        result = asyncio.run(middleware.aafter_model({"messages": []}, runtime=MagicMock()))
 
         assert result == {"title": "核心逻辑回归"}
 
         monkeypatch.setattr(middleware, "_should_generate_title", lambda state: False)
-        assert middleware.after_agent({"messages": []}, runtime=MagicMock()) is None
+        assert asyncio.run(middleware.aafter_model({"messages": []}, runtime=MagicMock())) is None