refactor(core): select system prompts through plugins (#37181)
This commit is contained in:
parent
b5177adb5b
commit
720f062cd0
29 changed files with 447 additions and 263 deletions
47
packages/core/src/plugin/system-prompt.ts
Normal file
47
packages/core/src/plugin/system-prompt.ts
Normal file
|
|
@ -0,0 +1,47 @@
|
|||
export * as SystemPromptPlugin from "./system-prompt"
|
||||
|
||||
import { define } from "@opencode-ai/plugin/v2/effect/plugin"
|
||||
import { Effect } from "effect"
|
||||
|
||||
import PROMPT_ANTHROPIC from "./system-prompt/anthropic.txt"
|
||||
import PROMPT_CODEX from "./system-prompt/codex.txt"
|
||||
import PROMPT_GEMINI from "./system-prompt/gemini.txt"
|
||||
import PROMPT_GPT from "./system-prompt/gpt.txt"
|
||||
import PROMPT_KIMI from "./system-prompt/kimi.txt"
|
||||
import PROMPT_META from "./system-prompt/meta.txt"
|
||||
import PROMPT_TRINITY from "./system-prompt/trinity.txt"
|
||||
|
||||
export const OpenAIPlugin = make("openai", (id) => {
|
||||
if (id.includes("gpt")) {
|
||||
if (id.includes("codex")) return PROMPT_CODEX
|
||||
return PROMPT_GPT
|
||||
}
|
||||
if (id.includes("o1") || id.includes("o3")) return PROMPT_GPT
|
||||
})
|
||||
|
||||
export const GooglePlugin = make("google", (id) => (id.includes("gemini-") ? PROMPT_GEMINI : undefined))
|
||||
export const AnthropicPlugin = make("anthropic", (id) => (id.includes("claude") ? PROMPT_ANTHROPIC : undefined))
|
||||
export const KimiPlugin = make("kimi", (id) => (id.includes("kimi") ? PROMPT_KIMI : undefined))
|
||||
export const ArceePlugin = make("arcee", (id) => (id.includes("trinity") ? PROMPT_TRINITY : undefined))
|
||||
export const MetaPlugin = make("meta", (id) => (id.includes("muse-spark") ? PROMPT_META : undefined))
|
||||
|
||||
export const Plugins = [OpenAIPlugin, GooglePlugin, AnthropicPlugin, KimiPlugin, ArceePlugin, MetaPlugin] as const
|
||||
|
||||
function make(id: string, select: (modelID: string) => string | undefined) {
|
||||
return define({
|
||||
id: `opencode.system-prompt.${id}`,
|
||||
effect: Effect.fn(`SystemPromptPlugin.${id}`)(function* (ctx) {
|
||||
yield* ctx.session.hook("context", (event) =>
|
||||
Effect.gen(function* () {
|
||||
if ((yield* ctx.agent.get(event.agent))?.system) return
|
||||
const system = event.system[0]
|
||||
if (!system) return
|
||||
const model = yield* ctx.catalog.model.get(event.model.providerID, event.model.id)
|
||||
const prompt = select(`${model?.modelID ?? event.model.id} ${model?.family ?? ""}`.toLowerCase())
|
||||
if (!prompt) return
|
||||
event.system[0] = { ...system, text: prompt }
|
||||
}),
|
||||
)
|
||||
}),
|
||||
})
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue