From 7d08bbd2624e3479c4154a2a7f6154fff2441a78 Mon Sep 17 00:00:00 2001
From: Yurun <admin@yurunsoft.com>
Date: Fri, 15 Sep 2023 11:19:05 +0800
Subject: [PATCH] =?UTF-8?q?OpenAI=20=E6=A8=A1=E5=9E=8B=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../views/config/components/config-openai.vue |  9 +-
 .../config/components/models-manager.vue      | 97 +++++++++++++++++++
 server/Module/Chat/Model/Redis/ChatConfig.php |  4 +-
 .../Embedding/Model/Redis/EmbeddingConfig.php |  4 +-
 .../OpenAI/Model/Redis/OpenAIConfig.php       |  4 +-
 5 files changed, 110 insertions(+), 8 deletions(-)
 create mode 100644 admin/src/views/config/components/models-manager.vue

diff --git a/admin/src/views/config/components/config-openai.vue b/admin/src/views/config/components/config-openai.vue
index bfbeed6..742c3ab 100644
--- a/admin/src/views/config/components/config-openai.vue
+++ b/admin/src/views/config/components/config-openai.vue
@@ -9,7 +9,10 @@
   >
     <n-space vertical>
       <n-card title="接口配置">
-        <api-manager v-model:apis="formData.apis" :models="props.value['config:openai'].config.models" />
+        <api-manager v-model:apis="formData.apis" :models="formData.models" />
+      </n-card>
+      <n-card title="模型管理">
+        <models-manager v-model:model-configs="formData.models" />
       </n-card>
     </n-space>
     <n-space class="w-full pt-16px" :size="24" justify="center">
@@ -24,12 +27,14 @@ import type { FormInst, FormRules } from 'naive-ui';
 import { defineConfigComponent } from '@/store';
 import type { ConfigComponentProps, ConfigComponentEmit } from '@/store';
 import ApiManager from './api-manager.vue';
+import ModelsManager from './models-manager.vue';
 
 const props = defineProps<ConfigComponentProps>();
 const emit = defineEmits<ConfigComponentEmit>();
 const rules: FormRules = {};
 const formData = ref({
-  apis: []
+  apis: [],
+  models: []
 });
 const form = ref<FormInst>();
 
diff --git a/admin/src/views/config/components/models-manager.vue b/admin/src/views/config/components/models-manager.vue
new file mode 100644
index 0000000..95c056d
--- /dev/null
+++ b/admin/src/views/config/components/models-manager.vue
@@ -0,0 +1,97 @@
+<template>
+  <n-scrollbar x-scrollable>
+    <n-table :single-line="false" striped class="w-max min-w-full">
+      <thead>
+        <tr class="text-center">
+          <th width="240">模型名称</th>
+          <th width="130">Token 输入倍率</th>
+          <th width="130">Token 输出倍率</th>
+          <th width="130">最大 Token</th>
+          <th width="70">付费标志</th>
+          <th>选中提示</th>
+          <th width="70">启用</th>
+          <th width="70">删除</th>
+        </tr>
+      </thead>
+      <tbody>
+        <tr v-for="(item, index) in modelConfigsData" :key="index">
+          <td>
+            <n-input v-model:value="item.model" />
+          </td>
+          <td>
+            <n-input-number v-model:value="item.inputTokenMultiple" :min="0" />
+          </td>
+          <td>
+            <n-input-number v-model:value="item.outputTokenMultiple" :min="0" />
+          </td>
+          <td><n-input-number v-model:value="item.maxTokens" :min="0" /></td>
+          <td class="text-center"><n-switch v-model:value="item.paying" /></td>
+          <td><n-input v-model:value="item.tips" type="textarea" :rows="2" /></td>
+          <td class="text-center"><n-switch v-model:value="item.enable" /></td>
+          <td>
+            <n-popconfirm :on-positive-click="() => handleDeleteModel(index)">
+              <template #default>确认删除？</template>
+              <template #trigger>
+                <n-button text block type="primary">删除</n-button>
+              </template>
+            </n-popconfirm>
+          </td>
+        </tr>
+      </tbody>
+    </n-table>
+    <n-space justify="center" class="mt-2">
+      <n-button type="primary" @click="handleAddModelConfig">
+        <icon-ic-round-plus class="mr-4px text-20px" />
+        增加一项
+      </n-button>
+    </n-space>
+  </n-scrollbar>
+</template>
+
+<script setup lang="tsx">
+import { watch, reactive } from 'vue';
+
+export interface Props {
+  modelConfigs: any[];
+}
+
+export interface Emit {
+  (e: 'update:modelConfigs', modelConfigs: any[]): void;
+}
+
+const props = defineProps<Props>();
+const emit = defineEmits<Emit>();
+
+const modelConfigsData = reactive(
+  (() => {
+    return props.modelConfigs.map(item => {
+      return {
+        ...item,
+        enable: item.enable === undefined ? true : item.enable,
+        inputTokenMultiple: parseFloat(item.inputTokenMultiple),
+        outputTokenMultiple: parseFloat(item.outputTokenMultiple)
+      };
+    });
+  })()
+);
+watch(modelConfigsData, value => {
+  emit('update:modelConfigs', value);
+});
+
+function handleDeleteModel(index: number) {
+  modelConfigsData.splice(index, 1);
+}
+
+function handleAddModelConfig() {
+  modelConfigsData.push({
+    model: '',
+    inputTokenMultiple: 1,
+    outputTokenMultiple: 1,
+    maxTokens: 4096,
+    tips: '',
+    enable: true
+  });
+}
+</script>
+
+<style scoped></style>
diff --git a/server/Module/Chat/Model/Redis/ChatConfig.php b/server/Module/Chat/Model/Redis/ChatConfig.php
index 71a4b81..ef4f742 100644
--- a/server/Module/Chat/Model/Redis/ChatConfig.php
+++ b/server/Module/Chat/Model/Redis/ChatConfig.php
@@ -43,8 +43,8 @@ public function getModelConfigs(): array
             return $this->modelConfigs = [
                 new ModelConfig(['model' => 'gpt-3.5-turbo', 'inputTokenMultiple' => '0.75', 'outputTokenMultiple' => '1.0', 'maxTokens' => 4096]),
                 new ModelConfig(['model' => 'gpt-3.5-turbo-16k', 'inputTokenMultiple' => '1.5', 'outputTokenMultiple' => '2.0', 'maxTokens' => 16384]),
-                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '150', 'outputTokenMultiple' => '3.0', 'maxTokens' => 8192]),
-                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '300', 'outputTokenMultiple' => '6.0', 'maxTokens' => 32768]),
+                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '15', 'outputTokenMultiple' => '30', 'maxTokens' => 8192]),
+                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '30', 'outputTokenMultiple' => '60', 'maxTokens' => 32768]),
             ];
         }
 
