feat: enhance text generation logic to conditionally trigger web search and manage tool preferences

src/lib/server/textGeneration/index.ts

@@ -21,6 +21,7 @@ import type { TextGenerationContext } from "./types";
 import type { ToolResult } from "$lib/types/Tool";
 import { toolHasName } from "../tools/utils";
 import directlyAnswer from "../tools/directlyAnswer";
+import { webSearchToolId, fetchUrlToolId } from "$lib/utils/toolIds";
 
 async function* keepAlive(done: AbortSignal): AsyncGenerator<MessageUpdate, undefined, undefined> {
 	while (!done.aborted) {
@@ -59,12 +60,28 @@ async function* textGenerationWithoutTitle(
 
 	let webSearchResult: WebSearch | undefined;
 
+	// Check if web search should be triggered
+	const hasWebSearchTool = toolsPreference?.includes(webSearchToolId) || false;
+	const shouldRunWebSearch =
+		!isContinue &&
+		((webSearch && !conv.assistantId) || // Legacy web search flag
+			assistantHasWebSearch(assistant) || // Assistant has web search enabled
+			(hasWebSearchTool && !conv.assistantId)); // Modern tool-based web search
+
 	// run websearch if:
 	// - it's not continuing a previous message
 	// - AND the model doesn't support tools and websearch is selected
 	// - OR the assistant has websearch enabled (no tools for assistants for now)
-	if (!isContinue && ((webSearch && !conv.assistantId) || assistantHasWebSearch(assistant))) {
+	if (shouldRunWebSearch) {
 		webSearchResult = yield* runWebSearch(conv, messages, assistant?.rag);
+
+		// Remove web search tool from tools preference after it's used
+		// This prevents the AI from calling web search repeatedly in the same conversation
+		if (hasWebSearchTool && ctx.toolsPreference) {
+			ctx.toolsPreference = ctx.toolsPreference.filter(
+				(toolId) => toolId !== webSearchToolId && toolId !== fetchUrlToolId
+			);
+		}
 	}
 
 	let preprompt = conv.preprompt;
@@ -72,10 +89,8 @@ async function* textGenerationWithoutTitle(
 		preprompt = await processPreprompt(preprompt, messages.at(-1)?.content);
 		if (messages[0].from === "system") messages[0].content = preprompt;
 	}
-
 	let toolResults: ToolResult[] = [];
-	let tools = model.tools ? await getTools(toolsPreference, ctx.assistant) : undefined;
-
+	let tools = model.tools ? await getTools(ctx.toolsPreference, ctx.assistant) : undefined;
 	if (tools) {
 		const toolCallsRequired = tools.some((tool) => !toolHasName(directlyAnswer.name, tool));
 		if (toolCallsRequired) {