Commit c5c38267 by archer

perf: vector over range error

parent 42fec3a9
......@@ -66,14 +66,14 @@ export async function getVector({
...axiosConfig()
}
)
.then((res) => {
.then(async (res) => {
if (!res.data?.data?.[0]?.embedding) {
// @ts-ignore
return Promise.reject(res.data?.error?.message || 'Embedding Error');
}
return {
tokenLen: res.data.usage.total_tokens || 0,
vectors: res.data.data.map((item) => unityDimensional(item.embedding))
vectors: await Promise.all(res.data.data.map((item) => unityDimensional(item.embedding)))
};
});
......@@ -88,6 +88,7 @@ export async function getVector({
}
function unityDimensional(vector: number[]) {
if (vector.length > 1536) return Promise.reject('向量维度不能超过 1536');
let resultVector = vector;
const vectorLen = vector.length;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment