Method _generate_local_env_setup

ms_agent/skill/container.py:771–817

Generate Python code to setup environment for local execution.

(self, input_spec: ExecutionInput)

            raise

    def _generate_local_env_setup(self, input_spec: ExecutionInput) -> str:
        """Generate Python code to setup environment for local execution."""
        lines = [
            'import os',
            'import sys',
            '',
            '# Setup environment for local execution',
            f"os.environ['SKILL_OUTPUT_DIR'] = {repr(str(self.output_dir))}",
            f"os.environ['SKILL_LOGS_DIR'] = {repr(str(self.logs_dir))}",
            '',
            '# Helper functions for I/O paths',
            'def get_output_path(filename):',
            '    """Get the full path for an output file. ALL outputs should use this."""',
            "    return os.path.join(os.environ['SKILL_OUTPUT_DIR'], filename)",
            '',
            f'SKILL_OUTPUT_DIR = {repr(str(self.output_dir))}',
            f'SKILL_LOGS_DIR = {repr(str(self.logs_dir))}',
        ]

        # Add working directory to sys.path for imports and change to it
        if input_spec.working_dir:
            work_dir = str(input_spec.working_dir)
            lines.extend([
                '',
                '# Setup working directory for resource access (READ-ONLY for resources)',
                f'_skill_dir = {repr(work_dir)}',
                "os.environ['SKILL_DIR'] = _skill_dir",
                'SKILL_DIR = _skill_dir',
                'if _skill_dir not in sys.path:',
                '    sys.path.insert(0, _skill_dir)',
                'os.chdir(_skill_dir)',
            ])

        # Add custom env vars
        for key, value in input_spec.env_vars.items():
            lines.append(f'os.environ[{repr(key)}] = {repr(value)}')

        # Add args
        if input_spec.args:
            lines.append('')
            lines.append('# Command line arguments')
            args_str = repr(input_spec.args)
            lines.append(f'ARGS = {args_str}')
            lines.append('sys.argv = ["script.py"] + [str(a) for a in ARGS]')

        lines.append('')
        return '\n'.join(lines)

    def _generate_local_js_env_setup(self, input_spec: ExecutionInput) -> str:
        """Generate JavaScript code to setup environment for local execution."""