import Foundation

public struct TextEmbeddingConfig: Codable, Sendable {
    public var modelType: String
    public var poolingMethod: PoolingMethod
    public var normalize: Bool

    public init(
        modelType: String = "gemma-2b",
        poolingMethod: PoolingMethod = .mean,
        normalize: Bool = true
    ) {
        self.modelType = modelType
        self.poolingMethod = poolingMethod
        self.normalize = normalize
    }
}