fix: use sample

JimmyLv · Mar 3, 2023 · 823e01d · 823e01d · vercel · Mar 3, 2023
1 parent 3732b41
commit 823e01d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/utils/3rd/bilibili.ts b/utils/3rd/bilibili.ts
@@ -1,4 +1,4 @@
-import { sample } from "lodash";
+import { sample } from "../fp";
 
 const run = async (bvId: string) => {
   const requestUrl = `https://api.bilibili.com/x/web-interface/view?bvid=${bvId}`;

diff --git a/utils/OpenAIResult.ts b/utils/OpenAIResult.ts
@@ -4,6 +4,7 @@ import {
   ReconnectInterval,
 } from "eventsource-parser";
 import { checkOpenaiApiKey } from "./3rd/openai";
+import { sample } from "./fp";
 
 // TODO: maybe chat with video?
 export type ChatGPTAgent = "user" | "system" | "assistant";
@@ -25,11 +26,6 @@ export interface OpenAIStreamPayload {
   n: number;
 }
 
-const sample = (arr: any[] = []) => {
-  const len = arr === null ? 0 : arr.length;
-  return len ? arr[Math.floor(Math.random() * len)] : undefined;
-};
-
 function formatResult(result: any) {
   const answer = result.choices[0].message?.content || "";
   if (answer.startsWith("\n\n")) {

diff --git a/utils/fp.ts b/utils/fp.ts
@@ -0,0 +1,4 @@
+export const sample = (arr: any[] = []) => {
+  const len = arr === null ? 0 : arr.length;
+  return len ? arr[Math.floor(Math.random() * len)] : undefined;
+};