@unkey/api 2.0.0-alpha.2 → 2.0.0-alpha.6

package/src/mcp-server/tools/ratelimitDeleteOverride.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ratelimitDeleteOverride } from "../../funcs/ratelimitDeleteOverride.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.V2RatelimitDeleteOverrideRequestBody$inboundSchema,
+};
+
+export const tool$ratelimitDeleteOverride: ToolDefinition<typeof args> = {
+  name: "ratelimit-delete-override",
+  description: ``,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await ratelimitDeleteOverride(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/ratelimitGetOverride.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ratelimitGetOverride } from "../../funcs/ratelimitGetOverride.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.V2RatelimitGetOverrideRequestBody$inboundSchema,
+};
+
+export const tool$ratelimitGetOverride: ToolDefinition<typeof args> = {
+  name: "ratelimit-get-override",
+  description: ``,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await ratelimitGetOverride(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/ratelimitLimit.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ratelimitLimit } from "../../funcs/ratelimitLimit.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.V2RatelimitLimitRequestBody$inboundSchema,
+};
+
+export const tool$ratelimitLimit: ToolDefinition<typeof args> = {
+  name: "ratelimit-limit",
+  description: `TODO`,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await ratelimitLimit(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools/ratelimitRatelimitListOverrides.ts

@@ -0,0 +1,36 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ratelimitRatelimitListOverrides } from "../../funcs/ratelimitRatelimitListOverrides.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.V2RatelimitListOverridesRequestBody$inboundSchema,
+};
+
+export const tool$ratelimitRatelimitListOverrides: ToolDefinition<typeof args> =
+  {
+    name: "ratelimit-ratelimit-list-overrides",
+    description: `List rate limit overrides`,
+    args,
+    tool: async (client, args, ctx) => {
+      const [result, apiCall] = await ratelimitRatelimitListOverrides(
+        client,
+        args.request,
+        { fetchOptions: { signal: ctx.signal } },
+      ).$inspect();
+
+      if (!result.ok) {
+        return {
+          content: [{ type: "text", text: result.error.message }],
+          isError: true,
+        };
+      }
+
+      const value = result.value;
+
+      return formatResult(value, apiCall);
+    },
+  };

package/src/mcp-server/tools/ratelimitSetOverride.ts

@@ -0,0 +1,35 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { ratelimitSetOverride } from "../../funcs/ratelimitSetOverride.js";
+import * as components from "../../models/components/index.js";
+import { formatResult, ToolDefinition } from "../tools.js";
+
+const args = {
+  request: components.V2RatelimitSetOverrideRequestBody$inboundSchema,
+};
+
+export const tool$ratelimitSetOverride: ToolDefinition<typeof args> = {
+  name: "ratelimit-set-override",
+  description: ``,
+  args,
+  tool: async (client, args, ctx) => {
+    const [result, apiCall] = await ratelimitSetOverride(
+      client,
+      args.request,
+      { fetchOptions: { signal: ctx.signal } },
+    ).$inspect();
+
+    if (!result.ok) {
+      return {
+        content: [{ type: "text", text: result.error.message }],
+        isError: true,
+      };
+    }
+
+    const value = result.value;
+
+    return formatResult(value, apiCall);
+  },
+};

package/src/mcp-server/tools.ts

