@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 +5 -1
- package/dist/index.js +26 -5
- package/package.json +1 -1
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
|
|
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
|
-
|
|
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
|
-
|
|
197
|
-
|
|
198
|
-
|
|
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
|
}
|