-
Notifications
You must be signed in to change notification settings - Fork 485
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
090e1e6
commit b9029ca
Showing
49 changed files
with
1,285 additions
and
362 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,102 @@ | ||
| import SwiftUI | ||
| import ChatService | ||
| import Persist | ||
| import ComposableArchitecture | ||
|
|
||
| public let SELECTED_LLM_KEY = "selectedLLM" | ||
|
|
||
| extension AppState { | ||
| func getSelectedModelFamily() -> String? { | ||
| if let savedModel = get(key: SELECTED_LLM_KEY), | ||
| let modelFamily = savedModel["modelFamily"]?.stringValue { | ||
| return modelFamily | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func getSelectedModelName() -> String? { | ||
| if let savedModel = get(key: SELECTED_LLM_KEY), | ||
| let modelName = savedModel["modelName"]?.stringValue { | ||
| return modelName | ||
| } | ||
| return nil | ||
| } | ||
|
|
||
| func setSelectedModel(_ model: LLMModel) { | ||
| update(key: SELECTED_LLM_KEY, value: model) | ||
| } | ||
| } | ||
|
|
||
| struct LLMModel: Codable, Hashable { | ||
| let modelName: String | ||
| let modelFamily: String | ||
| } | ||
|
|
||
| let defaultModel = LLMModel(modelName: "GPT-4o", modelFamily: "gpt-4o") | ||
| struct ModelPicker: View { | ||
| @State private var selectedModel = defaultModel.modelName | ||
| @State private var models: [LLMModel] = [ defaultModel ] | ||
| @State private var isHovered = false | ||
| @State private var isPressed = false | ||
|
|
||
| init() { | ||
| self.updateCurrentModel() | ||
| } | ||
|
|
||
| func updateCurrentModel() { | ||
| selectedModel = AppState.shared.getSelectedModelName() ?? defaultModel.modelName | ||
| } | ||
|
|
||
| var body: some View { | ||
| WithPerceptionTracking { | ||
| Menu(selectedModel) { | ||
| ForEach(models, id: \.self) { option in | ||
| Button { | ||
| selectedModel = option.modelName | ||
| AppState.shared.setSelectedModel(option) | ||
| } label: { | ||
| if selectedModel == option.modelName { | ||
| Text("✓ \(option.modelName)") | ||
| } else { | ||
| Text(" \(option.modelName)") | ||
| } | ||
| } | ||
| } | ||
| } | ||
| .menuStyle(BorderlessButtonMenuStyle()) | ||
| .frame(maxWidth: labelWidth()) | ||
| .padding(4) | ||
| .background( | ||
| RoundedRectangle(cornerRadius: 5) | ||
| .fill(isHovered ? Color.gray.opacity(0.1) : Color.clear) | ||
| ) | ||
| .onHover { hovering in | ||
| isHovered = hovering | ||
| } | ||
| .onAppear() { | ||
| Task { | ||
| updateCurrentModel() | ||
| self.models = await ChatService.shared.copilotModels().filter( | ||
| { $0.scopes.contains(.chatPanel) } | ||
| ).map { | ||
| LLMModel(modelName: $0.modelName, modelFamily: $0.modelFamily) | ||
| } | ||
| } | ||
| } | ||
| .help("Pick Model") | ||
| } | ||
| } | ||
|
|
||
| func labelWidth() -> CGFloat { | ||
| let font = NSFont.systemFont(ofSize: NSFont.systemFontSize) | ||
| let attributes = [NSAttributedString.Key.font: font] | ||
| let width = selectedModel.size(withAttributes: attributes).width | ||
| return CGFloat(width + 20) | ||
| } | ||
| } | ||
|
|
||
| struct ModelPicker_Previews: PreviewProvider { | ||
| static var previews: some View { | ||
| ModelPicker() | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.