@@ -0,0 +1,129 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { RequestHandlerExtra } from "@modelcontextprotocol/sdk/shared/protocol.js";
+import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { objectOutputType, ZodRawShape, ZodTypeAny } from "zod";
+import { UnkeyCore } from "../core.js";
+import { ConsoleLogger } from "./console-logger.js";
+import { MCPScope } from "./scopes.js";
+import { isAsyncIterable, isBinaryData, valueToBase64 } from "./shared.js";
+
+export type ToolDefinition<Args extends undefined | ZodRawShape = undefined> =
+  Args extends ZodRawShape ? {
+      name: string;
+      description: string;
+      scopes?: MCPScope[];
+      args: Args;
+      tool: (
+        client: UnkeyCore,
+        args: objectOutputType<Args, ZodTypeAny>,
+        extra: RequestHandlerExtra,
+      ) => CallToolResult | Promise<CallToolResult>;
+    }
+    : {
+      name: string;
+      description: string;
+      scopes?: MCPScope[];
+      args?: undefined;
+      tool: (
+        client: UnkeyCore,
+        extra: RequestHandlerExtra,
+      ) => CallToolResult | Promise<CallToolResult>;
+    };
+
+// Optional function to assist with formatting tool results
+export async function formatResult(
+  value: unknown,
+  init: { response?: Response | undefined },
+): Promise<CallToolResult> {
+  if (typeof value === "undefined") {
+    return { content: [] };
+  }
+
+  const { response } = init;
+  const contentType = response?.headers.get("content-type") ?? "";
+  let content: CallToolResult["content"] = [];
+
+  if (contentType.search(/\bjson\b/g)) {
+    content = [{ type: "text", text: JSON.stringify(value) }];
+  } else if (
+    contentType.startsWith("text/event-stream")
+    && isAsyncIterable(value)
+  ) {
+    content = await consumeSSE(value);
+  } else if (contentType.startsWith("text/") && typeof value === "string") {
+    content = [{ type: "text", text: value }];
+  } else if (isBinaryData(value) && contentType.startsWith("image/")) {
+    const data = await valueToBase64(value);
+    content = data == null
+      ? []
+      : [{ type: "image", data, mimeType: contentType }];
+  } else {
+    return {
+      content: [{
+        type: "text",
+        text: `Unsupported content type: "${contentType}"`,
+      }],
+      isError: true,
+    };
+  }
+
+  return { content };
+}
+
+async function consumeSSE(
+  value: AsyncIterable<unknown>,
+): Promise<CallToolResult["content"]> {
+  const content: CallToolResult["content"] = [];
+
+  for await (const chunk of value) {
+    if (typeof chunk === "string") {
+      content.push({ type: "text", text: chunk });
+    } else {
+      content.push({ type: "text", text: JSON.stringify(chunk) });
+    }
+  }
+
+  return content;
+}
+
+export function createRegisterTool(
+  logger: ConsoleLogger,
+  server: McpServer,
+  sdk: UnkeyCore,
+  allowedScopes: Set<MCPScope>,
+  allowedTools?: Set<string>,
+): <A extends ZodRawShape | undefined>(tool: ToolDefinition<A>) => void {
+  return <A extends ZodRawShape | 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 (tool.args) {
+      server.tool(tool.name, tool.description, tool.args, async (args, ctx) => {
+        return tool.tool(sdk, args, ctx);
+      });
+    } else {
+      server.tool(tool.name, tool.description, async (ctx) => {
+        return tool.tool(sdk, ctx);
+      });
+    }
+
+    logger.debug("Registered tool", { name: tool.name });
+  };
+}

package/src/models/components/apiscreateapiresponsedata.ts

