package com.yaoai.ai.providers.config;

import lombok.Data;

@Data
public class PlatformModelSettings {
    private PlatformModelItem text;
    private PlatformModelItem image;
    private PlatformModelItem video;

    public PlatformModelItem get(ModelCapability capability) {
        return switch (capability) {
            case TEXT -> text;
            case IMAGE -> image;
            case VIDEO -> video;
        };
    }

    public void set(ModelCapability capability, PlatformModelItem item) {
        switch (capability) {
            case TEXT -> text = item;
            case IMAGE -> image = item;
            case VIDEO -> video = item;
        }
    }
}
