pi-all-search 1.0.9 → 1.0.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-all-search",
-  "version": "1.0.9",
+  "version": "1.0.11",
   "description": "All-in-one web search extension for Pi — exa, tavily, anysearch, firecrawl, context7",
   "license": "MIT",
   "type": "module",

package/src/config.ts

@@ -1,21 +1,32 @@
 import { PROVIDERS } from "./providers/index.js";
+import { readFileSync } from "node:fs";
+import { join } from "node:path";
+import { homedir } from "node:os";
+
+const CONFIG_PATH = join(homedir(), ".pi", "web-search.json");
 
 export interface SearchConfig {
   apiKeys: Record<string, string>;
-  routing?: {
-    finance?: string;
-    academic?: string;
-    general?: string;
-  };
 }
 
 export function loadConfig(): SearchConfig {
-  const env = process.env;
   const apiKeys: Record<string, string> = {};
-  for (const meta of PROVIDERS) {
-    const key = env[meta.envVar];
-    if (key) apiKeys[meta.name] = key;
+
+  try {
+    const config = JSON.parse(readFileSync(CONFIG_PATH, "utf8"));
+    for (const meta of PROVIDERS) {
+      const key = config[meta.name + "ApiKey"];
+      if (key) apiKeys[meta.name] = key;
+    }
+  } catch {
+    // fallback to env vars
+    const env = process.env;
+    for (const meta of PROVIDERS) {
+      const key = env[meta.envVar];
+      if (key) apiKeys[meta.name] = key;
+    }
   }
+
   return { apiKeys };
 }
 

package/src/router.ts

@@ -10,23 +10,21 @@ export interface RoutingConfig {
 
 export function classifyIntent(query: string): SearchIntent {
   const q = query.toLowerCase();
-
   if (/\b(stock|price|ticker|forex|crypto|market|trade|earnings|fund|etf|ipo)\b/.test(q)) return "finance";
   if (/\b(paper|research|journal|doi|arxiv|scholar|academic|study|thesis)\b/.test(q)) return "academic";
   if (/\b(doc|docs|documentation|library|framework|api|sdk|how to|example|syntax|function|method|class|component)\b/.test(q)) return "docs";
   if (/\b(code|github|repo|repository|pull request|commit|branch|merge)\b/.test(q)) return "technical";
   if (/\b(news|latest|today|breaking|announced|update|release|happened)\b/.test(q)) return "news";
-
   return "general";
 }
 
-const INTENT_PROVIDERS: Record<SearchIntent, { primary: string; secondary: string[] }> = {
-  finance: { primary: "anysearch", secondary: ["exa", "tavily"] },
-  academic: { primary: "exa", secondary: ["anysearch", "tavily"] },
-  docs: { primary: "context7", secondary: ["exa", "tavily"] },
-  technical: { primary: "firecrawl", secondary: ["exa", "tavily"] },
-  news: { primary: "tavily", secondary: ["anysearch", "exa"] },
-  general: { primary: "tavily", secondary: ["anysearch", "exa", "firecrawl"] },
+const INTENT_PROVIDERS: Record<SearchIntent, string[]> = {
+  finance: ["anysearch", "exa", "tavily"],
+  academic: ["exa", "anysearch", "tavily"],
+  docs: ["context7", "exa", "tavily"],
+  technical: ["firecrawl", "exa", "tavily"],
+  news: ["tavily", "anysearch", "exa"],
+  general: ["tavily", "anysearch", "exa", "firecrawl"],
 };
 
 export function routeIntent(
@@ -35,20 +33,24 @@ export function routeIntent(
   requestedProvider?: string,
 ): RoutingConfig {
   if (requestedProvider && providers.has(requestedProvider)) {
+    const secondary = [...providers.keys()].filter((k) => k !== requestedProvider);
+    return { primary: requestedProvider, secondary, intent };
+  }
+
+  const candidates = INTENT_PROVIDERS[intent];
+  const available = candidates.filter((p) => providers.has(p));
+
+  if (available.length === 0) {
+    const allAvailable = [...providers.keys()];
     return {
-      primary: requestedProvider,
-      secondary: [...providers.keys()].filter((k) => k !== requestedProvider),
+      primary: allAvailable[0] ?? "tavily",
+      secondary: allAvailable.slice(1),
       intent,
     };
   }
 
-  const config = INTENT_PROVIDERS[intent];
-  const available = config.primary === "anysearch" && !providers.has("anysearch")
-    ? config.secondary.filter((p) => providers.has(p))
-    : [config.primary, ...config.secondary].filter((p) => providers.has(p));
-
   return {
-    primary: available[0] ?? "tavily",
+    primary: available[0],
     secondary: available.slice(1),
     intent,
   };

package/src/web-search.ts

@@ -121,18 +121,19 @@ export function registerWebSearchTool(pi: ExtensionAPI): void {
     name: "web_search",
     label: "Web Search",
     description:
-      "Search the web with automatic provider selection. For stocks/finance, uses Anysearch. For academic papers, uses Exa. For general web, uses Tavily. Falls back automatically if the primary provider fails. Use include_content with web_fetch for full page reading. Use queries (plural) for parallel multi-angle research.",
+      "Search the web with 5 providers (exa, tavily, anysearch, firecrawl, context7). Choose the right provider based on query type. Falls back automatically if the primary provider fails. Use web_fetch for full page content. Use queries (plural) for parallel multi-angle research.",
     promptSnippet:
       "Search the web with automatic or custom routing (set provider='exa' for papers, provider='anysearch' for finance, provider='tavily' for general, provider='context7' for docs).",
     get promptGuidelines() {
       return [
         "Use web_search for information beyond your training data — current events, recent docs, live data.",
-        "Set provider='anysearch' when searching for stock prices, tickers, forex, or CVE vulnerability hashes.",
-        "Set provider='exa' when searching for academic research papers, journals, or DOIs.",
-        "Set provider='tavily' for web pages, coding guides, and fast programming research.",
-        "Set provider='firecrawl' for scraping-heavy sites or when others fail.",
-        "Set provider='context7' for library/framework/API documentation, code examples, and how-to guides.",
-        "Set provider='auto' to let the fast local intent router decide automatically (default).",
+        "Choose the right provider based on the query type:",
+        "  • context7 — library/framework/API documentation, code examples, how-to guides, syntax questions",
+        "  • exa — academic research papers, journals, DOIs, scholarly articles, theses",
+        "  • anysearch — stock prices, tickers, forex, crypto, CVE vulnerabilities, financial data",
+        "  • firecrawl — scraping-heavy sites, code repos, GitHub content, when others fail",
+        "  • tavily — general web search, news, programming guides, fast results (default)",
+        "Set provider='auto' to let the local intent router decide automatically.",
         "After answering, include a \"Sources:\" section with markdown hyperlinks: [Title](URL).",
         "Use web_fetch after web_search to read full page content — web_search returns snippets only.",
         "Use {queries:[...]} with 2-4 varied angles for broader coverage.",