Skip to content

Commit 1f8a38c

Browse files
refactor: trim BubbleRole and BubbleMetadata from models exports
1 parent 1dbf635 commit 1f8a38c

4 files changed

Lines changed: 6 additions & 7 deletions

File tree

models/__init__.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from models.bubble_display import BubbleMetadata, BubbleRole, DisplayBubble
1+
from models.bubble_display import DisplayBubble
22
from models.cli_session import CliSessionMeta
33
from models.conversation import Bubble, Composer, Conversation, WorkspaceLocalComposer
44
from models.errors import SchemaError
@@ -9,8 +9,6 @@
99

1010
__all__ = [
1111
"Bubble",
12-
"BubbleMetadata",
13-
"BubbleRole",
1412
"CliSessionMeta",
1513
"DisplayBubble",
1614
"Composer",

services/workspace_tabs.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,13 +26,12 @@
2626
from utils.workspace_descriptor import read_json_file
2727
from models import (
2828
Bubble,
29-
BubbleMetadata,
30-
BubbleRole,
3129
Composer,
3230
DisplayBubble,
3331
ParseWarningCollector,
3432
SchemaError,
3533
)
34+
from models.bubble_display import BubbleMetadata, BubbleRole
3635
from models.raw_access import (
3736
conversation_header_bubble_id,
3837
message_request_context_project_layouts,

utils/cursor_md_exporter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@
2121
from pathlib import Path
2222
from typing import Any
2323

24-
from models import Bubble, BubbleRole, DisplayBubble
24+
from models import Bubble, DisplayBubble
25+
from models.bubble_display import BubbleRole
2526
from utils.cli_chat_reader import traverse_blobs, messages_to_bubbles
2627
from utils.display_bubble import (
2728
annotate_response_times,

utils/display_bubble.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
from datetime import datetime
66
from typing import Any, cast
77

8-
from models import Bubble, BubbleMetadata, BubbleRole, DisplayBubble
8+
from models import Bubble, DisplayBubble
9+
from models.bubble_display import BubbleMetadata, BubbleRole
910
from utils.path_helpers import to_epoch_ms
1011
from utils.text_extract import extract_text_from_bubble
1112
from utils.tool_parser import parse_tool_call

0 commit comments

Comments
 (0)