@knocklabs/agent-toolkit 0.1.1 → 0.1.3

package/dist/lib/tools/environments.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/index.d.ts

@@ -0,0 +1,4 @@
+export { f as allTools, t as toolPermissions, e as tools } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';

package/dist/lib/tools/index.js

@@ -0,0 +1,23 @@
+import {
+  allTools,
+  toolPermissions,
+  tools
+} from "../../chunk-NZB2TMJW.js";
+import "../../chunk-THLK376Y.js";
+import "../../chunk-OD5VHBGN.js";
+import "../../chunk-RYMJEH6P.js";
+import "../../chunk-BY4KQPUO.js";
+import "../../chunk-JTYQTWCX.js";
+import "../../chunk-VJFA3ZVW.js";
+import "../../chunk-XRQWR3LY.js";
+import "../../chunk-2CNJ4XX6.js";
+import "../../chunk-TLHETJRF.js";
+import "../../chunk-DSRTYV37.js";
+import "../../chunk-O6DWF42E.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  allTools,
+  toolPermissions,
+  tools
+};
+//# sourceMappingURL=index.js.map

package/dist/lib/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/message-types.d.ts

@@ -0,0 +1,15 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const messageTypes: {
+    listMessageTypes: KnockTool;
+    createOrUpdateMessageType: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+    manage: string[];
+};
+
+export { messageTypes, permissions };

package/dist/lib/tools/message-types.js

@@ -0,0 +1,10 @@
+import {
+  messageTypes,
+  permissions
+} from "../../chunk-THLK376Y.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  messageTypes,
+  permissions
+};
+//# sourceMappingURL=message-types.js.map

package/dist/lib/tools/message-types.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/messages.d.ts

@@ -0,0 +1,13 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const messages: {
+    getMessageContent: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+};
+
+export { messages, permissions };

package/dist/lib/tools/messages.js

@@ -0,0 +1,10 @@
+import {
+  messages,
+  permissions
+} from "../../chunk-OD5VHBGN.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  messages,
+  permissions
+};
+//# sourceMappingURL=messages.js.map

package/dist/lib/tools/messages.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/objects.d.ts

@@ -0,0 +1,19 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const objects: {
+    listObjects: KnockTool;
+    getObject: KnockTool;
+    createOrUpdateObject: KnockTool;
+    deleteObject: KnockTool;
+    subscribeUsersToObject: KnockTool;
+    unsubscribeUsersFromObject: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+    manage: string[];
+};
+
+export { objects, permissions };

package/dist/lib/tools/objects.js

@@ -0,0 +1,10 @@
+import {
+  objects,
+  permissions
+} from "../../chunk-RYMJEH6P.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  objects,
+  permissions
+};
+//# sourceMappingURL=objects.js.map

package/dist/lib/tools/objects.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/partials.d.ts

@@ -0,0 +1,13 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const partials: {
+    listPartials: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+};
+
+export { partials, permissions };

package/dist/lib/tools/partials.js

@@ -0,0 +1,10 @@
+import {
+  partials,
+  permissions
+} from "../../chunk-BY4KQPUO.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  partials,
+  permissions
+};
+//# sourceMappingURL=partials.js.map

package/dist/lib/tools/partials.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/tenants.d.ts

@@ -0,0 +1,16 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const tenants: {
+    getTenant: KnockTool;
+    setTenant: KnockTool;
+    deleteTenant: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+    manage: string[];
+};
+
+export { permissions, tenants };

package/dist/lib/tools/tenants.js

@@ -0,0 +1,10 @@
+import {
+  permissions,
+  tenants
+} from "../../chunk-JTYQTWCX.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  permissions,
+  tenants
+};
+//# sourceMappingURL=tenants.js.map

package/dist/lib/tools/tenants.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/users.d.ts

@@ -0,0 +1,19 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const users: {
+    getUser: KnockTool;
+    deleteUser: KnockTool;
+    createOrUpdateUser: KnockTool;
+    getUserPreferences: KnockTool;
+    setUserPreferences: KnockTool;
+    getUserMessages: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+    manage: string[];
+};
+
+export { permissions, users };

package/dist/lib/tools/users.js

