From 5b74f2bcfe6d831b373d536bf79739b750a5db02 Mon Sep 17 00:00:00 2001
From: VIVEK PATEL <patelvk2000@gmail.com>
Date: Mon, 21 Oct 2024 01:58:00 -0400
Subject: [PATCH] Added IBM Granite models

---
 packages/ollama/src/ollama-chat-settings.ts | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/packages/ollama/src/ollama-chat-settings.ts b/packages/ollama/src/ollama-chat-settings.ts
index df293ad..b6ea8e1 100644
--- a/packages/ollama/src/ollama-chat-settings.ts
+++ b/packages/ollama/src/ollama-chat-settings.ts
@@ -27,6 +27,12 @@ export type OllamaChatModelId =
   | 'gemma2:2b'
   | 'gemma2:9b'
   | 'gemma2:27b'
+  | 'granite3-dense'
+  | 'granite3-dense:2b'
+  | 'granite3-dense:8b'
+  | 'granite3-moe'
+  | 'granite3-moe:1b'
+  | 'granite3-moe:3b'
   | 'llama2'
   | 'llama2:7b'
   | 'llama2:13b'