From 8eea19a53987d5a99ac3f6d116bf58ec428a7498 Mon Sep 17 00:00:00 2001
From: Amith Koujalgi <amith.koujalgi@razorthink.com>
Date: Thu, 11 Jul 2024 23:35:39 +0530
Subject: [PATCH] Added model types - `gemma2` and `qwen2`

---
 .../amithkoujalgi/ollama4j/core/types/OllamaModelType.java   | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/main/java/io/github/amithkoujalgi/ollama4j/core/types/OllamaModelType.java b/src/main/java/io/github/amithkoujalgi/ollama4j/core/types/OllamaModelType.java
index 5733979..2613d86 100644
--- a/src/main/java/io/github/amithkoujalgi/ollama4j/core/types/OllamaModelType.java
+++ b/src/main/java/io/github/amithkoujalgi/ollama4j/core/types/OllamaModelType.java
@@ -9,6 +9,9 @@ package io.github.amithkoujalgi.ollama4j.core.types;
 @SuppressWarnings("ALL")
 public class OllamaModelType {
     public static final String GEMMA = "gemma";
+    public static final String GEMMA2 = "gemma2";
+
+
     public static final String LLAMA2 = "llama2";
     public static final String LLAMA3 = "llama3";
     public static final String MISTRAL = "mistral";
@@ -30,6 +33,8 @@ public class OllamaModelType {
     public static final String ZEPHYR = "zephyr";
     public static final String OPENHERMES = "openhermes";
     public static final String QWEN = "qwen";
+
+    public static final String QWEN2 = "qwen2";
     public static final String WIZARDCODER = "wizardcoder";
     public static final String LLAMA2_CHINESE = "llama2-chinese";
     public static final String TINYLLAMA = "tinyllama";