@@ -0,0 +1,10 @@
+import {
+  permissions,
+  users
+} from "../../chunk-VJFA3ZVW.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  permissions,
+  users
+};
+//# sourceMappingURL=users.js.map

package/dist/lib/tools/users.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/tools/workflows.d.ts

@@ -0,0 +1,18 @@
+import { a as KnockTool } from '../../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const workflows: {
+    listWorkflows: KnockTool;
+    triggerWorkflow: KnockTool;
+    createEmailWorkflow: KnockTool;
+    createOneOffWorkflowSchedule: KnockTool;
+};
+declare const permissions: {
+    read: string[];
+    manage: string[];
+    trigger: string[];
+};
+
+export { permissions, workflows };

package/dist/lib/tools/workflows.js

@@ -0,0 +1,10 @@
+import {
+  permissions,
+  workflows
+} from "../../chunk-XRQWR3LY.js";
+import "../../chunk-TOBLV65T.js";
+export {
+  permissions,
+  workflows
+};
+//# sourceMappingURL=workflows.js.map

package/dist/lib/tools/workflows.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/lib/utils.d.ts

@@ -0,0 +1,33 @@
+import { a as KnockTool, t as toolPermissions, T as ToolkitConfig, b as ToolCategory } from '../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+/**
+ * Given a list of tools, and some config may describe the tools that should be provided to the LLM,
+ * returns a filtered list of tools that match the config.
+ *
+ * Options:
+ * `*` - All tools
+ * `users.*` - All tools that start with `users.`
+ * `users.getUser` - A specific tool
+ *
+ */
+declare function filterTools(tools: Record<string, Record<string, KnockTool>>, pattern: string | undefined): KnockTool[];
+/**
+ * Given a category and a list of permissions, return a list of tools that the user has permission to use.
+ *
+ * @param category - The category to get tools for
+ * @param categoryPermissions - The permissions to use
+ * @returns A list of tools that the user has permission to use
+ */
+declare function getToolsWithPermissions(category: keyof typeof toolPermissions, categoryPermissions: Record<string, boolean | undefined>): KnockTool[];
+/**
+ * Given a config, return a list of tools for each category that the user has permission to use.
+ *
+ * @param config - The config to use
+ * @returns A list of tools for each category that the user has permission to use
+ */
+declare function getToolsByPermissionsInCategories(config: ToolkitConfig): Record<ToolCategory, KnockTool[]>;
+
+export { filterTools, getToolsByPermissionsInCategories, getToolsWithPermissions };

package/dist/lib/utils.js

@@ -0,0 +1,24 @@
+import {
+  filterTools,
+  getToolsByPermissionsInCategories,
+  getToolsWithPermissions
+} from "../chunk-4VR65G5W.js";
+import "../chunk-NZB2TMJW.js";
+import "../chunk-THLK376Y.js";
+import "../chunk-OD5VHBGN.js";
+import "../chunk-RYMJEH6P.js";
+import "../chunk-BY4KQPUO.js";
+import "../chunk-JTYQTWCX.js";
+import "../chunk-VJFA3ZVW.js";
+import "../chunk-XRQWR3LY.js";
+import "../chunk-2CNJ4XX6.js";
+import "../chunk-TLHETJRF.js";
+import "../chunk-DSRTYV37.js";
+import "../chunk-O6DWF42E.js";
+import "../chunk-TOBLV65T.js";
+export {
+  filterTools,
+  getToolsByPermissionsInCategories,
+  getToolsWithPermissions
+};
+//# sourceMappingURL=utils.js.map

package/dist/lib/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/modelcontextprotocol/adapter.d.ts

@@ -0,0 +1,11 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { K as KnockClient, C as Config, a as KnockTool } from '../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare class KnockMcpServer extends McpServer {
+    constructor(knockClient: KnockClient, config: Config, tools: KnockTool[]);
+}
+
+export { KnockMcpServer };

package/dist/modelcontextprotocol/adapter.js

@@ -0,0 +1,7 @@
+import {
+  KnockMcpServer
+} from "../chunk-XUYE3AGB.js";
+export {
+  KnockMcpServer
+};
+//# sourceMappingURL=adapter.js.map

