fileop.py

import os
from dataclasses import dataclass

from opendevin.observation import FileReadObservation, FileWriteObservation
from .base import ExecutableAction

# This is the path where the workspace is mounted in the container
# The LLM sometimes returns paths with this prefix, so we need to remove it
PATH_PREFIX = "/workspace/"

def resolve_path(base_path, file_path):
    if file_path.startswith(PATH_PREFIX):
        file_path = file_path[len(PATH_PREFIX):]
    return os.path.join(base_path, file_path)


@dataclass
class FileReadAction(ExecutableAction):
    path: str
    action: str = "read"

    def run(self, controller) -> FileReadObservation:
        path = resolve_path(controller.workdir, self.path)
        with open(path, 'r', encoding='utf-8') as file:
            return FileReadObservation(
                path=path,
                content=file.read())

    @property
    def message(self) -> str:
        return f"Reading file: {self.path}"

@dataclass
class FileWriteAction(ExecutableAction):
    path: str
    contents: str
    action: str = "write"

    def run(self, controller) -> FileWriteObservation:
        path = resolve_path(controller.workdir, self.path)
        with open(path, 'w', encoding='utf-8') as file:
            file.write(self.contents)
        return FileWriteObservation(content="", path=self.path)

    @property
    def message(self) -> str:
        return f"Writing file: {self.path}"