@oh-my-pi/pi-coding-agent 13.11.1 → 13.12.0

package/src/utils/title-generator.ts

@@ -9,7 +9,6 @@ import type { ModelRegistry } from "../config/model-registry";
 import { resolveModelRoleValue } from "../config/model-resolver";
 import { renderPromptTemplate } from "../config/prompt-templates";
 import type { Settings } from "../config/settings";
-import MODEL_PRIO from "../priority.json" with { type: "json" };
 import titleSystemPrompt from "../prompts/system/title-system.md" with { type: "text" };
 import { toReasoningEffort } from "../thinking";
 
@@ -17,45 +16,28 @@ const TITLE_SYSTEM_PROMPT = renderPromptTemplate(titleSystemPrompt);
 
 const MAX_INPUT_CHARS = 2000;
 
-function getTitleModelCandidates(
+function getTitleModel(
 	registry: ModelRegistry,
 	settings: Settings,
-): Array<{ model: Model<Api>; thinkingLevel?: ThinkingLevel }> {
+	currentModel?: Model<Api>,
+): { model: Model<Api>; thinkingLevel?: ThinkingLevel } | undefined {
 	const availableModels = registry.getAvailable();
-	if (availableModels.length === 0) return [];
-
-	const candidates: Array<{ model: Model<Api>; thinkingLevel?: ThinkingLevel }> = [];
-	const addCandidate = (model?: Model<Api>, thinkingLevel?: ThinkingLevel): void => {
-		if (!model) return;
-		const exists = candidates.some(
-			candidate => candidate.model.provider === model.provider && candidate.model.id === model.id,
-		);
-		if (!exists) {
-			candidates.push({ model, thinkingLevel });
-		}
-	};
+	if (availableModels.length === 0) return undefined;
 
 	const matchPreferences = { usageOrder: settings.getStorage()?.getModelUsageOrder() };
 	const configuredSmol = resolveModelRoleValue(settings.getModelRole("smol"), availableModels, {
 		settings,
 		matchPreferences,
 	});
-	addCandidate(configuredSmol.model, configuredSmol.thinkingLevel);
-
-	for (const pattern of MODEL_PRIO.smol) {
-		const needle = pattern.toLowerCase();
-		const exactMatch = availableModels.find(model => model.id.toLowerCase() === needle);
-		addCandidate(exactMatch);
-
-		const fuzzyMatch = availableModels.find(model => model.id.toLowerCase().includes(needle));
-		addCandidate(fuzzyMatch);
+	if (configuredSmol.model) {
+		return { model: configuredSmol.model, thinkingLevel: configuredSmol.thinkingLevel };
 	}
 
-	for (const model of availableModels) {
-		addCandidate(model);
+	if (currentModel) {
+		return { model: currentModel };
 	}
 
-	return candidates;
+	return undefined;
 }
 
 /**
@@ -71,10 +53,11 @@ export async function generateSessionTitle(
 	registry: ModelRegistry,
 	settings: Settings,
 	sessionId?: string,
+	currentModel?: Model<Api>,
 ): Promise<string | null> {
-	const candidates = getTitleModelCandidates(registry, settings);
-	if (candidates.length === 0) {
-		logger.debug("title-generator: no smol model found");
+	const candidate = getTitleModel(registry, settings, currentModel);
+	if (!candidate) {
+		logger.debug("title-generator: no title model found");
 		return null;
 	}
 
@@ -85,78 +68,73 @@ export async function generateSessionTitle(
 ${truncatedMessage}
 </user-message>`;
 
-	for (const candidate of candidates) {
-		const apiKey = await registry.getApiKey(candidate.model, sessionId);
-		if (!apiKey) {
-			logger.debug("title-generator: no API key for model", {
-				provider: candidate.model.provider,
-				id: candidate.model.id,
-			});
-			continue;
-		}
-
-		const request = {
-			model: `${candidate.model.provider}/${candidate.model.id}`,
-			systemPrompt: TITLE_SYSTEM_PROMPT,
-			userMessage,
-			maxTokens: 30,
-		};
-		logger.debug("title-generator: request", request);
-
-		try {
-			const response = await completeSimple(
-				candidate.model,
-				{
-					systemPrompt: request.systemPrompt,
-					messages: [{ role: "user", content: request.userMessage, timestamp: Date.now() }],
-				},
-				{
-					apiKey,
-					maxTokens: 30,
-					reasoning: toReasoningEffort(candidate.thinkingLevel),
-				},
-			);
-
-			if (response.stopReason === "error") {
-				logger.debug("title-generator: response error", {
-					model: request.model,
-					stopReason: response.stopReason,
-					errorMessage: response.errorMessage,
-				});
-				continue;
-			}
+	const apiKey = await registry.getApiKey(candidate.model, sessionId);
+	if (!apiKey) {
+		logger.debug("title-generator: no API key for smol model", {
+			provider: candidate.model.provider,
+			id: candidate.model.id,
+		});
+		return null;
+	}
 
-			// Extract title from response text content
-			let title = "";
-			for (const content of response.content) {
-				if (content.type === "text") {
-					title += content.text;
-				}
-			}
-			title = title.trim();
+	const request = {
+		model: `${candidate.model.provider}/${candidate.model.id}`,
+		systemPrompt: TITLE_SYSTEM_PROMPT,
+		userMessage,
+		maxTokens: 30,
+	};
+	logger.debug("title-generator: request", request);
+
+	try {
+		const response = await completeSimple(
+			candidate.model,
+			{
+				systemPrompt: request.systemPrompt,
+				messages: [{ role: "user", content: request.userMessage, timestamp: Date.now() }],
+			},
+			{
+				apiKey,
+				maxTokens: 30,
+				reasoning: toReasoningEffort(candidate.thinkingLevel),
+			},
+		);
 
-			logger.debug("title-generator: response", {
+		if (response.stopReason === "error") {
+			logger.debug("title-generator: response error", {
 				model: request.model,
-				title,
-				usage: response.usage,
 				stopReason: response.stopReason,
+				errorMessage: response.errorMessage,
 			});
+			return null;
+		}
 
-			if (!title) {
-				continue;
+		let title = "";
+		for (const content of response.content) {
+			if (content.type === "text") {
+				title += content.text;
 			}
+		}
+		title = title.trim();
 
-			// Clean up: remove quotes, trailing punctuation
-			return title.replace(/^["']|["']$/g, "").replace(/[.!?]$/, "");
-		} catch (err) {
-			logger.debug("title-generator: error", {
-				model: request.model,
-				error: err instanceof Error ? err.message : String(err),
-			});
+		logger.debug("title-generator: response", {
+			model: request.model,
+			title,
+			usage: response.usage,
+			stopReason: response.stopReason,
+		});
+
+		if (!title) {
+			return null;
 		}
-	}
 
-	return null;
+		return title.replace(/^["']|["']$/g, "").replace(/[.!?]$/, "");
+	} catch (err) {
+		logger.debug("title-generator: error", {
+			model: request.model,
+			error: err instanceof Error ? err.message : String(err),
+		});
+		return null;
+	}
 }
 
 /**

package/src/utils/tools-manager.ts

@@ -4,7 +4,7 @@ import * as path from "node:path";
 import { APP_NAME, getToolsDir, logger, ptree, TempDir } from "@oh-my-pi/pi-utils";
 
 const TOOLS_DIR = getToolsDir();
-const TOOL_DOWNLOAD_TIMEOUT_MS = 15000;
+const TOOL_DOWNLOAD_TIMEOUT_MS = 120_000;
 const TOOL_METADATA_TIMEOUT_MS = 5000;
 
 interface ToolConfig {

package/src/web/scrapers/index.ts

@@ -85,17 +85,16 @@ export type { RenderResult, SpecialHandler } from "./types";
 export {
 	fetchGitHubApi,
 	handleArtifactHub,
-	handleDocsRs,
 	handleArxiv,
 	handleAur,
 	handleBiorxiv,
 	handleBluesky,
 	handleBrew,
 	handleCheatSh,
-	handleCisaKev,
 	handleChocolatey,
-	handleClojars,
 	handleChooseALicense,
+	handleCisaKev,
+	handleClojars,
 	handleCoinGecko,
 	handleCratesIo,
 	handleCrossref,
@@ -103,9 +102,10 @@ export {
 	handleDiscogs,
 	handleDiscourse,
 	handleDockerHub,
+	handleDocsRs,
 	handleFdroid,
-	handleFlathub,
 	handleFirefoxAddons,
+	handleFlathub,
 	handleGitHub,
 	handleGitHubGist,
 	handleGitLab,
@@ -129,8 +129,8 @@ export {
 	handleOllama,
 	handleOpenCorporates,
 	handleOpenLibrary,
-	handleOrcid,
 	handleOpenVsx,
+	handleOrcid,
 	handleOsv,
 	handlePackagist,
 	handlePubDev,
@@ -142,13 +142,13 @@ export {
 	handleRepology,
 	handleRfc,
 	handleRubyGems,
-	handleSecEdgar,
 	handleSearchcode,
+	handleSecEdgar,
 	handleSemanticScholar,
 	handleSnapcraft,
 	handleSourcegraph,
-	handleSpotify,
 	handleSpdx,
+	handleSpotify,
 	handleStackOverflow,
 	handleTerraform,
 	handleTldr,

package/src/web/scrapers/utils.ts

@@ -1,4 +1,5 @@
 import { isRecord, ptree, TempDir } from "@oh-my-pi/pi-utils";
+
 export { isRecord };
 
 import { ToolAbortError } from "../../tools/tool-errors";