From 99de17162ec51e1a2fcdd9d80f67a84941d9792f Mon Sep 17 00:00:00 2001
From: chen <clean6378@gmail.com>
Date: Thu, 16 Jul 2026 05:34:53 +0800
Subject: [PATCH] refactor: trim BubbleRole and BubbleMetadata from models
 exports

---
 models/__init__.py          | 4 +---
 services/workspace_tabs.py  | 3 +--
 utils/cursor_md_exporter.py | 3 ++-
 utils/display_bubble.py     | 3 ++-
 4 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/models/__init__.py b/models/__init__.py
index a222bc3..0bf97bf 100644
--- a/models/__init__.py
+++ b/models/__init__.py
@@ -1,4 +1,4 @@
-from models.bubble_display import BubbleMetadata, BubbleRole, DisplayBubble
+from models.bubble_display import DisplayBubble
 from models.cli_session import CliSessionMeta
 from models.conversation import Bubble, Composer, Conversation, WorkspaceLocalComposer
 from models.errors import SchemaError
@@ -9,8 +9,6 @@
 
 __all__ = [
     "Bubble",
-    "BubbleMetadata",
-    "BubbleRole",
     "CliSessionMeta",
     "DisplayBubble",
     "Composer",
diff --git a/services/workspace_tabs.py b/services/workspace_tabs.py
index 7a2d9c3..ab2588c 100644
--- a/services/workspace_tabs.py
+++ b/services/workspace_tabs.py
@@ -26,13 +26,12 @@
 from utils.workspace_descriptor import read_json_file
 from models import (
     Bubble,
-    BubbleMetadata,
-    BubbleRole,
     Composer,
     DisplayBubble,
     ParseWarningCollector,
     SchemaError,
 )
+from models.bubble_display import BubbleMetadata, BubbleRole
 from models.raw_access import (
     conversation_header_bubble_id,
     message_request_context_project_layouts,
diff --git a/utils/cursor_md_exporter.py b/utils/cursor_md_exporter.py
index 12872f6..d85eb38 100644
--- a/utils/cursor_md_exporter.py
+++ b/utils/cursor_md_exporter.py
@@ -21,7 +21,8 @@
 from pathlib import Path
 from typing import Any
 
-from models import Bubble, BubbleRole, DisplayBubble
+from models import Bubble, DisplayBubble
+from models.bubble_display import BubbleRole
 from utils.cli_chat_reader import traverse_blobs, messages_to_bubbles
 from utils.display_bubble import (
     annotate_response_times,
diff --git a/utils/display_bubble.py b/utils/display_bubble.py
index 5dbec81..9c70a50 100644
--- a/utils/display_bubble.py
+++ b/utils/display_bubble.py
@@ -5,7 +5,8 @@
 from datetime import datetime
 from typing import Any, cast
 
-from models import Bubble, BubbleMetadata, BubbleRole, DisplayBubble
+from models import Bubble, DisplayBubble
+from models.bubble_display import BubbleMetadata, BubbleRole
 from utils.path_helpers import to_epoch_ms
 from utils.text_extract import extract_text_from_bubble
 from utils.tool_parser import parse_tool_call