@houtini/google-knowledge-graph-mcp 1.0.0 → 1.0.1

package/package.json

@@ -1,21 +1,31 @@
 {
   "name": "@houtini/google-knowledge-graph-mcp",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "MCP server for Google's free public Knowledge Graph Search API - search for structured entity information",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "bin": {
     "google-knowledge-graph-mcp": "dist/index.js"
   },
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "require": "./dist/index.js",
+      "import": "./dist/index.js"
+    },
+    "./package.json": "./package.json"
+  },
   "scripts": {
     "build": "tsc",
     "dev": "tsc --watch",
     "start": "node dist/index.js",
-    "prepublishOnly": "npm run build"
+    "prepublishOnly": "npm run build",
+    "prepack": "npm run build",
+    "clean": "rm -rf dist",
+    "type-check": "tsc --noEmit"
   },
   "files": [
     "dist",
-    "src",
     "README.md",
     "LICENSE",
     ".env.example"
@@ -30,7 +40,9 @@
     "claude",
     "ai-assistant",
     "semantic-search",
-    "entity-recognition"
+    "entity-recognition",
+    "knowledge-base",
+    "schema-org"
   ],
   "author": "Richard Baxter <richard@houtini.ai>",
   "license": "MIT",
@@ -43,14 +55,18 @@
   },
   "homepage": "https://github.com/houtini-ai/google-knowledge-graph-mcp#readme",
   "dependencies": {
-    "@modelcontextprotocol/sdk": "^1.0.4",
+    "@modelcontextprotocol/sdk": "^1.25.2",
     "zod": "^3.24.1"
   },
   "devDependencies": {
-    "@types/node": "^22.19.7",
+    "@types/node": "^25.0.9",
     "typescript": "^5.9.3"
   },
   "engines": {
     "node": ">=18.0.0"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org/"
   }
 }

package/src/client.ts

@@ -1,128 +0,0 @@
-export interface KnowledgeGraphEntity {
-  '@context'?: any;
-  '@type'?: string | string[];
-  '@id'?: string;
-  name?: string;
-  description?: string;
-  detailedDescription?: {
-    articleBody?: string;
-    url?: string;
-    license?: string;
-  };
-  image?: {
-    contentUrl?: string;
-    url?: string;
-  };
-  url?: string;
-  resultScore?: number;
-  [key: string]: any;
-}
-
-export interface SearchOptions {
-  query?: string;
-  ids?: string[];
-  languages?: string[];
-  types?: string[];
-  limit?: number;
-  indent?: boolean;
-  prefix?: boolean;
-}
-
-const API_KEY = process.env.GOOGLE_KNOWLEDGE_GRAPH_API_KEY || process.env.GOOGLE_CLOUD_API_KEY;
-const BASE_URL = 'https://kgsearch.googleapis.com/v1/entities:search';
-
-if (!API_KEY) {
-  throw new Error('GOOGLE_KNOWLEDGE_GRAPH_API_KEY or GOOGLE_CLOUD_API_KEY environment variable is required');
-}
-
-/**
- * Build URL for Knowledge Graph Search API
- * Docs: https://developers.google.com/knowledge-graph/reference/rest/v1/
- */
-function buildSearchUrl(options: SearchOptions): string {
-  const params = new URLSearchParams({
-    key: API_KEY!,
-  });
-
-  // Query parameter (for text search)
-  if (options.query) {
-    params.append('query', options.query);
-  }
-
-  // IDs parameter (for entity lookup by MID)
-  if (options.ids && options.ids.length > 0) {
-    params.append('ids', options.ids.join(','));
-  }
-
-  // Languages parameter
-  if (options.languages && options.languages.length > 0) {
-    params.append('languages', options.languages.join(','));
-  }
-
-  // Types parameter (schema.org types)
-  if (options.types && options.types.length > 0) {
-    params.append('types', options.types.join(','));
-  }
-
-  // Limit parameter (default: 20, max: 500)
-  if (options.limit) {
-    params.append('limit', String(options.limit));
-  }
-
-  // Indent parameter (for pretty-printed JSON)
-  if (options.indent) {
-    params.append('indent', 'True');
-  }
-
-  // Prefix parameter (for prefix matching)
-  if (options.prefix !== undefined) {
-    params.append('prefix', String(options.prefix));
-  }
-
-  return `${BASE_URL}?${params.toString()}`;
-}
-
-/**
- * Search Knowledge Graph by query string
- */
-export async function searchEntities(options: SearchOptions): Promise<KnowledgeGraphEntity[]> {
-  if (!options.query && !options.ids) {
-    throw new Error('Either query or ids parameter is required');
-  }
-
-  const url = buildSearchUrl(options);
-
-  const response = await fetch(url, {
-    method: 'GET',
-    headers: {
-      'Accept': 'application/json',
-    },
-  });
-
-  if (!response.ok) {
-    const errorText = await response.text();
-    throw new Error(`Knowledge Graph API request failed: ${response.status} ${response.statusText} - ${errorText}`);
-  }
-
-  const data: any = await response.json();
-
-  // API returns { itemListElement: [ { result: {...}, resultScore: number } ] }
-  if (data.itemListElement && Array.isArray(data.itemListElement)) {
-    return data.itemListElement.map((item: any) => ({
-      ...item.result,
-      resultScore: item.resultScore,
-    }));
-  }
-
-  return [];
-}
-
-/**
- * Lookup entities by Machine IDs (MIDs)
- */
-export async function lookupEntities(ids: string[], languages?: string[]): Promise<KnowledgeGraphEntity[]> {
-  return searchEntities({
-    ids,
-    languages,
-  });
-}

