@tacitai/mcp-server 0.1.1 → 0.3.0

package/dist/tools/ingest.js

@@ -1,9 +1,9 @@
-"use strict";
 // SPDX-License-Identifier: MIT
 /**
  * Ingest tools:
- *   tacit_ingest_file — POST /api/v1/portfolios/:id/ingest
- *   tacit_ingest_url  — POST /api/v1/portfolios/:id/ingest-url
+ *   tacit_ingest_file        — POST /api/v1/portfolios/:id/ingest
+ *   tacit_ingest_url         — POST /api/v1/portfolios/:id/ingest-url
+ *   tacit_ingest_url_general — POST /api/v1/portfolios/:id/ingest-url-general (FEAT-175)
  *
  * Both require portfolio_id input.
  *
@@ -16,50 +16,11 @@
  * The guard is synchronous and checks literal IPs/hostnames in the URL
  * (no DNS resolution required for the guard to function).
  */
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || (function () {
-    var ownKeys = function(o) {
-        ownKeys = Object.getOwnPropertyNames || function (o) {
-            var ar = [];
-            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
-            return ar;
-        };
-        return ownKeys(o);
-    };
-    return function (mod) {
-        if (mod && mod.__esModule) return mod;
-        var result = {};
-        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
-        __setModuleDefault(result, mod);
-        return result;
-    };
-})();
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.INGEST_URL_TOOL_DEF = exports.ingestUrlInputSchema = exports.INGEST_FILE_TOOL_DEF = exports.ingestFileInputSchema = void 0;
-exports.validateIngestUrl = validateIngestUrl;
-exports.buildIngestFileRequest = buildIngestFileRequest;
-exports.handleIngestFile = handleIngestFile;
-exports.buildIngestUrlRequest = buildIngestUrlRequest;
-exports.handleIngestUrl = handleIngestUrl;
-const zod_1 = require("zod");
-const fs = __importStar(require("fs"));
-const path = __importStar(require("path"));
-const client_js_1 = require("../client.js");
+import { z } from 'zod';
+import * as fs from 'fs';
+import * as path from 'path';
+import { getClient } from '../client.js';
+import { apiErrorResult } from './errors.js';
 // ─── SSRF guard — synchronous IP/hostname check ───────────────────────────────
 /**
  * Parse an IPv4 string into octets, returning null if invalid.
@@ -121,7 +82,7 @@ function isPrivateHost(host) {
  *   destination-IP filtering at request time, after resolution. Do not rely on
  *   this function alone for SSRF protection.
  */
