code refactor: able to create provider from OpenAIProvider

tddworks · Aug 20, 2024 · 862a86c · 862a86c
1 parent f2c0a79
commit 862a86c
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 8 deletions.
diff --git a/...teway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/DefaultOpenAIProvider.kt b/...teway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/DefaultOpenAIProvider.kt
@@ -36,3 +36,11 @@ class DefaultOpenAIProvider(
         return openAI.completions(request)
     }
 }
+
+fun OpenAIProvider.Companion.openAI(
+    config: OpenAIConfig,
+    models: List<Model>,
+    openAI: OpenAI = OpenAI.create(config)
+): OpenAIProvider {
+    return DefaultOpenAIProvider(config, models, openAI)
+}
diff --git a/...enai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/OpenAIProvider.kt b/...enai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/OpenAIProvider.kt
@@ -15,4 +15,6 @@ interface OpenAIProvider : Chat, Completions {
      * @return true if the model is supported, false otherwise
      */
     fun supports(model: Model): Boolean
+
+    companion object
 }
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
@@ -1,6 +1,5 @@
 package com.tddworks.openai.gateway.api.internal
 
-import com.tddworks.openai.api.OpenAI
 import com.tddworks.openai.api.chat.api.ChatCompletion
 import com.tddworks.openai.api.chat.api.ChatCompletionChunk
 import com.tddworks.openai.api.chat.api.ChatCompletionRequest
@@ -19,7 +18,6 @@ import kotlinx.serialization.ExperimentalSerializationApi
 @ExperimentalSerializationApi
 class DefaultOpenAIGateway(
     providers: List<OpenAIProvider>,
-    private val openAI: OpenAI,
 ) : OpenAIGateway {
     private val availableProviders: MutableList<OpenAIProvider> =
         providers.toMutableList()
@@ -40,9 +38,9 @@ class DefaultOpenAIGateway(
      * @return A ChatCompletion object containing the completions for the provided request.
      */
     override suspend fun chatCompletions(request: ChatCompletionRequest): ChatCompletion {
-        return availableProviders.firstOrNull {
-            it.supports(request.model)
-        }?.chatCompletions(request) ?: openAI.chatCompletions(request)
+        return availableProviders.firstOrNull { it.supports(request.model) }
+            ?.chatCompletions(request)
+            ?: throwNoProviderFound(request.model.value)
     }
 
     /**
@@ -55,7 +53,8 @@ class DefaultOpenAIGateway(
     override fun streamChatCompletions(request: ChatCompletionRequest): Flow<ChatCompletionChunk> {
         return availableProviders.firstOrNull {
             it.supports(request.model)
-        }?.streamChatCompletions(request) ?: openAI.streamChatCompletions(request)
+        }?.streamChatCompletions(request)
+            ?: throwNoProviderFound(request.model.value)
     }
 
     /**
@@ -67,6 +66,12 @@ class DefaultOpenAIGateway(
     override suspend fun completions(request: CompletionRequest): Completion {
         return availableProviders.firstOrNull {
             it.supports(request.model)
-        }?.completions(request) ?: openAI.completions(request)
+        }?.completions(request)
+            ?: throwNoProviderFound(request.model.value)
+    }
+
+
+    private fun throwNoProviderFound(model: String): Nothing {
+        throw UnsupportedOperationException("No provider found for model $model")
     }
 }
diff --git a/...-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/di/Koin.kt b/...-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/di/Koin.kt
@@ -10,6 +10,7 @@ import com.tddworks.openai.di.openAIModules
 import com.tddworks.openai.gateway.api.AnthropicOpenAIProvider
 import com.tddworks.openai.gateway.api.OllamaOpenAIProvider
 import com.tddworks.openai.gateway.api.OpenAIGateway
+import com.tddworks.openai.gateway.api.OpenAIProvider
 import com.tddworks.openai.gateway.api.internal.DefaultOpenAIGateway
 import kotlinx.serialization.ExperimentalSerializationApi
 import org.koin.core.context.startKoin
@@ -33,6 +34,22 @@ fun initOpenAIGateway(
     )
 }.koin.get<OpenAIGateway>()
 
+
+@ExperimentalSerializationApi
+fun createOpenAIGateway(providers: List<OpenAIProvider>) = startKoin {
+    modules(
+        commonModule(false) +
+                openAIGatewayModules(providers)
+    )
+}.koin.get<OpenAIGateway>()
+
+
+@OptIn(ExperimentalSerializationApi::class)
+fun openAIGatewayModules(providers: List<OpenAIProvider>) = module {
+    single { providers }
+    single<OpenAIGateway> { DefaultOpenAIGateway(get()) }
+}
+
 @ExperimentalSerializationApi
 fun openAIGatewayModules() = module {
     single<AnthropicOpenAIProvider> { AnthropicOpenAIProvider(get()) }
@@ -45,5 +62,5 @@ fun openAIGatewayModules() = module {
         )
     }
 
-    single<OpenAIGateway> { DefaultOpenAIGateway(get(), get()) }
+    single<OpenAIGateway> { DefaultOpenAIGateway(get()) }
 }