@kinginsun/mcp-drugsea 0.2.0 → 0.2.1

package/dist/index.js

@@ -2,7 +2,7 @@
 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 { clampLimit, clampOffset, fetchDetail, fetchFacets, listSearch, mcpDbDetail, mcpDbSearch, prefersMcpListApi, yaohaiPost, } from "./api.js";
+import { ApiError, clampLimit, clampOffset, fetchDetail, fetchFacets, listSearch, mcpDbDetail, mcpDbSearch, prefersMcpListApi, yaohaiPost, } from "./api.js";
 import { ATC_HINT, PRODUCT_CN_COMMON_FIELDS, PRODUCT_CN_DETAIL_PATH, PRODUCT_CN_FACET_FIELDS, PRODUCT_CN_FACET_PREFIX, PRODUCT_CN_VIEW_TYPES, REG_CN_COMMON_FIELDS, REG_CN_DETAIL_PATH, REG_CN_FACET_FIELDS, REG_CN_FACET_PREFIX, REG_CN_VIEW_TYPES, applyProductCnDefaults, applyRegCnDefaults, productCnSearchPath, regCnSearchPath, } from "./fields.js";
 import { EmptyObjectSchema, ProductCnDetailSchema, ProductCnFacetsSchema, ProductCnSearchSchema, RegCnDetailSchema, RegCnFacetsSchema, RegCnSearchSchema, YaohaiCatalogSchema, YaohaiDetailSchema, YaohaiGlobalSearchSchema, YaohaiSearchSchema, YaohaiSmartSearchSchema, } from "./types.js";
 const PACKAGE_VERSION = "0.2.0";
