修复m3e向量模型无法正常处理字符串的问题 (#693)

331d18c8 · 不做了睡大觉 · GitHub · 82882901 · 331d18c8
Commit 331d18c8 authored Jan 05, 2024 by 不做了睡大觉 Committed by GitHub Jan 05, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletions

packages/service/core/ai/embedding/index.ts
+3 -1

No files found.
--- a/packages/service/core/ai/embedding/index.ts
+++ b/packages/service/core/ai/embedding/index.ts
@@ -25,7 +25,9 @@ export async function getVectorsByText({
      }
    }
  }
+  if (typeof input === 'string') {
+    input = [input];
+  }
  try {
    // 获取 chatAPI
    const ai = getAIApi();