@henryqw/pi-task-models 0.4.0 → 0.5.1

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/index.d.ts CHANGED
@@ -48,7 +48,11 @@ export declare function resolveAvailableModel(models: readonly AvailableModel[],
48
48
  export declare function supportedThinkingLevels(model: AvailableModel): ThinkingLevel[];
49
49
  export declare function availableTaskModels(ctx: ExtensionContext): AvailableModel[];
50
50
  export declare function taskThinkingLevels(ctx: ExtensionContext, model: AvailableModel): ThinkingLevel[];
51
- export declare function resolveTaskModelRoute(ctx: ExtensionContext, route: TaskModelRoute): ResolvedTaskRoute | undefined;
51
+ export declare function rememberedThinkingLevel(model: {
52
+ provider: string;
53
+ id: string;
54
+ } | string, agentDir?: string): ThinkingLevel | undefined;
55
+ export declare function resolveTaskModelRoute(ctx: ExtensionContext, route: TaskModelRoute, agentDir?: string): ResolvedTaskRoute | undefined;
52
56
  export declare function resolveConfiguredTaskRoute(ctx: ExtensionContext, task: string, agentDir?: string): ResolvedTaskRoute;
53
57
  export declare function orderedProfileRoutes(profile: TaskModelProfile): TaskModelRoute[];
54
58
  export declare function activeTaskPackages(pi: Pick<ExtensionAPI, "getCommands" | "getAllTools">, tasks?: Readonly<Record<string, ProfileName>>): ActiveTaskPackage[];
package/dist/index.js CHANGED
@@ -14,6 +14,8 @@ export const DEFAULT_TASK_ASSIGNMENTS = {
14
14
  };
15
15
  const CODEX_ALIAS = /^openai-codex-(?:[2-9]|[1-9]\d+)$/;
16
16
  const CONFIG_FILE = "pi-task-models.json";
17
+ // Owned by @henryqw/pi-model-thinking; keys use the same canonical provider/model form.
18
+ const MODEL_THINKING_CONFIG_FILE = "pi-model-thinking.json";
17
19
  const defaultTaskAssignments = () => ({ ...DEFAULT_TASK_ASSIGNMENTS });
18
20
  export const configPath = (agentDir = getAgentDir()) => join(agentDir, "config", CONFIG_FILE);
19
21
  function isCodexProvider(provider) {
@@ -191,11 +193,30 @@ export function taskThinkingLevels(ctx, model) {
191
193
  return supported;
192
194
  return supported.includes(pinned) ? [pinned] : [];
193
195
  }
194
- export function resolveTaskModelRoute(ctx, route) {
196
+ // pi-model-thinking's config is untrusted user data; keep only recognized levels for this model.
197
+ export function rememberedThinkingLevel(model, agentDir = getAgentDir()) {
198
+ try {
199
+ const value = JSON.parse(readFileSync(join(agentDir, "config", MODEL_THINKING_CONFIG_FILE), "utf8"));
200
+ if (!value || typeof value !== "object" || Array.isArray(value))
201
+ return undefined;
202
+ const level = value[canonicalModelReference(model)];
203
+ return isThinkingLevel(level) ? level : undefined;
204
+ }
205
+ catch {
206
+ return undefined;
207
+ }
208
+ }
209
+ export function resolveTaskModelRoute(ctx, route, agentDir = getAgentDir()) {
195
210
  const model = resolveAvailableModel(availableTaskModels(ctx), route.model, ctx.model?.provider);
196
- return model && taskThinkingLevels(ctx, model).includes(route.thinkingLevel)
197
- ? { model, thinkingLevel: route.thinkingLevel }
198
- : undefined;
211
+ if (!model)
212
+ return undefined;
213
+ const levels = taskThinkingLevels(ctx, model);
214
+ // Remembered per-model thinking (pi-model-thinking) wins over the profile level when supported.
215
+ const remembered = rememberedThinkingLevel(model, agentDir);
216
+ const thinkingLevel = remembered && levels.includes(remembered) ? remembered : route.thinkingLevel;
217
+ if (!levels.includes(thinkingLevel))
218
+ return undefined;
219
+ return { model, thinkingLevel };
199
220
  }
200
221
  export function resolveConfiguredTaskRoute(ctx, task, agentDir = getAgentDir()) {
201
222
  let config;
@@ -212,7 +233,7 @@ export function resolveConfiguredTaskRoute(ctx, task, agentDir = getAgentDir())
212
233
  if (!profile)
213
234
  throw new Error(`Task ${task} profile ${profileName} is not configured. Run /task-models.`);
214
235
  for (const route of orderedProfileRoutes(profile)) {
215
- const resolved = resolveTaskModelRoute(ctx, route);
236
+ const resolved = resolveTaskModelRoute(ctx, route, agentDir);
216
237
  if (resolved)
217
238
  return resolved;
218
239
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@henryqw/pi-task-models",
3
- "version": "0.4.0",
3
+ "version": "0.5.1",
4
4
  "description": "Shared task model profiles and routing for HenryQW Pi extensions.",
5
5
  "keywords": [
6
6
  "pi-package",