@lightsage/mcp 0.1.7

package/src/mcp-server/tools.ts

@@ -0,0 +1,485 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import {
+  ShapeOutput,
+  ZodRawShapeCompat,
+} from "@modelcontextprotocol/sdk/server/zod-compat.js";
+import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
+import {
+  CallToolResult,
+  ServerNotification,
+  ServerRequest,
+} from "@modelcontextprotocol/sdk/types.js";
+import * as z from "zod";
+import { LightsageMcpCore } from "../core.js";
+import { ConsoleLogger } from "./console-logger.js";
+import {
+  MCPServerFlags,
+  resolveMCPServerAPIKey,
+} from "./flags.js";
+import { MCPScope, mcpScopes } from "./scopes.js";
+import { valueToBase64 } from "./shared.js";
+
+export type MCPToolAnnotationFilter = {
+  readOnlyHint?: boolean;
+  destructiveHint?: boolean;
+  idempotentHint?: boolean;
+  openWorldHint?: boolean;
+};
+
+const VALID_ANNOTATIONS = [
+  "readOnly",
+  "destructive",
+  "idempotent",
+  "openWorld",
+] as const;
+type AnnotationName = typeof VALID_ANNOTATIONS[number];
+
+const annotationToKey: Record<AnnotationName, keyof MCPToolAnnotationFilter> = {
+  readOnly: "readOnlyHint",
+  destructive: "destructiveHint",
+  idempotent: "idempotentHint",
+  openWorld: "openWorldHint",
+};
+
+export function buildAnnotationFilter(
+  annotations: string[] | undefined,
+): MCPToolAnnotationFilter {
+  const filter: MCPToolAnnotationFilter = {};
+  if (!annotations || annotations.length === 0) {
+    return filter;
+  }
+  for (const a of annotations) {
+    if (!VALID_ANNOTATIONS.includes(a as AnnotationName)) {
+      throw new Error(
+        `Invalid annotation filter: "${a}". Valid values are: ${
+          VALID_ANNOTATIONS.join(", ")
+        }`,
+      );
+    }
+  }
+  for (const name of VALID_ANNOTATIONS) {
+    filter[annotationToKey[name]] = annotations.includes(name);
+  }
+  return filter;
+}
+
+export type ToolDefinition<
+  Args extends undefined | ZodRawShapeCompat = undefined,
+> = Args extends ZodRawShapeCompat ? {
+    name: string;
+    description: string;
+    scopes?: MCPScope[];
+    args: Args;
+    annotations: {
+      title: string;
+      destructiveHint: boolean;
+      idempotentHint: boolean;
+      openWorldHint: boolean;
+      readOnlyHint: boolean;
+    };
+    tool: (
+      client: LightsageMcpCore,
+      args: ShapeOutput<Args>,
+      extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+    ) => CallToolResult | Promise<CallToolResult>;
+  }
+  : {
+    name: string;
+    description: string;
+    scopes?: MCPScope[];
+    args?: undefined;
+    annotations: {
+      title: string;
+      destructiveHint: boolean;
+      idempotentHint: boolean;
+      openWorldHint: boolean;
+      readOnlyHint: boolean;
+    };
+    tool: (
+      client: LightsageMcpCore,
+      extra: RequestHandlerExtra<ServerRequest, ServerNotification>,
+    ) => CallToolResult | Promise<CallToolResult>;
+  };
+
+// Optional function to assist with formatting tool results
+export async function formatResult(
+  response: Response,
+): Promise<CallToolResult> {
+  let content: CallToolResult["content"] = [];
+  const contentType = response?.headers.get("content-type") ?? "";
+
+  if (contentType.startsWith("image/")) {
+    const data = await valueToBase64(await response.arrayBuffer());
+    content = data == null
+      ? []
+      : [{ type: "image", data, mimeType: contentType }];
+  } else if (contentType.startsWith("audio/")) {
+    const data = await valueToBase64(await response.arrayBuffer());
+    content = data == null
+      ? []
+      : [{ type: "audio", data, mimeType: contentType }];
+  } else {
+    const text = await response.text();
+    content = [{ type: "text", text }];
+  }
+
+  return response.ok ? { content } : { content, isError: true };
+}
+
+export function createRegisterTool(
+  logger: ConsoleLogger,
+  server: McpServer,
+  getSDK: () => LightsageMcpCore,
+  allowedScopes: Set<MCPScope>,
+  allowedTools?: Set<string>,
+  dynamic?: boolean,
+  annotationFilter?: MCPToolAnnotationFilter,
+): [
+  <A extends ZodRawShapeCompat | undefined>(tool: ToolDefinition<A>) => void,
+  Array<{ name: string; description: string }>,
+  Map<string, ToolDefinition<ZodRawShapeCompat | undefined>>,
+] {
+  const tools: Array<{ name: string; description: string }> = [];
+  const toolMap = new Map<
+    string,
+    ToolDefinition<ZodRawShapeCompat | undefined>
+  >();
+  const registerTool = <A extends ZodRawShapeCompat | undefined>(
+    tool: ToolDefinition<A>,
+  ): void => {
+    if (allowedTools && !allowedTools.has(tool.name)) {
+      return;
+    }
+
+    const scopes = tool.scopes ?? [];
+    if (allowedScopes.size > 0 && scopes.length === 0) {
+      return;
+    }
+
+    if (
+      allowedScopes.size > 0
+      && !scopes.every((s: MCPScope) => allowedScopes.has(s))
+    ) {
+      return;
+    }
+
+    if (annotationFilter) {
+      const a = tool.annotations;
+      if (
+        annotationFilter.readOnlyHint !== undefined
+        && a.readOnlyHint !== annotationFilter.readOnlyHint
+      ) {
+        return;
+      }
+      if (
+        annotationFilter.destructiveHint !== undefined
+        && a.destructiveHint !== annotationFilter.destructiveHint
+      ) {
+        return;
+      }
+      if (
+        annotationFilter.idempotentHint !== undefined
+        && a.idempotentHint !== annotationFilter.idempotentHint
+      ) {
+        return;
+      }
+      if (
+        annotationFilter.openWorldHint !== undefined
+        && a.openWorldHint !== annotationFilter.openWorldHint
+      ) {
+        return;
+      }
+    }
+
+    toolMap.set(
+      tool.name,
+      tool as ToolDefinition<ZodRawShapeCompat | undefined>,
+    );
+
+    if (dynamic) {
+      logger.debug("Collected tool for dynamic mode", { name: tool.name });
+      tools.push({ name: tool.name, description: tool.description });
+      return;
+    }
+
+    if (tool.args) {
+      server.registerTool(
+        tool.name,
+        {
+          description: tool.description,
+          inputSchema: tool.args,
+          annotations: tool.annotations,
+        },
+        async (args, ctx) => {
+          return tool.tool(getSDK(), args, ctx);
+        },
+      );
+    } else {
+      server.registerTool(
+        tool.name,
+        {
+          description: tool.description,
+          annotations: tool.annotations,
+        },
+        async (ctx) => {
+          return tool.tool(getSDK(), ctx);
+        },
+      );
+    }
+
+    logger.debug("Registered tool", { name: tool.name });
+    tools.push({ name: tool.name, description: tool.description });
+  };
+
+  return [registerTool, tools, toolMap];
+}
+
+function matchesSearchTerms(
+  terms: string[],
+  name: string,
+  def: ToolDefinition<ZodRawShapeCompat | undefined>,
+): boolean {
+  if (terms.length === 0) {
+    return true;
+  }
+
+  const nameLower = name.toLowerCase();
+  const descLower = def.description.toLowerCase();
+  const scopesLower = def.scopes?.flatMap((s: unknown) => {
+    return typeof s === "string" ? [s.toLowerCase()] : [];
+  }) || [];
+
+  return terms.some((term) =>
+    nameLower.includes(term)
+    || descLower.includes(term)
+    || scopesLower.some((s) => s.includes(term))
+  );
+}
+
+export function registerDynamicTools(
+  logger: ConsoleLogger,
+  server: McpServer,
+  getSDK: () => LightsageMcpCore,
+  toolMap: Map<string, ToolDefinition<ZodRawShapeCompat | undefined>>,
+  allowedScopes: Set<MCPScope>,
+): void {
+  // 1. list_tools
+  server.registerTool("list_tools", {
+    description:
+      "List available tools. Optionally filter by search terms that match against tool name, description, and scopes.",
+    inputSchema: {
+      search_terms: z.array(z.string()).optional().describe(
+        [
+          "Filter the list of tools.",
+          "Each term is matched case-insensitively as a substring against tool name, description, and scopes.",
+          "Multiple terms are combined with a logical OR: a tool is included if ANY term matches ANY field.",
+          "If not provided, all tools are returned.",
+        ].join("\n"),
+      ),
+    },
+    annotations: {
+      title: "List Tools",
+      readOnlyHint: true,
+      destructiveHint: false,
+      idempotentHint: true,
+      openWorldHint: false,
+    },
+  }, async (args) => {
+    const terms = (args.search_terms ?? []).map((t) => t.toLowerCase());
+
+    const tools: Array<
+      { name: string; description: string; scopes?: MCPScope[] | undefined }
+    > = [];
+    for (const [name, def] of toolMap) {
+      if (!matchesSearchTerms(terms, name, def)) {
+        continue;
+      }
+
+      tools.push({
+        name: name,
+        description: def.description,
+        scopes: def.scopes,
+      });
+    }
+    return {
+      content: [{ type: "text", text: JSON.stringify(tools, null, 2) }],
+    };
+  });
+  logger.debug("Registered dynamic meta-tool", { name: "list_tools" });
+
+  // 2. describe_tool
+  server.registerTool("describe_tool_input", {
+    description:
+      "Get the input schema for one or more tools. It is a good idea to call this tool first to understand how to successfully call execute_tool.",
+    inputSchema: {
+      tool_names: z.array(z.string()).describe(
+        "The names of the tools to describe",
+      ),
+    },
+    annotations: {
+      title: "Describe Tool",
+      readOnlyHint: true,
+      destructiveHint: false,
+      idempotentHint: true,
+      openWorldHint: false,
+    },
+  }, async (args) => {
+    const parts: string[] = [];
+    const unknown: string[] = [];
+
+    if (args.tool_names.length === 0) {
+      return { content: [{ type: "text", text: "No tool names provided." }] };
+    }
+
+    for (const toolName of args.tool_names) {
+      const def = toolMap.get(toolName);
+      if (!def) {
+        unknown.push(toolName);
+        continue;
+      }
+
+      let schemaText = `<input_schema tool="${toolName}">\n\n`;
+      if (def.args) {
+        const jsonSchema = z.toJSONSchema(z.object(def.args), {
+          target: "draft-2020-12",
+          io: "input",
+          unrepresentable: "any",
+        });
+        schemaText += JSON.stringify(jsonSchema, null, 2);
+      } else {
+        schemaText += "This tool takes no input parameters.";
+      }
+      schemaText += `\n\n</input_schema>`;
+      parts.push(schemaText);
+    }
+
+    if (unknown.length > 0) {
+      parts.push(`Unknown tools: ${unknown.join(", ")}`);
+    }
+
+    return { content: [{ type: "text", text: parts.join("\n\n") }] };
+  });
+  logger.debug("Registered dynamic meta-tool", { name: "describe_tool_input" });
+
+  // 3. execute_tool
+  server.registerTool("execute_tool", {
+    description:
+      "Execute a tool by name with its arguments. If executing a given tool for the first time, it is recommended to call describe_tool_input first to understand the expected `arguments` shape.",
+    inputSchema: {
+      name: z.string().describe("The name of the tool to execute"),
+      arguments: z.looseObject({}).optional().describe(
+        "Arguments for the target tool as a JSON object, matching the schema returned by describe_tool_input.",
+      ),
+    },
+    annotations: {
+      title: "Execute Tool",
+      readOnlyHint: false,
+      destructiveHint: true,
+      idempotentHint: false,
+      openWorldHint: true,
+    },
+  }, async (args, ctx) => {
+    const def = toolMap.get(args.name);
+    if (!def) {
+      return {
+        content: [{ type: "text", text: `Unknown tool: ${args.name}` }],
+        isError: true,
+      };
+    }
+
+    let validatedInput: Record<string, unknown> = {};
+    if (def.args) {
+      const vres = z.object(def.args).safeParse(args.arguments ?? {});
+      if (vres.success) {
+        validatedInput = vres.data;
+      } else {
+        const issues = z.prettifyError(vres.error);
+
+        return {
+          content: [{
+            type: "text",
+            text:
+              `Invalid input for tool ${args.name}:\n<issues>\n${issues}\n</issues>`,
+          }],
+          isError: true,
+        };
+      }
+    }
+
+    try {
+      if (def.args) {
+        return await def.tool(getSDK(), validatedInput, ctx);
+      } else {
+        return await def.tool(getSDK(), ctx);
+      }
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      return {
+        content: [{
+          type: "text",
+          text: `Error executing tool ${args.name}: ${message}`,
+        }],
+        isError: true,
+      };
+    }
+  });
+  logger.debug("Registered dynamic meta-tool", { name: "execute_tool" });
+
+  // 4. list_scopes (only when the server has scopes defined)
+  if (mcpScopes.length > 0) {
+    const scopes = allowedScopes.size > 0
+      ? [...allowedScopes].sort()
+      : [...mcpScopes];
+
+    server.registerTool("list_scopes", {
+      description: [
+        "List the scopes available on this server.",
+        "Scopes are categories that group related tools together.",
+        "They can be used as search terms when filtering the list of tools.",
+      ].join("\n"),
+      annotations: {
+        title: "List Scopes",
+        readOnlyHint: true,
+        destructiveHint: false,
+        idempotentHint: true,
+        openWorldHint: false,
+      },
+    }, async () => {
+      const lines = scopes.map((s) => `- ${s}`);
+      return { content: [{ type: "text", text: lines.join("\n") }] };
+    });
+    logger.debug("Registered dynamic meta-tool", { name: "list_scopes" });
+  }
+}
+export function buildSDK(
+  headers: Headers,
+  cliFlags: MCPServerFlags,
+  disableStaticAuth: boolean,
+  logger: { level: string },
+) {
+  const flags = {
+    ...cliFlags,
+    "lightsage-api-key-auth": resolveMCPServerAPIKey(
+      cliFlags,
+      headers,
+      !disableStaticAuth,
+    ),
+  };
+
+  return new LightsageMcpCore({
+    security: { LightsageApiKeyAuth: flags["lightsage-api-key-auth"] ?? "" },
+    serverURL: cliFlags["server-url"],
+    serverIdx: cliFlags["server-index"],
+    debugLogger: logger.level === "debug"
+      ? {
+        log: (...args) => console.log(...args),
+        group: (...args) => console.group(...args),
+        groupEnd: (...args) => console.groupEnd(...args),
+      }
+      : undefined,
+  });
+}

