From 88cd7bee81c44cfa37705252147ceaba5d5fa4c5 Mon Sep 17 00:00:00 2001
From: francis2tm <franciscomelot@hotmail.com>
Date: Tue, 14 Jan 2025 20:03:06 +0000
Subject: [PATCH] feat: add new model classes for embedding and image
 processing

---
 packages/core/src/models.ts | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/packages/core/src/models.ts b/packages/core/src/models.ts
index 768ec2faab..8ce7763557 100644
--- a/packages/core/src/models.ts
+++ b/packages/core/src/models.ts
@@ -535,6 +535,8 @@ export const models: Models = {
             [ModelClass.SMALL]: "meta-llama/Llama-3.3-70B-Instruct",
             [ModelClass.MEDIUM]: "meta-llama/Llama-3.3-70B-Instruct",
             [ModelClass.LARGE]: "meta-llama/Llama-3.3-70B-Instruct",
+            [ModelClass.EMBEDDING]: "intfloat/multilingual-e5-large-instruct",
+            [ModelClass.IMAGE]: "black-forest-labs/FLUX.1-schnell",
         },
     },
 };