ChatModel.swift

import CodableWrappers
import Foundation

public struct ChatModel: Codable, Equatable, Identifiable {
    public var id: String
    public var name: String
    @FallbackDecoding<EmptyChatModelFormat>
    public var format: Format
    @FallbackDecoding<EmptyChatModelInfo>
    public var info: Info

    public init(id: String, name: String, format: Format, info: Info) {
        self.id = id
        self.name = name
        self.format = format
        self.info = info
    }

    public enum Format: String, Codable, Equatable, CaseIterable {
        case openAI
        case azureOpenAI
        case openAICompatible
        case googleAI
        case ollama
        case claude
    }

    public struct Info: Codable, Equatable {
        public struct OllamaInfo: Codable, Equatable {
            @FallbackDecoding<EmptyString>
            public var keepAlive: String

            public init(keepAlive: String = "") {
                self.keepAlive = keepAlive
            }
        }

        public struct OpenAIInfo: Codable, Equatable {
            @FallbackDecoding<EmptyString>
            public var organizationID: String

            public init(organizationID: String = "") {
                self.organizationID = organizationID
            }
        }
        
        public struct OpenAICompatibleInfo: Codable, Equatable {
            @FallbackDecoding<EmptyBool>
            public var enforceMessageOrder: Bool

            public init(enforceMessageOrder: Bool = false) {
                self.enforceMessageOrder = enforceMessageOrder
            }
        }
        
        public struct GoogleGenerativeAIInfo: Codable, Equatable {
            @FallbackDecoding<EmptyString>
            public var apiVersion: String

            public init(apiVersion: String = "") {
                self.apiVersion = apiVersion
            }
        }

        @FallbackDecoding<EmptyString>
        public var apiKeyName: String
        @FallbackDecoding<EmptyString>
        public var baseURL: String
        @FallbackDecoding<EmptyBool>
        public var isFullURL: Bool
        @FallbackDecoding<EmptyInt>
        public var maxTokens: Int
        @FallbackDecoding<EmptyBool>
        public var supportsFunctionCalling: Bool
        @FallbackDecoding<EmptyString>
        public var modelName: String

        @FallbackDecoding<EmptyChatModelOpenAIInfo>
        public var openAIInfo: OpenAIInfo
        @FallbackDecoding<EmptyChatModelOllamaInfo>
        public var ollamaInfo: OllamaInfo
        @FallbackDecoding<EmptyChatModelGoogleGenerativeAIInfo>
        public var googleGenerativeAIInfo: GoogleGenerativeAIInfo
        @FallbackDecoding<EmptyChatModelOpenAICompatibleInfo>
        public var openAICompatibleInfo: OpenAICompatibleInfo

        public init(
            apiKeyName: String = "",
            baseURL: String = "",
            isFullURL: Bool = false,
            maxTokens: Int = 4000,
            supportsFunctionCalling: Bool = true,
            modelName: String = "",
            openAIInfo: OpenAIInfo = OpenAIInfo(),
            ollamaInfo: OllamaInfo = OllamaInfo(),
            googleGenerativeAIInfo: GoogleGenerativeAIInfo = GoogleGenerativeAIInfo(),
            openAICompatibleInfo: OpenAICompatibleInfo = OpenAICompatibleInfo()
        ) {
            self.apiKeyName = apiKeyName
            self.baseURL = baseURL
            self.isFullURL = isFullURL
            self.maxTokens = maxTokens
            self.supportsFunctionCalling = supportsFunctionCalling
            self.modelName = modelName
            self.openAIInfo = openAIInfo
            self.ollamaInfo = ollamaInfo
            self.googleGenerativeAIInfo = googleGenerativeAIInfo
            self.openAICompatibleInfo = openAICompatibleInfo
        }
    }

    public var endpoint: String {
        switch format {
        case .openAI:
            let baseURL = info.baseURL
            if baseURL.isEmpty { return "https://api.openai.com/v1/chat/completions" }
            return "\(baseURL)/v1/chat/completions"
        case .openAICompatible:
            let baseURL = info.baseURL
            if baseURL.isEmpty { return "https://api.openai.com/v1/chat/completions" }
            if info.isFullURL { return baseURL }
            return "\(baseURL)/v1/chat/completions"
        case .azureOpenAI:
            let baseURL = info.baseURL
            let deployment = info.modelName
            let version = "2024-02-15-preview"
            if baseURL.isEmpty { return "" }
            return "\(baseURL)/openai/deployments/\(deployment)/chat/completions?api-version=\(version)"
        case .googleAI:
            let baseURL = info.baseURL
            if baseURL.isEmpty { return "https://generativelanguage.googleapis.com" }
            return "\(baseURL)"
        case .ollama:
            let baseURL = info.baseURL
            if baseURL.isEmpty { return "http://localhost:11434/api/chat" }
            return "\(baseURL)/api/chat"
        case .claude:
            let baseURL = info.baseURL
            if baseURL.isEmpty { return "https://api.anthropic.com/v1/messages" }
            return "\(baseURL)/v1/messages"
        }
    }
}

public struct EmptyChatModelInfo: FallbackValueProvider {
    public static var defaultValue: ChatModel.Info { .init() }
}

public struct EmptyChatModelFormat: FallbackValueProvider {
    public static var defaultValue: ChatModel.Format { .openAI }
}

public struct EmptyChatModelOllamaInfo: FallbackValueProvider {
    public static var defaultValue: ChatModel.Info.OllamaInfo { .init() }
}

public struct EmptyChatModelOpenAIInfo: FallbackValueProvider {
    public static var defaultValue: ChatModel.Info.OpenAIInfo { .init() }
}

public struct EmptyChatModelGoogleGenerativeAIInfo: FallbackValueProvider {
    public static var defaultValue: ChatModel.Info.GoogleGenerativeAIInfo { .init() }
}

public struct EmptyChatModelOpenAICompatibleInfo: FallbackValueProvider {
    public static var defaultValue: ChatModel.Info.OpenAICompatibleInfo { .init() }
}