ynosis-agent-mcp 0.1.0

package/README.md

@@ -0,0 +1,35 @@
+# ynosis-agent-mcp
+
+Ynosis Agent API 用の **MCP（stdio）**。Cursor など Node がある環境向け。
+
+ソース clone は不要です。CLI は別パッケージ [ynosis-agent](https://pypi.org/project/ynosis-agent/)（PyPI）。
+
+## 前提
+
+- Node.js 18 以降（`npx` が付属）
+- プロフィール（… → Agent API）で発行した `yna_…` キー
+
+`uv` / `uvx` は不要です。
+
+## Cursor（mcp.json）
+
+```json
+{
+  "mcpServers": {
+    "ynosis-agent": {
+      "command": "npx",
+      "args": ["-y", "ynosis-agent-mcp"],
+      "env": {
+        "YNOSIS_AGENT_API_KEY": "yna_REPLACE_ME",
+        "YNOSIS_AGENT_API_BASE": "https://ynosis.jp"
+      }
+    }
+  }
+}
+```
+
+保存後 **Cursor を再起動**し、Customize → MCPs で有効か、チャットのツールに `search_facilities` が出るかを確認してください。`npx` が Cursor から見えないときは `command` を `which npx` の絶対パスにします。
+
+接続先の既定は `https://ynosis.jp`（リダイレクトを追います）。
+
+手順の詳細: <https://ynosis.jp/notes/usage-037>

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "ynosis-agent-mcp",
+  "version": "0.1.0",
+  "description": "MCP stdio server for the Ynosis Agent API (facilities and spring analyses)",
+  "type": "module",
+  "bin": {
+    "ynosis-agent-mcp": "./src/server.mjs"
+  },
+  "files": [
+    "src",
+    "README.md"
+  ],
+  "engines": {
+    "node": ">=18"
+  },
+  "license": "UNLICENSED",
+  "homepage": "https://ynosis.jp/notes/usage-037",
+  "keywords": [
+    "ynosis",
+    "onsen",
+    "mcp",
+    "agent"
+  ],
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.12.1",
+    "zod": "^3.24.2"
+  }
+}

package/src/client.mjs

