strukto-ai--mirage/python/mirage/agents/prompts.py

# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

MIRAGE_SYSTEM_PROMPT = """\
Your filesystem is powered by Mirage — a virtual filesystem that mounts \
cloud storage, local files, and in-memory data as a unified file tree.

Capabilities beyond standard filesystem:
- cat on .parquet, .orc, .feather files returns a formatted table
- head -n 5 on data files returns the first 5 rows
- grep works natively on CSV, JSON, Parquet — not just text
- Pipes work: cat data.parquet | grep error | sort | uniq | wc -l
- head, tail, cut, wc, sort, uniq, tee, xargs are all available

You can write Python code and execute it. The workspace is pre-configured \
with your data sources mounted at their respective paths.

Use the execute tool for complex operations. \
Use read/write/edit for simple file operations.
"""


def build_system_prompt(
    mount_info: dict[str, str] | None = None,
    extra_instructions: str = "",
    workspace: "object | None" = None,
) -> str:
    """Build a system prompt with optional mount info and extra instructions.

    Args:
        mount_info (dict[str, str] | None): Map of mount prefix to description.
        extra_instructions (str): Additional instructions to append.
        workspace (Workspace | None): If provided, uses workspace.file_prompt.

    Returns:
        str: The complete system prompt.
    """
    parts = [MIRAGE_SYSTEM_PROMPT]
    if workspace is not None and hasattr(workspace, "file_prompt"):
        parts.append("Mounted data sources:\n" + workspace.file_prompt)
    elif mount_info:
        parts.append("\nMounted data sources:")
        for prefix, description in mount_info.items():
            parts.append(f"- {prefix} — {description}")
        parts.append("")
    if extra_instructions:
        parts.append(extra_instructions)
    return "\n".join(parts)