package/src/models/apiperformanceactionslistop.ts

@@ -0,0 +1,42 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicApiPerformanceActionsResponse,
+  PublicApiPerformanceActionsResponse$zodSchema,
+} from "./publicapiperformanceactionsresponse.js";
+
+export type ApiPerformanceActionsListRequest = {
+  source_id?: string | undefined;
+  status?: string | undefined;
+  action_type?: string | undefined;
+  priority?: string | undefined;
+  limit?: number | undefined;
+};
+
+export const ApiPerformanceActionsListRequest$zodSchema: z.ZodType<
+  ApiPerformanceActionsListRequest
+> = z.object({
+  action_type: z.string().optional(),
+  limit: z.int().default(100),
+  priority: z.string().optional(),
+  source_id: z.string().optional(),
+  status: z.string().optional(),
+});
+
+export type ApiPerformanceActionsListResponse =
+  | PublicApiPerformanceActionsResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceActionsListResponse$zodSchema: z.ZodType<
+  ApiPerformanceActionsListResponse
+> = z.union([
+  PublicApiPerformanceActionsResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);

package/src/models/apiperformanceactionsrefreshop.ts

@@ -0,0 +1,24 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicRefreshActionsResponse,
+  PublicRefreshActionsResponse$zodSchema,
+} from "./publicrefreshactionsresponse.js";
+
+export type ApiPerformanceActionsRefreshResponse =
+  | PublicRefreshActionsResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceActionsRefreshResponse$zodSchema: z.ZodType<
+  ApiPerformanceActionsRefreshResponse
+> = z.union([
+  PublicRefreshActionsResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);

package/src/models/apiperformanceactionsretrieveop.ts

@@ -0,0 +1,36 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicActionDetailResponse,
+  PublicActionDetailResponse$zodSchema,
+} from "./publicactiondetailresponse.js";
+
+export type ApiPerformanceActionsRetrieveRequest = {
+  action_id: string;
+  include_prompt?: boolean | undefined;
+};
+
+export const ApiPerformanceActionsRetrieveRequest$zodSchema: z.ZodType<
+  ApiPerformanceActionsRetrieveRequest
+> = z.object({
+  action_id: z.string(),
+  include_prompt: z.boolean().default(false),
+});
+
+export type ApiPerformanceActionsRetrieveResponse =
+  | PublicActionDetailResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceActionsRetrieveResponse$zodSchema: z.ZodType<
+  ApiPerformanceActionsRetrieveResponse
+> = z.union([
+  PublicActionDetailResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);

package/src/models/apiperformanceactionsupdateop.ts

@@ -0,0 +1,40 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicActionDetailResponse,
+  PublicActionDetailResponse$zodSchema,
+} from "./publicactiondetailresponse.js";
+import {
+  PublicUpdateActionRequest,
+  PublicUpdateActionRequest$zodSchema,
+} from "./publicupdateactionrequest.js";
+
+export type ApiPerformanceActionsUpdateRequest = {
+  action_id: string;
+  body: PublicUpdateActionRequest;
+};
+
+export const ApiPerformanceActionsUpdateRequest$zodSchema: z.ZodType<
+  ApiPerformanceActionsUpdateRequest
+> = z.object({
+  action_id: z.string(),
+  body: PublicUpdateActionRequest$zodSchema,
+});
+
+export type ApiPerformanceActionsUpdateResponse =
+  | PublicActionDetailResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceActionsUpdateResponse$zodSchema: z.ZodType<
+  ApiPerformanceActionsUpdateResponse
+> = z.union([
+  PublicActionDetailResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);

package/src/models/apiperformanceactionsverifyop.ts

@@ -0,0 +1,32 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicVerifyActionResponse,
+  PublicVerifyActionResponse$zodSchema,
+} from "./publicverifyactionresponse.js";
+
+export type ApiPerformanceActionsVerifyRequest = { action_id: string };
+
+export const ApiPerformanceActionsVerifyRequest$zodSchema: z.ZodType<
+  ApiPerformanceActionsVerifyRequest
+> = z.object({
+  action_id: z.string(),
+});
+
+export type ApiPerformanceActionsVerifyResponse =
+  | PublicVerifyActionResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceActionsVerifyResponse$zodSchema: z.ZodType<
+  ApiPerformanceActionsVerifyResponse
+> = z.union([
+  PublicVerifyActionResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);

package/src/models/apiperformanceconfigretrieveop.ts

@@ -0,0 +1,34 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import {
+  HTTPValidationError,
+  HTTPValidationError$zodSchema,
+} from "./httpvalidationerror.js";
+import {
+  PublicEvalConfigResponse,
+  PublicEvalConfigResponse$zodSchema,
+} from "./publicevalconfigresponse.js";
+
+export type ApiPerformanceConfigRetrieveRequest = {
+  source_id?: string | undefined;
+};
+
+export const ApiPerformanceConfigRetrieveRequest$zodSchema: z.ZodType<
+  ApiPerformanceConfigRetrieveRequest
+> = z.object({
+  source_id: z.string().optional(),
+});
+
+export type ApiPerformanceConfigRetrieveResponse =
+  | PublicEvalConfigResponse
+  | HTTPValidationError;
+
+export const ApiPerformanceConfigRetrieveResponse$zodSchema: z.ZodType<
+  ApiPerformanceConfigRetrieveResponse
+> = z.union([
+  PublicEvalConfigResponse$zodSchema,
+  HTTPValidationError$zodSchema,
+]);