pi-web-search 1.0.2 → 1.2.0

package/dist/url_context.js

@@ -1,102 +0,0 @@
-import { Type } from "@sinclair/typebox";
-import { callApiStream, getConfig, applyCitations } from "./api.js";
-import { getModel, missingConfigResult, errorResult, formatResult } from "./utils.js";
-export const UrlContextSchema = Type.Object({
-    query: Type.String({ description: "Question or task to perform on the URLs" }),
-    urls: Type.Array(Type.String(), {
-        description: "Public URLs to analyze (web pages, documents, images, YouTube videos, etc).",
-        minItems: 1,
-        maxItems: 20
-    }),
-});
-const YOUTUBE_REGEX = /^(?:https?:\/\/)?(?:www\.)?(?:youtube\.com\/(?:watch\?v=|embed\/)|youtu\.be\/)([a-zA-Z0-9_-]{11})/;
-export async function urlContext(id, params, signal, onUpdate, ctx) {
-    const model = await getModel(ctx);
-    if (!model)
-        return missingConfigResult(ctx);
-    const count = params.urls.length;
-    onUpdate?.({ content: [{ type: "text", text: `Analyzing ${count} URL${count > 1 ? 's' : ''}...` }], details: {} });
-    try {
-        const config = getConfig(model);
-        let contents = [];
-        let tools = [{ [config.urlContextTool]: {} }];
-        // Special handling for YouTube videos on Gemini
-        if (model.api === "google-generative-ai") {
-            const youtubeUrls = [];
-            const otherUrls = [];
-            for (const url of params.urls) {
-                if (YOUTUBE_REGEX.test(url)) {
-                    youtubeUrls.push(url);
-                }
-                else {
-                    otherUrls.push(url);
-                }
-            }
-            // If we have YouTube URLs, construct file_data parts
-            if (youtubeUrls.length > 0) {
-                const parts = [];
-                for (const url of youtubeUrls) {
-                    parts.push({
-                        file_data: { file_uri: url, mime_type: "video/mp4" }
-                    });
-                }
-                let prompt = params.query;
-                if (otherUrls.length > 0) {
-                    prompt += `\n\nURLs:\n${otherUrls.join("\n")}`;
-                }
-                else {
-                    // If no other URLs, we might not need the tool, but keep it just in case
-                    // or maybe the tool is required for grounding even with video?
-                    // "google_search_retrieval" tool might confuse if there are no URLs to retrieve.
-                    // But if we remove the tool, we might lose grounding capabilities (like search).
-                    // Let's keep the tool enabled.
-                }
-                parts.push({ text: prompt });
-                contents = [{ role: "user", parts }];
-            }
-            else {
-                // No YouTube URLs, standard behavior
-                const combinedPrompt = `${params.query}\n\nURLs:\n${params.urls.join("\n")}`;
-                contents = [{ role: "user", parts: [{ text: combinedPrompt }] }];
-            }
-        }
-        else {
-            // Not Gemini, standard behavior
-            const combinedPrompt = `${params.query}\n\nURLs:\n${params.urls.join("\n")}`;
-            contents = [{ role: "user", parts: [{ text: combinedPrompt }] }];
-        }
-        const result = await callApiStream(ctx, model, {
-            contents,
-            tools
-        }, onUpdate);
-        const { text, sources } = applyCitations(result.text, result.groundingMetadata);
-        // Handle both camelCase and snake_case metadata
-        const urlMeta = result.urlContextMetadata?.urlMetadata
-            || result.urlContextMetadata?.url_metadata || [];
-        const retrieved = urlMeta
-            .filter((m) => (m.urlRetrievalStatus || m.url_retrieval_status) === "URL_RETRIEVAL_STATUS_SUCCESS")
-            .map((m) => m.retrievedUrl || m.retrieved_url || m.url);
-        const failed = urlMeta
-            .filter((m) => (m.urlRetrievalStatus || m.url_retrieval_status) !== "URL_RETRIEVAL_STATUS_SUCCESS")
-            .map((m) => ({
-            url: m.retrievedUrl || m.retrieved_url || m.url,
-            status: m.urlRetrievalStatus || m.url_retrieval_status
-        }));
-        let summary = text;
-        if (failed.length > 0) {
-            summary += `\n\n## URL Status\n✅ Retrieved: ${retrieved.length}\n❌ Failed: ${failed.length}`;
-            failed.forEach((f) => { summary += `\n- ${f.url}: ${f.status}`; });
-        }
-        if (sources.length > 0 && !summary.includes("## Sources")) {
-            summary += `\n\n## Sources\n${sources.map((s, i) => `${i + 1}. [${s.title}](${s.url})`).join("\n")}`;
-        }
-        return formatResult(summary, {
-            retrieved,
-            failed: failed.length > 0 ? failed : undefined,
-            model: model.id
-        });
-    }
-    catch (e) {
-        return errorResult(e);
-    }
-}

