Skip to content

Commit 48d0a10

Browse files
authored
Fix mcp-server build typing for template merge (#309)
1 parent d4b5099 commit 48d0a10

1 file changed

Lines changed: 18 additions & 3 deletions

File tree

packages/mcp-server/src/server.ts

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,10 @@
11
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
22
import type { CallToolResult, TextContent } from '@modelcontextprotocol/sdk/types.js'
3-
import type { CreateAssemblyParams, LintAssemblyInstructionsResult } from '@transloadit/node'
3+
import type {
4+
AssemblyInstructionsInput,
5+
CreateAssemblyParams,
6+
LintAssemblyInstructionsResult,
7+
} from '@transloadit/node'
48
import {
59
extractFieldNamesFromTemplate,
610
getRobotHelp,
@@ -541,6 +545,14 @@ const parseInstructions = (input: unknown): CreateAssemblyParams | undefined =>
541545
return undefined
542546
}
543547

548+
const toAssemblyInstructionsInput = (params: CreateAssemblyParams): AssemblyInstructionsInput => {
549+
if (!params.auth || params.auth.key) {
550+
return params as AssemblyInstructionsInput
551+
}
552+
const { auth: _auth, ...rest } = params
553+
return rest as AssemblyInstructionsInput
554+
}
555+
544556
export const createTransloaditMcpServer = (
545557
options: TransloaditMcpServerOptions = {},
546558
): McpServer => {
@@ -631,7 +643,10 @@ export const createTransloaditMcpServer = (
631643
const template = await client.getTemplate(params.template_id)
632644
allowStepsOverride = template.content.allow_steps_override !== false
633645
try {
634-
const merged = mergeTemplateContent(template.content, params)
646+
const merged = mergeTemplateContent(
647+
template.content,
648+
toAssemblyInstructionsInput(params),
649+
)
635650
mergedInstructions = merged as CreateAssemblyParams
636651
mergedSteps = isRecord(merged.steps) ? (merged.steps as Record<string, unknown>) : {}
637652
mergedFields = isRecord(merged.fields) ? (merged.fields as Record<string, unknown>) : {}
@@ -681,7 +696,7 @@ export const createTransloaditMcpServer = (
681696
mergedSteps,
682697
isRecord(params.steps) ? params.steps : undefined,
683698
analysis.importStepNames,
684-
)
699+
) as CreateAssemblyParams['steps']
685700
}
686701
}
687702
}

0 commit comments

Comments
 (0)