chatdesk-ui/chatdesk-ui/lib/generate-bgem3-embedding.ts

20 lines
583 B
TypeScript

export async function generateBgeM3Embedding(text: string): Promise<number[] | null> {
try {
const response = await fetch("http://localhost:8000/embedding", {
method: "POST",
headers: { "Content-Type": "application/json" },
body: JSON.stringify({ input: text })
})
if (!response.ok) {
throw new Error(`Failed to fetch BGE-M3 embedding: ${response.status}`)
}
const result = await response.json()
return result.embedding as number[]
} catch (err) {
console.error("Error in generateBgeM3Embedding:", err)
return null
}
}