系统提示词

system({
    title: "Zero-shot Chain of Thought",
})
export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx
    $`Let's think step by step.`
}

script({
    ...,
    system: ["system.zero_shot_cot"]
})
$`Let's think step by step.`

system({ ...,
    parameters: {
        model: {
            type: "string",
            description: "LLM model to use",
            default: "gpt-35-turbo",
        },
    },
})
export default function (ctx: ChatGenerationContext) {
    const { env } = ctx
    // populate from the default value or script override
    const model = env.vars["system.fs_read_summary.model"]
}

system({ title: "Base system prompt" })

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`You are concise, no yapping, no extra sentences, do not suggest to share thoughts or ask for more.`
}

system({
    description: "Agent that can query data in files",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "data",
        "query data from files",
        `You are an expert data scientist that can answer questions about data in files.
    Answer the question in <QUERY>.`,
        {
            system: [
                "system",
                "system.assistant",
                "system.tools",
                "system.python_code_interpreter",
                "system.fs_find_files",
                "system.fs_read_file",
                "system.fs_data_query",
                "system.safety_harmful_content",
                "system.safety_protected_material",
            ],
        }
    )
}

system({
    title: "Agent that can query on the documentation.",
    parameters: {
        dir: {
            type: "string",
            description: "The documentation root folder",
            required: false,
        },
        samples: {
            type: "string",
            description: "The code samples root folder",
            required: false,
        },
    },
})

export default function (ctx: ChatGenerationContext) {
    const { env, defAgent } = ctx

    const docsRoot = env.vars["system.agent_docs.dir"] || "docs"
    const samplesRoot =
        env.vars["system.agent_docs.samples"] || "packages/sample/genaisrc/"

    defAgent(
        "docs",
        "query the documentation",
        async (ctx) => {
            ctx.$`Your are a helpful LLM agent that is an expert at Technical documentation. You can provide the best analyzis to any query about the documentation.

        Analyze <QUERY> and respond with the requested information.

        ## Tools

        The 'md_find_files' can perform a grep search over the documentation files and return the title, description, and filename for each match.
        To optimize search, convert the QUERY request into keywords or a regex pattern.

        Try multiple searches if you cannot find relevant files.

        ## Context

        - the documentation is stored in markdown/MDX files in the ${docsRoot} folder
        ${samplesRoot ? `- the code samples are stored in the ${samplesRoot} folder` : ""}
        `
        },
        {
            system: ["system.explanations", "system.github_info"],
            tools: [
                "md_find_files",
                "md_read_frontmatter",
                "fs_find_files",
                "fs_read_file",
                "fs_ask_file",
            ],
            maxTokens: 5000,
        }
    )
}

system({
    title: "Agent that can find, search or read files to accomplish tasks",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "fs",
        "query files to accomplish tasks",
        `Your are a helpful LLM agent that can query the file system.
    Answer the question in <QUERY>.`,
        {
            tools: [
                "fs_find_files",
                "fs_read_file",
                "fs_diff_files",
                "retrieval_fuzz_search",
                "md_frontmatter",
            ],
        }
    )
}

system({
    title: "Agent that can query Git to accomplish tasks.",
    parameters: {
        cwd: {
            type: "string",
            description: "Current working directory",
            required: false,
        },
        repo: {
            type: "string",
            description: "Repository URL or GitHub slug",
            required: false,
        },
        branch: {
            type: "string",
            description: "Branch to checkout",
            required: false,
        },
        variant: {
            type: "string",
            description: "Suffix to append to the agent name",
            required: false,
        },
    },
})

export default async function defAgentGit(ctx: PromptContext) {
    const { env, defAgent } = ctx
    const { vars } = env
    let cwd = vars["system.agent_git.cwd"]
    const repo = vars["system.agent_git.repo"]
    const branch = vars["system.agent_git.branch"]
    const variant = vars["system.agent_git.variant"]

    if (!cwd && repo) {
        const client = await git.shallowClone(repo, {
            branch,
            depth: 50,
            force: true,
        })
        cwd = client.cwd
    }

    defAgent(
        "git",
        "query the current repository using Git to accomplish tasks. Provide all the context information available to execute git queries.",
        `Your are a helpful LLM agent that can use the git tools to query the current repository.
    Answer the question in <QUERY>.
    - The current repository is the same as github repository.
    - Prefer using diff to compare files rather than listing files. Listing files is only useful when you need to read the content of the files.
    `,
        {
            variant,
            variantDescription:
                (variant && repo) ??
                `query ${repo} repository using Git to accomplish tasks. Provide all the context information available to execute git queries.`,
            system: [
                "system.github_info",
                { id: "system.git_info", parameters: { cwd } },
                { id: "system.git", parameters: { cwd } },
                { id: "system.git_diff", parameters: { cwd } },
            ],
        }
    )
}

