refactor - removeProvider update to use id

tddworks · Aug 24, 2024 · 906e8f3 · 906e8f3
1 parent 9386056
commit 906e8f3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/...penai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/OpenAIGateway.kt b/...penai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/OpenAIGateway.kt
@@ -16,6 +16,6 @@ interface OpenAIGateway : Chat, Completions {
     )
 
     fun addProvider(provider: OpenAIProvider): OpenAIGateway
-    fun removeProvider(name: String)
+    fun removeProvider(id: String)
     fun getProviders(): List<OpenAIProvider>
 }
diff --git a/...re/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/DefaultOpenAIGateway.kt b/...re/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/DefaultOpenAIGateway.kt
@@ -77,8 +77,8 @@ class DefaultOpenAIGateway(
         return this
     }
 
-    override fun removeProvider(name: String) {
-        availableProviders.removeAll { it.name == name }
+    override fun removeProvider(id: String) {
+        availableProviders.removeAll { it.id == id }
     }
 
     override fun getProviders(): List<OpenAIProvider> {

diff --git a/...e/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/internal/DefaultOpenAIGatewayTest.kt b/...e/src/jvmTest/kotlin/com/tddworks/openai/gateway/api/internal/DefaultOpenAIGatewayTest.kt
@@ -150,7 +150,7 @@ class DefaultOpenAIGatewayTest {
 
     @Test
     fun `should able to remove provider`() {
-        openAIGateway.removeProvider(anthropic.name)
+        openAIGateway.removeProvider(anthropic.id)
         // Then
         assertEquals(3, openAIGateway.getProviders().size)
         assertEquals(azure, openAIGateway.getProviders().last())