@@ -0,0 +1,238 @@
+const DEFAULT_BASE = "https://ynosis.jp";
+
+function resolveKey() {
+  return process.env.YNOSIS_AGENT_API_KEY || process.env.AGENT_API_KEY || "";
+}
+
+function resolveBase() {
+  const raw =
+    process.env.YNOSIS_AGENT_API_BASE ||
+    process.env.AGENT_API_BASE ||
+    process.env.API_BASE ||
+    DEFAULT_BASE;
+  return raw.replace(/\/$/, "");
+}
+
+function imageContentType(filePath) {
+  const suffix = filePath.toLowerCase().split(".").pop();
+  if (suffix === "png") return "image/png";
+  if (["webp", "bmp", "tif", "tiff"].includes(suffix)) return `image/${suffix}`;
+  if (suffix === "heic" || suffix === "heif") return "image/heic";
+  if (suffix === "pdf") return "application/pdf";
+  return "image/jpeg";
+}
+
+export class AgentApiError extends Error {
+  constructor(statusCode, detail) {
+    super(`HTTP ${statusCode}: ${detail}`);
+    this.statusCode = statusCode;
+    this.detail = detail;
+  }
+}
+
+export class AgentApiClient {
+  constructor({ baseUrl, apiKey, timeoutMs = 120_000 } = {}) {
+    this.baseUrl = (baseUrl || resolveBase()).replace(/\/$/, "");
+    this.apiKey = apiKey ?? resolveKey();
+    this.timeoutMs = timeoutMs;
+  }
+
+  requireKey() {
+    if (!this.apiKey) {
+      throw new Error(
+        "API キーが未設定です。YNOSIS_AGENT_API_KEY または AGENT_API_KEY に yna_… を設定してください"
+      );
+    }
+  }
+
+  headers(jsonBody = true) {
+    this.requireKey();
+    const headers = { "X-API-Key": this.apiKey };
+    if (jsonBody) headers["Content-Type"] = "application/json";
+    return headers;
+  }
+
+  async request(method, path, { json, form } = {}) {
+    const url = `${this.baseUrl}${path}`;
+    const ctrl = new AbortController();
+    const timer = setTimeout(() => ctrl.abort(), form ? Math.max(this.timeoutMs, 180_000) : this.timeoutMs);
+    try {
+      const init = { method, headers: form ? { "X-API-Key": this.apiKey } : this.headers(json != null), signal: ctrl.signal };
+      if (form) init.body = form;
+      else if (json != null) init.body = JSON.stringify(json);
+      const res = await fetch(url, init);
+      const text = await res.text();
+      if (!res.ok) {
+        let detail = text;
+        try {
+          const body = JSON.parse(text);
+          if (body && typeof body === "object" && "detail" in body) detail = String(body.detail);
+        } catch {
+          /* keep text */
+        }
+        throw new AgentApiError(res.status, detail);
+      }
+      if (!text) return { status: res.status, json: null, headers: res.headers };
+      return { status: res.status, json: JSON.parse(text), headers: res.headers };
+    } finally {
+      clearTimeout(timer);
+    }
+  }
+
+  async registerFacility(payload) {
+    const res = await this.request("POST", "/api/agent/facilities", { json: payload });
+    return { status_code: res.status, facility: res.json };
+  }
+
+  async searchFacilities(params = {}) {
+    const clean = Object.fromEntries(
+      Object.entries(params).filter(([, v]) => v != null && v !== "")
+    );
+    const qs = new URLSearchParams(clean).toString();
+    const path = `/api/agent/facilities/search${qs ? `?${qs}` : ""}`;
+    const res = await this.request("GET", path);
+    const total = res.headers.get("X-Total-Count");
+    const items = res.json;
+    return { items, total_count: total != null ? Number(total) : (items?.length ?? 0) };
+  }
+
+  async getFacility(facilityId) {
+    const res = await this.request("GET", `/api/agent/facilities/${facilityId}`);
+    return res.json;
+  }
+
+  async registerFacilitiesBatch(facilities, { continueOnError = true } = {}) {
+    const results = [];
+    let created = 0;
+    let updated = 0;
+    let failed = 0;
+    for (let i = 0; i < facilities.length; i += 1) {
+      const payload = facilities[i];
+      try {
+        const one = await this.registerFacility(payload);
+        const statusCode = Number(one.status_code);
+        if (statusCode === 201) {
+          created += 1;
+        } else {
+          updated += 1;
+        }
+        results.push({
+          index: i,
+          ok: true,
+          action: statusCode === 201 ? "created" : "updated",
+          status_code: statusCode,
+          facility_id: one.facility?.id,
+          name: one.facility?.name,
+        });
+      } catch (e) {
+        failed += 1;
+        results.push({
+          index: i,
+          ok: false,
+          error: e instanceof AgentApiError ? e.detail : String(e),
+          status_code: e instanceof AgentApiError ? e.statusCode : undefined,
+          name: payload?.name,
+        });
+        if (!continueOnError) break;
+      }
+    }
+    return { created, updated, failed, results };
+  }
+
+  async updateFacility(facilityId, payload) {
+    const res = await this.request("PUT", `/api/facilities/${facilityId}`, { json: payload });
+    return res.json;
+  }
+
+  async registerSource(payload, imagePath) {
+    const { readFile } = await import("node:fs/promises");
+    const { basename } = await import("node:path");
+    if (!imagePath) {
+      const res = await this.request("POST", "/api/agent/register", { json: payload });
+      return res.json;
+    }
+    const buf = await readFile(imagePath);
+    const form = new FormData();
+    form.append("payload", JSON.stringify(payload));
+    form.append("file", new Blob([buf], { type: imageContentType(imagePath) }), basename(imagePath));
+    const res = await this.request("POST", "/api/agent/register", { form });
+    return res.json;
+  }
+
+  async analyzeImage(imagePath) {
+    const { readFile } = await import("node:fs/promises");
+    const { basename } = await import("node:path");
+    const buf = await readFile(imagePath);
+    const form = new FormData();
+    form.append("file", new Blob([buf], { type: imageContentType(imagePath) }), basename(imagePath));
+    const res = await this.request("POST", "/api/agent/analyze", { form });
+    return res.json;
+  }
+
+  async getParsePrompt(rawText) {
+    const res = await this.request("POST", "/api/agent/parse-prompt", { json: { raw_text: rawText } });
+    return res.json;
+  }
+
+  async parseTranscript(rawText, llmJson) {
+    const body = { raw_text: rawText };
+    if (llmJson != null) body.llm_json = llmJson;
+    const res = await this.request("POST", "/api/agent/parse", { json: body });
+    return res.json;
+  }
+
+  async findSimilarSources(payload) {
+    const res = await this.request("POST", "/api/sources/similar", { json: payload });
+    return res.json;
+  }
+
+  async uploadSourceImage(sourceId, imagePath) {
+    const { readFile } = await import("node:fs/promises");
+    const { basename } = await import("node:path");
+    const buf = await readFile(imagePath);
+    const form = new FormData();
+    form.append("file", new Blob([buf], { type: imageContentType(imagePath) }), basename(imagePath));
+    const res = await this.request("POST", `/api/sources/${sourceId}/image`, { form });
+    return res.json;
+  }
+}
+
+export function prepareParsedForSave(parsed) {
+  const out = { ...parsed };
+  const unit = out.dissolved_material_unit || "mg";
+  const dm = out.dissolved_material;
+  if (typeof dm === "number") {
+    out.dissolved_material = unit === "g" ? Math.round(dm * 1000 * 10) / 10 : Math.round(dm * 10) / 10;
+  }
+  delete out.dissolved_material_unit;
+  for (const key of [
+    "is_composition_table",
+    "ion_total_warnings",
+    "bisulfate_keyword_detected",
+    "metasilicate_ion_keyword_detected",
+    "metaborate_ion_keyword_detected",
+    "trace_quantified_detected",
+    "name",
+    "address",
+    "prefecture",
+  ]) {
+    delete out[key];
+  }
+  return out;
+}
+
+export const FACILITY_TAG_OPTIONS = [
+  "足元湧出",
+  "掛け流し",
+  "秘湯",
+  "野湯",
+  "露天あり",
+  "貸切湯あり",
+  "日帰り入浴可",
+  "共同浴場",
+  "造成泉",
+  "モール泉",
+  "混浴",
+  "ぬる湯",
+  "冷鉱泉",
+];

