fix: tool name tweaks

Author: cjbell

src/lib/tools/email-layouts.ts

@@ -9,6 +9,8 @@ import { KnockTool } from "../knock-tool.js";
 type SerializedEmailLayout = {
   key: string;
   name: string;
+  htmlContent: string;
+  textContent: string;
 };
 
 function serializeEmailLayoutResponse(
@@ -17,6 +19,8 @@ function serializeEmailLayoutResponse(
   return {
     key: emailLayout.key,
     name: emailLayout.name,
+    htmlContent: emailLayout.html_layout,
+    textContent: emailLayout.text_layout,
   };
 }
 
@@ -46,7 +50,7 @@ const listEmailLayouts = KnockTool({
 });
 
 const createOrUpdateEmailLayout = KnockTool({
-  method: "create_or_update_email_layout",
+  method: "upsert_email_layout",
   name: "Create or update email layout",
   description: `Create or update a new email layout within the environment given. Use this tool when you need to define shared pieces of content across multiple email templates, like a header/footer. The email layout will be used to render the email template.
 

src/lib/tools/message-types.ts

@@ -48,7 +48,7 @@ const listMessageTypes = KnockTool({
 });
 
 const createOrUpdateMessageType = KnockTool({
-  method: "create_or_update_message_type",
+  method: "upsert_message_type",
   name: "Create or update message type",
   description: `
   Create or update a message type. A message type is a schema that defines fields available to an editor within Knock. Message types always have at least one variant, that MUST be named "default". Use this tool when you need to create a new message type, or update an existing message type.

src/lib/tools/objects.ts

@@ -48,7 +48,7 @@ const getObject = KnockTool({
 });
 
 const createOrUpdateObject = KnockTool({
-  method: "create_or_update_object",
+  method: "upsert_object",
   name: "Create or update object",
   description: `Create or update an object in a specific collection. Objects are used to model custom collections in Knock that are NOT users or tenants. If the object does not exist, it will be created. If the object exists, it will be updated with the provided properties. The update will always perform an upsert operation, so you do not need to provide the full properties each time.
   

src/lib/tools/tenants.ts

@@ -49,9 +49,9 @@ const listTenants = KnockTool({
   },
 });
 
-const setTenant = KnockTool({
-  method: "set_tenant",
-  name: "Set tenant",
+const createOrUpdateTenant = KnockTool({
+  method: "upsert_tenant",
+  name: "Create or update tenant",
   description: `
   Creates or updates a tenant using the properties provided. Tenants in Knock are used to model organizations, teams, and other groups of users. They are a special type of object.
   
@@ -83,10 +83,10 @@ const setTenant = KnockTool({
 export const tenants = {
   getTenant,
   listTenants,
-  setTenant,
+  createOrUpdateTenant,
 };
 
 export const permissions = {
   read: ["getTenant", "listTenants"],
-  manage: ["setTenant"],
+  manage: ["createOrUpdateTenant"],
 };

src/lib/tools/users.ts

@@ -40,7 +40,7 @@ const getUser = KnockTool({
 });
 
 const createOrUpdateUser = KnockTool({
-  method: "create_or_update_user",
+  method: "upsert_user",
   name: "Create or update user",
   description: `
   Creates a new user if they don't exist, or updates the user object for the given userId, including email, name, phone number, and any custom properties.

src/lib/tools/workflow-steps.ts

@@ -137,7 +137,7 @@ const contentBlockSchema = z.union([
 ]);
 
 const createOrUpdateEmailStepInWorkflow = KnockTool({
-  method: "create_or_update_email_step_in_workflow",
+  method: "upsert_workflow_email_step",
   name: "Create or update email step in workflow",
   description: `
   Creates or updates an email step in a workflow. Use this tool when you're asked to create an email notification and you need to specify the content of the email.
@@ -302,7 +302,7 @@ const createOrUpdateEmailStepInWorkflow = KnockTool({
 });
 
 const createOrUpdateSmsStepInWorkflow = KnockTool({
-  method: "create_or_update_sms_step_in_workflow",
+  method: "upsert_workflow_sms_step",
   name: "Create or update sms step in workflow",
   description: `
   Creates an SMS step in a workflow. Use this tool when you're asked to create an SMS notification and you need to specify the content of the SMS.
@@ -355,7 +355,7 @@ const createOrUpdateSmsStepInWorkflow = KnockTool({
 });
 
 const createOrUpdatePushStepInWorkflow = KnockTool({
-  method: "create_or_update_push_step_in_workflow",
+  method: "upsert_workflow_push_step",
   name: "Create or update push step in workflow",
   description: `
   Creates a push step in a workflow. Use this tool when you're asked to create a push notification and you need to specify the content of the push notification.
@@ -415,7 +415,7 @@ const createOrUpdatePushStepInWorkflow = KnockTool({
 
 // TODO: Add support for action buttons, not just the action URL
 const createOrUpdateInAppFeedStepInWorkflow = KnockTool({
-  method: "create_or_update_in_app_feed_step_in_workflow",
+  method: "upsert_workflow_in_app_step",
   name: "Create or update in app feed step in workflow",
   description: `
   Creates an in app feed step in a workflow. Use this tool when you're asked to create an in app feed notification and you need to specify the content of the in app feed notification.
@@ -476,7 +476,7 @@ const createOrUpdateInAppFeedStepInWorkflow = KnockTool({
 });
 
 const createOrUpdateChatStepInWorkflow = KnockTool({
-  method: "create_or_update_chat_step_in_workflow",
+  method: "upsert_workflow_chat_step",
   name: "Create or update chat step in workflow",
   description: `
   Creates a chat step in a workflow. Use this tool when you're asked to create a chat, Slack, Discord, or Microsoft Teams notification and you need to specify the content of the chat notification.
@@ -530,7 +530,7 @@ const createOrUpdateChatStepInWorkflow = KnockTool({
 });
 
 const createOrUpdateDelayStepInWorkflow = KnockTool({
-  method: "create_or_update_delay_step_in_workflow",
+  method: "upsert_workflow_delay_step",
   name: "Create or update delay step in workflow",
   description: `
   Creates a delay step in a workflow. Use this tool when you're asked to add a delay to the workflow that pauses, or waits for a period of time before continuing.
@@ -598,7 +598,7 @@ const createOrUpdateDelayStepInWorkflow = KnockTool({
 });
 
 const createOrUpdateBatchStepInWorkflow = KnockTool({
-  method: "create_or_update_batch_step_in_workflow",
+  method: "upsert_workflow_batch_step",
   name: "Create or update batch step in workflow",
   description: `
   Creates a batch step in a workflow. Use this tool when you're asked to create a batch step or asked to add digesting behavior to a workflow. The batch step collects multiple workflow triggers for a single recipient over a period of time and then flushes the content to the next step.

src/lib/tools/workflows.ts

@@ -1,4 +1,4 @@
-import { Workflow } from "@knocklabs/mgmt/resources/index.js";
+import { Workflow, WorkflowStep } from "@knocklabs/mgmt/resources/index.js";
 import { z } from "zod";
 
 import { KnockTool } from "../knock-tool.js";
@@ -28,6 +28,19 @@ export function serializeWorkflowResponse(
   };
 }
 
+export function serializeFullWorkflowResponse(
+  workflow: Workflow
+): SerializedWorkflow & { steps: WorkflowStep[] } {
+  return {
+    key: workflow.key,
+    name: workflow.name,
+    description: workflow.description,
+    categories: workflow.categories,
+    schema: workflow.trigger_data_json_schema,
+    steps: workflow.steps,
+  };
+}
+
 const listWorkflows = KnockTool({
   method: "list_workflows",
   name: "List workflows",
@@ -80,7 +93,7 @@ const getWorkflow = KnockTool({
       environment: params.environment ?? config.environment ?? "development",
     });
 
-    return serializeWorkflowResponse(workflow);
+    return serializeFullWorkflowResponse(workflow);
   },
 });