fix: enhance return type in defineRpcFunction to include args and return schemas

Gianthard-cyh · Gianthard-cyh · commit 7ae96aa3589a · 2026-01-14T16:42:10.000+08:00
diff --git a/packages/kit/src/utils/define.ts b/packages/kit/src/utils/define.ts
@@ -31,5 +31,9 @@ export function defineRpcFunction<
 
   const fn = birpc(rest)
 
-  return fn as typeof fn & { argsSchema?: AS, returnSchema?: RS }
+  const augmentedFn = fn as typeof fn & { argsSchema?: AS, returnSchema?: RS }
+  augmentedFn.argsSchema = args
+  augmentedFn.returnSchema = ret
+
+  return augmentedFn
 }

Original file line number	Diff line number	Diff line change
`@@ -31,5 +31,9 @@ export function defineRpcFunction<`
`31`	`31`
`32`	`32`	`const fn = birpc(rest)`
`33`	`33`
`34`		`- return fn as typeof fn & { argsSchema?: AS, returnSchema?: RS }`
	`34`	`+ const augmentedFn = fn as typeof fn & { argsSchema?: AS, returnSchema?: RS }`
	`35`	`+ augmentedFn.argsSchema = args`
	`36`	`+ augmentedFn.returnSchema = ret`
	`37`	`+`
	`38`	`+ return augmentedFn`
`35`	`39`	`}`