1

yangbin

2024-08-15 fc38e2635216775a80210d0df109dc6174d66813

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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);
        }
    }
}