Upload 8 files

Dockerfile

@@ -8,6 +8,8 @@ RUN git clone https://github.com/yokingma/search_with_ai.git /app
 COPY ./logo.png /app/web/src/assets/logo.png
 COPY ./logo.png /app/web/src/public/favicon.ico
 COPY ./home.vue /app/web/src/pages/home.vue
+COPY ./toolbar.vue /app/web/src/components/toolbar.vue
+COPY ./constant.ts /app/backend/constant.ts
 
 WORKDIR /app
 RUN yarn install && yarn run build

app.ts

@@ -0,0 +1,82 @@
+import Koa from 'koa';
+import Router from '@koa/router';
+import cors from '@koa/cors';
+import { bodyParser } from '@koa/bodyparser';
+import serve from 'koa-static';
+import path from 'path';
+import { whiteListMiddleware } from './middlewares';
+import history from 'koa2-connect-history-api-fallback';
+import { chatStreamController, localChatStreamController, localModelsController, modelsController, searchController, sogouSearchController } from './controllers';
+import { initializeModels } from './constant'; // 请确保路径正确
+
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-ignore
+import dotenvx from '@dotenvx/dotenvx';
+dotenvx.config();
+
+const app = new Koa();
+const router = new Router();
+
+const port = process.env.PORT || 3000;
+
+app.use(history({
+  index: '/index.html',
+  whiteList: ['/api']
+}));
+
+// static path
+const staticPath = path.join(__dirname, '../web/build');
+app.use(serve(staticPath, {
+  gzip: true,
+  index: 'index.html'
+}));
+
+app.use(cors({
+  origin: '*'
+}));
+
+app.use(bodyParser());
+
+// Error handler
+app.use(async (ctx, next) => {
+  try {
+    await next();
+  } catch(err) {
+    console.error('[server error]', err);
+    ctx.res.statusCode = 422;
+    ctx.body = err;
+  }
+});
+
+// router
+app.use(router.routes()).use(router.allowedMethods());
+
+// controller
+router.post('/api/search', whiteListMiddleware(), searchController);
+router.post('/api/sogou/search', sogouSearchController);
+router.post('/api/chat', chatStreamController);
+router.get('/api/models', modelsController);
+
+// local llm
+router.get('/api/local/models', localModelsController);
+router.post('/api/local/chat', localChatStreamController);
+
+// 创建异步启动函数
+async function startApp() {
+  try {
+    // 初始化模型
+    await initializeModels();
+    console.log('Models initialized successfully');
+
+    // 启动服务器
+    app.listen(port, () => {
+      console.log(`Server is running on port ${port}`);
+    });
+  } catch (error) {
+    console.error('Failed to start the application:', error);
+    process.exit(1);
+  }
+}
+
+// 调用启动函数
+startApp();

constant.ts

@@ -0,0 +1,95 @@
+import { IModelInfo } from './interface';
+import axios from 'axios';
+
+// Search engine related. You don't really need to change this.
+export const EndPoint = {
+  BING_SEARCH_V7_ENDPOINT: 'https://api.bing.microsoft.com/v7.0/search',
+  GOOGLE_SEARCH_ENDPOINT: 'https://www.googleapis.com/customsearch/v1'
+};
+
+export const BING_MKT = 'en-US';
+
+// default timeout ms
+export const DEFAULT_SEARCH_ENGINE_TIMEOUT = 20000;
+
+// default search keywords
+export const DefaultQuery = 'Who said \'live long and prosper';
+export const DefaultSystem = 'You are a helpful assistant.';
+
+// 创建一个异步函数来获取模型列表
+async function fetchOpenAIModels(): Promise<string[]> {
+  try {
+    const response = await axios.get('https://api.deem.love/v1/models');
+    const data = response.data;
+    return data.data.map((model: { id: string }) => model.id);
+  } catch (error) {
+    console.error('获取模型列表失败:', error);
+    return [];
+  }
+}
+
+// 修改 Models 数组,将 'openai' 平台的模型设置为动态获取
+export const Models: IModelInfo[] = [
+  {
+    platform: 'aliyun',
+    type: '',
+    models: ['qwen-max', 'qwen-max-1201', 'qwen-turbo', 'qwen-plus']
+  },
+  {
+    platform: 'openai',
+    type: 'openai',
+    //models: ['gpt-3.5-turbo', 'gpt-4-0125-preview', 'gpt-4-turbo-preview', 'gpt-4o']
+    models: [] // 初始为空数组,稍后动态填充
+  },
+  {
+    platform: 'baidu',
+    type: 'baidu',
+    models: ['eb-instant', 'completions_pro', 'ernie_bot_8k']
+  },
+  {
+    platform: 'google',
+    type: 'gemini',
+    models: ['gemini-pro', 'gemini-1.5-pro-latest']
+  },
+  {
+    platform: 'yi',
+    type: 'openai',
+    models: ['yi-34b-chat-0205', 'yi-34b-chat-200k']
+  },
+  {
+    platform: 'moonshot',
+    type: 'openai',
+    models: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k']
+  },
+  {
+    platform: 'lepton',
+    type: 'openai',
+    models: ['llama2-7b', 'llama2-13b', 'llama2-70b', 'mixtral-8*7b', 'mixtral-8*22b']
+  },
+  {
+    platform: 'deepseek',
+    type: 'openai',
+    models: ['deepseek-chat', 'deepseek-coder']
+  },
+  {
+    platform: 'chatglm',
+    type: 'openai',
+    models: ['glm-4', 'glm-4v', 'glm-3-turbo']
+  },
+  {
+    platform: 'tencent',
+    type: 'tencent',
+    models: ['std', 'pro']
+  }
+];
+
+// 创建一个初始化函数来更新 Models 数组
+export async function initializeModels() {
+  const openAIModels = await fetchOpenAIModels();
+  Models = Models.map(platform => {
+    if (platform.platform === 'openai') {
+      return { ...platform, models: openAIModels };
+    }
+    return platform;
+  });
+}

home.vue

@@ -4,7 +4,6 @@
       <div class="flex items-center justify-center gap-2">
         <img :src="logoUrl" class="w-10" />
         <span class="text-3xl font-bold dark:text-gray-100">AI Search</span>
-        <t-tag variant="light" class="text-xs text-gray-500"></t-tag>
       </div>
       <SearchInputBar :autofocus="true" :loading="false" @search="search" />
       <div class="my-2 flex flex-wrap items-center justify-center gap-4">

toolbar.vue

@@ -0,0 +1,32 @@
+<script setup lang="ts">
+import { RiSettingsLine, RiGithubLine } from '@remixicon/vue';
+
+type Emit = {
+  (e: 'show'): void;
+}
+
+const emits = defineEmits<Emit>();
+
+
+</script>
+
+<script lang="ts">
+export default {
+  name: 'ToolBar'
+};
+</script>
+
+<template>
+  <div class="fixed bottom-1/3 right-4 z-50 flex flex-col items-center justify-center gap-4">
+    <div class="flex w-9 justify-center gap-2 rounded-xl bg-gray-200 p-1 shadow-lg dark:bg-gray-600">
+      <t-button href="https://deem.love" target="_blank" shape="circle" theme="default">
+        <template #icon> <RiGithubLine /></template>
+      </t-button>
+    </div>
+    <div class="flex w-9 justify-center gap-2 rounded-xl bg-gray-200 p-1 shadow-lg dark:bg-gray-600">
+      <t-button shape="circle" theme="default" @click="emits('show')">
+        <template #icon> <RiSettingsLine /></template>
+      </t-button>
+    </div>
+  </div>
+</template>