63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
"""Mode and ModeOption dataclasses for the Modes system.
|
|
|
|
The Modes system allows developers to define multiple picker categories
|
|
(e.g., Model, Approach, Reasoning Effort) that users can select from
|
|
in the chat composer.
|
|
"""
|
|
|
|
from dataclasses import dataclass, field
|
|
from typing import List, Optional
|
|
|
|
from dataclasses_json import DataClassJsonMixin
|
|
|
|
|
|
@dataclass
|
|
class ModeOption(DataClassJsonMixin):
|
|
"""A single selectable option within a Mode.
|
|
|
|
Attributes:
|
|
id: Unique identifier for this option (e.g., "gpt-5", "planning")
|
|
name: Display name shown in the UI (e.g., "GPT-5", "Planning")
|
|
description: Optional description shown in the dropdown
|
|
icon: Optional icon - can be a Lucide icon name, local path, or URL
|
|
default: Whether this is the default selected option for its mode
|
|
"""
|
|
|
|
id: str
|
|
name: str
|
|
description: Optional[str] = None
|
|
icon: Optional[str] = None
|
|
default: bool = False
|
|
|
|
|
|
@dataclass
|
|
class Mode(DataClassJsonMixin):
|
|
"""A category of options the user can select from.
|
|
|
|
Each Mode represents a picker dropdown in the chat composer.
|
|
Users select exactly one option per mode.
|
|
|
|
Attributes:
|
|
id: Unique identifier for this mode (e.g., "llm", "approach")
|
|
name: Display name shown in the UI (e.g., "Model", "Approach")
|
|
options: List of available options for this mode
|
|
"""
|
|
|
|
id: str
|
|
name: str
|
|
options: List[ModeOption] = field(default_factory=list)
|
|
|
|
def get_default_option(self) -> Optional[ModeOption]:
|
|
"""Get the default option for this mode, or the first option if none is default."""
|
|
for option in self.options:
|
|
if option.default:
|
|
return option
|
|
return self.options[0] if self.options else None
|
|
|
|
def get_option_by_id(self, option_id: str) -> Optional[ModeOption]:
|
|
"""Get an option by its ID."""
|
|
for option in self.options:
|
|
if option.id == option_id:
|
|
return option
|
|
return None
|