From 7100c5e6fee9a0127534d564dfe09132d8b7c30d Mon Sep 17 00:00:00 2001
From: yuancjun <153783375+yuancjun@users.noreply.github.com>
Date: Tue, 17 Mar 2026 11:11:56 +0800
Subject: [PATCH] Use agent's framework workspace instead of current working
 directory

---
 src/bub/channels/cli/__init__.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/bub/channels/cli/__init__.py b/src/bub/channels/cli/__init__.py
index 99acedb6..21606583 100644
--- a/src/bub/channels/cli/__init__.py
+++ b/src/bub/channels/cli/__init__.py
@@ -40,9 +40,9 @@ def __init__(self, on_receive: MessageHandler, agent: Agent) -> None:
         self._mode = "agent"  # or "shell"
         self._main_task: asyncio.Task | None = None
         self._renderer = CliRenderer(get_console())
-        self._prompt = self._build_prompt(Path.cwd())
         self._last_tape_info: TapeInfo | None = None
-        self._workspace = Path.cwd()
+        self._workspace = self._agent.framework.workspace
+        self._prompt = self._build_prompt(self._workspace)
 
     async def _refresh_tape_info(self) -> None:
         tape = self._agent.tapes.session_tape(self._message_template["session_id"], self._workspace)