package cc.mrbird.febs.ai.strategy;

import com.google.common.collect.Maps;
import lombok.Getter;
import org.springframework.stereotype.Component;

import java.util.Map;

@Component
@Getter
public class LlmStrategyFactory {

    public final Map<String, LlmStrategyService> llmStrategyMap = Maps.newHashMapWithExpectedSize(8);

    public LlmStrategyFactory(Map<String, LlmStrategyService> strategyMap) {
        this.llmStrategyMap.clear();
        this.llmStrategyMap.putAll(strategyMap);
    }

    public Map<String, LlmStrategyService> getCalculationStrategyMap() {
        return llmStrategyMap;
    }
}