办学质量监测教学评价系统
ageer
2025-05-11 84b8d6f675c2f3aed83e0503e79cf1fa05e888e5
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
@@ -7,6 +7,8 @@
import io.github.ollama4j.models.chat.OllamaChatRequestModel;
import io.github.ollama4j.models.generate.OllamaStreamHandler;
import lombok.extern.slf4j.Slf4j;
import org.ruoyi.chat.enums.ChatModeType;
import org.ruoyi.chat.service.chat.IChatService;
import org.ruoyi.chat.util.SSEUtil;
import org.ruoyi.common.chat.entity.chat.Message;
import org.ruoyi.common.chat.request.ChatRequest;
@@ -22,14 +24,18 @@
import java.util.concurrent.CompletableFuture;
/**
 * @author ageer
 */
@Service
@Slf4j
public class OllamaServiceImpl  {
public class OllamaServiceImpl implements IChatService {
   @Autowired
   private  IChatModelService chatModelService;
    @Autowired
    private IChatModelService chatModelService;
    public SseEmitter chat(ChatRequest chatRequest,SseEmitter emitter) {
    @Override
    public SseEmitter chat(ChatRequest chatRequest, SseEmitter emitter) {
        ChatModelVo chatModelVo = chatModelService.selectModelByName(chatRequest.getModel());
        String host = chatModelVo.getApiHost();
        List<Message> msgList = chatRequest.getMessages();
@@ -73,4 +79,8 @@
        return emitter;
    }
    @Override
    public String getCategory() {
        return ChatModeType.OLLAMA.getCode();
    }
}