@@ -0,0 +1,76 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type ApisCreateApiResponseData = {
+  /**
+   * The id of the API
+   */
+  apiId: string;
+  /**
+   * The name of the API
+   */
+  name: string;
+};
+
+/** @internal */
+export const ApisCreateApiResponseData$inboundSchema: z.ZodType<
+  ApisCreateApiResponseData,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  apiId: z.string(),
+  name: z.string(),
+});
+
+/** @internal */
+export type ApisCreateApiResponseData$Outbound = {
+  apiId: string;
+  name: string;
+};
+
+/** @internal */
+export const ApisCreateApiResponseData$outboundSchema: z.ZodType<
+  ApisCreateApiResponseData$Outbound,
+  z.ZodTypeDef,
+  ApisCreateApiResponseData
+> = z.object({
+  apiId: z.string(),
+  name: z.string(),
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace ApisCreateApiResponseData$ {
+  /** @deprecated use `ApisCreateApiResponseData$inboundSchema` instead. */
+  export const inboundSchema = ApisCreateApiResponseData$inboundSchema;
+  /** @deprecated use `ApisCreateApiResponseData$outboundSchema` instead. */
+  export const outboundSchema = ApisCreateApiResponseData$outboundSchema;
+  /** @deprecated use `ApisCreateApiResponseData$Outbound` instead. */
+  export type Outbound = ApisCreateApiResponseData$Outbound;
+}
+
+export function apisCreateApiResponseDataToJSON(
+  apisCreateApiResponseData: ApisCreateApiResponseData,
+): string {
+  return JSON.stringify(
+    ApisCreateApiResponseData$outboundSchema.parse(apisCreateApiResponseData),
+  );
+}
+
+export function apisCreateApiResponseDataFromJSON(
+  jsonString: string,
+): SafeParseResult<ApisCreateApiResponseData, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => ApisCreateApiResponseData$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'ApisCreateApiResponseData' from JSON`,
+  );
+}

package/src/models/components/badrequesterrordetails.ts

@@ -0,0 +1,110 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+import {
+  ValidationError,
+  ValidationError$inboundSchema,
+  ValidationError$Outbound,
+  ValidationError$outboundSchema,
+} from "./validationerror.js";
+
+export type BadRequestErrorDetails = {
+  /**
+   * A human-readable explanation specific to this occurrence of the problem.
+   */
+  detail: string;
+  /**
+   * A URI reference that identifies the specific occurrence of the problem.
+   */
+  instance?: string | undefined;
+  /**
+   * HTTP status code
+   */
+  status: number;
+  /**
+   * A short, human-readable summary of the problem type. This value should not change between occurrences of the error.
+   */
+  title: string;
+  /**
+   * A URI reference to human-readable documentation for the error.
+   */
+  type: string;
+  /**
+   * Optional list of individual error details
+   */
+  errors: Array<ValidationError>;
+};
+
+/** @internal */
+export const BadRequestErrorDetails$inboundSchema: z.ZodType<
+  BadRequestErrorDetails,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  detail: z.string(),
+  instance: z.string().optional(),
+  status: z.number().int(),
+  title: z.string(),
+  type: z.string(),
+  errors: z.array(ValidationError$inboundSchema),
+});
+
+/** @internal */
+export type BadRequestErrorDetails$Outbound = {
+  detail: string;
+  instance?: string | undefined;
+  status: number;
+  title: string;
+  type: string;
+  errors: Array<ValidationError$Outbound>;
+};
+
+/** @internal */
+export const BadRequestErrorDetails$outboundSchema: z.ZodType<
+  BadRequestErrorDetails$Outbound,
+  z.ZodTypeDef,
+  BadRequestErrorDetails
+> = z.object({
+  detail: z.string(),
+  instance: z.string().optional(),
+  status: z.number().int(),
+  title: z.string(),
+  type: z.string(),
+  errors: z.array(ValidationError$outboundSchema),
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace BadRequestErrorDetails$ {
+  /** @deprecated use `BadRequestErrorDetails$inboundSchema` instead. */
+  export const inboundSchema = BadRequestErrorDetails$inboundSchema;
+  /** @deprecated use `BadRequestErrorDetails$outboundSchema` instead. */
+  export const outboundSchema = BadRequestErrorDetails$outboundSchema;
+  /** @deprecated use `BadRequestErrorDetails$Outbound` instead. */
+  export type Outbound = BadRequestErrorDetails$Outbound;
+}
+
+export function badRequestErrorDetailsToJSON(
+  badRequestErrorDetails: BadRequestErrorDetails,
+): string {
+  return JSON.stringify(
+    BadRequestErrorDetails$outboundSchema.parse(badRequestErrorDetails),
+  );
+}
+
+export function badRequestErrorDetailsFromJSON(
+  jsonString: string,
+): SafeParseResult<BadRequestErrorDetails, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => BadRequestErrorDetails$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'BadRequestErrorDetails' from JSON`,
+  );
+}

package/src/models/components/baseerror.ts