@@ -111,6 +111,7 @@ server.setRequestHandler(ListToolsRequestSchema, async () => {
             {
                 name: "yaohai-smart-search",
                 description: "Natural-language Yaohai search: auto-routes the question to up to 3 databases. Use when the user question is broad or the target DB is unclear. " +
+                    "Falls back to global search when the router matches nothing or finds no rows. " +
                     ROUTING_HINT,
                 inputSchema: {
                     type: "object",
@@ -259,6 +260,170 @@ function encodeId(id) {
 function asQuery(query) {
     return { ...(query ?? {}) };
 }
+async function globalSearchContent(q, query, limit) {
+    const gquery = asQuery(query);
+    if (q && (gquery.term === undefined || gquery.term === "")) {
+        gquery.term = q;
+    }
+    const content = await yaohaiPost("/g/mcp/yaohai/global-search", {
+        query: gquery,
+        limit,
+        offset: 0,
+    });
+    return content;
+}
+function allSmartResultsEmpty(content) {
+    if (!content || typeof content !== "object") {
+        return true;
+    }
+    const results = content.results;
+    if (!Array.isArray(results) || results.length === 0) {
+        return true;
+    }
+    return results.every((entry) => {
+        const result = entry?.result;
+        if (!result || typeof result !== "object") {
+            return true;
+        }
+        const total = result.total;
+        return typeof total === "number" ? total === 0 : true;
+    });
+}
+/**
+ * The backend router searches the entire question string in one field per DB,
+ * which rarely matches (e.g. item="医保目录 阿司匹林" → 0 rows), and sometimes
+ * picks a field the DB does not support (e.g. `item` on jicai). Retry each
+ * empty matched DB with the individual tokens of the question across the
+ * router's field plus the catalog's search_fields for that DB.
+ */
+async function retrySmartResultsWithTokens(content, q, limit) {
+    const results = content.results;
+    if (!Array.isArray(results)) {
+        return null;
+    }
+    const tokens = q.split(/\s+/).filter((t) => t && t !== q);
+    if (tokens.length === 0) {
+        return null;
+    }
+    const catalogFields = new Map();
+    async function searchFieldsFor(dbname) {
+        if (catalogFields.has(dbname)) {
+            return catalogFields.get(dbname);
+        }
+        let keys = [];
+        try {
+            const cat = (await yaohaiPost("/g/mcp/yaohai/catalog", { q: dbname }));
+            const db = (cat.databases ?? []).find((d) => d.id === dbname);
+            keys = (db?.search_fields ?? [])
+                .map((f) => f.key)
+                .filter((k) => typeof k === "string");
+        }
+        catch {
+            keys = [];
+        }
+        catalogFields.set(dbname, keys);
+        return keys;
+    }
+    let changed = false;
+    const patched = [];
+    for (const entry of results) {
+        const inner = entry?.result;
+        if (!inner || typeof inner !== "object") {
+            patched.push(entry);
+            continue;
+        }
+        const r = inner;
+        if (r.total !== 0 || typeof r.dbname !== "string") {
+            patched.push(entry);
+            continue;
+        }
+        const queryApplied = r.query_applied;
+        const appliedKeys = queryApplied ? Object.keys(queryApplied) : [];
+        const singleFieldWholeQuestion = appliedKeys.length === 1 && queryApplied[appliedKeys[0]] === q;
+        if (!singleFieldWholeQuestion) {
+            patched.push(entry);
+            continue;
+        }
+        const routerField = appliedKeys[0];
+        const fields = [routerField, ...(await searchFieldsFor(r.dbname))];
+        const uniqueFields = [...new Set(fields)].slice(0, 6);
+        let replaced = false;
+        outer: for (const token of tokens.slice(0, 3)) {
+            for (const field of uniqueFields) {
+                try {
+                    const retry = (await yaohaiPost("/g/mcp/yaohai/search", {
+                        dbname: r.dbname,
+                        query: { [field]: token },
+                        limit,
+                        offset: 0,
+                    }));
+                    if (typeof retry.total === "number" && retry.total > 0) {
+                        patched.push({
+                            ...entry,
+                            result: { ...retry, retry: { token, field } },
+                        });
+                        changed = true;
+                        replaced = true;
+                        break outer;
+                    }
+                }
+                catch {
+                    // ignore per-token retry errors; try next field/token
+                }
+            }
+        }
+        if (!replaced) {
+            patched.push(entry);
+        }
+    }
+    if (!changed) {
+        return null;
+    }
+    return { ...content, results: patched };
+}
+/**
+ * Last resort when the router matched databases but every search came back
+ * empty (e.g. the question is just a DB keyword like 集采): browse the matched
+ * databases without a query so the caller still sees representative rows.
+ */
+async function browseMatchedDatabases(content, limit) {
+    const matched = content.matched_databases;
+    if (!Array.isArray(matched) || matched.length === 0) {
+        return null;
+    }
+    const results = [];
+    for (const db of matched.slice(0, 3)) {
+        const id = db?.id;
+        if (typeof id !== "string") {
+            continue;
+        }
+        try {
+            const browse = (await yaohaiPost("/g/mcp/yaohai/search", {
+                dbname: id,
+                query: {},
+                limit,
+                offset: 0,
+            }));
+            if (typeof browse.total === "number" && browse.total > 0) {
+                results.push({
+                    score: db.score,
+                    result: { ...browse, browse: true },
+                });
+            }
+        }
+        catch {
+            // DB not browsable — skip
+        }
+    }
+    if (results.length === 0) {
+        return null;
+    }
+    return {
+        ...content,
+        results,
+        note: "Smart-search found no rows for the full question; showing sample rows from the matched databases instead.",
+    };
+}
 server.setRequestHandler(CallToolRequestSchema, async (request) => {
     const { name, arguments: args } = request.params;
     try {
@@ -306,15 +471,53 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
             }
             case "yaohai-smart-search": {
                 const validated = YaohaiSmartSearchSchema.parse(args);
-                const body = {
-                    q: validated.q,
-                    limit: clampLimit(validated.limit, YAOHAI_LIMIT_DEFAULT, YAOHAI_LIMIT_MAX),
-                };
+                const limit = clampLimit(validated.limit, YAOHAI_LIMIT_DEFAULT, YAOHAI_LIMIT_MAX);
+                const body = { q: validated.q, limit };
                 if (validated.query) {
                     body.query = validated.query;
                 }
-                const content = await yaohaiPost("/g/mcp/yaohai/smart-search", body);
-                return ok(content);
+                try {
+                    const content = await yaohaiPost("/g/mcp/yaohai/smart-search", body);
+                    if (allSmartResultsEmpty(content)) {
+                        // Router matched DBs but searched the whole question in one field
+                        // (or picked an unsupported field). Retry with individual tokens
+                        // across the router's field + the DB's catalog search_fields.
+                        const retried = await retrySmartResultsWithTokens(content, validated.q, limit);
+                        if (retried) {
+                            return ok(retried);
+                        }
+                        // Keyword-only question (e.g. "集采"): show sample rows from the
+                        // matched databases.
+                        const browsed = await browseMatchedDatabases(content, limit);
+                        if (browsed) {
+                            return ok(browsed);
+                        }
+                        // Still nothing — try the global panorama.
+                        const fallback = await globalSearchContent(validated.q, validated.query, limit);
+                        return ok({
+                            question: validated.q,
+                            matched_databases: content.matched_databases,
+                            fallback: "global-search",
+                            note: "Smart-search router matched databases but returned no rows; fell back to global search.",
+                            ...fallback,
+                        });
+                    }
+                    return ok(content);
+                }
+                catch (error) {
+                    if (error instanceof ApiError) {
+                        // Backend router could not match any database (e.g. plain drug-name
+                        // question) — fall back to the global panorama search.
+                        const fallback = await globalSearchContent(validated.q, validated.query, limit);
+                        return ok({
+                            question: validated.q,
+                            fallback: "global-search",
+                            note: `Smart-search router failed (${error.message}); fell back to global search.`,
+                            ...fallback,
+                        });
+                    }
+                    throw error;
+                }
             }
             case "product-cn-fields": {
                 EmptyObjectSchema.parse(args ?? {});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kinginsun/mcp-drugsea",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "MCP server for DrugSea / Yaohai pharmaceutical databases: cross-db search, China marketed products (product_cn), and CDE registration review (reg_cn).",
   "type": "module",
   "main": "dist/index.js",