From 9219593ee11818adc21bdf8b14413eb2a66507fc Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Thu, 16 Oct 2025 17:56:57 +0530
Subject: [PATCH] feat(model-registry): add loading method

---
 src/lib/models/registry.ts | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/src/lib/models/registry.ts b/src/lib/models/registry.ts
index a434dcc..a2a7aa8 100644
--- a/src/lib/models/registry.ts
+++ b/src/lib/models/registry.ts
@@ -53,6 +53,26 @@ class ModelRegistry {
 
     return providers;
   }
+
+  async loadChatModel(providerId: string, modelName: string) {
+    const provider = this.activeProviders.find((p) => p.id === providerId);
+
+    if (!provider) throw new Error('Invalid provider id');
+
+    const model = await provider.provider.loadChatModel(modelName);
+
+    return model;
+  }
+
+  async loadEmbeddingModel(providerId: string, modelName: string) {
+    const provider = this.activeProviders.find((p) => p.id === providerId);
+
+    if (!provider) throw new Error('Invalid provider id');
+
+    const model = await provider.provider.loadEmbeddingModel(modelName);
+
+    return model;
+  }
 }
 
 export default ModelRegistry;