@opengeni/api-router 0.5.2 → 0.5.4

package/src/integrations/provider-domain.ts

@@ -7,7 +7,10 @@ import { HTTPException } from "hono/http-exception";
  * domain silently breaks the enable-time connectionRef domain match.
  */
 export function canonicalProviderDomain(value: string): string {
-  const canonical = value.trim().toLowerCase().replace(/^www\./, "");
+  const canonical = value
+    .trim()
+    .toLowerCase()
+    .replace(/^www\./, "");
   if (!canonical) {
     throw new HTTPException(400, { message: "providerDomain must not be empty" });
   }

package/src/mcp/documents.ts

@@ -4,11 +4,7 @@ import {
   searchDocuments,
   type DocumentServices,
 } from "@opengeni/documents";
-import {
-  createKnowledgeMemory,
-  listKnowledgeMemories,
-  type Database,
-} from "@opengeni/db";
+import { createKnowledgeMemory, listKnowledgeMemories, type Database } from "@opengeni/db";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import * as z from "zod/v4";
 
@@ -17,7 +13,20 @@ const SearchInputSchema = {
   baseIds: z.array(z.string().uuid()).optional(),
   limit: z.number().int().positive().max(50).optional(),
   mode: z.enum(["hybrid", "vector", "keyword"]).optional(),
-  sourceKinds: z.array(z.enum(["manual_upload", "meeting_transcript", "repository", "email", "chat", "document", "web", "other"])).optional(),
+  sourceKinds: z
+    .array(
+      z.enum([
+        "manual_upload",
+        "meeting_transcript",
+        "repository",
+        "email",
+        "chat",
+        "document",
+        "web",
+        "other",
+      ]),
+    )
+    .optional(),
   aclTags: z.array(z.string().min(1)).optional(),
 };
 
@@ -42,91 +51,142 @@ export function buildDocumentsMcpServer(
     version: "1.0.0",
   });
 
-  server.registerTool("list_document_bases", {
-    description: "List document bases available for retrieval.",
-    inputSchema: {},
-  }, async () => ({
-    content: [{ type: "text", text: JSON.stringify(await listDocumentBases(db, workspaceId)) }],
-  }));
+  server.registerTool(
+    "list_document_bases",
+    {
+      description: "List document bases available for retrieval.",
+      inputSchema: {},
+    },
+    async () => ({
+      content: [{ type: "text", text: JSON.stringify(await listDocumentBases(db, workspaceId)) }],
+    }),
+  );
 
-  server.registerTool("search_documents", {
-    description: "Search indexed documents with hybrid, vector, or keyword retrieval.",
-    inputSchema: SearchInputSchema,
-  }, async (input) => searchContent(db, workspaceId, documentServices, input));
+  server.registerTool(
+    "search_documents",
+    {
+      description: "Search indexed documents with hybrid, vector, or keyword retrieval.",
+      inputSchema: SearchInputSchema,
+    },
+    async (input) => searchContent(db, workspaceId, documentServices, input),
+  );
 
-  server.registerTool("knowledge_search", {
-    description: "Search company knowledge sources with optional base, source-kind, ACL, and retrieval-mode filters.",
-    inputSchema: SearchInputSchema,
-  }, async (input) => searchContent(db, workspaceId, documentServices, input));
+  server.registerTool(
+    "knowledge_search",
+    {
+      description:
+        "Search company knowledge sources with optional base, source-kind, ACL, and retrieval-mode filters.",
+      inputSchema: SearchInputSchema,
+    },
+    async (input) => searchContent(db, workspaceId, documentServices, input),
+  );
 
