forked from intitni/CopilotForXcode
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathChatGPTConfiguration.swift
More file actions
39 lines (33 loc) · 1.03 KB
/
ChatGPTConfiguration.swift
File metadata and controls
39 lines (33 loc) · 1.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import Foundation
import AIModel
import Preferences
import Keychain
public protocol ChatGPTConfiguration {
var model: ChatModel { get }
var temperature: Double { get }
var apiKey: String { get }
var stop: [String] { get }
var maxTokens: Int { get }
var minimumReplyTokens: Int { get }
var runFunctionsAutomatically: Bool { get }
}
public extension ChatGPTConfiguration {
var endpoint: String {
model.endpoint
}
var apiKey: String {
(try? Keychain.apiKey.get(model.info.apiKeyName)) ?? ""
}
func overriding(
_ overrides: OverridingChatGPTConfiguration.Overriding
) -> OverridingChatGPTConfiguration {
.init(overriding: self, with: overrides)
}
func overriding(
_ update: (inout OverridingChatGPTConfiguration.Overriding) -> Void = { _ in }
) -> OverridingChatGPTConfiguration {
var overrides = OverridingChatGPTConfiguration.Overriding()
update(&overrides)
return .init(overriding: self, with: overrides)
}
}