package/src/index.ts

@@ -1,186 +0,0 @@
-#!/usr/bin/env node
-import { Server } from '@modelcontextprotocol/sdk/server/index.js';
-import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
-import {
-  CallToolRequestSchema,
-  ListToolsRequestSchema,
-} from '@modelcontextprotocol/sdk/types.js';
-import { searchEntities, lookupEntities, SearchOptions } from './client.js';
-
-const server = new Server(
-  {
-    name: 'google-knowledge-graph-mcp',
-    version: '1.0.0',
-  },
-  {
-    capabilities: {
-      tools: {},
-    },
-  }
-);
-
-server.setRequestHandler(ListToolsRequestSchema, async () => {
-  return {
-    tools: [
-      {
-        name: 'search_knowledge_graph',
-        description: 'Search Google Knowledge Graph for entities by name or topic. Returns structured information about real-world entities like people, places, organizations, and concepts from Google\'s public knowledge base.',
-        inputSchema: {
-          type: 'object',
-          properties: {
-            query: {
-              type: 'string',
-              description: 'Search query for entities (e.g., "Taylor Swift", "Eiffel Tower", "Python programming")',
-            },
-            languages: {
-              type: 'array',
-              items: { type: 'string' },
-              description: 'Language codes (ISO 639, e.g., ["en", "es", "fr"]). Default: ["en"]',
-            },
-            types: {
-              type: 'array',
-              items: { type: 'string' },
-              description: 'Filter by schema.org types (e.g., ["Person", "Organization", "Place"])',
-            },
-            limit: {
-              type: 'number',
-              minimum: 1,
-              maximum: 500,
-              description: 'Maximum results to return (1-500). Default: 20',
-            },
-          },
-          required: ['query'],
-        },
-      },
-      {
-        name: 'lookup_knowledge_graph_entities',
-        description: 'Look up specific Knowledge Graph entities by their Machine IDs (MIDs). Use this when you already know the entity IDs from a previous search. MIDs look like /m/0dl567 or /g/11b6vwtjpg.',
-        inputSchema: {
-          type: 'object',
-          properties: {
-            ids: {
-              type: 'array',
-              items: { type: 'string' },
-              minItems: 1,
-              description: 'Entity Machine IDs (MIDs) to lookup (e.g., ["/m/0dl567"])',
-            },
-            languages: {
-              type: 'array',
-              items: { type: 'string' },
-              description: 'Language codes for results. Default: ["en"]',
-            },
-          },
-          required: ['ids'],
-        },
-      },
-    ],
-  };
-});
-
-server.setRequestHandler(CallToolRequestSchema, async (request) => {
-  try {
-    const { name, arguments: args } = request.params;
-
-    if (name === 'search_knowledge_graph') {
-      const { query, languages, types, limit } = args as any;
-
-      const options: SearchOptions = {
-        query,
-        languages: languages || ['en'],
-        types,
-        limit: limit || 20,
-      };
-
-      const entities = await searchEntities(options);
-
-      const output = {
-        entities: entities.map(e => {
-          const rawMid = e['@id'] || '';
-          const mid = rawMid.replace(/^kg:/, '');
-          
-          return {
-            mid,
-            name: e.name || '',
-            type: Array.isArray(e['@type']) ? e['@type'] : [e['@type'] || 'Thing'],
-            description: e.description || undefined,
-            detailedDescription: e.detailedDescription?.articleBody || undefined,
-            image: e.image?.contentUrl || undefined,
-            url: e.url || undefined,
-            resultScore: e.resultScore || undefined,
-          };
-        }),
-        count: entities.length,
-      };
-
-      return {
-        content: [
-          {
-            type: 'text',
-            text: JSON.stringify(output, null, 2),
-          },
-        ],
-      };
-    }
-
-    if (name === 'lookup_knowledge_graph_entities') {
-      const { ids, languages } = args as any;
-
-      const entities = await lookupEntities(ids, languages || ['en']);
-
-      const output = {
-        entities: entities.map(e => {
-          const rawMid = e['@id'] || '';
-          const mid = rawMid.replace(/^kg:/, '');
-          
-          return {
-            mid,
-            name: e.name || '',
-            type: Array.isArray(e['@type']) ? e['@type'] : [e['@type'] || 'Thing'],
-            description: e.description || undefined,
-            detailedDescription: e.detailedDescription?.articleBody || undefined,
-            image: e.image?.contentUrl || undefined,
-            url: e.url || undefined,
-            resultScore: e.resultScore || undefined,
-          };
-        }),
-        count: entities.length,
-      };
-
-      return {
-        content: [
-          {
-            type: 'text',
-            text: JSON.stringify(output, null, 2),
-          },
-        ],
-      };
-    }
-
-    throw new Error(`Unknown tool: ${name}`);
-  } catch (error) {
-    return {
-      content: [
-        {
-          type: 'text',
-          text: `Error: ${error instanceof Error ? error.message : String(error)}`,
-        },
-      ],
-      isError: true,
-    };
-  }
-});
-
-async function main() {
-  const transport = new StdioServerTransport();
-  await server.connect(transport);
-
-  process.on('SIGINT', async () => {
-    await server.close();
-    process.exit(0);
-  });
-}
-
-main().catch((error) => {
-  process.stderr.write(`Fatal error: ${error}\n`);
-  process.exit(1);
-});