package/dist/modelcontextprotocol/adapter.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/modelcontextprotocol/index.d.ts

@@ -0,0 +1,27 @@
+import { a as KnockTool, K as KnockClient, C as Config } from '../types-B-9pppCQ.js';
+import { KnockMcpServer } from './adapter.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+import '@modelcontextprotocol/sdk/server/mcp.js';
+
+type CreateKnockMcpServerParams = {
+    /**
+     * Array of Knock tools to enable in the server.
+     */
+    tools?: KnockTool[];
+    /**
+     * A Knock client to use for the server.
+     */
+    knockClient: KnockClient;
+    /**
+     * The config to use for the server.
+     */
+    config: Config;
+};
+/**
+ * Creates a Knock MCP Server with the given parameters.
+ */
+declare const createKnockMcpServer: (params: CreateKnockMcpServerParams) => Promise<KnockMcpServer>;
+
+export { createKnockMcpServer };

package/dist/modelcontextprotocol/index.js

@@ -0,0 +1,21 @@
+import {
+  createKnockMcpServer
+} from "../chunk-PDMA6RIB.js";
+import "../chunk-NZB2TMJW.js";
+import "../chunk-THLK376Y.js";
+import "../chunk-OD5VHBGN.js";
+import "../chunk-RYMJEH6P.js";
+import "../chunk-BY4KQPUO.js";
+import "../chunk-JTYQTWCX.js";
+import "../chunk-VJFA3ZVW.js";
+import "../chunk-XRQWR3LY.js";
+import "../chunk-2CNJ4XX6.js";
+import "../chunk-TLHETJRF.js";
+import "../chunk-DSRTYV37.js";
+import "../chunk-O6DWF42E.js";
+import "../chunk-TOBLV65T.js";
+import "../chunk-XUYE3AGB.js";
+export {
+  createKnockMcpServer
+};
+//# sourceMappingURL=index.js.map

package/dist/modelcontextprotocol/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":[],"sourcesContent":[],"mappings":"","names":[]}

package/dist/modelcontextprotocol/local-server.d.ts

@@ -0,0 +1 @@
+#!/usr/bin/env node

package/dist/modelcontextprotocol/local-server.js

@@ -0,0 +1,86 @@
+#!/usr/bin/env node
+import {
+  createKnockClient
+} from "../chunk-3LZK6RH6.js";
+import {
+  filterTools
+} from "../chunk-4VR65G5W.js";
+import {
+  createKnockMcpServer
+} from "../chunk-PDMA6RIB.js";
+import {
+  tools
+} from "../chunk-NZB2TMJW.js";
+import "../chunk-THLK376Y.js";
+import "../chunk-OD5VHBGN.js";
+import "../chunk-RYMJEH6P.js";
+import "../chunk-BY4KQPUO.js";
+import "../chunk-JTYQTWCX.js";
+import "../chunk-VJFA3ZVW.js";
+import "../chunk-XRQWR3LY.js";
+import "../chunk-2CNJ4XX6.js";
+import "../chunk-TLHETJRF.js";
+import "../chunk-DSRTYV37.js";
+import "../chunk-O6DWF42E.js";
+import "../chunk-TOBLV65T.js";
+import "../chunk-XUYE3AGB.js";
+
+// src/modelcontextprotocol/local-server.ts
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { default as yargs } from "yargs";
+import { hideBin } from "yargs/helpers";
+var main = async () => {
+  const {
+    serviceToken,
+    environment,
+    userId,
+    tenantId,
+    tools: patterns
+  } = await yargs(hideBin(process.argv)).version("0.1.0").option("tools", {
+    alias: "t",
+    type: "string",
+    array: true,
+    description: `List of tools to enable. Use "*" to enable all tools. Use "category" or "category.*" to enable all tools from a category. Use "category.toolName" to pick a single tool. Available categories: ${Object.keys(
+      tools
+    )}`
+  }).option("service-token", {
+    alias: "st",
+    type: "string",
+    description: `Knock service token`
+  }).option("environment", {
+    alias: "e",
+    type: "string",
+    description: `The environment to operate in from your Knock account`
+  }).option("user-id", {
+    alias: "u",
+    type: "string",
+    description: `The user id to operate as`
+  }).option("tenant-id", {
+    alias: "t",
+    type: "string",
+    description: `The tenant id to operate as`
+  }).parse();
+  const SERVICE_TOKEN = serviceToken || process.env.KNOCK_SERVICE_TOKEN;
+  if (!SERVICE_TOKEN) {
+    throw new Error("Knock service token is required");
+  }
+  const config = {
+    serviceToken: SERVICE_TOKEN,
+    userId,
+    tenantId,
+    environment
+  };
+  const knockClient = createKnockClient(config);
+  const filteredTools = patterns ? patterns.map((pattern) => filterTools(tools, pattern)).flat() : void 0;
+  const mcpServer = await createKnockMcpServer({
+    config,
+    knockClient,
+    tools: filteredTools
+  });
+  const transport = new StdioServerTransport();
+  await mcpServer.connect(transport);
+};
+main().catch((error) => {
+  console.error("\nKnock: Error initializing MCP server:\n", error.message);
+});
+//# sourceMappingURL=local-server.js.map

