package org.ruoyi.chat.factory;


import lombok.extern.slf4j.Slf4j;
import org.ruoyi.chat.service.chat.IChatService;
import org.ruoyi.chat.service.chat.impl.OllamaServiceImpl;
import org.ruoyi.chat.service.chat.impl.OpenAIServiceImpl;

import org.springframework.stereotype.Component;

@Component
@Slf4j
public class SseServiceFactory {

    public IChatService getSseService(String type) {
        if ("openai".equals(type)) {
            return new OpenAIServiceImpl();
        } else if ("ollama".equals(type)) {
            return new OllamaServiceImpl();
        } else {
            throw new IllegalArgumentException("Unknown type: " + type);
        }
    }
}