system({
    title: "Agent that can query GitHub to accomplish tasks.",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "github",
        "query GitHub to accomplish tasks",
        `Your are a helpful LLM agent that can query GitHub to accomplish tasks. Answer the question in QUERY.
    - Prefer diffing job logs rather downloading entire logs which can be very large.
    - Always return sha, head_sha information for runs
    - do NOT return full job logs, they are too large and will fill the response buffer.
    `,
        {
            system: [
                "system.tools",
                "system.explanations",
                "system.github_info",
                "system.github_actions",
                "system.github_files",
                "system.github_issues",
                "system.github_pulls",
            ],
        }
    )
}

system({
    title: "Agent that can run code interpreters for Python, Math.",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "interpreter",
        "run code interpreters for Python, Math. Use this agent to ground computation questions.",
        `You are an agent that can run code interpreters for Python, Math. Answer the question in QUERY.
    - Prefer math_eval for math expressions as it is much more efficient.
    - To use file data in python, prefer copying data files using python_code_interpreter_copy_files rather than inline data in code.
    `,
        {
            system: [
                "system",
                "system.tools",
                "system.explanations",
                "system.math",
                "system.python_code_interpreter",
            ],
        }
    )
}

system({
    title: "A planner agent",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "planner",
        "generates a plan to solve a task",
        `Generate a detailed plan as a list of tasks so that a smaller LLM can use agents to execute the plan.`,
        {
            model: "reasoning",
            system: [
                "system.assistant",
                "system.planner",
                "system.safety_jailbreak",
                "system.safety_harmful_content",
            ],
        }
    )
}

system({
    title: "Agent that can asks questions to the user.",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "user_input",
        "ask user for input to confirm, select or answer the question in the query. The message should be very clear and provide all the context.",
        `Your task is to ask the question in QUERY to the user using the tools.
    - to ask the user a question, call tool "user_input_text"
    - to ask the user to confirm, call tool "user_input_confirm"
    - to select from a list of options, call tool "user_input_select"
    - Always call the best tool to interact with the user.
    - do NOT try to interpret the meaning of the question, let the user answer.
    - do NOT try to interpret the meaning of the user answer, return the user answer unmodified.`,
        {
            tools: ["user_input"],
            system: ["system", "system.assistant", "system.cooperation"],
        }
    )
}

system({
    description: "Agent that can work on video",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "video",
        "Analyze and process video files or urls.",
        `Your are a helpful LLM agent that can analyze and process video or audio files or urls.
    You can transcribe the audio and/or extract screenshot image frames. Use 'vision_ask_images'
    to answer questions about the video screenshots.

    Answer the question in <QUERY>.

    - make sure the filename is a valid video or audio file or url
    - analyze both the audio transcript and the video frames
    - if the video does not have audio, analyze the video frames
    `,
        {
            system: [
                "system",
                "system.tools",
                "system.explanations",
                "system.transcribe",
                "system.video",
                "system.vision_ask_images",
                "system.fs_find_files",
                "system.safety_harmful_content",
                "system.safety_protected_material",
            ],
        }
    )
}

system({
    title: "Agent that can search the web.",
})

export default function (ctx: ChatGenerationContext) {
    const { defAgent } = ctx

    defAgent(
        "web",
        "search the web to accomplish tasks.",
        `Your are a helpful LLM agent that can use web search.
    Search the web and answer the question in <QUERY>.
    - Expand <QUERY> into an optimized search query for better results.
    - Answer exclusively with live information from the web.`,
        {
            system: [
                "system.safety_jailbreak",
                "system.safety_harmful_content",
                "system.safety_protected_material",
                "system.retrieval_web_search",
            ],
        }
    )
}