package/dist/modelcontextprotocol/local-server.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/modelcontextprotocol/local-server.ts"],"sourcesContent":["#!/usr/bin/env node\n\nimport { StdioServerTransport } from \"@modelcontextprotocol/sdk/server/stdio.js\";\nimport { default as yargs } from \"yargs\";\nimport { hideBin } from \"yargs/helpers\";\n\nimport { createKnockClient } from \"../lib/knock-client.js\";\nimport { Config } from \"../types.js\";\nimport { createKnockMcpServer } from \"./index.js\";\nimport { tools } from \"../lib/tools/index.js\";\nimport { filterTools } from \"../lib/utils.js\";\n\n/**\n * Main entry point for the Knock MCP server.\n * Runs as a standalone process, as defined in package.json#bin.\n * An entrypoint for this file exists in the tsup configuration of the package.\n */\nconst main = async () => {\n  const {\n    serviceToken,\n    environment,\n    userId,\n    tenantId,\n    tools: patterns,\n  } = await yargs(hideBin(process.argv))\n    .version(\"0.1.0\")\n    .option(\"tools\", {\n      alias: \"t\",\n      type: \"string\",\n      array: true,\n      description: `List of tools to enable. Use \"*\" to enable all tools. Use \"category\" or \"category.*\" to enable all tools from a category. Use \"category.toolName\" to pick a single tool. Available categories: ${Object.keys(\n        tools\n      )}`,\n    })\n    .option(\"service-token\", {\n      alias: \"st\",\n      type: \"string\",\n      description: `Knock service token`,\n    })\n    .option(\"environment\", {\n      alias: \"e\",\n      type: \"string\",\n      description: `The environment to operate in from your Knock account`,\n    })\n    .option(\"user-id\", {\n      alias: \"u\",\n      type: \"string\",\n      description: `The user id to operate as`,\n    })\n    .option(\"tenant-id\", {\n      alias: \"t\",\n      type: \"string\",\n      description: `The tenant id to operate as`,\n    })\n    .parse();\n\n  const SERVICE_TOKEN = serviceToken || process.env.KNOCK_SERVICE_TOKEN;\n\n  if (!SERVICE_TOKEN) {\n    throw new Error(\"Knock service token is required\");\n  }\n\n  const config: Config = {\n    serviceToken: SERVICE_TOKEN,\n    userId,\n    tenantId,\n    environment,\n  };\n\n  const knockClient = createKnockClient(config);\n\n  const filteredTools = patterns\n    ? patterns.map((pattern) => filterTools(tools, pattern)).flat()\n    : undefined;\n\n  const mcpServer = await createKnockMcpServer({\n    config,\n    knockClient,\n    tools: filteredTools,\n  });\n\n  const transport = new StdioServerTransport();\n  await mcpServer.connect(transport);\n};\n\nmain().catch((error) => {\n  console.error(\"\\nKnock: Error initializing MCP server:\\n\", error.message);\n});\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEA,SAAS,4BAA4B;AACrC,SAAS,WAAW,aAAa;AACjC,SAAS,eAAe;AAaxB,IAAM,OAAO,YAAY;AACvB,QAAM;AAAA,IACJ;AAAA,IACA;AAAA,IACA;AAAA,IACA;AAAA,IACA,OAAO;AAAA,EACT,IAAI,MAAM,MAAM,QAAQ,QAAQ,IAAI,CAAC,EAClC,QAAQ,OAAO,EACf,OAAO,SAAS;AAAA,IACf,OAAO;AAAA,IACP,MAAM;AAAA,IACN,OAAO;AAAA,IACP,aAAa,kMAAkM,OAAO;AAAA,MACpN;AAAA,IACF,CAAC;AAAA,EACH,CAAC,EACA,OAAO,iBAAiB;AAAA,IACvB,OAAO;AAAA,IACP,MAAM;AAAA,IACN,aAAa;AAAA,EACf,CAAC,EACA,OAAO,eAAe;AAAA,IACrB,OAAO;AAAA,IACP,MAAM;AAAA,IACN,aAAa;AAAA,EACf,CAAC,EACA,OAAO,WAAW;AAAA,IACjB,OAAO;AAAA,IACP,MAAM;AAAA,IACN,aAAa;AAAA,EACf,CAAC,EACA,OAAO,aAAa;AAAA,IACnB,OAAO;AAAA,IACP,MAAM;AAAA,IACN,aAAa;AAAA,EACf,CAAC,EACA,MAAM;AAET,QAAM,gBAAgB,gBAAgB,QAAQ,IAAI;AAElD,MAAI,CAAC,eAAe;AAClB,UAAM,IAAI,MAAM,iCAAiC;AAAA,EACnD;AAEA,QAAM,SAAiB;AAAA,IACrB,cAAc;AAAA,IACd;AAAA,IACA;AAAA,IACA;AAAA,EACF;AAEA,QAAM,cAAc,kBAAkB,MAAM;AAE5C,QAAM,gBAAgB,WAClB,SAAS,IAAI,CAAC,YAAY,YAAY,OAAO,OAAO,CAAC,EAAE,KAAK,IAC5D;AAEJ,QAAM,YAAY,MAAM,qBAAqB;AAAA,IAC3C;AAAA,IACA;AAAA,IACA,OAAO;AAAA,EACT,CAAC;AAED,QAAM,YAAY,IAAI,qBAAqB;AAC3C,QAAM,UAAU,QAAQ,SAAS;AACnC;AAEA,KAAK,EAAE,MAAM,CAAC,UAAU;AACtB,UAAQ,MAAM,6CAA6C,MAAM,OAAO;AAC1E,CAAC;","names":[]}

