@mybricks/plugin-ai 0.0.1 → 0.0.2

package/src/preset/requestAsStream.ts

@@ -0,0 +1,246 @@
+import forge from "node-forge";
+
+function generateRandomKey(length: number) {
+  const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
+  let result = "";
+  for (let i = 0; i < length; i++) {
+    const randomIndex = Math.floor(Math.random() * chars.length);
+    result += chars[randomIndex];
+  }
+  return result;
+}
+
+function getAiEncryptData(params: { mode: any; data: unknown }) {
+  const { mode, data } = params;
+  if (mode === "development") {
+    return data;
+  }
+  const PUBLIC_KEY = `-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA1ITWRl6ePMu7Fhusup2d
+FEz/hCRTE5mUIeGIjtezG5g8ewBdTaR2FRxtTFONYTaaSR6yFXm9k74tkS1/i0Z8
+7eIV130XydOn4zFhk2sOkG46mQ+lZwJkyVwvMaAOCnHluTIaPMPMV3sYpp3cWspl
+2H++R5/kOGVm6EG9HivrimQEKDDJLg9owbfWO2kSEM9ZpUHUt29msYq+lDtBrivG
+oodvC8p5H4a/jXKvLtPRGO09ZO3xk1ktS8isc376Ec9L9Zo8wSwaj5Z/Pg7nd7Sa
+tqj5BEj3YH8rSr1dg77ZMMH1lsuzdA0NHmRGYEvWnUoD6dMqjJjufNwAw9D47DQH
+lwIDAQAB
+-----END PUBLIC KEY-----`;
+
+  // 生成一个随机的AES密钥
+  const AESKey = generateRandomKey(16);
+
+  // 用AES密钥加密数据
+  const cipher = forge.cipher.createCipher("AES-CBC", AESKey);
+  cipher.start({ iv: AESKey });
+  cipher.update(
+    forge.util.createBuffer(forge.util.encodeUtf8(JSON.stringify(data))),
+  );
+  cipher.finish();
+  const encryptedData = forge.util.encode64(cipher.output.getBytes());
+
+  // 使用RSA公钥加密AES密钥
+  const publicKey = forge.pki.publicKeyFromPem(PUBLIC_KEY);
+  const encryptedAESKey = forge.util.encode64(publicKey.encrypt(AESKey));
+
+  return { chatContent: encryptedData, chatKey: encryptedAESKey };
+}
+
+const logger = {
+  info(message: string) {
+    console.log(
+      "%c%s%c %s",
+      "background-color: #fa6400; color: #ffffff;padding: 0px 6px",
+      "AI-SDK",
+      "color: #ffffff",
+      message,
+    );
+  },
+};
+
+enum FetchTarget {
+  CustomApp = "CustomApp",
+  Platform = "Platform",
+  Center = "Center",
+}
+
+let fetchTaget: FetchTarget;
+
+async function checkFetchTarget(): Promise<FetchTarget> {
+  if (fetchTaget) {
+    return Promise.resolve(fetchTaget);
+  }
+
+  /** 如果安装了自定义的AI服务，请求自定义服务 */
+  const hasAICustomApp = await fetch("/api/ai-service/check-config")
+    .then((res) => {
+      return res.json();
+    })
+    .then((data: any) => {
+      if (data?.code === 1) {
+        return true;
+      } else {
+        return false;
+      }
+    })
+    .catch((e: any) => {
+      return false;
+    });
+
+  if (hasAICustomApp) {
+    logger.info("使用自定义服务");
+    return (fetchTaget = FetchTarget.CustomApp);
+  }
+
+  /** 如果配置了平台token，请求平台服务 */
+  const hasPlatformToken = await fetch("/api/assistant/status")
+    .then((res) => {
+      return res.json();
+    })
+    .then((data: any) => {
+      if (data?.code === 1) {
+        return true;
+      } else {
+        return false;
+      }
+    })
+    .catch((e: any) => {
+      return false;
+    });
+
+  if (hasPlatformToken) {
+    logger.info("使用平台服务");
+    return (fetchTaget = FetchTarget.Platform);
+  }
+
+  logger.info("使用AI服务");
+  return (fetchTaget = FetchTarget.Center);
+}
+
+const transfromExtendParams = (extendParams: { aiRole?: string }) => {
+  const { aiRole } = extendParams;
+  let model = "google/gemini-3-flash-preview";
+  let role = "default";
+
+  if (!aiRole) {
+    return {
+      model,
+      role,
+    };
+  }
+
+  switch (true) {
+    case ["image"].includes(aiRole): {
+      model = "anthropic/claude-sonnet-4.5";
+      role = "image";
+      break;
+    }
+    case ["junior"].includes(aiRole): {
+      model = "z-ai/glm-4.7";
+      role = "junior";
+      break;
+    }
+    case ["architect"].includes(aiRole): {
+      model = "google/gemini-3-pro-preview";
+      role = "architect";
+      break;
+    }
+    case ["expert"].includes(aiRole): {
+      model = "anthropic/claude-sonnet-4.5";
+      role = "expert";
+      break;
+    }
+    default: {
+      role = "default";
+      break;
+    }
+  }
+
+  return {
+    model,
+    role,
+  };
+};
+
+export type RequestAsStreamMode = "development" | "production";
+
+const createRequestAsStream = (mode: RequestAsStreamMode) =>
+  async (params: {
+    messages: any;
+    emits: any;
+    aiRole?: any;
+  }) => {
+    const { messages, emits, aiRole } = params;
+    const { cancel, write, complete, error } = emits;
+
+    if (mode !== "development") {
+      await checkFetchTarget();
+    }
+
+    const extendParams = transfromExtendParams({ aiRole });
+
+    try {
+      const controller = new AbortController();
+
+      let streamUrl = "//ai.mybricks.world/stream-with-tools";
+      if (fetchTaget === FetchTarget.CustomApp) {
+        streamUrl = "/api/ai-service/stream";
+      } else if (fetchTaget === FetchTarget.Platform) {
+        streamUrl = "/api/assistant/stream";
+      }
+
+      fetch(
+        mode === "development" ? "//ai.mybricks.world/stream-test" : streamUrl,
+        {
+          signal: controller.signal,
+          method: "POST",
+          credentials: "include", // 跨域时携带 cookie
+          headers: {
+            "Content-Type": "application/json",
+            ...(extendParams.role
+              ? {
+                  "M-Request-Role": extendParams.role,
+                }
+              : {}),
+          },
+          body: JSON.stringify(
+            mode === "development"
+              ? {
+                  messages,
+                  ...extendParams,
+                }
+              : getAiEncryptData({
+                  mode,
+                  data: {
+                    messages,
+                    ...extendParams,
+                  },
+                }),
+          ),
+        },
+      ).then(async (response) => {
+        cancel(() => {
+          //注册回调
+          controller.abort(); //取消请求
+        });
+
+        const reader = response.body!.getReader();
+        const decoder = new TextDecoder();
+
+        while (true) {
+          const { done, value } = await reader.read();
+
+          if (done) {
+            break;
+          }
+
+          const chunk = decoder.decode(value, { stream: true });
+          write(chunk);
+        }
+
+        complete("");
+      });
+    } catch (ex) {
+      error(ex as any);
+    }
+  };
+
+export { createRequestAsStream };