-function validateIngestUrl(rawUrl) {
+export function validateIngestUrl(rawUrl) {
     let parsed;
     try {
         parsed = new URL(rawUrl);
@@ -141,12 +102,12 @@ function validateIngestUrl(rawUrl) {
     return { ok: true };
 }
 // ─── tacit_ingest_file ────────────────────────────────────────────────────────
-exports.ingestFileInputSchema = zod_1.z.object({
-    portfolio_id: zod_1.z.string().min(1).describe('Portfolio UUID to ingest into'),
-    path: zod_1.z.string().min(1).describe('Local file path to ingest'),
-    title: zod_1.z.string().optional().describe('Optional title override'),
+export const ingestFileInputSchema = z.object({
+    portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+    path: z.string().min(1).describe('Local file path to ingest'),
+    title: z.string().optional().describe('Optional title override'),
 });
-exports.INGEST_FILE_TOOL_DEF = {
+export const INGEST_FILE_TOOL_DEF = {
     name: 'tacit_ingest_file',
     description: 'Ingest a local file into a Tacit portfolio. Route: POST /api/v1/portfolios/:id/ingest.',
     inputSchema: {
@@ -159,7 +120,7 @@ exports.INGEST_FILE_TOOL_DEF = {
         required: ['portfolio_id', 'path'],
     },
 };
-function buildIngestFileRequest(client, input) {
+export function buildIngestFileRequest(client, input) {
     return {
         async execute() {
             const { portfolio_id, path: filePath, title } = input;
@@ -187,20 +148,76 @@ function buildIngestFileRequest(client, input) {
         },
     };
 }
-async function handleIngestFile(args) {
-    const parsed = exports.ingestFileInputSchema.safeParse(args);
+export async function handleIngestFile(args) {
+    const parsed = ingestFileInputSchema.safeParse(args);
+    if (!parsed.success) {
+        return { content: [{ type: 'text', text: `Invalid input: ${parsed.error.message}` }], isError: true };
+    }
+    return buildIngestFileRequest(getClient(), parsed.data).execute();
+}
+// ─── tacit_ingest_url_general ────────────────────────────────────────────────
+export const ingestUrlGeneralInputSchema = z.object({
+    portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+    url: z.string().min(1).describe('Public HTTPS URL to ingest (PDF, HTML, text — any content type)'),
+    program_id: z.string().optional().describe('Optional program UUID within the portfolio'),
+    title: z.string().optional().describe('Optional title override'),
+});
+export const INGEST_URL_GENERAL_TOOL_DEF = {
+    name: 'tacit_ingest_url_general',
+    description: 'Ingest any public URL (PDF, HTML, text) into a Tacit portfolio. Not domain-restricted. ' +
+        'Route: POST /api/v1/portfolios/:id/ingest-url-general. ' +
+        'URL must use https:// scheme and must not resolve to private/loopback addresses (SSRF protection).',
+    inputSchema: {
+        type: 'object',
+        properties: {
+            portfolio_id: { type: 'string', description: 'Portfolio UUID' },
+            url: { type: 'string', description: 'Public HTTPS URL to ingest (PDF, HTML, text)' },
+            program_id: { type: 'string', description: 'Optional program UUID within the portfolio' },
+            title: { type: 'string', description: 'Optional title override' },
+        },
+        required: ['portfolio_id', 'url'],
+    },
+};
+export function buildIngestUrlGeneralRequest(client, input) {
+    return {
+        async execute() {
+            const { portfolio_id, url, program_id, title } = input;
+            const guard = validateIngestUrl(url);
+            if (!guard.ok) {
+                return {
+                    content: [{ type: 'text', text: `SSRF guard rejected URL: ${guard.reason}` }],
+                    isError: true,
+                };
+            }
+            try {
+                const body = { source_url: url };
+                if (program_id)
+                    body['program_id'] = program_id;
+                if (title)
+                    body['title'] = title;
+                const result = await client.post(`/api/v1/portfolios/${portfolio_id}/ingest-url-general`, body);
+                return { content: [{ type: 'text', text: JSON.stringify(result) }] };
+            }
+            catch (err) {
+                return apiErrorResult(err, `/api/v1/portfolios/:id/ingest-url-general`);
+            }
+        },
+    };
+}
+export async function handleIngestUrlGeneral(args) {
+    const parsed = ingestUrlGeneralInputSchema.safeParse(args);
     if (!parsed.success) {
         return { content: [{ type: 'text', text: `Invalid input: ${parsed.error.message}` }], isError: true };
     }
-    return buildIngestFileRequest((0, client_js_1.getClient)(), parsed.data).execute();
+    return buildIngestUrlGeneralRequest(getClient(), parsed.data).execute();
 }
 // ─── tacit_ingest_url ─────────────────────────────────────────────────────────
-exports.ingestUrlInputSchema = zod_1.z.object({
-    portfolio_id: zod_1.z.string().min(1).describe('Portfolio UUID to ingest into'),
-    url: zod_1.z.string().min(1).describe('Public HTTPS URL to ingest (must be https://, no private/loopback hosts)'),
-    program_id: zod_1.z.string().min(1).describe('Program UUID within the portfolio'),
+export const ingestUrlInputSchema = z.object({
+    portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+    url: z.string().min(1).describe('Public HTTPS URL to ingest (must be https://, no private/loopback hosts)'),
+    program_id: z.string().min(1).describe('Program UUID within the portfolio'),
 });
-exports.INGEST_URL_TOOL_DEF = {
+export const INGEST_URL_TOOL_DEF = {
     name: 'tacit_ingest_url',
     description: 'Ingest a public URL into a Tacit portfolio. Route: POST /api/v1/portfolios/:id/ingest-url. ' +
         'URL must use https:// scheme and must not resolve to private/loopback addresses (SSRF protection).',
@@ -214,7 +231,7 @@ exports.INGEST_URL_TOOL_DEF = {
         required: ['portfolio_id', 'url', 'program_id'],
     },
 };
-function buildIngestUrlRequest(client, input) {
+export function buildIngestUrlRequest(client, input) {
     return {
         async execute() {
             const { portfolio_id, url, program_id } = input;
@@ -239,21 +256,37 @@ function buildIngestUrlRequest(client, input) {
         },
     };
 }
-async function handleIngestUrl(args) {
-    const parsed = exports.ingestUrlInputSchema.safeParse(args);
+export async function handleIngestUrl(args) {
+    const parsed = ingestUrlInputSchema.safeParse(args);
     if (!parsed.success) {
         return { content: [{ type: 'text', text: `Invalid input: ${parsed.error.message}` }], isError: true };
     }
-    return buildIngestUrlRequest((0, client_js_1.getClient)(), parsed.data).execute();
+    return buildIngestUrlRequest(getClient(), parsed.data).execute();
 }
-// ─── Shared error helper ──────────────────────────────────────────────────────
-function apiErrorResult(err, endpoint) {
-    if (err instanceof client_js_1.TacitApiError) {
-        return {
-            content: [{ type: 'text', text: `API error ${err.statusCode} on ${endpoint}: ${err.body}` }],
-            isError: true,
-        };
-    }
-    const msg = err instanceof Error ? err.message : String(err);
-    return { content: [{ type: 'text', text: `Error: ${msg}` }], isError: true };
+export function registerIngestTools(server, getClientFn) {
+    server.tool('tacit_ingest_file', 'Ingest a local file into a Tacit portfolio. Route: POST /api/v1/portfolios/:id/ingest.', {
+        portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+        path: z.string().min(1).describe('Local file path to ingest'),
+        title: z.string().optional().describe('Optional title override'),
+    }, async (args) => {
+        const client = await getClientFn();
+        return buildIngestFileRequest(client, args).execute();
+    });
+    server.tool('tacit_ingest_url', 'Ingest a public URL into a Tacit portfolio (SSRF-guarded). Route: POST /api/v1/portfolios/:id/ingest-url.', {
+        portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+        url: z.string().min(1).describe('Public HTTPS URL to ingest'),
+        program_id: z.string().min(1).describe('Program UUID within the portfolio'),
+    }, async (args) => {
+        const client = await getClientFn();
+        return buildIngestUrlRequest(client, args).execute();
+    });
+    server.tool('tacit_ingest_url_general', 'Ingest any public URL (PDF, HTML, text) into a Tacit portfolio — not domain-restricted. Route: POST /api/v1/portfolios/:id/ingest-url-general.', {
+        portfolio_id: z.string().min(1).describe('Portfolio UUID to ingest into'),
+        url: z.string().min(1).describe('Public HTTPS URL to ingest (PDF, HTML, text)'),
+        program_id: z.string().optional().describe('Optional program UUID within the portfolio'),
+        title: z.string().optional().describe('Optional title override'),
+    }, async (args) => {
+        const client = await getClientFn();
+        return buildIngestUrlGeneralRequest(client, args).execute();
+    });
 }