@wolido/pi-anysearch 0.1.0 → 0.2.0

package/README.md

@@ -49,7 +49,7 @@ Get a key at [anysearch.ai](https://anysearch.ai).
 
 ```bash
 npm install
-npm test           # vitest, 36 tests
+npm test           # vitest, 63 tests
 npm run typecheck  # tsc --noEmit
 ```
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@wolido/pi-anysearch",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "AnySearch AI search API extension for pi",
   "type": "module",
   "main": "./src/index.ts",

package/src/anysearch.ts

@@ -50,7 +50,7 @@ export interface AnySearchClient {
 	search(params: SearchParams): Promise<SearchResult>;
 	batchSearch(params: BatchSearchParams): Promise<SearchResult[]>;
 	extract(url: string): Promise<string>;
-	getDomains(domain: string): Promise<string>;
+	getDomains(domain: string | string[]): Promise<string>;
 }
 
 const DEFAULT_BASE_URL = "https://api.anysearch.com";
@@ -168,7 +168,7 @@ export function createAnySearchClient(options: AnySearchClientOptions = {}): Any
 				params: {
 					protocolVersion: "2025-03-26",
 					capabilities: {},
-					clientInfo: { name: "pi-anysearch", version: "0.1.0" },
+					clientInfo: { name: "pi-anysearch", version: "0.2.0" },
 				},
 			}),
 		});
@@ -244,8 +244,9 @@ export function createAnySearchClient(options: AnySearchClientOptions = {}): Any
 			return mcpCall("extract", { url });
 		},
 
-		getDomains(domain: string): Promise<string> {
-			return mcpCall("get_sub_domains", { domain });
+		getDomains(domainOrDomains: string | string[]): Promise<string> {
+			const args = Array.isArray(domainOrDomains) ? { domains: domainOrDomains } : { domain: domainOrDomains };
+			return mcpCall("get_sub_domains", args);
 		},
 	};
 }

package/src/index.ts

@@ -107,12 +107,52 @@ const batchSearchSchema = Type.Object({
 	...sharedSearchParams,
 });
 
+// 17 top-level capability domains accepted by the MCP get_sub_domains tool.
+const DOMAIN_VALUES = [
+	"general",
+	"resource",
+	"social_media",
+	"finance",
+	"academic",
+	"legal",
+	"health",
+	"business",
+	"security",
+	"ip",
+	"code",
+	"energy",
+	"environment",
+	"agriculture",
+	"travel",
+	"film",
+	"gaming",
+] as const;
+
+const DOMAIN_LIST_TEXT = DOMAIN_VALUES.join(", ");
+
+const DOMAIN_DESCRIPTION =
+	"Top-level capability domain. Valid values: " + DOMAIN_LIST_TEXT;
+
+const DOMAINS_ARRAY_DESCRIPTION =
+	"List of top-level capability domains to inspect. Valid values: " +
+	DOMAIN_LIST_TEXT +
+	". Pass all 17 values to list the full capability table. The official recommendation is to request at most 5 domains per call.";
+
 const extractSchema = Type.Object({
 	url: Type.String({ description: "URL of the page to extract readable content from" }),
 });
 
 const domainsSchema = Type.Object({
-	domain: Type.String({ description: 'Top-level capability domain to inspect, e.g. "finance", "academic"' }),
+	domain: Type.Optional(
+		StringEnum(DOMAIN_VALUES, {
+			description: DOMAIN_DESCRIPTION,
+		}),
+	),
+	domains: Type.Optional(
+		Type.Array(StringEnum(DOMAIN_VALUES), {
+			description: DOMAINS_ARRAY_DESCRIPTION,
+		}),
+	),
 });
 
 /** Byte-level truncation of a single field value (e.g. one result's content). */
@@ -313,10 +353,14 @@ export default function anysearchExtension(pi: ExtensionAPI) {
 		parameters: domainsSchema,
 		async execute(_toolCallId, params, _signal, _onUpdate, ctx) {
 			const client = await getClient(ctx);
-			const text = await client.getDomains(params.domain);
+			const target = params.domains?.length ? params.domains : params.domain;
+			if (!target) {
+				throw new Error(`domain or domains is required. Valid values: ${DOMAIN_LIST_TEXT}`);
+			}
+			const text = await client.getDomains(target);
 			return {
 				content: [{ type: "text", text: truncateForOutput(text) }],
-				details: { domain: params.domain },
+				details: { domain: params.domain, domains: params.domains },
 			};
 		},
 	});