package org.jeecg.modules.ai.service.imp;

import org.jeecg.modules.ai.service.IWebSocketAiService;
import org.jeecg.modules.ai.websocket.AiWebSocket;
import org.jeecg.modules.utils.AiTestUils;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;

/**
 * @author clown
 * * @date 2024/8/8
 */
@Service
public class WebSocketAiService implements IWebSocketAiService {


    @Override
    @Scheduled(cron = "0/10 * * * * ?")
    public void selectChatIA() {
        String name = "{\"id\":\"683a65fd-8feb-4446-ad32-714c4785f667\",\"messages\":[{\"role\":\"user\",\"content\":\"给我讲个故事？\"}],\"stream\":true,\"max_tokens\":500}"; // JSON数据
        String url = "https://836u458t54.vicp.fun/chat/test_chat";
        String aiPost =  AiTestUils.sendJsonToHttpsPost(url, name);
        System.out.println(aiPost);
        for (String s : AiWebSocket.getSessionPoolAi().keySet()) {
            AiWebSocket.pushMessage(s,aiPost);
        }
    }
}