package org.ruoyi.chat.config;

import jakarta.annotation.PostConstruct;
import lombok.Getter;
import lombok.RequiredArgsConstructor;
import org.ruoyi.common.core.utils.OkHttpUtil;

import org.ruoyi.domain.vo.ChatModelVo;
import org.ruoyi.service.IChatModelService;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;

@Component
@RequiredArgsConstructor
public class OkHttpConfig {

    private final IChatModelService chatModelService;
    private final Map<String, OkHttpUtil> okHttpUtilMap = new HashMap<>();
    @Getter
    private String generate;

    @PostConstruct
    public void init() {
        initializeOkHttpUtil("suno");
        initializeOkHttpUtil("luma");
    }

    private void initializeOkHttpUtil(String modelName) {
        ChatModelVo chatModelVo = chatModelService.selectModelByName(modelName);
        if(chatModelVo==null){
            return;
        }
        OkHttpUtil okHttpUtil = new OkHttpUtil();
        okHttpUtil.setApiHost(chatModelVo.getApiHost());
        okHttpUtil.setApiKey(chatModelVo.getApiKey());
        generate = String.valueOf(chatModelVo.getModelPrice());
        okHttpUtilMap.put(modelName, okHttpUtil);
    }

    public OkHttpUtil getOkHttpUtil(String modelName) {
        return okHttpUtilMap.get(modelName);
    }
}