v4.4.6 (#377)

packages/core/ai/config.ts

@@ -0,0 +1,28 @@
+import { UserModelSchema } from '../user/type';
+import { Configuration, OpenAIApi } from 'openai';
+
+export const openaiBaseUrl = process.env.OPENAI_BASE_URL || 'https://api.openai.com/v1';
+export const baseUrl = process.env.ONEAPI_URL || openaiBaseUrl;
+
+export const systemAIChatKey = process.env.CHAT_API_KEY || '';
+
+export const getAIChatApi = (props?: UserModelSchema['openaiAccount']) => {
+  return new OpenAIApi(
+    new Configuration({
+      basePath: props?.baseUrl || baseUrl,
+      apiKey: props?.key || systemAIChatKey
+    })
+  );
+};
+
+/* openai axios config */
+export const axiosConfig = (props?: UserModelSchema['openaiAccount']) => {
+  return {
+    baseURL: props?.baseUrl || baseUrl, // 此处仅对非 npm 模块有效
+    httpsAgent: global.httpsAgent,
+    headers: {
+      Authorization: `Bearer ${props?.key || systemAIChatKey}`,
+      auth: process.env.OPENAI_BASE_URL_AUTH || ''
+    }
+  };
+};

packages/core/ai/constant.ts

@@ -0,0 +1 @@
+export { ChatCompletionRequestMessageRoleEnum } from 'openai';

packages/core/ai/functions/createQuestionGuide.ts

@@ -0,0 +1,57 @@
+import { ChatCompletionRequestMessage } from '../type';
+import { getAIChatApi } from '../config';
+
+export const Prompt_QuestionGuide = `我不太清楚问你什么问题，请帮我生成 3 个问题，引导我继续提问。问题的长度应小于20个字符，按 JSON 格式返回: ["问题1", "问题2", "问题3"]`;
+
+export async function createQuestionGuide({
+  messages,
+  model
+}: {
+  messages: ChatCompletionRequestMessage[];
+  model: string;
+}) {
+  const chatAPI = getAIChatApi();
+  const { data } = await chatAPI.createChatCompletion({
+    model: model,
+    temperature: 0,
+    max_tokens: 200,
+    messages: [
+      ...messages,
+      {
+        role: 'user',
+        content: Prompt_QuestionGuide
+      }
+    ],
+    stream: false
+  });
+
+  const answer = data.choices?.[0].message?.content || '';
+  const totalTokens = data.usage?.total_tokens || 0;
+
+  const start = answer.indexOf('[');
+  const end = answer.lastIndexOf(']');
+
+  if (start === -1 || end === -1) {
+    return {
+      result: [],
+      tokens: totalTokens
+    };
+  }
+
+  const jsonStr = answer
+    .substring(start, end + 1)
+    .replace(/(\\n|\\)/g, '')
+    .replace(/  /g, '');
+
+  try {
+    return {
+      result: JSON.parse(jsonStr),
+      tokens: totalTokens
+    };
+  } catch (error) {
+    return {
+      result: [],
+      tokens: totalTokens
+    };
+  }
+}

packages/core/ai/type.d.ts

@@ -0,0 +1 @@
+export type { CreateChatCompletionRequest, ChatCompletionRequestMessage } from 'openai';