Skip to content

Commit e36eb0b

Browse files
committed
refactor(mcp-server): use builtin template id
1 parent 912faee commit e36eb0b

1 file changed

Lines changed: 5 additions & 43 deletions

File tree

packages/mcp-server/src/server.ts

Lines changed: 5 additions & 43 deletions
Original file line numberDiff line numberDiff line change
@@ -1,17 +1,7 @@
11
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
22
import type { CallToolResult, TextContent } from '@modelcontextprotocol/sdk/types.js'
3-
import type {
4-
CreateAssemblyParams,
5-
LintAssemblyInstructionsResult,
6-
TemplateContent,
7-
} from '@transloadit/node'
8-
import {
9-
getRobotHelp,
10-
listRobots,
11-
mergeTemplateContent,
12-
prepareInputFiles,
13-
Transloadit,
14-
} from '@transloadit/node'
3+
import type { CreateAssemblyParams, LintAssemblyInstructionsResult } from '@transloadit/node'
4+
import { getRobotHelp, listRobots, prepareInputFiles, Transloadit } from '@transloadit/node'
155
import { z } from 'zod'
166
import packageJson from '../package.json' with { type: 'json' }
177
import { extractBearerToken } from './http-helpers.ts'
@@ -444,19 +434,6 @@ const fetchBuiltinTemplates = async (client: Transloadit): Promise<BuiltinTempla
444434
.filter((template): template is BuiltinTemplate => Boolean(template))
445435
}
446436

447-
const fetchBuiltinTemplateContent = async (
448-
client: Transloadit,
449-
slug: string,
450-
version?: string,
451-
): Promise<TemplateContent | undefined> => {
452-
const templateId = buildBuiltinTemplateId(slug, version)
453-
const template = (await client.getTemplate(templateId)) as {
454-
content?: unknown
455-
}
456-
if (!template?.content || !isRecord(template.content)) return undefined
457-
return template.content as TemplateContent
458-
}
459-
460437
const parseInstructions = (input: unknown): CreateAssemblyParams | undefined => {
461438
if (input == null) return undefined
462439
if (typeof input === 'string') {
@@ -555,25 +532,10 @@ export const createTransloaditMcpServer = (
555532
let params = parseInstructions(instructions) ?? ({} as CreateAssemblyParams)
556533

557534
if (builtin_template) {
558-
const templateContent = await fetchBuiltinTemplateContent(
559-
client,
560-
builtin_template.slug,
561-
builtin_template.version,
562-
)
563-
564-
if (!templateContent) {
565-
return buildToolError(
566-
'mcp_unknown_template',
567-
`Unknown builtin template: ${builtin_template.slug}`,
568-
{ path: 'builtin_template.slug' },
569-
)
570-
}
571-
535+
const templateId = buildBuiltinTemplateId(builtin_template.slug, builtin_template.version)
572536
const overrides = builtin_template.overrides
573-
params = mergeTemplateContent(
574-
templateContent,
575-
overrides && isRecord(overrides) ? (overrides as Record<string, unknown>) : undefined,
576-
) as Record<string, unknown>
537+
params = (overrides && isRecord(overrides) ? overrides : {}) as CreateAssemblyParams
538+
params.template_id = templateId
577539
}
578540
const prep = await prepareInputFiles({
579541
inputFiles: fileInputs,

0 commit comments

Comments
 (0)