免费接入DeepSeek等多种大模型

核心代码

import cn.hutool.core.collection.CollUtil;
import com.tool4j.pasteshare.entity.params.AiParams;
import com.tool4j.pasteshare.entity.params.Message;
import com.tool4j.pasteshare.service.AiCompletionService;
import com.tool4j.pasteshare.util.deepseek.DeepSeekResult;
import com.tool4j.pasteshare.util.deepseek.DeepSeekUtil;
import com.tool4j.pasteshare.util.http.HttpUtil;
import io.reactivex.Flowable;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletResponse;
import java.util.List;
import java.util.Map;

@Service("deepSeekCompletionService")
@Slf4j
public class DeepSeekCompletionServiceImpl implements AiCompletionService {
    private static final String api = "https://api.deepseek.com/chat/completions";

    @Override
    public String completions(List<Message> messages, float temperature) {
        AiParams aiParams = new AiParams();
        aiParams.setMessages(messages);
        aiParams.setTemperature(temperature);
        return HttpUtil.post(api, aiParams, Map.of("Authorization", DeepSeekUtil.API_KEY));
    }

    @Override
    public void completionsStream(HttpServletResponse response, List<Message> messages, float temperature, Long logId, Boolean enableSearch) {
        Flowable<DeepSeekResult> flow = DeepSeekUtil.completionsStream(messages, temperature);
        flow.blockingForEach(item -> {
            List<DeepSeekResult.Choices> choices = item.getChoices();
            if (CollUtil.isNotEmpty(choices)) {
                String content = choices.get(0).getDelta().getContent();
                response.getWriter().write(content == null ? "" : content);
                response.getWriter().flush();
            }
            try {
                Thread.sleep(20);
            } catch (Exception ex) {
            }
        });
    }

}