feat(BE-190): As a user, i want able to use ollama-client

- support llama3
tddworks · Apr 19, 2024 · 5108aa6 · 5108aa6
1 parent d625cb9
commit 5108aa6
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/gradle.properties b/gradle.properties
@@ -12,7 +12,7 @@ android.useAndroidX=true
 android.nonTransitiveRClass=true
 
 #publishing for kmmbridge
-LIBRARY_VERSION=0.1.2
+LIBRARY_VERSION=0.1.3
 GROUP=com.tddworks
 
 # POM

diff --git a/...ma-client/ollama-client-core/src/commonMain/kotlin/com/tddworks/ollama/api/OllamaModel.kt b/...ma-client/ollama-client-core/src/commonMain/kotlin/com/tddworks/ollama/api/OllamaModel.kt
@@ -8,6 +8,7 @@ import kotlin.jvm.JvmInline
 value class OllamaModel(val value: String) {
     companion object {
         val LLAMA2 = OllamaModel("llama2")
+        val LLAMA3 = OllamaModel("llama3")
         val CODE_LLAMA = OllamaModel("codellama")
         val MISTRAL = OllamaModel("mistral")
         val availableModels = listOf(LLAMA2, CODE_LLAMA, MISTRAL)

diff --git a/...a-client/ollama-client-core/src/jvmTest/kotlin/com/tddworks/ollama/api/OllamaModelTest.kt b/...a-client/ollama-client-core/src/jvmTest/kotlin/com/tddworks/ollama/api/OllamaModelTest.kt
@@ -7,6 +7,7 @@ class OllamaModelTest {
 
     @Test
     fun `should return correct latest API model name`() {
+        assertEquals("llama3", OllamaModel.LLAMA3.value)
         assertEquals("llama2", OllamaModel.LLAMA2.value)
         assertEquals("codellama", OllamaModel.CODE_LLAMA.value)
         assertEquals("mistral", OllamaModel.MISTRAL.value)