Class ModelRouter

back/src/main/java/com/smarthr/service/ai/ModelRouter.java:22–137

import java.util.stream.Collectors;

@Slf4j
@Service
@RequiredArgsConstructor
public class ModelRouter {

    private final ModelRegistry modelRegistry;
    private final AIModelProperties aiModelProperties;
    private final UserRepository userRepository;

    /**
     * 获取当前模型适配器（无用户上下文，走默认模型）
     */
    public AIModelAdapter route() {
        return getDefaultAdapter();
    }

    /**
     * 获取指定用户的模型适配器
     */
    public AIModelAdapter route(Long userId) {
        return resolveAdapter(userId);
    }

    /**
     * 获取当前模型信息
     */
    public Map<String, Object> getCurrentModelInfo() {
        AIModelAdapter adapter = resolveAdapter(null);
        return Map.of(
                "modelId", adapter.getModelId(),
                "modelName", adapter.getModelName(),
                "enabled", adapter.isEnabled()
        );
    }

    /**
     * 获取指定用户的当前模型信息
     */
    public Map<String, Object> getCurrentModelInfo(Long userId) {
        AIModelAdapter adapter = resolveAdapter(userId);
        return Map.of(
                "modelId", adapter.getModelId(),
                "modelName", adapter.getModelName(),
                "enabled", adapter.isEnabled()
        );
    }

    /**
     * 获取所有可用的模型列表
     */
    public List<Map<String, Object>> getAvailableModels() {
        String currentId = getDefaultAdapter().getModelId();
        return buildModelList(currentId);
    }

    /**
     * 获取指定用户可用模型列表，并标记当前模型
     */