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