pi-all-search 1.0.2 → 1.0.4

package/README.md

@@ -1,6 +1,6 @@
 # pi-all-search
 
-**All-in-one web search extension for Pi — exa, tavily, anysearch, firecrawl, brave.**
+**All-in-one web search extension for Pi — exa, tavily, anysearch, firecrawl, context7.**
 
 ## Install
 
@@ -18,11 +18,11 @@ Set API keys in environment or `~/.pi/web-search.json`:
   "tavilyApiKey": "tvly-...",
   "anysearchApiKey": "as_sk_...",
   "firecrawlApiKey": "fc-...",
-  "braveApiKey": "BSA_..."
+  "context7ApiKey": "ctx7sk_..."
 }
 ```
 
-Or set environment variables: `EXA_API_KEY`, `TAVILY_API_KEY`, `ANYSEARCH_API_KEY`, `FIRECRAWL_API_KEY`, `BRAVE_API_KEY`.
+Or set environment variables: `EXA_API_KEY`, `TAVILY_API_KEY`, `ANYSEARCH_API_KEY`, `FIRECRAWL_API_KEY`, `CONTEXT7_API_KEY`.
 
 ## Usage
 
@@ -31,6 +31,7 @@ web_search({ query: "TypeScript best practices" })
 web_search({ queries: ["React vs Vue", "Angular vs Svelte"] })
 web_search({ query: "AAPL stock price", provider: "anysearch" })
 web_search({ query: "rust async programming", provider: "tavily" })
+web_search({ query: "Next.js caching", provider: "context7" })
 ```
 
 ## Providers
@@ -41,14 +42,15 @@ web_search({ query: "rust async programming", provider: "tavily" })
 | **tavily** | General web, programming, fast results | `TAVILY_API_KEY` |
 | **anysearch** | Finance, stocks, structured data | `ANYSEARCH_API_KEY` |
 | **firecrawl** | Scraping-heavy sites, fallback | `FIRECRAWL_API_KEY` |
-| **brave** | General web, good coverage | `BRAVE_API_KEY` |
+| **context7** | Library/framework/API documentation | `CONTEXT7_API_KEY` |
 
 ## Routing
 
 Automatic intent-based routing:
-- **Finance queries** → anysearch → brave → exa
-- **Academic queries** → exa → anysearch → brave
-- **General queries** → tavily → brave → exa → anysearch
+- **Finance queries** → anysearch → exa → tavily
+- **Academic queries** → exa → anysearch → tavily
+- **General queries** → tavily → exa → anysearch → firecrawl
+- **Docs queries** → context7 → exa → tavily
 
 Override with `provider="exa"` etc.
 

package/package.json

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

package/src/providers/context7.ts

@@ -0,0 +1,27 @@
+import type { SearchProvider, SearchResult } from "./types.js";
+
+export const CONTEXT7_META = { name: "context7", label: "Context7", envVar: "CONTEXT7_API_KEY" };
+
+export class Context7Provider implements SearchProvider {
+  name = "context7";
+  constructor(private apiKey: string) {}
+
+  async search(query: string, maxResults: number): Promise<{ results: SearchResult[] }> {
+    const resp = await fetch("https://api.context7.com/v1/search", {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        Authorization: `Bearer ${this.apiKey}`,
+      },
+      body: JSON.stringify({ query, limit: maxResults }),
+    });
+    const data = await resp.json();
+    const results: SearchResult[] = (data.results ?? []).map((r: any) => ({
+      title: r.title ?? r.name ?? "",
+      url: r.url ?? r.documentation_url ?? "",
+      snippet: r.description ?? r.snippet ?? "",
+      score: r.score,
+    }));
+    return { results };
+  }
+}

package/src/providers/index.ts

@@ -2,7 +2,7 @@ import { EXA_META, ExaProvider } from "./exa.js";
 import { TAVILY_META, TavilyProvider } from "./tavily.js";
 import { ANYSEARCH_META, AnysearchProvider } from "./anysearch.js";
 import { FIRECRAWL_META, FirecrawlProvider } from "./firecrawl.js";