package/dist/openai/index.d.ts

@@ -0,0 +1,36 @@
+import { ChatCompletionTool, ChatCompletionMessageToolCall, ChatCompletionToolMessageParam } from 'openai/resources.mjs';
+import { T as ToolkitConfig, b as ToolCategory } from '../types-B-9pppCQ.js';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+declare const createKnockToolkit: (config: ToolkitConfig) => {
+    /**
+     * Get all tools as a flat list.
+     *
+     * @returns An array of all tools
+     */
+    getAllTools: () => ChatCompletionTool[];
+    /**
+     * Get all tools for a specific category.
+     *
+     * @param category - The category of too  ls to get
+     * @returns An array of tools for the given category
+     */
+    getTools: (category: ToolCategory) => ChatCompletionTool[];
+    /**
+     * Get a map of all tools by method name.
+     *
+     * @returns A map of all tools by method name
+     */
+    getToolMap: () => Record<string, ChatCompletionTool>;
+    /**
+     * Handle a tool call from the OpenAI API. Call this to invoke the tool.
+     *
+     * @param toolCall - The tool call to handle
+     * @returns The result of the tool call
+     */
+    handleToolCall: (toolCall: ChatCompletionMessageToolCall) => Promise<ChatCompletionToolMessageParam>;
+};
+
+export { createKnockToolkit };

package/dist/openai/index.js

