feat: support different cachedId

lib/openai/fetchOpenAIResult.ts

@@ -1,12 +1,9 @@
 import { Redis } from "@upstash/redis";
-import {
-  createParser,
-  ParsedEvent,
-  ReconnectInterval,
-} from "eventsource-parser";
+import { createParser, ParsedEvent, ReconnectInterval } from "eventsource-parser";
 import { trimOpenAiResult } from "~/lib/openai/trimOpenAiResult";
 import { VideoConfig } from "~/lib/types";
 import { isDev } from "~/utils/env";
+import { getCacheId } from "~/utils/getCacheId";
 
 export enum ChatGPTAgent {
   user = "user",
@@ -54,9 +51,8 @@ export async function fetchOpenAIResult(
     throw new Error(`OpenAI API Error [${res.statusText}]: ${errorJson.error?.message}`);
   }
 
-  const { showTimestamp, videoId } = videoConfig;
   const redis = Redis.fromEnv();
-  const cacheId = showTimestamp ? `timestamp-${videoId}` : videoId;
+  const cacheId = getCacheId(videoConfig);
 
   if (!payload.stream) {
     const result = await res.json();

middleware.ts

@@ -3,13 +3,10 @@ import { Redis } from "@upstash/redis";
 import type { NextFetchEvent, NextRequest } from "next/server";
 import { NextResponse } from "next/server";
 import { SummarizeParams } from "~/lib/types";
+import { getCacheId } from "~/utils/getCacheId";
 import { validateLicenseKey } from "./lib/lemon";
 import { checkOpenaiApiKeys } from "./lib/openai/checkOpenaiApiKey";
-import {
-  ratelimitForApiKeyIps,
-  ratelimitForFreeAccounts,
-  ratelimitForIps,
-} from "./lib/upstash";
+import { ratelimitForApiKeyIps, ratelimitForFreeAccounts, ratelimitForIps } from "./lib/upstash";
 import { isDev } from "./utils/env";
 
 const redis = Redis.fromEnv();
@@ -32,9 +29,8 @@ export async function middleware(req: NextRequest, context: NextFetchEvent) {
   try {
     const { userConfig, videoConfig } = (await req.json()) as SummarizeParams;
     // TODO: update shouldShowTimestamp to use videoConfig
-    const { userKey, shouldShowTimestamp } = userConfig || {};
-    const { videoId } = videoConfig || {};
-    const cacheId = shouldShowTimestamp ? `timestamp-${videoId}` : videoId;
+    const { userKey } = userConfig || {};
+    const cacheId = getCacheId(videoConfig);
     const ipIdentifier = req.ip ?? "127.0.0.11";
 
     // licenseKeys

utils/getCacheId.ts

@@ -0,0 +1,11 @@
+import { VideoConfig } from "~/lib/types";
+
+export function getCacheId({
+  showTimestamp,
+  videoId,
+  detailLevel,
+}: VideoConfig) {
+  return showTimestamp
+    ? `timestamp-${videoId}-${detailLevel}`
+    : `${videoId}-${detailLevel}`;
+}