minion-code 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

minion_code/{types.py → type_defs.py}

@@ -27,13 +27,21 @@ class InputMode(Enum):
 @dataclass
 class MessageContent:
     """Represents message content - can be text or structured content"""
+
     content: Union[str, List[Dict[str, Any]]]
     type: str = "text"
 
+    def __init__(
+        self, content: Union[str, List[Dict[str, Any]]] = "", type: str = "text"
+    ):
+        self.content = content
+        self.type = type
+
 
 @dataclass
 class Message:
     """Core message structure equivalent to TypeScript MessageType"""
+
     uuid: str = field(default_factory=lambda: str(uuid.uuid4()))
     type: MessageType = MessageType.USER
     message: MessageContent = field(default_factory=lambda: MessageContent(""))
@@ -44,6 +52,7 @@ class Message:
 @dataclass
 class ToolUseConfirm:
     """Tool use confirmation context"""
+
     tool_name: str
     parameters: Dict[str, Any]
     on_confirm: Any  # Callable[[], None]
@@ -53,6 +62,7 @@ class ToolUseConfirm:
 @dataclass
 class BinaryFeedbackContext:
     """Binary feedback context for comparing two assistant messages"""
+
     m1: Message
     m2: Message
     resolve: Any  # Callable[[str], None]
@@ -61,6 +71,7 @@ class BinaryFeedbackContext:
 @dataclass
 class ToolJSXContext:
     """Tool JSX rendering context"""
+
     jsx: Optional[Any] = None
     should_hide_prompt_input: bool = False
 
@@ -68,6 +79,7 @@ class ToolJSXContext:
 @dataclass
 class ModelInfo:
     """Model information display"""
+
     name: str
     provider: str
     context_length: int
@@ -77,12 +89,13 @@ class ModelInfo:
 
 class REPLConfig:
     """Configuration equivalent to getGlobalConfig()"""
+
     def __init__(self):
         self.verbose: bool = False
         self.debug: bool = False
         self.safe_mode: bool = False
         self.has_acknowledged_cost_threshold: bool = False
         self.model_name: str = "claude-3-5-sonnet-20241022"
-    
+
     def get_model_name(self, context: str = "main") -> str:
-        return self.model_name
+        return self.model_name

minion_code/utils/__init__.py

@@ -22,23 +22,51 @@ from .todo_storage import (
     clear_todos,
 )
 
+from .output_truncator import (
+    truncate_output,
+    check_file_size_before_read,
+    check_mcp_output,
+    save_large_output,
+    cleanup_cache,
+    OutputTooLargeError,
+    MCPContentTooLargeError,
+    FileTooLargeError,
+    MAX_OUTPUT_SIZE,
+    MAX_FILE_SIZE,
+    MAX_TOKEN_LIMIT,
+    CACHE_DIR,
+)
+
 __all__ = [
     # Todo file utilities
-    'get_todo_file_path',
-    'get_default_storage_dir',
-    'ensure_storage_dir_exists',
-    'list_todo_files',
-    'extract_agent_id_from_todo_file',
-    'is_todo_file',
+    "get_todo_file_path",
+    "get_default_storage_dir",
+    "ensure_storage_dir_exists",
+    "list_todo_files",
+    "extract_agent_id_from_todo_file",
+    "is_todo_file",
     # Todo storage
-    'TodoItem',
-    'TodoStatus',
-    'TodoPriority',
-    'TodoStorage',
-    'get_todos',
-    'set_todos',
-    'add_todo',
-    'update_todo',
-    'remove_todo',
-    'clear_todos',
-]
+    "TodoItem",
+    "TodoStatus",
+    "TodoPriority",
+    "TodoStorage",
+    "get_todos",
+    "set_todos",
+    "add_todo",
+    "update_todo",
+    "remove_todo",
+    "clear_todos",
+    # Output truncator
+    "truncate_output",
+    "check_file_size_before_read",
+    "check_mcp_output",
+    "save_large_output",
+    "cleanup_cache",
+    "OutputTooLargeError",
+    "MCPContentTooLargeError",
+    "FileTooLargeError",
+    "MAX_OUTPUT_SIZE",
+    "MAX_FILE_SIZE",
+    "MAX_TOKEN_LIMIT",
+    "CACHE_DIR",
+]