CopilotForXcode/Core/Sources/ServiceUpdateMigration/MigrateTo240.swift at main · Piitty/CopilotForXcode · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
import AIModel
import Foundation
import Keychain
import Preferences

func migrateTo240(
    defaults: UserDefaults = .shared,
    keychain: KeychainType = Keychain.apiKey
) throws {
    let finishedMigrationKey = "MigrateTo240Finished"
    if defaults.bool(forKey: finishedMigrationKey) { return }

    do {
        let chatModelOpenAIId = UUID().uuidString
        let chatModelAzureOpenAIId = UUID().uuidString
        let embeddingModelOpenAIId = UUID().uuidString
        let embeddingModelAzureOpenAIId = UUID().uuidString

        let openAIAPIKeyName = "OpenAI"
        let openAIAPIKey = defaults.deprecatedValue(for: \.openAIAPIKey)
        if !openAIAPIKey.isEmpty {
            try keychain.update(openAIAPIKey, key: openAIAPIKeyName)
        }

        let azureOpenAIAPIKeyName = "Azure OpenAI"
        let azureOpenAIAPIKey = defaults.deprecatedValue(for: \.azureOpenAIAPIKey)
        if !azureOpenAIAPIKey.isEmpty {
            try keychain.update(azureOpenAIAPIKey, key: azureOpenAIAPIKeyName)
        }

        defaults.set({
            let openAIModel = ChatGPTModel(rawValue: defaults.deprecatedValue(for: \.chatGPTModel))

            let openAI = ChatModel(
                id: chatModelOpenAIId,
                name: "OpenAI",
                format: .openAI,
                info: .init(
                    apiKeyName: openAIAPIKeyName,
                    baseURL: defaults.deprecatedValue(for: \.openAIBaseURL),
                    maxTokens: openAIModel?.maxToken ?? defaults
                        .deprecatedValue(for: \.chatGPTMaxToken),
                    modelName: openAIModel?.rawValue ?? defaults
                        .deprecatedValue(for: \.chatGPTModel)
                )
            )
            let azureOpenAI = ChatModel(
                id: chatModelAzureOpenAIId,
                name: "Azure OpenAI",
                format: .azureOpenAI,
                info: .init(
                    apiKeyName: azureOpenAIAPIKeyName,
                    baseURL: defaults.deprecatedValue(for: \.azureOpenAIBaseURL),
                    maxTokens: defaults.deprecatedValue(for: \.chatGPTMaxToken),
                    modelName: defaults
                        .deprecatedValue(for: \.azureChatGPTDeployment)
                )
            )

            return [openAI, azureOpenAI]
        }(), for: \.chatModels)

        defaults.set({
            if defaults.deprecatedValue(for: \.chatFeatureProvider) == .azureOpenAI {
                return chatModelAzureOpenAIId
            }
            return chatModelOpenAIId
        }(), for: \.defaultChatFeatureChatModelId)

        defaults.set({
            let openAIModel = OpenAIEmbeddingModel(
                rawValue: defaults.deprecatedValue(for: \.embeddingModel)
            )

            let openAI = EmbeddingModel(
                id: embeddingModelOpenAIId,
                name: "OpenAI",
                format: .openAI,
                info: .init(
                    apiKeyName: openAIAPIKeyName,
                    baseURL: defaults.deprecatedValue(for: \.openAIBaseURL),
                    maxTokens: openAIModel?.maxToken ?? 8191,
                    modelName: openAIModel?.rawValue ?? defaults.deprecatedValue(for: \.embeddingModel)
                )
            )

            let azureOpenAI = EmbeddingModel(
                id: embeddingModelAzureOpenAIId,
                name: "Azure OpenAI",
                format: .azureOpenAI,
                info: .init(
                    apiKeyName: azureOpenAIAPIKeyName,
                    baseURL: defaults.deprecatedValue(for: \.azureOpenAIBaseURL),
                    maxTokens: 8191,
                    modelName: defaults
                        .deprecatedValue(for: \.azureEmbeddingDeployment)
                )
            )

            return [openAI, azureOpenAI]
        }(), for: \.embeddingModels)

        defaults.set({
            if defaults.deprecatedValue(for: \.embeddingFeatureProvider) == .azureOpenAI {
                return embeddingModelAzureOpenAIId
            }
            return embeddingModelOpenAIId
        }(), for: \.defaultChatFeatureEmbeddingModelId)

        defaults.set(true, forKey: finishedMigrationKey)
    } catch {
        print(error.localizedDescription)
        throw error
    }
}