@@ -0,0 +1,86 @@
+import {
+  knockToolToChatCompletionTool
+} from "../chunk-WB4XCCUS.js";
+import {
+  createKnockClient
+} from "../chunk-3LZK6RH6.js";
+import {
+  getToolsByPermissionsInCategories
+} from "../chunk-4VR65G5W.js";
+import "../chunk-NZB2TMJW.js";
+import "../chunk-THLK376Y.js";
+import "../chunk-OD5VHBGN.js";
+import "../chunk-RYMJEH6P.js";
+import "../chunk-BY4KQPUO.js";
+import "../chunk-JTYQTWCX.js";
+import "../chunk-VJFA3ZVW.js";
+import "../chunk-XRQWR3LY.js";
+import "../chunk-2CNJ4XX6.js";
+import "../chunk-TLHETJRF.js";
+import "../chunk-DSRTYV37.js";
+import "../chunk-O6DWF42E.js";
+import "../chunk-TOBLV65T.js";
+
+// src/openai/index.ts
+var createKnockToolkit = (config) => {
+  const knockClient = createKnockClient(config);
+  const allowedToolsByCategory = getToolsByPermissionsInCategories(config);
+  const toolsByMethod = Object.values(allowedToolsByCategory).flat().reduce(
+    (acc, tool) => ({ ...acc, [tool.method]: tool }),
+    {}
+  );
+  return {
+    /**
+     * Get all tools as a flat list.
+     *
+     * @returns An array of all tools
+     */
+    getAllTools: () => Object.values(toolsByMethod).map((t) => knockToolToChatCompletionTool(t)),
+    /**
+     * Get all tools for a specific category.
+     *
+     * @param category - The category of too  ls to get
+     * @returns An array of tools for the given category
+     */
+    getTools: (category) => allowedToolsByCategory[category].map(
+      (t) => knockToolToChatCompletionTool(t)
+    ),
+    /**
+     * Get a map of all tools by method name.
+     *
+     * @returns A map of all tools by method name
+     */
+    getToolMap: () => Object.entries(toolsByMethod).reduce(
+      (acc, [method, tool]) => {
+        acc[method] = knockToolToChatCompletionTool(tool);
+        return acc;
+      },
+      {}
+    ),
+    /**
+     * Handle a tool call from the OpenAI API. Call this to invoke the tool.
+     *
+     * @param toolCall - The tool call to handle
+     * @returns The result of the tool call
+     */
+    handleToolCall: async (toolCall) => {
+      const tool = toolsByMethod[toolCall.function.name];
+      if (!tool) {
+        throw new Error(`Tool ${toolCall.function.name} not found`);
+      }
+      const result = await tool.bindExecute(
+        knockClient,
+        config
+      )(toolCall.function.arguments);
+      return {
+        role: "tool",
+        tool_call_id: toolCall.id,
+        content: result
+      };
+    }
+  };
+};
+export {
+  createKnockToolkit
+};
+//# sourceMappingURL=index.js.map