package/src/server.mjs

@@ -0,0 +1,216 @@
+#!/usr/bin/env node
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+import {
+  AgentApiClient,
+  AgentApiError,
+  FACILITY_TAG_OPTIONS,
+  prepareParsedForSave,
+} from "./client.mjs";
+
+function client() {
+  return new AgentApiClient();
+}
+
+function dump(data) {
+  return JSON.stringify(data, null, 2);
+}
+
+function errText(e) {
+  return dump({ error: e instanceof Error ? e.message : String(e) });
+}
+
+function textResult(text) {
+  return { content: [{ type: "text", text }] };
+}
+
+const server = new McpServer({
+  name: "ynosis-agent",
+  version: "0.1.0",
+  instructions:
+    "Ynosis（温泉データベース）への施設・成分表登録・検索用ツール。" +
+    "検索は search_facilities / get_facility。" +
+    "施設登録は register_facilities_batch または register_facility。" +
+    "成分表は手元 Vision なら get_parse_prompt → 手元 LLM → parse_transcript、" +
+    "または analyze_source_image（Ynosis OCR）。" +
+    "その後 register_source。新規源泉は image_path 必須。" +
+    "OCR / サーバー抽出の parse は並列禁止。" +
+    "YNOSIS_AGENT_API_KEY（または AGENT_API_KEY）が必要。" +
+    "CLI は PyPI の ynosis-agent。CSV ファイル一括は CLI / Python MCP。",
+});
+
+const dict = z.record(z.any());
+
+server.tool("register_facility", "施設を1件登録する（FacilityCreate 形式）。", { facility: dict }, async ({ facility }) => {
+  try {
+    return textResult(dump(await client().registerFacility(facility)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool(
+  "search_facilities",
+  "施設を検索する（未承認の Agent 登録も含む）。",
+  {
+    q: z.string().optional(),
+    prefecture: z.string().optional(),
+    area: z.string().optional(),
+    onsen_name: z.string().optional(),
+    spring_type: z.string().optional(),
+    tags: z.string().optional(),
+    has_analysis: z.string().optional(),
+    status: z.string().optional(),
+    limit: z.number().optional(),
+    skip: z.number().optional(),
+    sort: z.string().optional(),
+  },
+  async (args) => {
+    try {
+      return textResult(
+        dump(
+          await client().searchFacilities({
+            q: args.q,
+            prefecture: args.prefecture,
+            area: args.area,
+            onsen_name: args.onsen_name,
+            spring_type: args.spring_type,
+            tags: args.tags,
+            has_analysis: args.has_analysis,
+            status: args.status,
+            limit: args.limit ?? 20,
+            skip: args.skip ?? 0,
+            sort: args.sort ?? "default",
+          })
+        )
+      );
+    } catch (e) {
+      return textResult(errText(e));
+    }
+  }
+);
+
+server.tool("get_facility", "施設詳細を取得する（未承認可）。", { facility_id: z.number() }, async ({ facility_id }) => {
+  try {
+    return textResult(dump(await client().getFacility(facility_id)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool(
+  "register_facilities_batch",
+  "施設を一括登録する（JSON 配列。CSV ファイルは ynosis-agent CLI）。",
+  {
+    facilities: z.array(dict).optional(),
+    continue_on_error: z.boolean().optional(),
+    dry_run: z.boolean().optional(),
+  },
+  async ({ facilities, continue_on_error, dry_run }) => {
+    try {
+      if (!facilities?.length) {
+        return textResult(dump({ error: "facilities 配列を指定してください。CSV は ynosis-agent CLI を使ってください" }));
+      }
+      if (dry_run) {
+        return textResult(dump({ dry_run: true, count: facilities.length, facilities }));
+      }
+      return textResult(
+        dump(await client().registerFacilitiesBatch(facilities, { continueOnError: continue_on_error ?? true }))
+      );
+    } catch (e) {
+      return textResult(errText(e));
+    }
+  }
+);
+
+server.tool(
+  "update_facility",
+  "既存施設を部分更新する（Agent キー時は編集依頼）。",
+  { facility_id: z.number(), updates: dict },
+  async ({ facility_id, updates }) => {
+    try {
+      return textResult(dump(await client().updateFacility(facility_id, updates)));
+    } catch (e) {
+      return textResult(errText(e));
+    }
+  }
+);
+
+server.tool(
+  "register_source",
+  "成分表付きで施設・源泉を登録する。新規源泉は image_path（成分表画像）必須。",
+  { payload: dict, image_path: z.string().optional() },
+  async ({ payload, image_path }) => {
+    try {
+      return textResult(dump(await client().registerSource(payload, image_path)));
+    } catch (e) {
+      return textResult(e instanceof AgentApiError ? dump({ error: String(e) }) : errText(e));
+    }
+  }
+);
+
+server.tool("get_parse_prompt", "書き起こし全文から本番抽出プロンプト（system/user）を返す。LLM は呼ばない。", { raw_text: z.string() }, async ({ raw_text }) => {
+  try {
+    return textResult(dump(await client().getParsePrompt(raw_text)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool(
+  "parse_transcript",
+  "書き起こし全文を parsed にする。llm_json があれば後処理のみ。DI は使わない。",
+  { raw_text: z.string(), llm_json: dict.optional() },
+  async ({ raw_text, llm_json }) => {
+    try {
+      return textResult(dump(await client().parseTranscript(raw_text, llm_json)));
+    } catch (e) {
+      return textResult(errText(e));
+    }
+  }
+);
+
+server.tool("analyze_source_image", "成分表画像を Ynosis OCR で解析する（DB 未書込）。並列禁止。", { image_path: z.string() }, async ({ image_path }) => {
+  try {
+    return textResult(dump(await client().analyzeImage(image_path)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool("prepare_parsed", "register_source 用に dissolved_material を mg 換算する。", { parsed: dict }, async ({ parsed }) => {
+  try {
+    return textResult(dump(prepareParsedForSave(parsed)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool("find_similar_sources", "登録前に成分が近い既存源泉を検索する。", { source: dict }, async ({ source }) => {
+  try {
+    return textResult(dump(await client().findSimilarSources(source)));
+  } catch (e) {
+    return textResult(errText(e));
+  }
+});
+
+server.tool("get_tag_options", "登録可能な施設タグ一覧を返す。", {}, async () =>
+  textResult(dump({ tags: FACILITY_TAG_OPTIONS }))
+);
+
+server.tool(
+  "upload_source_image",
+  "成分表の証憑画像をアップロードする。",
+  { source_id: z.number(), image_path: z.string() },
+  async ({ source_id, image_path }) => {
+    try {
+      return textResult(dump(await client().uploadSourceImage(source_id, image_path)));
+    } catch (e) {
+      return textResult(errText(e));
+    }
+  }
+);
+
+const transport = new StdioServerTransport();
+await server.connect(transport);