From a11b3d0c4f3b3d393de64cb1e22c9a38eeb63c07 Mon Sep 17 00:00:00 2001
From: Kirk Lin <linkirk@163.com>
Date: Sun, 25 Feb 2024 13:53:33 +0800
Subject: [PATCH] feat: add mock ai

---
 .../src/pages/chat/components/ChatLayout.vue  |  1 -
 .../MessageInputArea/ChatInputAreaInner.vue   |  2 +-
 .../languageModelAgent/azureOpenai/index.ts   |  6 +-
 .../src/languageModelAgent/celeris/index.ts   | 58 +++++++++++++++++++
 .../src/languageModelAgent/openai/index.ts    |  8 +--
 .../utils/error/constants.ts                  |  1 +
 packages/ai/services/openai/agent/index.ts    | 11 +++-
 7 files changed, 75 insertions(+), 12 deletions(-)
 create mode 100644 packages/ai/core/src/languageModelAgent/celeris/index.ts

diff --git a/apps/admin/src/pages/chat/components/ChatLayout.vue b/apps/admin/src/pages/chat/components/ChatLayout.vue
index a5203ce8..28bdbe32 100644
--- a/apps/admin/src/pages/chat/components/ChatLayout.vue
+++ b/apps/admin/src/pages/chat/components/ChatLayout.vue
@@ -45,7 +45,6 @@ watchEffect(() => {
           :width="360"
           :native-scrollbar="true"
           show-trigger="arrow-circle"
-          bordered
         >
           <ChatHistorySidebar />
         </NLayoutSider>
diff --git a/apps/admin/src/pages/chat/components/ChatPanel/MessageInputArea/ChatInputAreaInner.vue b/apps/admin/src/pages/chat/components/ChatPanel/MessageInputArea/ChatInputAreaInner.vue
index ca29ac02..5b836a81 100644
--- a/apps/admin/src/pages/chat/components/ChatPanel/MessageInputArea/ChatInputAreaInner.vue
+++ b/apps/admin/src/pages/chat/components/ChatPanel/MessageInputArea/ChatInputAreaInner.vue
@@ -3,7 +3,7 @@ defineModel();
 </script>
 
 <template>
-  <NInput type="textarea" placeholder="请输入聊天内容。。。。" round show-count :resizable="false" class="h-full" :bordered="false" />
+  <NInput type="textarea" placeholder="请输入聊天内容。。。。" round show-count :resizable="false" class="h-full bg-[--action-color]" />
 </template>
 
 <style scoped>
diff --git a/packages/ai/core/src/languageModelAgent/azureOpenai/index.ts b/packages/ai/core/src/languageModelAgent/azureOpenai/index.ts
index 7771e2cc..1fb6c83d 100644
--- a/packages/ai/core/src/languageModelAgent/azureOpenai/index.ts
+++ b/packages/ai/core/src/languageModelAgent/azureOpenai/index.ts
@@ -8,12 +8,10 @@ import {
 } from "@azure/openai";
 import { OpenAIStream, StreamingTextResponse } from "ai";
 
-import { LanguageModelAgentRuntimeErrorType } from "../utils/error/constants";
+import { AgentRuntimeError, DEBUG_CHAT_COMPLETION, LanguageModelAgentRuntimeErrorType, debugStream } from "../utils";
 import type { OpenAIChatStreamPayload } from "../../types";
 import { ModelBrandProvider } from "../types";
-import { AgentRuntimeError } from "../utils/error/createError";
-import { debugStream } from "../utils/debugStream";
-import { DEBUG_CHAT_COMPLETION } from "../utils/env";
+
 import { AbstractLanguageModel } from "../abstractAI";
 
 export class CelerisAzureOpenAI extends AbstractLanguageModel<any> {
diff --git a/packages/ai/core/src/languageModelAgent/celeris/index.ts b/packages/ai/core/src/languageModelAgent/celeris/index.ts
new file mode 100644
index 00000000..be9b45c9
--- /dev/null
+++ b/packages/ai/core/src/languageModelAgent/celeris/index.ts
@@ -0,0 +1,58 @@
+import { StreamingTextResponse } from "ai";
+import { AbstractLanguageModel } from "../abstractAI";
+import { ModelBrandProvider } from "../types";
+import { AgentRuntimeError, DEBUG_CHAT_COMPLETION, LanguageModelAgentRuntimeErrorType, debugStream } from "../utils";
+import type { OpenAIChatMessage, OpenAIChatStreamPayload } from "../../types";
+
+export class CelerisMockLanguageModel extends AbstractLanguageModel<any> {
+  constructor() {
+    super(ModelBrandProvider.Celeris);
+  }
+
+  async streamChatCompletions(messages: OpenAIChatMessage[]) {
+    // 模拟将响应转换为流
+    const stream = new ReadableStream({
+      start(controller) {
+        messages.map(message =>
+          // 模拟处理消息
+          controller.enqueue(`Mock completion for ${message.content}`));
+        controller.close();
+      },
+    });
+
+    return stream;
+  }
+
+  async chat(payload: OpenAIChatStreamPayload) {
+    // ============  1. preprocess messages   ============ //
+    const { messages } = payload;
+
+    // ============  2. send api   ============ //
+
+    try {
+      const response = await this.streamChatCompletions(
+        messages,
+      );
+
+      const [debug, prod] = response.tee();
+
+      if (DEBUG_CHAT_COMPLETION) {
+        debugStream(debug).catch(console.error);
+      }
+
+      return new StreamingTextResponse(prod);
+    } catch (e) {
+      const error = e as { [key: string]: any; code: string; message: string };
+
+      const errorType = error.code
+        ? LanguageModelAgentRuntimeErrorType.MockAIBusinessError
+        : LanguageModelAgentRuntimeErrorType.AgentRuntimeError;
+
+      throw AgentRuntimeError.chat({
+        error,
+        errorType,
+        provider: this._brandId,
+      });
+    }
+  }
+}
diff --git a/packages/ai/core/src/languageModelAgent/openai/index.ts b/packages/ai/core/src/languageModelAgent/openai/index.ts
index 5dcefa15..b472dcb3 100644
--- a/packages/ai/core/src/languageModelAgent/openai/index.ts
+++ b/packages/ai/core/src/languageModelAgent/openai/index.ts
@@ -3,12 +3,10 @@ import type { ClientOptions } from "openai";
 import urlJoin from "url-join";
 import { OpenAIStream, StreamingTextResponse } from "ai";
 import { AbstractLanguageModel } from "../abstractAI";
-import { AgentRuntimeError } from "../utils/error/createError";
-import { LanguageModelAgentRuntimeErrorType } from "../utils/error/constants";
+import { AgentRuntimeError, DEBUG_CHAT_COMPLETION, LanguageModelAgentRuntimeErrorType, debugStream, handleOpenAIError } from "../utils";
+
 import type { OpenAIChatStreamPayload } from "../../types";
-import { DEBUG_CHAT_COMPLETION } from "../utils/env";
-import { debugStream } from "../utils/debugStream";
-import { handleOpenAIError } from "../utils/error/handleOpenAIError";
+
 import { desensitizeUrl } from "../utils/desensitizeUrl";
 import { ModelBrandProvider } from "../types";
 
diff --git a/packages/ai/core/src/languageModelAgent/utils/error/constants.ts b/packages/ai/core/src/languageModelAgent/utils/error/constants.ts
index f6acf1ba..39b7f7f1 100644
--- a/packages/ai/core/src/languageModelAgent/utils/error/constants.ts
+++ b/packages/ai/core/src/languageModelAgent/utils/error/constants.ts
@@ -2,6 +2,7 @@ export const LanguageModelAgentRuntimeErrorType = {
   AgentRuntimeError: "AgentRuntimeError", // Agent Runtime 模块运行时错误
   LocationNotSupportedError: "LocationNotSupportedError", // 不支持的位置错误
   OpenAIBusinessError: "OpenAIBusinessError", // OpenAI 业务错误
+  MockAIBusinessError: "MockAIBusinessError", // Mock 业务错误
   NoOpenAIAPIKey: "NoOpenAIAPIKey", // 缺少 OpenAI API 密钥
   InvalidAzureAPIKey: "InvalidAzureAPIKey", // 无效的 Azure API 密钥
   AzureBusinessError: "AzureBusinessError", // Azure 业务错误
diff --git a/packages/ai/services/openai/agent/index.ts b/packages/ai/services/openai/agent/index.ts
index 0c2eb0fd..ca7514f8 100644
--- a/packages/ai/services/openai/agent/index.ts
+++ b/packages/ai/services/openai/agent/index.ts
@@ -7,6 +7,7 @@ import type {
   AbstractLanguageModel,
   OpenAIChatStreamPayload,
 } from "@celeris/ai-core";
+import { CelerisMockLanguageModel } from "@celeris/ai-core/src/languageModelAgent/celeris";
 import { getServerConfig } from "../config/server";
 import type { JWTPayload } from "../constants/auth";
 
@@ -37,7 +38,6 @@ class LanguageModelAgent {
     switch (provider) {
       // eslint-disable-next-line default-case-last
       default:
-      case "oneapi":
       case ModelBrandProvider.OpenAI: {
         runtimeModel = this.initOpenAI(payload, azureOpenAI);
         break;
@@ -47,6 +47,11 @@ class LanguageModelAgent {
         runtimeModel = this.initAzureOpenAI(payload);
         break;
       }
+
+      case ModelBrandProvider.Celeris:{
+        runtimeModel = this.initMockAI();
+        break;
+      }
     }
 
     return new LanguageModelAgent(runtimeModel);
@@ -73,6 +78,10 @@ class LanguageModelAgent {
     });
   }
 
+  private static initMockAI() {
+    return new CelerisMockLanguageModel();
+  }
+
   private static initAzureOpenAI(payload: JWTPayload) {
     const { AZURE_API_KEY, AZURE_API_VERSION, AZURE_ENDPOINT } = getServerConfig();
     const apiKey = payload?.apiKey || AZURE_API_KEY;