@@ -0,0 +1,93 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type BaseError = {
+  /**
+   * A human-readable explanation specific to this occurrence of the problem.
+   */
+  detail: string;
+  /**
+   * A URI reference that identifies the specific occurrence of the problem.
+   */
+  instance?: string | undefined;
+  /**
+   * HTTP status code
+   */
+  status: number;
+  /**
+   * A short, human-readable summary of the problem type. This value should not change between occurrences of the error.
+   */
+  title: string;
+  /**
+   * A URI reference to human-readable documentation for the error.
+   */
+  type: string;
+};
+
+/** @internal */
+export const BaseError$inboundSchema: z.ZodType<
+  BaseError,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  detail: z.string(),
+  instance: z.string().optional(),
+  status: z.number().int(),
+  title: z.string(),
+  type: z.string(),
+});
+
+/** @internal */
+export type BaseError$Outbound = {
+  detail: string;
+  instance?: string | undefined;
+  status: number;
+  title: string;
+  type: string;
+};
+
+/** @internal */
+export const BaseError$outboundSchema: z.ZodType<
+  BaseError$Outbound,
+  z.ZodTypeDef,
+  BaseError
+> = z.object({
+  detail: z.string(),
+  instance: z.string().optional(),
+  status: z.number().int(),
+  title: z.string(),
+  type: z.string(),
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace BaseError$ {
+  /** @deprecated use `BaseError$inboundSchema` instead. */
+  export const inboundSchema = BaseError$inboundSchema;
+  /** @deprecated use `BaseError$outboundSchema` instead. */
+  export const outboundSchema = BaseError$outboundSchema;
+  /** @deprecated use `BaseError$Outbound` instead. */
+  export type Outbound = BaseError$Outbound;
+}
+
+export function baseErrorToJSON(baseError: BaseError): string {
+  return JSON.stringify(BaseError$outboundSchema.parse(baseError));
+}
+
+export function baseErrorFromJSON(
+  jsonString: string,
+): SafeParseResult<BaseError, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) => BaseError$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'BaseError' from JSON`,
+  );
+}

package/src/models/components/identitiescreateidentityresponsedata.ts

@@ -0,0 +1,74 @@
+/*
+ * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
+ */
+
+import * as z from "zod";
+import { safeParse } from "../../lib/schemas.js";
+import { Result as SafeParseResult } from "../../types/fp.js";
+import { SDKValidationError } from "../errors/sdkvalidationerror.js";
+
+export type IdentitiesCreateIdentityResponseData = {
+  /**
+   * The id of the identity. Used internally, you do not need to store this.
+   */
+  identityId: string;
+};
+
+/** @internal */
+export const IdentitiesCreateIdentityResponseData$inboundSchema: z.ZodType<
+  IdentitiesCreateIdentityResponseData,
+  z.ZodTypeDef,
+  unknown
+> = z.object({
+  identityId: z.string(),
+});
+
+/** @internal */
+export type IdentitiesCreateIdentityResponseData$Outbound = {
+  identityId: string;
+};
+
+/** @internal */
+export const IdentitiesCreateIdentityResponseData$outboundSchema: z.ZodType<
+  IdentitiesCreateIdentityResponseData$Outbound,
+  z.ZodTypeDef,
+  IdentitiesCreateIdentityResponseData
+> = z.object({
+  identityId: z.string(),
+});
+
+/**
+ * @internal
+ * @deprecated This namespace will be removed in future versions. Use schemas and types that are exported directly from this module.
+ */
+export namespace IdentitiesCreateIdentityResponseData$ {
+  /** @deprecated use `IdentitiesCreateIdentityResponseData$inboundSchema` instead. */
+  export const inboundSchema =
+    IdentitiesCreateIdentityResponseData$inboundSchema;
+  /** @deprecated use `IdentitiesCreateIdentityResponseData$outboundSchema` instead. */
+  export const outboundSchema =
+    IdentitiesCreateIdentityResponseData$outboundSchema;
+  /** @deprecated use `IdentitiesCreateIdentityResponseData$Outbound` instead. */
+  export type Outbound = IdentitiesCreateIdentityResponseData$Outbound;
+}
+
+export function identitiesCreateIdentityResponseDataToJSON(
+  identitiesCreateIdentityResponseData: IdentitiesCreateIdentityResponseData,
+): string {
+  return JSON.stringify(
+    IdentitiesCreateIdentityResponseData$outboundSchema.parse(
+      identitiesCreateIdentityResponseData,
+    ),
+  );
+}
+
+export function identitiesCreateIdentityResponseDataFromJSON(
+  jsonString: string,
+): SafeParseResult<IdentitiesCreateIdentityResponseData, SDKValidationError> {
+  return safeParse(
+    jsonString,
+    (x) =>
+      IdentitiesCreateIdentityResponseData$inboundSchema.parse(JSON.parse(x)),
+    `Failed to parse 'IdentitiesCreateIdentityResponseData' from JSON`,
+  );
+}

package/src/models/components/index.ts

@@ -2,15 +2,32 @@
  * Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT.
  */
 
-export * from "./httpmetadata.js";
+export * from "./apiscreateapiresponsedata.js";
+export * from "./badrequesterrordetails.js";
+export * from "./baseerror.js";
+export * from "./identitiescreateidentityresponsedata.js";
+export * from "./livenessresponsedata.js";
+export * from "./meta.js";
+export * from "./pagination.js";
+export * from "./ratelimitdeleteoverrideresponsedata.js";
+export * from "./ratelimitlimitresponsedata.js";
+export * from "./ratelimitoverride.js";
+export * from "./ratelimitsetoverrideresponsedata.js";
 export * from "./security.js";
+export * from "./v2apiscreateapirequestbody.js";
+export * from "./v2apiscreateapiresponsebody.js";
+export * from "./v2identitiescreateidentityrequestbody.js";
+export * from "./v2identitiescreateidentityresponsebody.js";
 export * from "./v2livenessresponsebody.js";
+export * from "./v2ratelimit.js";
 export * from "./v2ratelimitdeleteoverriderequestbody.js";
 export * from "./v2ratelimitdeleteoverrideresponsebody.js";
 export * from "./v2ratelimitgetoverriderequestbody.js";
 export * from "./v2ratelimitgetoverrideresponsebody.js";
 export * from "./v2ratelimitlimitrequestbody.js";
 export * from "./v2ratelimitlimitresponsebody.js";
+export * from "./v2ratelimitlistoverridesrequestbody.js";
+export * from "./v2ratelimitlistoverridesresponsebody.js";
 export * from "./v2ratelimitsetoverriderequestbody.js";
 export * from "./v2ratelimitsetoverrideresponsebody.js";
 export * from "./validationerror.js";