修复向量模型无法正常处理字符串的问题

stakeswky · Jan 5, 2024 · b49972e · b49972e
1 parent a47b89d
commit b49972e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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();