diff --git a/server/server-route.ts b/server/server-route.ts index 4364af5..4c1f4c7 100755 --- a/server/server-route.ts +++ b/server/server-route.ts @@ -31,7 +31,9 @@ export const newServerRouter = < return arg; }; -export const createClientForServer = >( +export const createClientForServer = < + T extends ReturnType +>( router: T ) => { return new Proxy( @@ -52,7 +54,7 @@ export const createClientForServer = >( +export const useServerRouter = >( router: T ) => { const rou = createRouter<{ @@ -63,7 +65,7 @@ export const useServerRouter = async (arg: { }): ServerSession => { const internal = { has_router: false, - router: null as null | Awaited>, + router: null as null | ReturnType, }; if (typeof arg.router === "object" && arg.router instanceof Promise) { internal.has_router = true;