Commit 6c70f060 by archer

perf: vector unit

parent 6a39b514
...@@ -67,13 +67,13 @@ export async function getVector({ ...@@ -67,13 +67,13 @@ export async function getVector({
} }
) )
.then((res) => { .then((res) => {
if (!res.data?.usage?.total_tokens) { if (!res.data?.data?.[0]?.embedding) {
// @ts-ignore // @ts-ignore
return Promise.reject(res.data?.error?.message || 'Embedding Error'); return Promise.reject(res.data?.error?.message || 'Embedding Error');
} }
return { return {
tokenLen: res.data.usage.total_tokens || 0, tokenLen: res.data.usage.total_tokens || 0,
vectors: res.data.data.map((item) => item.embedding) vectors: res.data.data.map((item) => unityDimensional(item.embedding))
}; };
}); });
...@@ -86,3 +86,12 @@ export async function getVector({ ...@@ -86,3 +86,12 @@ export async function getVector({
return result; return result;
} }
function unityDimensional(vector: number[]) {
let resultVector = vector;
const vectorLen = vector.length;
const zeroVector = new Array(1536 - vectorLen).fill(0);
return resultVector.concat(zeroVector);
}
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