diff --git a/server/Module/Embedding/Model/Redis/EmbeddingConfig.php b/server/Module/Embedding/Model/Redis/EmbeddingConfig.php
index 4289995..fc4f03c 100644
--- a/server/Module/Embedding/Model/Redis/EmbeddingConfig.php
+++ b/server/Module/Embedding/Model/Redis/EmbeddingConfig.php
@@ -225,8 +225,8 @@ public function getChatModelConfigs(): ?array
             return $this->chatModelConfigs = [
                 new ModelConfig(['model' => 'gpt-3.5-turbo', 'inputTokenMultiple' => '0.75', 'outputTokenMultiple' => '1.0', 'maxTokens' => 4096]),
                 new ModelConfig(['model' => 'gpt-3.5-turbo-16k', 'inputTokenMultiple' => '1.5', 'outputTokenMultiple' => '2.0', 'maxTokens' => 16384]),
-                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '150', 'outputTokenMultiple' => '3.0', 'maxTokens' => 8192]),
-                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '300', 'outputTokenMultiple' => '6.0', 'maxTokens' => 32768]),
+                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '15', 'outputTokenMultiple' => '30', 'maxTokens' => 8192]),
+                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '30', 'outputTokenMultiple' => '60', 'maxTokens' => 32768]),
             ];
         }
 
diff --git a/server/Module/OpenAI/Model/Redis/OpenAIConfig.php b/server/Module/OpenAI/Model/Redis/OpenAIConfig.php
index 328d7ba..de352d6 100644
--- a/server/Module/OpenAI/Model/Redis/OpenAIConfig.php
+++ b/server/Module/OpenAI/Model/Redis/OpenAIConfig.php
@@ -95,8 +95,8 @@ public function getModels(): ?array
             return $this->models = [
                 new ModelConfig(['model' => 'gpt-3.5-turbo', 'inputTokenMultiple' => '0.75', 'outputTokenMultiple' => '1.0', 'maxTokens' => 4096]),
                 new ModelConfig(['model' => 'gpt-3.5-turbo-16k', 'inputTokenMultiple' => '1.5', 'outputTokenMultiple' => '2.0', 'maxTokens' => 16384]),
-                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '150', 'outputTokenMultiple' => '3.0', 'maxTokens' => 8192]),
-                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '300', 'outputTokenMultiple' => '6.0', 'maxTokens' => 32768]),
+                new ModelConfig(['model' => 'gpt-4', 'enable' => false, 'inputTokenMultiple' => '15', 'outputTokenMultiple' => '30', 'maxTokens' => 8192]),
+                new ModelConfig(['model' => 'gpt-4-32k', 'enable' => false, 'inputTokenMultiple' => '30', 'outputTokenMultiple' => '60', 'maxTokens' => 32768]),
                 new ModelConfig(['model' => 'text-embedding-ada-002', 'inputTokenMultiple' => '0.05', 'outputTokenMultiple' => '0.05', 'maxTokens' => 8191]),
             ];
         }