system.ts 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. import { Ripgrep } from "../file/ripgrep"
  2. import { Instance } from "../project/instance"
  3. import PROMPT_ANTHROPIC from "./prompt/anthropic.txt"
  4. import PROMPT_ANTHROPIC_WITHOUT_TODO from "./prompt/qwen.txt"
  5. import PROMPT_BEAST from "./prompt/beast.txt"
  6. import PROMPT_GEMINI from "./prompt/gemini.txt"
  7. import PROMPT_CODEX from "./prompt/codex_header.txt"
  8. import type { Provider } from "@/provider/provider"
  9. export namespace SystemPrompt {
  10. export function instructions() {
  11. return PROMPT_CODEX.trim()
  12. }
  13. export function provider(model: Provider.Model) {
  14. if (model.api.id.includes("gpt-5")) return [PROMPT_CODEX]
  15. if (model.api.id.includes("gpt-") || model.api.id.includes("o1") || model.api.id.includes("o3"))
  16. return [PROMPT_BEAST]
  17. if (model.api.id.includes("gemini-")) return [PROMPT_GEMINI]
  18. if (model.api.id.includes("claude")) return [PROMPT_ANTHROPIC]
  19. return [PROMPT_ANTHROPIC_WITHOUT_TODO]
  20. }
  21. export async function environment(model: Provider.Model) {
  22. const project = Instance.project
  23. return [
  24. [
  25. `You are powered by the model named ${model.api.id}. The exact model ID is ${model.providerID}/${model.api.id}`,
  26. `Here is some useful information about the environment you are running in:`,
  27. `<env>`,
  28. ` Working directory: ${Instance.directory}`,
  29. ` Is directory a git repo: ${project.vcs === "git" ? "yes" : "no"}`,
  30. ` Platform: ${process.platform}`,
  31. ` Today's date: ${new Date().toDateString()}`,
  32. `</env>`,
  33. `<directories>`,
  34. ` ${
  35. project.vcs === "git" && false
  36. ? await Ripgrep.tree({
  37. cwd: Instance.directory,
  38. limit: 50,
  39. })
  40. : ""
  41. }`,
  42. `</directories>`,
  43. ].join("\n"),
  44. ]
  45. }
  46. }