Vitest context router caller

Hi, Im trying to setup vitest to test trpc. I would like to have a trpc approuter caller to be accessible from vitest context but im struggling with typings of appRouter.createCaller(ctx) return type

//vitest.setup.ts

// eslint-disable-next-line import/no-extraneous-dependencies
import { afterEach, beforeEach, vi } from 'vitest'
import { createInnerTRPCContext } from './src/server/api/trpc'
import { appRouter } from './src/server/api/root'

const contextCaller = async (schemaFilters) => {
  const options = {
    req: {
      authorisation: { authorised: true },
      logger: {
        info: vi.fn(),
        warn: vi.fn(),
        debug: vi.fn(),
        error: vi.fn(),
      },
    },
    res: vi.fn(),
    currentSchema: schemaFilters ? { filters: { ...schemaFilters } } : {},
  }

  const ctx = await createInnerTRPCContext(options)
  return appRouter.createCaller(ctx)
}

beforeEach((context) => {
  vi.mock('./src/server/tlsHttpClient', () => ({
    httpClient: {
      get: vi.fn().mockReturnValue({ execute: vi.fn() }),
      post: vi.fn(),
      delete: vi.fn(),
      put: vi.fn(),
    },
  }))

  context.trpcContextCaller = contextCaller
})

afterEach(() => {
  vi.clearAllMocks()
})

How I could type the contextCaller function? It currently inffers Promise<any>