package/src/preset/user.ts

@@ -0,0 +1,6 @@
+const user = {
+  name: "user",
+  avatar: "https://my.mybricks.world/default_avatar.png"
+}
+
+export { user }

package/src/startView/components/header/header.less

@@ -0,0 +1,17 @@
+.header {
+  color: var(--mybricks-color-primary);
+  flex-wrap: wrap;
+  justify-content: center;
+  align-items: center;
+  margin-bottom: 8px;
+  font-size: 24px;
+  font-weight: 600;
+  line-height: 32px;
+  display: flex;
+
+  img {
+    width: 40px;
+    height: 40px;
+    object-fit: cover;
+  }
+}

package/src/startView/components/header/header.tsx

@@ -0,0 +1,15 @@
+import React from "react"
+import classNames from "classnames"
+import css from "./header.less"
+
+const Header = () => {
+  return (
+    <div className={classNames(css.header)}>
+      欢迎使用
+      {/* <img src="https://my.mybricks.world/image/icon.png"/>
+      MyBricks.ai */}
+    </div>
+  )
+}
+
+export { Header }

package/src/startView/components/index.ts

@@ -0,0 +1 @@
+export * from "./header/header";

package/src/startView/index.less

@@ -1,216 +1,34 @@
-.editor {
-  z-index: 13;
-  position: relative;
-  background-color: #ffffff;
-  border: solid 1px rgba(0, 0, 0, .13);
+.start-view {
+  width: 80%;
   max-width: 768px;
-  width: 100%;
-  border-radius: 20px;
-  box-shadow: 0 5px 16px -4px #00000012;
-  color: rgba(0, 0, 0, .9);
-  cursor: default;
-
-  .topArea {
-    padding: 12px;
-    display: flex;
-    gap: 8px;
-    align-items: center;
-    box-shadow: 0 .5px 0 0 rgba(0, 0, 0, .13);
-
-    .imageThumbnail {
-      cursor: pointer;
-      pointer-events: auto;
-      flex-shrink: 0;
-      width: 56px;
-      aspect-ratio: 1;
-      min-width: 56px;
-      border-radius: 8px;
-      border: 1px solid rgba(0, 0, 0, .13);
-      position: relative;
-
-      img {
-        width: 100%;
-        height: 100%;
-        display: inline-block;
-        background-repeat: no-repeat;
-        background-size: cover;
-        overflow: hidden;
-      }
-
-      .imageDeleteContainer {
-        cursor: pointer;
-        position: absolute;
-        top: 0;
-        right: 0;
-        padding: 4px;
-        z-index: 2;
-
-        .imageDeleteIcon {
-          width: 12px;
-          height: 12px;
-          border-radius: 50%;
-          background-color: rgba(0, 0, 0, .4);
-          display: flex;
-          align-items: center;
-          justify-content: center;
-          color: #ffffff;
 
-          svg {
-            width: 8px;
-            height: 8px;
-          }
-
-          &:hover {
-            opacity: 0.8;
-            background-color: var(--mybricks-color-primary);
-          }
-        }
-      }
-    }
-  }
-
-  .input {
-    padding: 12px 16px 10px;
-    line-height: 20px;
-    min-height: 60px;
-    position: relative;
-
-    .inputEditorContainer {
-      font-size: 12px;
-      max-height: 140px;
-      scrollbar-width: none;
-      position: relative;
-      min-height: inherit;
-      overflow: auto;
-
-      .inputEditor {
-        user-select: text;
-        white-space: pre-wrap;
-        word-break: break-word;
-        width: 100%;
-        max-height: fit-content;
-        min-height: inherit;
-        overflow: hidden;
-        outline: none;
-        cursor: text;
-
-        &[contentEditable="false"] {
-          cursor: default;
-        }
-      }
-
-      .inputPlaceholder {
-        position: absolute;
-        overflow: hidden;
-        text-overflow: ellipsis;
-        top: 1px;
-        left: 1px;
-        user-select: none;
-        pointer-events: none;
-        color: rgba(0, 0, 0, .6);
-      }
-    }
-  }
-
-  .editorAction {
-    display: flex;
-    justify-content: space-between;
-    align-items: flex-end;
-    padding: 0 8px;
-    margin-top: 2px;
-    margin-bottom: 8px;
-    position: relative;
-
-    .leftArea {
-      display: flex;
-      align-items: center;
-
-      .attachmentButton {
-        overflow: hidden;
-        display: flex;
-        align-items: center;
-        justify-content: center;
-        width: 32px;
-        border-radius: 20px;
-        height: 32px;
-        cursor: pointer;
-
-        svg {
-          width: 20px;
-          height: 20px;
-          color: rgba(0, 0, 0, .6);
-        }
-
-        &:hover {
-          background-color: rgba(0, 0, 0, .09);
-        }
-
-        input {
-          display: none;
-        }
-      }
-    }
-
-    .rightArea {
-      display: flex;
-      align-items: center;
-
-      .sendButtonContainer {
-        cursor: pointer;
-        background-color: var(--mybricks-color-primary);
-        border-radius: 12px;
-        display: flex;
-        align-items: center;
-        color: rgba(255, 255, 255, .9);
-
-        .sendButton {
-          overflow: hidden;
-          display: flex;
-          align-items: center;
-          justify-content: center;
-          width: 32px;
-          height: 32px;
-
-          svg {
-            width: 28px;
-            height: 28px;
-          }
-        }
-
-        .loadingButton {
-          svg {
-            width: 14px;
-            height: 14px;
-          }
-        }
-      }
-
-      .disabled {
-        cursor: default;
-        opacity: 0.5;
-      }
-    }
-  }
-
-  :global {
-    * {
-      scrollbar-color: transparent transparent;
-    }
+  .editor {
+    width: 100%;
   }
 }
 
-.temp {
-  z-index: 13;
-  position: relative;
-  background-color: #ffffff;
-  border: solid 1px rgba(0, 0, 0, .13);
+.view {
+  width: 80%;
   max-width: 768px;
-  width: 100%;
+  // background-color: #ffffff;
+  background-color: transparent;
+  border: solid 1px rgba(0, 0, 0, .13);
   border-radius: 20px;
   box-shadow: 0 5px 16px -4px #00000012;
   color: rgba(0, 0, 0, .9);
   cursor: default;
   height: 80%;
-  padding: 10px;
+  // padding: 10px;
+
   display: flex;
-}
+  flex-direction: column;
+
+  &.empty {
+    display: flex;
+    flex-direction: column;
+    // background-color: transparent;
+    border: none;
+    height: fit-content;
+    box-shadow: none;
+  }
+}