Add custom header filed for Ollama APIs

Author: intitni

Core/Sources/HostApp/AccountSettings/ChatModelManagement/ChatModelEditView.swift

@@ -243,7 +243,7 @@ struct ChatModelEditView: View {
 
                 MaxTokensTextField(store: store)
                 SupportsFunctionCallingToggle(store: store)
-                
+
                 TextField(text: $store.openAIOrganizationID, prompt: Text("Optional")) {
                     Text("Organization ID")
                 }
@@ -321,11 +321,11 @@ struct ChatModelEditView: View {
                 Toggle(isOn: $store.enforceMessageOrder) {
                     Text("Enforce message order to be user/assistant alternated")
                 }
-                
+
                 Toggle(isOn: $store.openAICompatibleSupportsMultipartMessageContent) {
                     Text("Support multi-part message content")
                 }
-                
+
                 Button("Custom Headers") {
                     isEditingCustomHeader.toggle()
                 }
@@ -375,13 +375,15 @@ struct ChatModelEditView: View {
 
     struct OllamaForm: View {
         @Perception.Bindable var store: StoreOf<ChatModelEdit>
+        @State var isEditingCustomHeader = false
+
         var body: some View {
             WithPerceptionTracking {
                 BaseURLTextField(store: store, prompt: Text("http://127.0.0.1:11434")) {
                     Text("/api/chat")
                 }
-				
-				ApiKeyNamePicker(store: store)
+
+                ApiKeyNamePicker(store: store)
 
                 TextField("Model Name", text: $store.modelName)
 
@@ -397,6 +399,12 @@ struct ChatModelEditView: View {
                     )
                 }
                 .padding(.vertical)
+
+                Button("Custom Headers") {
+                    isEditingCustomHeader.toggle()
+                }
+            }.sheet(isPresented: $isEditingCustomHeader) {
+                CustomHeaderSettingsView(headers: $store.customHeaders)
             }
         }
     }

Core/Sources/HostApp/AccountSettings/EmbeddingModelManagement/EmbeddingModelEditView.swift

@@ -339,6 +339,8 @@ struct EmbeddingModelEditView: View {
 
     struct OllamaForm: View {
         @Perception.Bindable var store: StoreOf<EmbeddingModelEdit>
+        @State var isEditingCustomHeader = false
+        
         var body: some View {
             WithPerceptionTracking {
                 BaseURLTextField(store: store, prompt: Text("http://127.0.0.1:11434")) {
@@ -364,6 +366,12 @@ struct EmbeddingModelEditView: View {
                     )
                 }
                 .padding(.vertical)
+                
+                Button("Custom Headers") {
+                    isEditingCustomHeader.toggle()
+                }
+            }.sheet(isPresented: $isEditingCustomHeader) {
+                CustomHeaderSettingsView(headers: $store.customHeaders)
             }
         }
     }

Tool/Sources/OpenAIService/APIs/OlamaChatCompletionsService.swift

@@ -59,11 +59,15 @@ extension OllamaChatCompletionsService: ChatCompletionsAPI {
         let encoder = JSONEncoder()
         request.httpBody = try encoder.encode(requestBody)
         request.setValue("application/json", forHTTPHeaderField: "Content-Type")
-		
-		if !apiKey.isEmpty {
-			request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
-		}
-		
+
+        if !apiKey.isEmpty {
+            request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
+        }
+
+        for field in model.info.customHeaderInfo.headers where !field.key.isEmpty {
+            request.setValue(field.value, forHTTPHeaderField: field.key)
+        }
+
         let (result, response) = try await URLSession.shared.data(for: request)
 
         guard let response = response as? HTTPURLResponse else {
@@ -140,11 +144,15 @@ extension OllamaChatCompletionsService: ChatCompletionsStreamAPI {
         let encoder = JSONEncoder()
         request.httpBody = try encoder.encode(requestBody)
         request.setValue("application/json", forHTTPHeaderField: "Content-Type")
-		
-		if !apiKey.isEmpty {
-			request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
-		}
-		
+
+        if !apiKey.isEmpty {
+            request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
+        }
+
+        for field in model.info.customHeaderInfo.headers where !field.key.isEmpty {
+            request.setValue(field.value, forHTTPHeaderField: field.key)
+        }
+
         let (result, response) = try await URLSession.shared.bytes(for: request)
 
         guard let response = response as? HTTPURLResponse else {

Tool/Sources/OpenAIService/APIs/OllamaEmbeddingService.swift

@@ -30,6 +30,10 @@ struct OllamaEmbeddingService: EmbeddingAPI {
         if !apiKey.isEmpty {
             request.setValue("Bearer \(apiKey)", forHTTPHeaderField: "Authorization")
         }
+        
+        for field in model.info.customHeaderInfo.headers where !field.key.isEmpty {
+            request.setValue(field.value, forHTTPHeaderField: field.key)
+        }
 
         let (result, response) = try await URLSession.shared.data(for: request)
         guard let response = response as? HTTPURLResponse else {