Moved dataclass to different file

libs/computer/computer/interface/base.py

@@ -3,8 +3,7 @@
 from abc import ABC, abstractmethod
 from typing import Optional, Dict, Any, Tuple, List
 from ..logger import Logger, LogLevel
-from ..models import Computer, CommandResult, MouseButton
-
+from .models import MouseButton, CommandResult
 
 class BaseComputerInterface(ABC):
     """Base class for computer control interfaces."""

libs/computer/computer/interface/linux.py

@@ -9,7 +9,7 @@ import websockets
 from ..logger import Logger, LogLevel
 from .base import BaseComputerInterface
 from ..utils import decode_base64_image, encode_base64_image, bytes_to_image, draw_box, resize_image
-from ..models import Computer, CommandResult, Key, KeyType, MouseButton
+from .models import Key, KeyType, MouseButton, CommandResult
 
 class LinuxComputerInterface(BaseComputerInterface):
     """Interface for Linux."""

libs/computer/computer/interface/macos.py

@@ -9,8 +9,7 @@ import websockets
 from ..logger import Logger, LogLevel
 from .base import BaseComputerInterface
 from ..utils import decode_base64_image, encode_base64_image, bytes_to_image, draw_box, resize_image
-from ..models import Computer, CommandResult, Key, KeyType, MouseButton
-
+from .models import Key, KeyType, MouseButton, CommandResult
 
 class MacOSComputerInterface(BaseComputerInterface):
     """Interface for macOS."""

libs/computer/computer/interface/models.py

@@ -1,5 +1,17 @@
 from enum import Enum
 from typing import Dict, List, Any, TypedDict, Union, Literal
+from dataclasses import dataclass
+
+@dataclass
+class CommandResult:
+    stdout: str
+    stderr: str  
+    returncode: int
+    
+    def __init__(self, stdout: str, stderr: str, returncode: int):
+        self.stdout = stdout
+        self.stderr = stderr
+        self.returncode = returncode
 
 # Navigation key literals
 NavigationKey = Literal['pagedown', 'pageup', 'home', 'end', 'left', 'right', 'up', 'down']

libs/computer/computer/models.py

@@ -6,18 +6,6 @@ from typing import Optional, Any, Dict
 # Import base provider interface
 from .providers.base import BaseVMProvider
 
-@dataclass
-class CommandResult:
-    """Command result."""
-    stdout: str
-    stderr: str  
-    returncode: int
-    
-    def __init__(self, stdout: str, stderr: str, returncode: int):
-        self.stdout = stdout
-        self.stderr = stderr
-        self.returncode = returncode
-    
 @dataclass
 class Display:
     """Display configuration."""