package com.kwan.springbootkwan.utils;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson2.JSON;
import com.kwan.springbootkwan.entity.chat.ChoiceMessage;
import com.kwan.springbootkwan.entity.chat.Choices;
import com.kwan.springbootkwan.entity.chat.CompletionChatRequest;
import com.kwan.springbootkwan.entity.chat.CompletionChatResponse;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 人工智能chatgpt-api
 *
 * @author : qinyingjie
 * @version : 2.2.0
 * @date : 2023/3/22 20:40
 */
public class OpenAIChatGptUtil {
    /**
     * 请求地址
     */
    final static private String chatCompletionUrl = "https://api.openai.com/v1/chat/completions";
    /**
     * 模型
     */
    final static private String model = "gpt-3.5-turbo";
    /**
     * 这里使用你自己的认证信息
     */
    final static private String Authorization = "Bearer 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";


    public static void main(String[] args) {
        String question = "默写静夜思";
        //这里是我根据模型请求封装的模型实体
        CompletionChatRequest completionChatRequest = new CompletionChatRequest();
        //封装http请求
        HttpRequest post = HttpUtil.createPost(chatCompletionUrl).timeout(300000);
        Map<String, String> headers = new HashMap<>();
        headers.put("Authorization", Authorization);
        post.addHeaders(headers);
        post.contentType("application/json;charset=UTF-8");
        completionChatRequest.setModel(model);
        ChoiceMessage message = new ChoiceMessage("user", question);
        List<Object> messages = new ArrayList<>();
        messages.add(message);
        completionChatRequest.setMessages(messages);
        post.body(JSON.toJSONString(completionChatRequest));
        String body = post.execute().body();
        CompletionChatResponse completionChatResponse = JSON.parseObject(body, CompletionChatResponse.class);
        final List<Choices> choices = completionChatResponse.getChoices();
        for (Choices choice : choices) {
            final ChoiceMessage message1 = choice.getMessage();
            final String content = message1.getContent();
            System.out.println(content);
        }
        System.out.println("成功");
    }

//    public static CompletionChatResponse chatCompletion(String prompt) {//prompt即要将要发送至gtp的内容
//        //这里我把用户发送和机器人返回的20条内容内容存在了redis中，用于关联对话上下文场景，如果不需要，注释掉即可，直接new一个ArrayList
//        List<Object> messages = getUserCacheMessages();
//        int size = messages.size();
//
//        if (size > 19) {
//            messages.remove(0);
//        }
//
//        messages.add(message);
//        setUserCacheMessages(messages);//把新的放到redis
//       completionChatRequest.setMessages(messages);
//        post.body(JSON.toJSONString(completionChatRequest));
//        String body = post.execute().body();
//        CompletionChatResponse completionChatResponse = JSON.parseObject(body, CompletionChatResponse.class);
//        return completionChatResponse;
//    }
//
//    public static List<Object> getUserCacheMessages(){
//        RedisCache redisCache = SpringUtils.getBean(RedisCache.class);
//        List<Object> cacheMessages = redisCache.getCacheObject(JwtTokenUtil.getUserId + ":messages");
//        if (StringUtils.isEmpty(cacheMessages)){
//            cacheMessages= new ArrayList<>();
//        }
//        return cacheMessages;
//    }
//
//    public static void setUserCacheMessages(List<Object> messages){
//        while(messages.size()>20){
//            messages.remove(0);
//        }
//        RedisCache redisCache = SpringUtils.getBean(RedisCache.class);
//        redisCache.setCacheObject(JwtTokenUtil.getUserId + ":messages", messages, 30, TimeUnit.MINUTES);
//    }
}