-import { BRAVE_META, BraveProvider } from "./brave.js";
+import { CONTEXT7_META, Context7Provider } from "./context7.js";
 import type { SearchProvider } from "./types.js";
 
 export interface ProviderMeta {
@@ -16,7 +16,7 @@ export const PROVIDERS: readonly ProviderMeta[] = [
   TAVILY_META,
   ANYSEARCH_META,
   FIRECRAWL_META,
-  BRAVE_META,
+  CONTEXT7_META,
 ];
 
 export function createProvider(name: string, apiKey: string): SearchProvider {
@@ -29,8 +29,8 @@ export function createProvider(name: string, apiKey: string): SearchProvider {
       return new AnysearchProvider(apiKey);
     case "firecrawl":
       return new FirecrawlProvider(apiKey);
-    case "brave":
-      return new BraveProvider(apiKey);
+    case "context7":
+      return new Context7Provider(apiKey);
     default:
       throw new Error(`Unknown provider: "${name}". Available: ${PROVIDERS.map((p) => p.name).join(", ")}`);
   }

package/src/router.ts

@@ -1,24 +1,26 @@
 import type { SearchProvider } from "./providers/types.js";
 
-export type SearchIntent = "finance" | "academic" | "general";
+export type SearchIntent = "finance" | "academic" | "general" | "docs";
 
 export function classifyIntent(query: string): SearchIntent {
   const q = query.toLowerCase();
   if (/\b(stock|price|ticker|forex|crypto|market|trade|earnings)\b/.test(q)) return "finance";
   if (/\b(paper|research|journal|doi|arxiv|scholar|academic|study)\b/.test(q)) return "academic";
+  if (/\b(doc|docs|documentation|library|framework|api|sdk|how to|example|syntax)\b/.test(q)) return "docs";
   return "general";
 }
 
-const INTENT PROVIDERS: Record<SearchIntent, { primary: string; secondary: string[] }> = {
-  finance: { primary: "anysearch", secondary: ["brave", "exa"] },
-  academic: { primary: "exa", secondary: ["anysearch", "brave"] },
-  general: { primary: "tavily", secondary: ["brave", "exa", "anysearch"] },
+const INTENT_PROVIDERS: Record<SearchIntent, { primary: string; secondary: string[] }> = {
+  finance: { primary: "anysearch", secondary: ["exa", "tavily"] },
+  academic: { primary: "exa", secondary: ["anysearch", "tavily"] },
+  general: { primary: "tavily", secondary: ["exa", "anysearch", "firecrawl"] },
+  docs: { primary: "context7", secondary: ["exa", "tavily"] },
 };
 
 export function routeIntent(
   intent: SearchIntent,
   providers: Map<string, SearchProvider>,
-  requestedProvider?: string
+  requestedProvider?: string,
 ): { primary: string; secondary: string[] } {
   if (requestedProvider && providers.has(requestedProvider)) {
     return { primary: requestedProvider, secondary: [...providers.keys()].filter((k) => k !== requestedProvider) };

package/src/web-search.ts

@@ -128,9 +128,9 @@ export function registerWebSearchTool(pi: ExtensionAPI): void {
         "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, library docs, and fast programming research.",
-        "Set provider='brave' for general web search with good coverage.",
+        "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).",
         "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.",
@@ -148,7 +148,7 @@ export function registerWebSearchTool(pi: ExtensionAPI): void {
         }),
       ),
       provider: Type.Optional(
-        StringEnum(["auto", "exa", "tavily", "anysearch", "firecrawl", "brave"], {
+        StringEnum(["auto", "exa", "tavily", "anysearch", "firecrawl", "context7"], {
           description:
             "Directly override the search provider. 'auto' uses local intent routing (default).",
           default: "auto",

package/src/providers/brave.ts

@@ -1,25 +0,0 @@
-import type { SearchProvider, SearchResult } from "./types.js";
-
-export const BRAVE_META = { name: "brave", label: "Brave", envVar: "BRAVE_API_KEY" };
-
-export class BraveProvider implements SearchProvider {
-  name = "brave";
-  constructor(private apiKey: string) {}
-
-  async search(query: string, maxResults: number, signal?: AbortSignal): Promise<{ results: SearchResult[] }> {
-    const resp = await fetch(
-      `https://api.search.brave.com/res/v1/web/search?q=${encodeURIComponent(query)}&count=${maxResults}`,
-      {
-        headers: { "X-Subscription-Token": this.apiKey, Accept: "application/json" },
-        signal,
-      }
-    );
-    const data = await resp.json();
-    const results: SearchResult[] = ((data.web?.results ?? []) as any[]).slice(0, maxResults).map((r) => ({
-      title: r.title ?? "",
-      url: r.url ?? "",
-      snippet: r.description ?? "",
-    }));
-    return { results };
-  }
-}