feat(gateway): fix uts.

tddworks · Aug 21, 2024 · 43e8546 · 43e8546
1 parent becf14d
commit 43e8546
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/...ient/anthropic-client-core/src/jvmTest/kotlin/com/tddworks/anthropic/api/AnthropicTest.kt b/...ient/anthropic-client-core/src/jvmTest/kotlin/com/tddworks/anthropic/api/AnthropicTest.kt
@@ -1,10 +1,7 @@
 package com.tddworks.anthropic.api
 
-import com.tddworks.anthropic.di.iniAnthropic
 import org.junit.jupiter.api.Assertions.assertEquals
-import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
-import org.koin.test.junit5.AutoCloseKoinTest
 
 class AnthropicTest {
 
@@ -16,7 +13,7 @@ class AnthropicTest {
 
         assertEquals("CONFIG_API_KEY", anthropic.apiKey())
 
-        assertEquals("CONFIG_API_KEY", anthropic.apiKey())
+        assertEquals("api.anthropic.com", anthropic.baseUrl())
 
         assertEquals("2023-06-01", anthropic.anthropicVersion())
     }

diff --git a/ollama-client/ollama-client-core/src/jvmTest/kotlin/com/tddworks/ollama/api/OllamaTest.kt b/ollama-client/ollama-client-core/src/jvmTest/kotlin/com/tddworks/ollama/api/OllamaTest.kt
@@ -37,7 +37,7 @@ class OllamaTestTest : AutoCloseKoinTest() {
 
     @Test
     fun `should return default settings`() {
-        val target = Ollama()
+        val target = Ollama.create(ollamaConfig = OllamaConfig())
 
         assertEquals("localhost", target.baseUrl())
 

diff --git a/...ateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/anthropic/api/AnthropicTest.kt b/...ateway/openai-gateway-core/src/jvmTest/kotlin/com/tddworks/anthropic/api/AnthropicTest.kt
@@ -17,9 +17,9 @@ class AnthropicTest : AutoCloseKoinTest() {
     @Test
     fun `should create anthropic with default settings`() {
 
-        val anthropic = Anthropic()
+        val anthropic = Anthropic.create(anthropicConfig = AnthropicConfig())
 
-        assertEquals("CONFIGURE_ME", anthropic.apiKey())
+        assertEquals("CONFIG_API_KEY", anthropic.apiKey())
         assertEquals(Anthropic.BASE_URL, anthropic.baseUrl())
         assertEquals("2023-06-01", anthropic.anthropicVersion())
     }