package/dist/openai/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/openai/index.ts"],"sourcesContent":["import {\n  ChatCompletionTool,\n  ChatCompletionMessageToolCall,\n  ChatCompletionToolMessageParam,\n} from \"openai/resources.mjs\";\nimport { ToolCategory, ToolkitConfig } from \"../types.js\";\nimport { getToolsByPermissionsInCategories } from \"../lib/utils.js\";\nimport { createKnockClient } from \"../lib/knock-client.js\";\nimport { knockToolToChatCompletionTool } from \"./tool-converter.js\";\nimport { KnockTool } from \"../lib/knock-tool.js\";\n\nconst createKnockToolkit = (config: ToolkitConfig) => {\n  const knockClient = createKnockClient(config);\n  const allowedToolsByCategory = getToolsByPermissionsInCategories(config);\n\n  const toolsByMethod = Object.values(allowedToolsByCategory)\n    .flat()\n    .reduce(\n      (acc, tool) => ({ ...acc, [tool.method]: tool }),\n      {} as Record<string, KnockTool>\n    );\n\n  return {\n    /**\n     * Get all tools as a flat list.\n     *\n     * @returns An array of all tools\n     */\n    getAllTools: (): ChatCompletionTool[] =>\n      Object.values(toolsByMethod).map((t) => knockToolToChatCompletionTool(t)),\n\n    /**\n     * Get all tools for a specific category.\n     *\n     * @param category - The category of too  ls to get\n     * @returns An array of tools for the given category\n     */\n    getTools: (category: ToolCategory): ChatCompletionTool[] =>\n      allowedToolsByCategory[category].map((t) =>\n        knockToolToChatCompletionTool(t)\n      ),\n\n    /**\n     * Get a map of all tools by method name.\n     *\n     * @returns A map of all tools by method name\n     */\n    getToolMap: () =>\n      Object.entries(toolsByMethod).reduce(\n        (acc, [method, tool]) => {\n          acc[method] = knockToolToChatCompletionTool(tool);\n          return acc;\n        },\n        {} as Record<string, ChatCompletionTool>\n      ),\n    /**\n     * Handle a tool call from the OpenAI API. Call this to invoke the tool.\n     *\n     * @param toolCall - The tool call to handle\n     * @returns The result of the tool call\n     */\n    handleToolCall: async (toolCall: ChatCompletionMessageToolCall) => {\n      const tool = toolsByMethod[toolCall.function.name];\n\n      if (!tool) {\n        throw new Error(`Tool ${toolCall.function.name} not found`);\n      }\n\n      const result = await tool.bindExecute(\n        knockClient,\n        config\n      )(toolCall.function.arguments);\n\n      return {\n        role: \"tool\",\n        tool_call_id: toolCall.id,\n        content: result,\n      } as ChatCompletionToolMessageParam;\n    },\n  };\n};\n\nexport { createKnockToolkit };\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;AAWA,IAAM,qBAAqB,CAAC,WAA0B;AACpD,QAAM,cAAc,kBAAkB,MAAM;AAC5C,QAAM,yBAAyB,kCAAkC,MAAM;AAEvE,QAAM,gBAAgB,OAAO,OAAO,sBAAsB,EACvD,KAAK,EACL;AAAA,IACC,CAAC,KAAK,UAAU,EAAE,GAAG,KAAK,CAAC,KAAK,MAAM,GAAG,KAAK;AAAA,IAC9C,CAAC;AAAA,EACH;AAEF,SAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAML,aAAa,MACX,OAAO,OAAO,aAAa,EAAE,IAAI,CAAC,MAAM,8BAA8B,CAAC,CAAC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAQ1E,UAAU,CAAC,aACT,uBAAuB,QAAQ,EAAE;AAAA,MAAI,CAAC,MACpC,8BAA8B,CAAC;AAAA,IACjC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAOF,YAAY,MACV,OAAO,QAAQ,aAAa,EAAE;AAAA,MAC5B,CAAC,KAAK,CAAC,QAAQ,IAAI,MAAM;AACvB,YAAI,MAAM,IAAI,8BAA8B,IAAI;AAChD,eAAO;AAAA,MACT;AAAA,MACA,CAAC;AAAA,IACH;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,IAOF,gBAAgB,OAAO,aAA4C;AACjE,YAAM,OAAO,cAAc,SAAS,SAAS,IAAI;AAEjD,UAAI,CAAC,MAAM;AACT,cAAM,IAAI,MAAM,QAAQ,SAAS,SAAS,IAAI,YAAY;AAAA,MAC5D;AAEA,YAAM,SAAS,MAAM,KAAK;AAAA,QACxB;AAAA,QACA;AAAA,MACF,EAAE,SAAS,SAAS,SAAS;AAE7B,aAAO;AAAA,QACL,MAAM;AAAA,QACN,cAAc,SAAS;AAAA,QACvB,SAAS;AAAA,MACX;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/openai/tool-converter.d.ts

@@ -0,0 +1,15 @@
+import { a as KnockTool } from '../types-B-9pppCQ.js';
+import { ChatCompletionTool } from 'openai/resources.mjs';
+import 'zod';
+import '@knocklabs/node';
+import '@knocklabs/mgmt';
+
+/**
+ * Convert a KnockTool to a ChatCompletionTool, ready to be used with the OpenAI API.
+ *
+ * @param knockTool - The KnockTool to convert
+ * @returns The converted ChatCompletionTool
+ */
+declare function knockToolToChatCompletionTool(knockTool: KnockTool): ChatCompletionTool;
+
+export { knockToolToChatCompletionTool };