@mtayfur/opencode-prompt-enhancer 1.0.1 → 1.0.2

package/README.md

@@ -49,23 +49,24 @@ OpenCode `>=1.18.12 <2` is required.
 
 Restart OpenCode after changing the plugin configuration.
 
-## Model override
+## Model
 
-By default, the enhancer uses OpenCode's `small_model`, falling back to `model`. To use a different model only for prompt enhancement, configure the plugin as a tuple and pass a fully qualified `provider/model` ID:
+The enhancer uses OpenCode's `small_model` when `model` is not set.
 
 ```jsonc
 {
   "plugin": [
     [
       "@mtayfur/opencode-prompt-enhancer@latest",
-      { "model": "anthropic/claude-sonnet-4-6" }
+      {
+        "model": "anthropic/claude-sonnet-4-6",
+        "variant": "high"
+      }
     ]
   ]
 }
 ```
 
-The same options object can be added to a local plugin entry. Restart OpenCode after changing the model.
-
 ## Use
 
 1. Open OpenCode in a workspace.

package/dist/plugins/prompt-enhancer.js

@@ -79,16 +79,18 @@ function indentContextContinuation(text, indentation) {
 }
 function resolveEnhancerModel(api, options) {
   const modelOverride = typeof options?.model === "string" ? options.model : undefined;
-  if (modelOverride?.trim()) {
-    const override = parseModelString(modelOverride);
-    if (override) return override;
-    api.ui.toast({
-      variant: "warning",
-      title: TOAST_TITLE,
-      message: `Invalid model override ${JSON.stringify(modelOverride)}; expected "provider/model". Using default model.`
-    });
+  const model = modelOverride ? parseModelString(modelOverride) : parseModelString(api.state.config.small_model);
+  if (modelOverride && !model) {
+    throw new Error(`Invalid model override ${JSON.stringify(modelOverride)}; expected "provider/model".`);
   }
-  return parseModelString(api.state.config.small_model || api.state.config.model);
+  if (!model) throw new Error("OpenCode small_model is not configured.");
+  const variant = typeof options?.variant === "string" && options.variant.trim() ? options.variant.trim() : undefined;
+  return {
+    model,
+    ...(variant ? {
+      variant
+    } : {})
+  };
 }
 function clonePromptInfo(prompt) {
   return {
@@ -290,7 +292,7 @@ function gatherContext(api) {
 }
 async function enhanceWithModel(api, options, input, signal) {
   const directory = api.state.path.directory;
-  const model = resolveEnhancerModel(api, options);
+  const selection = resolveEnhancerModel(api, options);
   const context = gatherContext(api);
   const userMessage = [`<CONTEXT>\n${context}\n</CONTEXT>`, `<DRAFT>\n${input}\n</DRAFT>`].join("\n\n");
   const created = await api.client.session.create({
@@ -311,7 +313,10 @@ async function enhanceWithModel(api, options, input, signal) {
     const response = await withRequestTimeout(signal, ENHANCEMENT_TIMEOUT_MS, requestSignal => api.client.session.prompt({
       sessionID: tempSessionID,
       directory,
-      model,
+      model: selection.model,
+      ...(selection.variant ? {
+        variant: selection.variant
+      } : {}),
       system: ENHANCER_SYSTEM_PROMPT,
       parts: [{
         type: "text",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mtayfur/opencode-prompt-enhancer",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "OpenCode plugin that rewrites rough drafts into stronger prompts.",
   "repository": {
     "type": "git",

package/plugins/prompt-enhancer.tsx

@@ -160,20 +160,21 @@ function indentContextContinuation(text: string, indentation: string): string {
   return text.replaceAll("\n", `\n${indentation}`)
 }
 
-function resolveEnhancerModel(api: Api, options: PluginOptions | undefined): ModelRef | undefined {
+function resolveEnhancerModel(
+  api: Api,
+  options: PluginOptions | undefined,
+): { model: ModelRef, variant?: string } {
   const modelOverride = typeof options?.model === "string" ? options.model : undefined
-  if (modelOverride?.trim()) {
-    const override = parseModelString(modelOverride)
-    if (override) return override
-
-    api.ui.toast({
-      variant: "warning",
-      title: TOAST_TITLE,
-      message: `Invalid model override ${JSON.stringify(modelOverride)}; expected "provider/model". Using default model.`,
-    })
+  const model = modelOverride ? parseModelString(modelOverride) : parseModelString(api.state.config.small_model)
+  if (modelOverride && !model) {
+    throw new Error(`Invalid model override ${JSON.stringify(modelOverride)}; expected "provider/model".`)
   }
+  if (!model) throw new Error("OpenCode small_model is not configured.")
 
-  return parseModelString(api.state.config.small_model || api.state.config.model)
+  const variant = typeof options?.variant === "string" && options.variant.trim()
+    ? options.variant.trim()
+    : undefined
+  return { model, ...(variant ? { variant } : {}) }
 }
 
 function clonePromptInfo(prompt: TuiPromptInfo): TuiPromptInfo {
@@ -448,7 +449,7 @@ async function enhanceWithModel(
   signal: AbortSignal,
 ): Promise<string> {
   const directory = api.state.path.directory
-  const model = resolveEnhancerModel(api, options)
+  const selection = resolveEnhancerModel(api, options)
   const context = gatherContext(api)
   const userMessage = [
     `<CONTEXT>\n${context}\n</CONTEXT>`,
@@ -475,7 +476,8 @@ async function enhanceWithModel(
         {
           sessionID: tempSessionID,
           directory,
-          model,
+          model: selection.model,
+          ...(selection.variant ? { variant: selection.variant } : {}),
           system: ENHANCER_SYSTEM_PROMPT,
           parts: [
             {