-  server.registerTool("fetch_document_chunk", {
-    description: "Fetch one indexed document chunk by id.",
-    inputSchema: {
-      chunkId: z.string().uuid(),
+  server.registerTool(
+    "fetch_document_chunk",
+    {
+      description: "Fetch one indexed document chunk by id.",
+      inputSchema: {
+        chunkId: z.string().uuid(),
+      },
     },
-  }, async ({ chunkId }) => {
-    const found = await getDocumentChunk(db, workspaceId, chunkId);
-    return {
-      content: [{ type: "text", text: found ? JSON.stringify(found) : `chunk not found: ${chunkId}` }],
-      isError: !found,
-    };
-  });
+    async ({ chunkId }) => {
+      const found = await getDocumentChunk(db, workspaceId, chunkId);
+      return {
+        content: [
+          { type: "text", text: found ? JSON.stringify(found) : `chunk not found: ${chunkId}` },
+        ],
+        isError: !found,
+      };
+    },
+  );
 
-  server.registerTool("knowledge_fetch", {
-    description: "Fetch one knowledge source chunk by id.",
-    inputSchema: {
-      chunkId: z.string().uuid(),
+  server.registerTool(
+    "knowledge_fetch",
+    {
+      description: "Fetch one knowledge source chunk by id.",
+      inputSchema: {
+        chunkId: z.string().uuid(),
+      },
     },
-  }, async ({ chunkId }) => {
-    const found = await getDocumentChunk(db, workspaceId, chunkId);
-    return {
-      content: [{ type: "text", text: found ? JSON.stringify(found) : `chunk not found: ${chunkId}` }],
-      isError: !found,
-    };
-  });
+    async ({ chunkId }) => {
+      const found = await getDocumentChunk(db, workspaceId, chunkId);
+      return {
+        content: [
+          { type: "text", text: found ? JSON.stringify(found) : `chunk not found: ${chunkId}` },
+        ],
+        isError: !found,
+      };
+    },
+  );
 
-  server.registerTool("memory_search", {
-    description: "Search approved company memory records.",
-    inputSchema: {
-      query: z.string().min(1).optional(),
-      kind: MemoryKindSchema.optional(),
-      scope: z.string().min(1).optional(),
-      limit: z.number().int().positive().max(100).optional(),
+  server.registerTool(
+    "memory_search",
+    {
+      description: "Search approved company memory records.",
+      inputSchema: {
+        query: z.string().min(1).optional(),
+        kind: MemoryKindSchema.optional(),
+        scope: z.string().min(1).optional(),
+        limit: z.number().int().positive().max(100).optional(),
+      },
     },
-  }, async ({ query, kind, scope, limit }) => ({
-    content: [{ type: "text", text: JSON.stringify(await listKnowledgeMemories(db, workspaceId, {
-      ...(query ? { query } : {}),
-      status: "approved",
-      ...(kind ? { kind } : {}),
-      ...(scope ? { scope } : {}),
-      ...(limit ? { limit } : {}),
-    })) }],
-  }));
+    async ({ query, kind, scope, limit }) => ({
+      content: [
+        {
+          type: "text",
+          text: JSON.stringify(
+            await listKnowledgeMemories(db, workspaceId, {
+              ...(query ? { query } : {}),
+              status: ["active", "approved"],
+              ...(kind ? { kind } : {}),
+              ...(scope ? { scope } : {}),
+              ...(limit ? { limit } : {}),
+            }),
+          ),
+        },
+      ],
+    }),
+  );
 
-  server.registerTool("memory_propose", {
-    description: "Propose a company memory record for human review.",
-    inputSchema: {
-      text: z.string().min(1),
-      kind: MemoryKindSchema.optional(),
-      scope: z.string().min(1).optional(),
-      sourceRefs: z.array(SourceRefSchema).optional(),
-      confidence: z.number().min(0).max(1).optional(),
-      metadata: z.record(z.string(), z.unknown()).optional(),
+  server.registerTool(
+    "memory_propose",
+    {
+      description: "Propose a company memory record for human review.",
+      inputSchema: {
+        text: z.string().min(1),
+        kind: MemoryKindSchema.optional(),
+        scope: z.string().min(1).optional(),
+        sourceRefs: z.array(SourceRefSchema).optional(),
+        confidence: z.number().min(0).max(1).optional(),
+        metadata: z.record(z.string(), z.unknown()).optional(),
+      },
     },
-  }, async ({ text, kind, scope, sourceRefs, confidence, metadata }) => ({
-    content: [{ type: "text", text: JSON.stringify(await createKnowledgeMemory(db, {
-      accountId,
-      workspaceId,
-      status: "proposed",
-      kind: kind ?? "semantic",
-      scope: scope ?? "workspace",
-      text,
-      sourceRefs: sourceRefs?.map((sourceRef) => ({ ...sourceRef, metadata: sourceRef.metadata ?? {} })) ?? [],
-      confidence: confidence ?? 0.5,
-      metadata: metadata ?? {},
-      createdBySessionId: options.createdBySessionId,
-    })) }],
-  }));
+    async ({ text, kind, scope, sourceRefs, confidence, metadata }) => ({
+      content: [
+        {
+          type: "text",
+          text: JSON.stringify(
+            await createKnowledgeMemory(db, {
+              accountId,
+              workspaceId,
+              status: "proposed",
+              kind: kind ?? "semantic",
+              scope: scope ?? "workspace",
+              text,
+              sourceRefs:
+                sourceRefs?.map((sourceRef) => ({
+                  ...sourceRef,
+                  metadata: sourceRef.metadata ?? {},
+                })) ?? [],
+              confidence: confidence ?? 0.5,
+              metadata: metadata ?? {},
+              createdBySessionId: options.createdBySessionId,
+            }),
+          ),
+        },
+      ],
+    }),
+  );
 
   return server;
 }
@@ -140,22 +200,41 @@ async function searchContent(
     baseIds?: string[] | undefined;
     limit?: number | undefined;
     mode?: "hybrid" | "vector" | "keyword" | undefined;
-    sourceKinds?: Array<"manual_upload" | "meeting_transcript" | "repository" | "email" | "chat" | "document" | "web" | "other"> | undefined;
+    sourceKinds?:
+      | Array<
+          | "manual_upload"
+          | "meeting_transcript"
+          | "repository"
+          | "email"
+          | "chat"
+          | "document"
+          | "web"
+          | "other"
+        >
+      | undefined;
     aclTags?: string[] | undefined;
   },
 ) {
   return {
-    content: [{
-      type: "text" as const,
-      text: JSON.stringify(await searchDocuments(db, {
-        workspaceId,
-        query: input.query,
-        ...(input.baseIds ? { baseIds: input.baseIds } : {}),
-        ...(input.limit ? { limit: input.limit } : {}),
-        ...(input.mode ? { mode: input.mode } : {}),
-        ...(input.sourceKinds ? { sourceKinds: input.sourceKinds } : {}),
-        ...(input.aclTags ? { aclTags: input.aclTags } : {}),
-      }, documentServices)),
-    }],
+    content: [
+      {
+        type: "text" as const,
+        text: JSON.stringify(
+          await searchDocuments(
+            db,
+            {
+              workspaceId,
+              query: input.query,
+              ...(input.baseIds ? { baseIds: input.baseIds } : {}),
+              ...(input.limit ? { limit: input.limit } : {}),
+              ...(input.mode ? { mode: input.mode } : {}),
+              ...(input.sourceKinds ? { sourceKinds: input.sourceKinds } : {}),
+              ...(input.aclTags ? { aclTags: input.aclTags } : {}),
+            },
+            documentServices,
+          ),
+        ),
+      },
+    ],
   };
 }