package/dist/utils.d.ts

@@ -1,6 +0,0 @@
-import type { ExtensionContext, AgentToolResult } from "@mariozechner/pi-coding-agent";
-import { type Model } from "@mariozechner/pi-ai";
-export declare function formatResult(text: string, details: any): AgentToolResult<any>;
-export declare function getModel(ctx: ExtensionContext): Promise<Model<any> | undefined>;
-export declare function missingConfigResult(ctx: ExtensionContext): AgentToolResult<any>;
-export declare function errorResult(e: Error): AgentToolResult<any>;

package/dist/utils.js

@@ -1,63 +0,0 @@
-import { truncateHead, DEFAULT_MAX_BYTES, DEFAULT_MAX_LINES } from "@mariozechner/pi-coding-agent";
-// --- Formatting ---
-export function formatResult(text, details) {
-    const { content, truncated } = truncateHead(text, { maxLines: DEFAULT_MAX_LINES, maxBytes: DEFAULT_MAX_BYTES });
-    return {
-        content: [{ type: "text", text: content + (truncated ? "\n\n[Truncated]" : "") }],
-        details
-    };
-}
-// --- Model Selection ---
-export async function getModel(ctx) {
-    // flash first, big first: 3-flash -> 2.5-flash -> 2.0-flash
-    // provider priority: google-gemini-cli -> google-antigravity -> google -> google-generative-ai
-    const models = ctx.modelRegistry.getAvailable();
-    const flashModels = [
-        /gemini-3.*flash/i,
-        /gemini-2\.5.*flash/i,
-        /gemini-2\.0.*flash/i,
-        /gemini.*flash/i,
-    ];
-    const providers = [
-        "google-gemini-cli",
-        "google-antigravity",
-        "google",
-        "google-generative-ai",
-    ];
-    // Filter to only Google-compatible models (those with supported api/provider)
-    const googleModels = models.filter(m => providers.includes(m.provider) ||
-        m.api === "google-generative-ai" ||
-        m.api === "google-gemini-cli");
-    // Try each flash pattern in priority order
-    for (const pattern of flashModels) {
-        const matching = googleModels.filter(m => pattern.test(m.id));
-        if (matching.length === 0)
-            continue;
-        // Among matches, pick by provider priority
-        for (const provider of providers) {
-            const model = matching.find(m => m.provider === provider);
-            if (model)
-                return model;
-        }
-        // Fall back to first match if no priority provider found
-        return matching[0];
-    }
-    // No flash model found, try any Google model by provider priority
-    for (const provider of providers) {
-        const model = googleModels.find(m => m.provider === provider);
-        if (model)
-            return model;
-    }
-    // Return first available Google model if any
-    return googleModels[0];
-}
-// --- Error Results ---
-export function missingConfigResult(ctx) {
-    const msg = ctx.model && ["google-gemini-cli", "google-antigravity"].includes(ctx.model.provider)
-        ? `Provider ${ctx.model.provider} requires valid OAuth credentials.`
-        : "No Google Gemini configuration found. Please configure GEMINI_API_KEY.";
-    return { content: [{ type: "text", text: `Failed: ${msg}` }], details: { error: "missing_config" } };
-}
-export function errorResult(e) {
-    return { content: [{ type: "text", text: `Error: ${e.message}` }], details: { error: true } };
-}

package/dist/web_search.d.ts

