diff --git a/packages/plugins/trpc/src/generator.ts b/packages/plugins/trpc/src/generator.ts index 9a4f2e894..4532cb803 100644 --- a/packages/plugins/trpc/src/generator.ts +++ b/packages/plugins/trpc/src/generator.ts @@ -131,10 +131,10 @@ function createAppRouter( appRouter .addFunction({ - name: 'createRouter', + name: 'createRouter, Proc extends ProcBuilder>', parameters: [ - { name: 'router', type: 'RouterFactory' }, - { name: 'procedure', type: 'ProcBuilder' }, + { name: 'router', type: 'Router' }, + { name: 'procedure', type: 'Proc' }, ], isExported: true, }) @@ -159,7 +159,9 @@ function createAppRouter( moduleSpecifier: `./${model}.router`, }); - writer.writeLine(`${lowerCaseFirst(model)}: create${model}Router(router, procedure),`); + writer.writeLine( + `${lowerCaseFirst(model)}: create${model}Router(router, procedure),` + ); } }); writer.write(');'); @@ -243,10 +245,10 @@ function generateModelCreateRouter( } const createRouterFunc = modelRouter.addFunction({ - name: 'createRouter', + name: 'createRouter, Proc extends ProcBuilder>', parameters: [ - { name: 'router', type: 'RouterFactory' }, - { name: 'procedure', type: 'ProcBuilder' }, + { name: 'router', type: 'Router' }, + { name: 'procedure', type: 'Proc' }, ], isExported: true, isDefaultExport: true,