@henryqw/pi-task-models 0.4.1 → 0.6.0

package/dist/index.d.ts

@@ -48,8 +48,12 @@ export declare function resolveAvailableModel(models: readonly AvailableModel[],
 export declare function supportedThinkingLevels(model: AvailableModel): ThinkingLevel[];
 export declare function availableTaskModels(ctx: ExtensionContext): AvailableModel[];
 export declare function taskThinkingLevels(ctx: ExtensionContext, model: AvailableModel): ThinkingLevel[];
-export declare function resolveTaskModelRoute(ctx: ExtensionContext, route: TaskModelRoute): ResolvedTaskRoute | undefined;
-export declare function resolveConfiguredTaskRoute(ctx: ExtensionContext, task: string, agentDir?: string): ResolvedTaskRoute;
+export declare function rememberedThinkingLevel(model: {
+    provider: string;
+    id: string;
+} | string, agentDir?: string): ThinkingLevel | undefined;
+export declare function resolveTaskModelRoute(ctx: ExtensionContext, route: TaskModelRoute, agentDir?: string, thinking?: ThinkingLevel): ResolvedTaskRoute | undefined;
+export declare function resolveConfiguredTaskRoute(ctx: ExtensionContext, task: string, agentDir?: string, thinking?: ThinkingLevel): ResolvedTaskRoute;
 export declare function orderedProfileRoutes(profile: TaskModelProfile): TaskModelRoute[];
 export declare function activeTaskPackages(pi: Pick<ExtensionAPI, "getCommands" | "getAllTools">, tasks?: Readonly<Record<string, ProfileName>>): ActiveTaskPackage[];
 export declare function createTaskModelsExtension(pi: ExtensionAPI, options?: {

package/dist/index.js

@@ -14,6 +14,8 @@ export const DEFAULT_TASK_ASSIGNMENTS = {
 };
 const CODEX_ALIAS = /^openai-codex-(?:[2-9]|[1-9]\d+)$/;
 const CONFIG_FILE = "pi-task-models.json";
+// Owned by @henryqw/pi-model-thinking; keys use the same canonical provider/model form.
+const MODEL_THINKING_CONFIG_FILE = "pi-model-thinking.json";
 const defaultTaskAssignments = () => ({ ...DEFAULT_TASK_ASSIGNMENTS });
 export const configPath = (agentDir = getAgentDir()) => join(agentDir, "config", CONFIG_FILE);
 function isCodexProvider(provider) {
@@ -191,13 +193,37 @@ export function taskThinkingLevels(ctx, model) {
         return supported;
     return supported.includes(pinned) ? [pinned] : [];
 }
-export function resolveTaskModelRoute(ctx, route) {
+// pi-model-thinking's config is untrusted user data; keep only recognized levels for this model.
+export function rememberedThinkingLevel(model, agentDir = getAgentDir()) {
+    try {
+        const value = JSON.parse(readFileSync(join(agentDir, "config", MODEL_THINKING_CONFIG_FILE), "utf8"));
+        if (!value || typeof value !== "object" || Array.isArray(value))
+            return undefined;
+        const level = value[canonicalModelReference(model)];
+        return isThinkingLevel(level) ? level : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+export function resolveTaskModelRoute(ctx, route, agentDir = getAgentDir(), thinking) {
     const model = resolveAvailableModel(availableTaskModels(ctx), route.model, ctx.model?.provider);
-    return model && taskThinkingLevels(ctx, model).includes(route.thinkingLevel)
-        ? { model, thinkingLevel: route.thinkingLevel }
-        : undefined;
+    if (!model)
+        return undefined;
+    const levels = taskThinkingLevels(ctx, model);
+    // An explicit requested level wins over remembered and profile levels; a model
+    // that cannot honor it is skipped so later fallback routes get considered.
+    if (thinking !== undefined) {
+        return levels.includes(thinking) ? { model, thinkingLevel: thinking } : undefined;
+    }
+    // Remembered per-model thinking (pi-model-thinking) wins over the profile level when supported.
+    const remembered = rememberedThinkingLevel(model, agentDir);
+    const thinkingLevel = remembered && levels.includes(remembered) ? remembered : route.thinkingLevel;
+    if (!levels.includes(thinkingLevel))
+        return undefined;
+    return { model, thinkingLevel };
 }
-export function resolveConfiguredTaskRoute(ctx, task, agentDir = getAgentDir()) {
+export function resolveConfiguredTaskRoute(ctx, task, agentDir = getAgentDir(), thinking) {
     let config;
     try {
         config = readTaskModelsConfig(agentDir);
@@ -212,11 +238,11 @@ export function resolveConfiguredTaskRoute(ctx, task, agentDir = getAgentDir())
     if (!profile)
         throw new Error(`Task ${task} profile ${profileName} is not configured. Run /task-models.`);
     for (const route of orderedProfileRoutes(profile)) {
-        const resolved = resolveTaskModelRoute(ctx, route);
+        const resolved = resolveTaskModelRoute(ctx, route, agentDir, thinking);
         if (resolved)
             return resolved;
     }
-    throw new Error(`Task ${task} profile ${profileName} has no available route. Run /task-models.`);
+    throw new Error(`Task ${task} profile ${profileName} has no available route${thinking ? ` supporting thinking ${thinking}` : ""}. Run /task-models.`);
 }
 export function orderedProfileRoutes(profile) {
     return profile.fallback ? [profile.primary, profile.fallback] : [profile.primary];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@henryqw/pi-task-models",
-  "version": "0.4.1",
+  "version": "0.6.0",
   "description": "Shared task model profiles and routing for HenryQW Pi extensions.",
   "keywords": [
     "pi-package",
@@ -30,7 +30,7 @@
   "scripts": {
     "build": "tsc --project tsconfig.build.json",
     "test": "npm run build && node --test test/*.test.ts",
-    "typecheck": "tsc --noEmit --allowImportingTsExtensions --target ES2022 --module NodeNext --moduleResolution NodeNext --types node --skipLibCheck src/*.ts extensions/*.ts",
+    "typecheck": "tsc --noEmit --allowImportingTsExtensions --target ES2022 --module NodeNext --moduleResolution NodeNext --types node --skipLibCheck src/*.ts extensions/*.ts test/*.ts",
     "prepack": "npm run build",
     "pack:check": "npm pack --dry-run"
   },