Function customToolFor

server/src/training-agent/framework-server.ts:327–357

(name: string, description: string, inputSchema: Record<string, z.ZodTypeAny>, handler: LegacyHandler)

  // eslint-disable-next-line @typescript-eslint/no-explicit-any
  function customToolFor(name: string, description: string, inputSchema: Record<string, z.ZodTypeAny>, handler: LegacyHandler): AdcpCustomToolConfig<any, undefined> {
    return {
      description,
      inputSchema,
      handler: async (args: unknown, extra: unknown) => {
        const params = (args as Record<string, unknown>) ?? {};
        const authInfo = ((extra as { authInfo?: { clientId?: string } } | undefined)?.authInfo) ?? undefined;
        const trainingCtx: TrainingContext = {
          mode: 'open',
          principal: authInfo?.clientId ?? 'anonymous',
        };
        const { context: callerContext, ...handlerArgs } = params;
        return runWithSessionContext(async () => {
          let result: unknown;
          try {
            result = await Promise.resolve(handler(handlerArgs as ToolArgs, trainingCtx));
          } catch (err) {
            logger.error({ err, tool: name }, 'framework custom-tool handler threw');
            return serviceUnavailable(err, callerContext);
          }
          try {
            await flushDirtySessions();
          } catch (err) {
            logger.error({ err, tool: name }, 'framework custom-tool flushDirtySessions threw');
            return serviceUnavailable(err, callerContext);
          }
          return toAdaptedResponse(result, callerContext);
        });
      },
    };
  }

  const ACCOUNT_REF = z.object({
    publisher_id: z.string().optional(),