@@ -1,8 +0,0 @@
-import type { ExtensionContext, AgentToolUpdateCallback } from "@mariozechner/pi-coding-agent";
-import { type Static } from "@sinclair/typebox";
-export declare const WebSearchSchema: import("@sinclair/typebox").TObject<{
-    query: import("@sinclair/typebox").TString;
-    urls: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TArray<import("@sinclair/typebox").TString>>;
-}>;
-export type WebSearchInput = Static<typeof WebSearchSchema>;
-export declare function webSearch(id: string, params: WebSearchInput, signal: AbortSignal, onUpdate: AgentToolUpdateCallback | undefined, ctx: ExtensionContext): Promise<import("@mariozechner/pi-coding-agent").AgentToolResult<any>>;

package/dist/web_search.js

@@ -1,75 +0,0 @@
-import { Type } from "@sinclair/typebox";
-import { callApiStream, getConfig, applyCitations } from "./api.js";
-import { getModel, missingConfigResult, errorResult, formatResult } from "./utils.js";
-export const WebSearchSchema = Type.Object({
-    query: Type.String({ description: "The search query or question to answer" }),
-    urls: Type.Optional(Type.Array(Type.String(), {
-        description: "Additional URLs to analyze along with search (up to 20)",
-        maxItems: 20
-    })),
-});
-export async function webSearch(id, params, signal, onUpdate, ctx) {
-    const model = await getModel(ctx);
-    if (!model)
-        return missingConfigResult(ctx);
-    const hasUrls = params.urls && params.urls.length > 0;
-    const urlCount = hasUrls ? params.urls.length : 0;
-    onUpdate?.({
-        content: [{
-                type: "text",
-                text: hasUrls
-                    ? `Searching and analyzing ${urlCount} URL(s)...`
-                    : `Searching for "${params.query}"...`
-            }],
-        details: {}
-    });
-    try {
-        const config = getConfig(model);
-        // Build prompt: include URLs if provided
-        const prompt = hasUrls
-            ? `${params.query}\n\nAlso analyze these URLs:\n${params.urls.join("\n")}`
-            : params.query;
-        // Enable google_search, add url_context if URLs provided
-        const tools = hasUrls
-            ? [{ [config.searchTool]: {} }, { [config.urlContextTool]: {} }]
-            : [{ [config.searchTool]: {} }];
-        const result = await callApiStream(ctx, model, {
-            contents: [{ role: "user", parts: [{ text: prompt }] }],
-            tools
-        }, onUpdate);
-        const { text, sources } = applyCitations(result.text, result.groundingMetadata);
-        // Handle URL context metadata
-        const urlMeta = result.urlContextMetadata?.urlMetadata
-            || result.urlContextMetadata?.url_metadata || [];
-        const retrieved = urlMeta
-            .filter((m) => (m.urlRetrievalStatus || m.url_retrieval_status) === "URL_RETRIEVAL_STATUS_SUCCESS")
-            .map((m) => m.retrievedUrl || m.retrieved_url || m.url);
-        const failed = urlMeta
-            .filter((m) => (m.urlRetrievalStatus || m.url_retrieval_status) !== "URL_RETRIEVAL_STATUS_SUCCESS")
-            .map((m) => ({
-            url: m.retrievedUrl || m.retrieved_url || m.url,
-            status: m.urlRetrievalStatus || m.url_retrieval_status
-        }));
-        let summary = text;
-        // Add URL status if there were failures
-        if (failed.length > 0) {
-            summary += `\n\n## URL Status\n✅ Retrieved: ${retrieved.length}\n❌ Failed: ${failed.length}`;
-            failed.forEach((f) => { summary += `\n- ${f.url}: ${f.status}`; });
-        }
-        // Add sources
-        if (sources.length > 0) {
-            summary += `\n\n## Sources\n${sources.map((s, i) => `${i + 1}. [${s.title}](${s.url})`).join("\n")}`;
-        }
-        return formatResult(summary, {
-            sources,
-            searchQueries: result.groundingMetadata?.webSearchQueries,
-            retrieved: retrieved.length > 0 ? retrieved : undefined,
-            failed: failed.length > 0 ? failed : undefined,
-            model: model.id,
-            grounded: sources.length > 0
-        });
-    }
-    catch (e) {
-        return errorResult(e);
-    }
-}

package/tsconfig.json

@@ -1,14 +0,0 @@
-{
-  "compilerOptions": {
-    "target": "ES2022",
-    "module": "NodeNext",
-    "moduleResolution": "NodeNext",
-    "outDir": "./dist",
-    "rootDir": "./src",
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "declaration": true
-  },
-  "include": ["src/**/*"]
-}