feat: add flag for better prompt

pages/api/summarize.ts

@@ -1,6 +1,7 @@
-import { NextResponse } from "next/server";
 import { Redis } from "@upstash/redis";
 import type { NextFetchEvent, NextRequest } from "next/server";
+import { NextResponse } from "next/server";
+import pRetry from "p-retry";
 import { isDev } from "../../utils/env";
 import { OpenAIResult } from "../../utils/OpenAIResult";
 import { getChunckedTranscripts, getSummaryPrompt } from "../../utils/prompt";
@@ -13,6 +14,21 @@ if (!process.env.OPENAI_API_KEY) {
   throw new Error("Missing env var from OpenAI");
 }
 
+const run = async (bvId: string) => {
+  const requestUrl = `https://api.bilibili.com/x/web-interface/view?bvid=${bvId}`;
+  console.log(`fetch`, requestUrl);
+  const response = await fetch(requestUrl, {
+    method: "GET",
+  });
+  const json = await response.json();
+  const subtitleList = json.data?.subtitle?.list;
+  if (!subtitleList || subtitleList?.length < 1) {
+    throw new Error(response.statusText);
+  }
+
+  return json;
+};
+
 export default async function handler(
   req: NextRequest,
   context: NextFetchEvent
@@ -25,16 +41,19 @@ export default async function handler(
   if (!bvId) {
     return new Response("No bvid in the request", { status: 500 });
   }
-  const requestUrl = `https://api.bilibili.com/x/web-interface/view?bvid=${bvId}`;
-  console.log(`fetch`, requestUrl);
-  const response = await fetch(requestUrl, {
-    method: "GET",
+  const res = await pRetry(() => run(bvId), {
+    onFailedAttempt: (error) => {
+      console.log(
+        `Attempt ${error.attemptNumber} failed. There are ${error.retriesLeft} retries left.`
+      );
+    },
+    retries: 3,
   });
-  const res = await response.json();
   // @ts-ignore
   const title = res.data?.title;
   const subtitleList = res.data?.subtitle?.list;
-  if (!subtitleList || subtitleList?.length === 0) {
+  if (!subtitleList || subtitleList?.length < 1) {
+    console.error("No subtitle in the video: ", bvId);
     return new Response("No subtitle in the video", { status: 501 });
   }
   const betterSubtitle =
@@ -54,10 +73,10 @@ export default async function handler(
   });
   // console.log("========transcripts========", transcripts);
   const text = getChunckedTranscripts(transcripts, transcripts);
-  const prompt = getSummaryPrompt(title, text);
+  const prompt = getSummaryPrompt(title, text, true);
 
   try {
-    apiKey && console.log("========use user key========");
+    apiKey && console.log("========use user apiKey========");
     isDev && console.log("prompt", prompt);
     const payload = {
       model: "gpt-3.5-turbo",

utils/prompt.ts

@@ -1,10 +1,13 @@
 
-export function getSummaryPrompt(title: string,transcript: any) {
-    return `标题: "${title
-      .replace(/\n+/g, " ")
-      .trim()}"\n视频字幕: "${truncateTranscript(transcript)
-      .replace(/\n+/g, " ")
-      .trim()}"\n我希望你是一名专业的视频内容编辑，帮我总结视频的内容精华。请先用一句简短的话总结视频梗概。然后再请你将视频字幕文本进行总结，在每句话的最前面加上时间戳，然后以无序列表的方式返回，不要超过5条。记得不要重复句子，确保所有的句子都足够精简，清晰完整，祝你好运！`;
+export function getSummaryPrompt(title: string, transcript: any, hasNoTimestamp?: boolean) {
+  const betterPrompt = `我希望你是一名专业的视频内容编辑，帮我总结视频的内容精华。请你将视频字幕文本进行总结，然后以无序列表的方式返回，不要超过5条。记得不要重复句子，确保所有的句子都足够精简，清晰完整，祝你好运！`
+  const promptWithTimestamp = `我希望你是一名专业的视频内容编辑，帮我总结视频的内容精华。请先用一句简短的话总结视频梗概。然后再请你将视频字幕文本进行总结，在每句话的最前面加上开始的时间戳，然后以无序列表的方式返回。请注意不要超过5条哦，确保所有的句子都足够精简，清晰完整，祝你好运！`;
+
+  return `标题: "${title
+    .replace(/\n+/g, " ")
+    .trim()}"\n视频字幕: "${truncateTranscript(transcript)
+    .replace(/\n+/g, " ")
+    .trim()}"\n${hasNoTimestamp ? betterPrompt : promptWithTimestamp}`;
   }
 
   // Seems like 15,000 bytes is the limit for the prompt