|
1 | 1 | import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js' |
2 | 2 | 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' |
15 | 5 | import { z } from 'zod' |
16 | 6 | import packageJson from '../package.json' with { type: 'json' } |
17 | 7 | import { extractBearerToken } from './http-helpers.ts' |
@@ -444,19 +434,6 @@ const fetchBuiltinTemplates = async (client: Transloadit): Promise<BuiltinTempla |
444 | 434 | .filter((template): template is BuiltinTemplate => Boolean(template)) |
445 | 435 | } |
446 | 436 |
|
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 | | - |
460 | 437 | const parseInstructions = (input: unknown): CreateAssemblyParams | undefined => { |
461 | 438 | if (input == null) return undefined |
462 | 439 | if (typeof input === 'string') { |
@@ -555,25 +532,10 @@ export const createTransloaditMcpServer = ( |
555 | 532 | let params = parseInstructions(instructions) ?? ({} as CreateAssemblyParams) |
556 | 533 |
|
557 | 534 | 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) |
572 | 536 | 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 |
577 | 539 | } |
578 | 540 | const prep = await prepareInputFiles({ |
579 | 541 | inputFiles: fileInputs, |
|
0 commit comments