Remove redundant extensions

HealthGPT/HealthGPT/HealthGPTView.swift

@@ -28,10 +28,10 @@
     var body: some View {
         NavigationStack {
             if let llm = healthDataInterpreter.llm {
-                let contextBinding = Binding { llm.context.chatEntity } set: { llm.context = $0.llmContextEntity }
+                let contextBinding = Binding { llm.context.chat } set: { llm.context.chat = $0 }
 
                 ChatView(contextBinding, exportFormat: .text)
-                    .speak(llm.context.chatEntity, muted: !textToSpeech)
+                    .speak(llm.context.chat, muted: !textToSpeech)
                     .speechToolbarButton(muted: !$textToSpeech)
                     .viewStateAlert(state: llm.state)
                     .navigationTitle("WELCOME_TITLE")
@@ -71,7 +71,7 @@
             if FeatureFlags.mockMode {
                 await healthDataInterpreter.prepareLLM(with: LLMMockSchema())
             } else if FeatureFlags.localLLM || llmSource == .local {
                 await healthDataInterpreter.prepareLLM(with: LLMLocalSchema(modelPath: .cachesDirectory.appending(path: "llm.gguf")))
             } else {
                 await healthDataInterpreter.prepareLLM(with: LLMOpenAISchema(parameters: .init(modelType: openAIModel)))
             }
@@ -113,53 +113,3 @@
         }
     }
 }
-
-extension Array where Element == LLMContextEntity {
-    var chatEntity: [ChatEntity] {
-        self.map { llmContextEntity in
-            let role: ChatEntity.Role
-
-            switch llmContextEntity.role {
-            case .user:
-                role = .user
-            case .assistant:
-                role = .assistant
-            case .system, .tool:
-                role = .hidden(type: .unknown)
-            }
-
-            return ChatEntity(
-                role: role,
-                content: llmContextEntity.content,
-                complete: llmContextEntity.complete,
-                id: llmContextEntity.id,
-                date: llmContextEntity.date
-            )
-        }
-    }
-}
-
-extension Array where Element == ChatEntity {
-    var llmContextEntity: [LLMContextEntity] {
-        self.map { chatEntity in
-            let role: LLMContextEntity.Role
-
-            switch chatEntity.role {
-            case .user:
-                role = .user
-            case .assistant:
-                role = .assistant()
-            case .hidden:
-                role = .system
-            }
-
-            return LLMContextEntity(
-                role: role,
-                content: chatEntity.content,
-                complete: chatEntity.complete,
-                id: chatEntity.id,
-                date: chatEntity.date
-            )
-        }
-    }
-}

HealthGPT/Onboarding/LLMSource.swift

@@ -7,6 +7,7 @@
 //
 
 import Foundation
+import SpeziLLMOpenAI
 
 
 enum LLMSource: String, CaseIterable, Identifiable, Codable {