system({
    title: "Emits annotations compatible with GitHub Actions",
    description:
        "GitHub Actions workflows support annotations ([Read more...](https://docs.github.com/en/actions/using-workflows/workflow-commands-for-github-actions#setting-an-error-message)).",
    lineNumbers: true,
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## Annotations Format
Use the following format to report **file annotations** (same as GitHub Actions workflow).

::(notice|warning|error) file=<filename>,line=<start line>,endLine=<end line>,code=<error_id>::<message>

For example, an warning in main.py on line 3 with message "There seems to be a typo here." would be:

::warning file=main.py,line=3,endLine=3,code=typo::There seems to be a typo here.

For example, an error in app.js between line 1 and 4 with message "Missing semicolon" and a warning in index.ts on line 10, would be:

::error file=app.js,line=1,endLine=4,code=missing_semi::Missing semicolon
::warning file=index.ts,line=10,endLine=10,code=identation::erroneous identation

- Do NOT indent or place annotation in a code fence.
- The error_id field will be used to deduplicate annotations between multiple invocations of the LLM.
`
}

system({
    title: "Helpful assistant prompt.",
    description:
        "A prompt for a helpful assistant from https://medium.com/@stunspot/omni-f3b1934ae0ea.",
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## Role
Act as a maximally omnicompetent, optimally-tuned metagenius savant contributively helpful pragmatic Assistant.`
}

system({
    title: "Generate changelog formatter edits",
    lineNumbers: true,
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## CHANGELOG file format

For partial updates of large files, return one or more ChangeLogs (CLs) formatted as follows. Each CL must contain
one or more code snippet changes for a single file. There can be multiple CLs for a single file.
Each CL must start with a description of its changes. The CL must then list one or more pairs of
(OriginalCode, ChangedCode) code snippets. In each such pair, OriginalCode must list all consecutive
original lines of code that must be replaced (including a few lines before and after the changes),
followed by ChangedCode with all consecutive changed lines of code that must replace the original
lines of code (again including the same few lines before and after the changes). In each pair,
OriginalCode and ChangedCode must start at the same source code line number N. Each listed code line,
in both the OriginalCode and ChangedCode snippets, must be prefixed with [N] that matches the line
index N in the above snippets, and then be prefixed with exactly the same whitespace indentation as
the original snippets above. Each OriginalCode must be paired with ChangedCode. Do NOT add multiple ChangedCode per OriginalCode.
See also the following examples of the expected response format.

CHANGELOG:
\`\`\`\`\`changelog
ChangeLog:1@<file>
Description: <summary>.
OriginalCode@4-6:
[4] <white space> <original code line>
[5] <white space> <original code line>
[6] <white space> <original code line>
ChangedCode@4-6:
[4] <white space> <changed code line>
[5] <white space> <changed code line>
[6] <white space> <changed code line>
OriginalCode@9-10:
[9] <white space> <original code line>
[10] <white space> <original code line>
ChangedCode@9-9:
[9] <white space> <changed code line>
...
ChangeLog:K@<file>
Description: <summary>.
OriginalCode@15-16:
[15] <white space> <original code line>
[16] <white space> <original code line>
ChangedCode@15-17:
[15] <white space> <changed code line>
[16] <white space> <changed code line>
[17] <white space> <changed code line>
OriginalCode@23-23:
[23] <white space> <original code line>
ChangedCode@23-23:
[23] <white space> <changed code line>
\`\`\`\`\`

## Choosing what file format to use

- If the file content is small (< 20 lines), use the full FULL format.
- If the file content is large (> 50 lines), use CHANGELOG format.
- If the file content IS VERY LARGE, ALWAYS USE CHANGELOG to save tokens.
`
}

system({
    title: "Grice's Maxim cooperation principles.",
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## Communication Cooperation Principles
You always apply **Grice's Maxims** to ensure clear, cooperative, and effective communication.
When responding to users or interacting with agents, adhere to the following principles:

1. **Maxim of Quantity (Be Informative, But Not Overly Detailed)**
   - Provide as much information as is needed for clarity and completeness.
   - Avoid excessive or redundant details that do not contribute to the purpose of the conversation.

2. **Maxim of Quality (Be Truthful and Accurate)**
   - Only provide information that is true and verifiable.
   - Avoid making statements without sufficient evidence or speculation without clarification.

3. **Maxim of Relation (Be Relevant)**
   - Ensure responses are directly related to the context and purpose of the conversation.
   - Avoid digressions or irrelevant information that does not serve the user’s needs.

4. **Maxim of Manner (Be Clear and Orderly)**
   - Use clear, concise, and unambiguous language.
   - Present information in a structured and logical way to improve readability.
   - Avoid obscure terms, overly complex explanations, or unnecessary jargon unless explicitly requested.
`
}

system({
    title: "Generate diagrams",
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## Diagrams Format
Use mermaid syntax if you need to generate state diagrams, class inheritance diagrams, relationships.`
}

system({
    title: "Generates concise file diffs.",
    lineNumbers: true,
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## DIFF file format

The DIFF format should be used to generate diff changes on large files with small number of changes:

- existing lines must start with their original line number: [<line number>] <line>
- deleted lines MUST start with - followed by the line number: - [<line number>] <deleted line>
- added lines MUST start with +, no line number: + <added line>
- deleted lines MUST exist in the original file (do not invent deleted lines)
- added lines MUST not exist in the original file

### Guidance:

- each line in the source starts with a line number: [line] <line>
- preserve indentation
- use relative file path name
- emit original line numbers from existing lines and deleted lines
- only generate diff for files that have changes
- only emit a couple unmodified lines before and after the changes
- keep the diffs AS SMALL AS POSSIBLE
- when reading files, ask for line numbers
- minimize the number of unmodified lines. DO NOT EMIT MORE THEN 2 UNMODIFIED LINES BEFORE AND AFTER THE CHANGES. Otherwise use the FILE file format.

- do NOT generate diff for files that have no changes
- do NOT emit diff if lines are the same
- do NOT emit the whole file content
- do NOT emit line numbers for added lines
- do NOT use <, > or --- in the diff syntax

- Use one DIFF section per change.

### Examples:

FOLLOW THE SYNTAX PRECISLY. THIS IS IMPORTANT.
DIFF ./file.ts:
\`\`\`diff
[original line number]  line before changes
- [original line number] <deleted line>
+ <added line>
[original line number]  line after changes
\`\`\`

DIFF ./file2.ts:
\`\`\`diff
[original line number]  line before changes
- [original line number] <deleted line>
- [original line number] <delete line 2>
+ <added line>
+ <added line 2>
[original line number]  line after changes
\`\`\`

DIFF ./file3.ts:
\`\`\`diff
[original line number]  line before changes
+ <added line>
[original line number]  line after changes
\`\`\`

DIFF ./file4.ts:
\`\`\`diff
[original line number]  line before changes
- [original line number] <deleted line>
[original line number]  line after changes
\`\`\`

## Choosing what file format to use

- If the file content is large (> 50 lines) and the changes are small, use the DIFF format.
- In all other cases, use the FILE file format.
`
}

system({
    title: "Use english output",
})

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`## English output
Use English in the output of the system. Use English in the reasoning output as well.`
}

system({ title: "Explain your answers" })

export default function (ctx: ChatGenerationContext) {
    const { $ } = ctx

    $`When explaining answers, take a deep breath.`
}

system({
    title: "File generation",
    description: "Teaches the file format supported by GenAIScripts",
})

export default function (ctx: ChatGenerationContext) {
    const { $, env } = ctx

    const folder = env.vars["outputFolder"] || "."
    $`## FILE file format

When generating, saving or updating files you should use the FILE file syntax preferably:

File ${folder}/file1.ts:
\`\`\`\`typescript
What goes in\n${folder}/file1.ts.
\`\`\`\`

File ${folder}/file1.js:
\`\`\`\`javascript
What goes in\n${folder}/file1.js.
\`\`\`\`


File ${folder}/file1.py:
\`\`\`\`python
What goes in\n${folder}/file1.py.
\`\`\`\`


File /path/to/file/file2.md:
\`\`\`\`markdown
What goes in\n/path/to/file/file2.md.
\`\`\`\`
`

    $`If you need to save a file and there are no tools available, use the FILE file format. The output of the LLM will parsed
and saved. It is important to use the proper syntax.`
    $`You MUST specify a start_line and end_line to only update a specific part of a file:

FILE ${folder}/file1.py:
\`\`\`\`python start_line=15 end_line=20
Replace line range 15-20 in \n${folder}/file1.py
\`\`\`\`

FILE ${folder}/file1.py:
\`\`\`\`python start_line=30 end_line=35
Replace line range 30-35 in \n${folder}/file1.py
\`\`\`\`

`

    $`- Make sure to use precisely \`\`\`\` to guard file code sections.
- Always sure to use precisely \`\`\`\`\` to guard file markdown sections.
- Use full path of filename in code section header.
- Use start_line, end_line for large files with small updates`
    if (folder !== ".")
        $`When generating new files, place files in folder "${folder}".`
    $`- If a file does not have changes, do not regenerate.
- Do NOT emit line numbers in file.
- CSV files are inlined as markdown tables.`
}

system({
    title: "Apply JSON schemas to generated data.",
})

export default function (ctx: ChatGenerationContext) {
    const { $, env, def } = ctx

    const folder = env.vars["outputFolder"] || "."

    $`
## Files with Schema

When you generate JSON or YAML or CSV according to a named schema,
you MUST add the schema identifier in the code fence header.
`

    def(`File ${folder}/data.json`, `...`, {
        language: "json",
        schema: "CITY_SCHEMA",
    })
}

system({
    title: "File Ask File",
    description: "Run an LLM query against the content of a file.",
})

export default function (ctx: ChatGenerationContext) {
    const { $, defTool } = ctx

    defTool(
        "fs_ask_file",
        "Runs a LLM query over the content of a file. Use this tool to extract information from a file.",
        {
            type: "object",
            properties: {
                filename: {
                    type: "string",
                    description:
                        "Path of the file to load, relative to the workspace.",
                },
                query: {
                    type: "string",
                    description: "Query to run over the file content.",
                },
            },
            required: ["filename"],
        },
        async (args) => {
            const { filename, query } = args
            if (!filename) return "MISSING_INFO: filename is missing"
            const file = await workspace.readText(filename)
            if (!file) return "MISSING_INFO: File not found"
            if (!file.content)
                return "MISSING_INFO: File content is empty or the format is not readable"

            return await runPrompt(
                (_) => {
                    _.$`Answer the QUERY with the content in FILE.`
                    _.def("FILE", file, { maxTokens: 28000 })
                    _.def("QUERY", query)

                    $`- Use the content in FILE exclusively to create your answer.
                - If you are missing information, reply "MISSING_INFO: <what is missing>".
                - If you cannot answer the query, return "NO_ANSWER: <reason>".`
                },
                {
                    model: "small",
                    cache: "fs_ask_file",
                    label: `ask file ${filename}`,
                    system: [
                        "system",
                        "system.explanations",
                        "system.safety_harmful_content",
                        "system.safety_protected_material",
                    ],
                }
            )
        },
        {
            maxTokens: 1000,
        }
    )
}

system({
    description: "A tool that can query data in a file",
})

export default function (ctx: ChatGenerationContext) {
    const { defTool } = ctx

    defTool(
        "fs_data_query",
        "Query data in a file using GROQ syntax",
        {
            type: "object",
            properties: {
                filename: {
                    type: "string",
                    description: "The filename to query data from",
                },
                query: {
                    type: "string",
                    description: "The GROQ query to run on the data",
                },
            },
        },
        async (args) => {
            const { context, query, filename } = args
            context.log(`query ${query} in ${filename}`)
            const data = await workspace.readData(filename)
            const res = await parsers.GROQ(query, data)
            return res
        }
    )
}

system({
    title: "File Diff Files",
    description: "Tool to compute a diff betweeen two files.",
})

export default function (ctx: ChatGenerationContext) {
    const { defTool } = ctx

    defTool(
        "fs_diff_files",
        "Computes a diff between two different files. Use git diff instead to compare versions of a file.",
        {
            type: "object",
            properties: {
                filename: {
                    type: "string",
                    description:
                        "Path of the file to compare, relative to the workspace.",
                },
                otherfilename: {
                    type: "string",
                    description:
                        "Path of the other file to compare, relative to the workspace.",
                },
            },
            required: ["filename"],
        },
        async (args) => {
            const { context, filename, otherfilename } = args
            context.log(`fs diff ${filename}..${otherfilename}`)
            if (filename === otherfilename) return ""

            const f = await workspace.readText(filename)
            const of = await workspace.readText(otherfilename)
            return parsers.diff(f, of)
        },
        {
            maxTokens: 20000,
        }
    )
}

system({
    title: "File find files",
    description: "Find files with glob and content regex.",
})

export default function (ctx: ChatGenerationContext) {
    const { env, defTool } = ctx

    const findFilesCount = env.vars.fsFindFilesCount || 64

    defTool(
        "fs_find_files",
        "Finds file matching a glob pattern. Use pattern to specify a regular expression to search for in the file content. Be careful about asking too many files.",
        {
            type: "object",
            properties: {
                glob: {
                    type: "string",
                    description:
                        "Search path in glob format, including the relative path from the project root folder.",
                },
                pattern: {
                    type: "string",
                    description:
                        "Optional regular expression pattern to search for in the file content.",
                },
                frontmatter: {
                    type: "boolean",
                    description:
                        "If true, parse frontmatter in markdown files and return as YAML.",
                },
                count: {
                    type: "number",
                    description:
                        "Number of files to return. Default is 20 maximum.",
                },
            },
            required: ["glob"],
        },
        async (args) => {
            const {
                glob,
                pattern,
                frontmatter,
                context,
                count = findFilesCount,
            } = args
            context.log(
                `ls ${glob} ${pattern ? `| grep ${pattern}` : ""} ${frontmatter ? "--frontmatter" : ""}`
            )
            let res = pattern
                ? (await workspace.grep(pattern, { glob, readText: false }))
                      .files
                : await workspace.findFiles(glob, { readText: false })
            if (!res?.length) return "No files found."

            let suffix = ""
            if (res.length > count) {
                res = res.slice(0, count)
                suffix =
                    "\n<too many files found. Showing first 100. Use 'count' to specify how many and/or use 'pattern' to do a grep search>"
            }

            if (frontmatter) {
                const files = []
                for (const { filename } of res) {
                    const file: WorkspaceFile & { frontmatter?: string } = {
                        filename,
                    }
                    files.push(file)
                    if (/\.mdx?$/i.test(filename)) {
                        try {
                            const content = await workspace.readText(filename)
                            const fm = await parsers.frontmatter(content)
                            if (fm) file.frontmatter = fm
                        } catch (e) {}
                    }
                }
                const preview = files
                    .map((f) =>
                        [f.filename, f.frontmatter?.title]
                            .filter((p) => !!p)
                            .join(", ")
                    )
                    .join("\n")
                context.log(preview)
                return YAML.stringify(files) + suffix
            } else {
                const filenames = res.map((f) => f.filename).join("\n") + suffix
                context.log(filenames)
                return filenames
            }
        }
    )
}

system({
    title: "File Read File",
    description: "Function to read file content as text.",
})

export default function (ctx: ChatGenerationContext) {
    const { defTool } = ctx

    defTool(
        "fs_read_file",
        "Reads a file as text from the file system. Returns undefined if the file does not exist.",
        {
            type: "object",
            properties: {
                filename: {
                    type: "string",
                    description:
                        "Path of the file to load, relative to the workspace.",
                },
                line: {
                    type: "integer",
                    description:
                        "Line number (starting at 1) to read with a few lines before and after.",
                },
                line_start: {
                    type: "integer",
                    description:
                        "Line number (starting at 1) to start reading from.",
                },
                line_end: {
                    type: "integer",
                    description:
                        "Line number (starting at 1) to end reading at.",
                },
                line_numbers: {
                    type: "boolean",
                    description:
                        "Whether to include line numbers in the output.",
                },
            },
            required: ["filename"],
        },
        async (args) => {
            let {
                filename,
                line,
                line_start,
                line_end,
                line_numbers,
                context,
            } = args
            if (!filename) return "<MISSING>filename</MISSING>"
            if (!isNaN(line)) {
                line_start = Math.max(1, line - 5)
                line_end = Math.max(1, line + 5)
            }
            const hasRange = !isNaN(line_start) && !isNaN(line_end)
            if (hasRange) {
                line_start = Math.max(1, line_start)
                line_end = Math.max(1, line_end)
            }
            let content
            try {
                context.log(
                    `cat ${filename}${hasRange ? ` | sed -n '${line_start},${line_end}p'` : ""}`
                )
                const res = await workspace.readText(filename)
                content = res.content ?? ""
            } catch (e) {
                return "<FILE_NOT_FOUND>"
            }
            if (line_numbers || hasRange) {
                const lines = content.split("\n")
                content = lines
                    .map((line, i) => `[${i + 1}] ${line}`)
                    .join("\n")
            }
            if (!isNaN(line_start) && !isNaN(line_end)) {
                const lines = content.split("\n")
                content = lines.slice(line_start, line_end).join("\n")
            }
            return content
        },
        {
            maxTokens: 10000,
        }
    )
}

system({
    title: "git read operations",
    description: "Tools to query a git repository.",
    parameters: {
        cwd: {
            type: "string",
            description: "Current working directory",
            required: false,
        },
    },
})

export default function (ctx: ChatGenerationContext) {
    const { env, defTool } = ctx
    const { vars } = env
    const cwd = vars["system.git.cwd"]
    const client = cwd ? git.client(cwd) : git

    defTool(
        "git_branch_default",
        "Gets the default branch using client.",
        {},
        async () => {
            return await client.defaultBranch()
        }
    )

    defTool(
        "git_branch_current",
        "Gets the current branch using client.",
        {},
        async () => {
            return await client.branch()
        }
    )

    defTool(
        "git_branch_list",
        "List all branches using client.",
        {},
        async () => {
            return await client.exec("branch")
        }
    )

    defTool(
        "git_list_commits",
        "Generates a history of commits using the git log command.",
        {
            type: "object",
            properties: {
                base: {
                    type: "string",
                    description: "Base branch to compare against.",
                },
                head: {
                    type: "string",
                    description: "Head branch to compare",
                },
                count: {
                    type: "number",
                    description: "Number of commits to return",
                },
                author: {
                    type: "string",
                    description: "Author to filter by",
                },
                until: {
                    type: "string",
                    description:
                        "Display commits until the given date. Formatted yyyy-mm-dd",
                },
                after: {
                    type: "string",
                    description:
                        "Display commits after the given date. Formatted yyyy-mm-dd",
                },
                paths: {
                    type: "array",
                    description: "Paths to compare",
                    items: {
                        type: "string",
                        description: "File path or wildcard supported by git",
                    },
                },
                excludedPaths: {
                    type: "array",
                    description: "Paths to exclude",
                    items: {
                        type: "string",
                        description: "File path or wildcard supported by git",
                    },
                },
            },
        },
        async (args) => {
            const {
                context,
                base,
                head,
                paths,
                excludedPaths,
                count,
                author,
                until,
                after,
            } = args
            const commits = await client.log({
                base,
                head,
                author,
                paths,
                until,
                after,
                excludedPaths,
                count,
            })
            const res = commits
                .map(({ sha, date, message }) => `${sha} ${date} ${message}`)
                .join("\n")
            context.debug(res)
            return res
        }
    )

    defTool(
        "git_status",
        "Generates a status of the repository using client.",
        {},
        async () => {
            return await client.exec(["status", "--porcelain"])
        }
    )

    defTool("git_last_tag", "Gets the last tag using client.", {}, async () => {
        return await client.lastTag()
    })
}

system({
    title: "git diff",
    description: "Tools to query a git repository.",
    parameters: {
        cwd: {
            type: "string",
            description: "Current working directory",
            required: false,
        },
    },
})

export default function (ctx: ChatGenerationContext) {
    const { env, defTool } = ctx
    const { vars } = env
    const cwd = vars["system.git_diff.cwd"]
    const client = cwd ? git.client(cwd) : git

    defTool(
        "git_diff",
        "Computes file diffs using the git diff command. If the diff is too large, it returns the list of modified/added files.",
        {
            type: "object",
            properties: {
                base: {
                    type: "string",
                    description:
                        "Base branch, ref, commit sha to compare against.",
                },
                head: {
                    type: "string",
                    description:
                        "Head branch, ref, commit sha to compare. Use 'HEAD' to compare against the current branch.",
                },
                staged: {
                    type: "boolean",
                    description: "Compare staged changes",
                },
                nameOnly: {
                    type: "boolean",
                    description: "Show only file names",
                },
                paths: {
                    type: "array",
                    description: "Paths to compare",
                    items: {
                        type: "string",
                        description: "File path or wildcard supported by git",
                    },
                },
                excludedPaths: {
                    type: "array",
                    description: "Paths to exclude",
                    items: {
                        type: "string",
                        description: "File path or wildcard supported by git",
                    },
                },
            },
        },
        async (args) => {
            const { context, ...rest } = args
            const res = await client.diff({
                llmify: true,
                ...rest,
            })
            return res
        },
        {
            maxTokens: 20000,
        }
    )
}