1
const remainingBalance = ...;
2
const result = streamText({
3
      model: ...,
4

5
      messages,
6
      system: ...,
7
      maxOutputTokens,
8
      onFinish: async ({ totalUsage }) => {
9
        logger.info('On-page AI Assistant: Response finished', {
10
          userId: user.id,
11
          ...totalUsage,
12
        });
13

14
        const totalTokens = totalUsage.totalTokens;
15
        await AIService.trackAITokensUsage(user.id, {
16
          totalTokens,
17
        });
18

19
        if (remainingBalance - totalTokens <= 0) {
20
          logger.warn('On-page AI Assistant: Page owner has no remaining balance after request', {
21
            userId: user.id,
22
            slug,
23
            remainingBalance,
24
            totalTokens
25
          });
26

27
          logger.info(
28
            `[REVALIDATION]: HITTING REVALIDATION FOR ${url.composePathname(Path.Page, slug)}`,
29
          );
30
     
31
          // no results from this call, I would expect revalidation of `/page/${slug}`
32
          revalidatePath(url.composePathname(Path.Page, slug));
33
        }
34
      },
35
    });
36

37
    return result.toUIMessageStreamResponse();