fixup!

Signed-off-by: Tomas Pilar <[email protected]>
i-am-bee · Jan 16, 2025 · 141f397 · 141f397
1 parent ecc294c
commit 141f397
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/embeddings/embeddings.service.ts b/src/embeddings/embeddings.service.ts
@@ -31,7 +31,10 @@ export async function embeddings({
   input
 }: EmbeddingsCreateBody): Promise<EmbeddingsCreateResponse> {
   const llm = defaultAIProvider.createEmbeddingBackend({ model });
-  const embeddings = new Embeddings({ model, inputs: typeof input === 'string' ? [input] : input });
+  const embeddings = new Embeddings({
+    model: llm.modelId,
+    inputs: typeof input === 'string' ? [input] : input
+  });
   await ORM.em.persistAndFlush(embeddings);
   try {
     const output = await llm.embed(embeddings.inputs);