diff --git a/openai-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/OllamaOpenAIProviderConfig.kt b/openai-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/OllamaOpenAIProviderConfig.kt
index 6762bff..91647e0 100644
--- a/openai-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/OllamaOpenAIProviderConfig.kt
+++ b/openai-gateway/openai-gateway-core/src/commonMain/kotlin/com/tddworks/openai/gateway/api/internal/OllamaOpenAIProviderConfig.kt
@@ -4,14 +4,18 @@ import com.tddworks.ollama.api.OllamaConfig
 import com.tddworks.openai.gateway.api.OpenAIProviderConfig
 
 data class OllamaOpenAIProviderConfig(
+    val port: () -> Int = { 11434 },
+    val protocol: () -> String = { "http" },
     override val baseUrl: () -> String = { "http//:localhost:11434" },
     override val apiKey: () -> String = { "ollama-ignore-this" }
 ) : OpenAIProviderConfig
 
 fun OllamaOpenAIProviderConfig.toOllamaConfig() =
-    OllamaConfig(baseUrl = baseUrl)
+    OllamaConfig(baseUrl = baseUrl, protocol = protocol, port = port)
 
 fun OpenAIProviderConfig.Companion.ollama(
     apiKey: () -> String = { "ollama-ignore-this" },
-    baseUrl: () -> String = { "http//:localhost:11434" },
-) = OllamaOpenAIProviderConfig(baseUrl, apiKey)
\ No newline at end of file
+    baseUrl: () -> String = { "localhost" },
+    protocol: () -> String = { "http" },
+    port: () -> Int = { 11434 }
+) = OllamaOpenAIProviderConfig(port, protocol, baseUrl, apiKey)
\ No newline at end of file