spring boot 嵌入chatGPT步骤

一、需要良好的网络

二、需要在OpenAI官网https://openai.com/注册用户,并获取一个api-key,sk开头的

验证是否可用网站:http://tools.lbbit.top/check_key_valid/

三、spring boot 配置文件

openai.proxyHost=127.0.0.1
openai.proxyPort=7890
openai.keys=sk-xxxxxxxxxx
openai.proxy=https://xxxxxxx/

四、新建配置类AiServiceFactory

import com.fasterxml.jackson.databind.ObjectMapper;
import com.theokanning.openai.client.OpenAiApi;
import com.theokanning.openai.service.OpenAiService;
import okhttp3.OkHttpClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;
import retrofit2.Retrofit;

import java.net.InetSocketAddress;
import java.net.Proxy;
import java.time.Duration;

@Component
public class AiServiceFactory {


    @Value("${openai.proxyHost}")
    private String proxyHost;
    /**
     * 代理端口
     */
    @Value("${openai.proxyPort}")
    private Integer proxyPort;
    /**
     * openai apikey
     */
    @Value("${openai.keys}")
    private String token;

    @Value("${openai.proxy}")
    private String proxyIp;

    private static final Duration DEFAULT_TIMEOUT = Duration.ofSeconds(10L);

    public OpenAiService createService() {

        ObjectMapper mapper = OpenAiService.defaultObjectMapper();
        // 设置代理
        Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(proxyHost, proxyPort));
        OkHttpClient client = OpenAiService.defaultClient(token, DEFAULT_TIMEOUT).newBuilder()
                .proxy(proxy)
                .build();
        Retrofit retrofit = OpenAiService.defaultRetrofit(client, mapper).newBuilder().baseUrl(proxyIp).build();
        return new OpenAiService(retrofit.create(OpenAiApi.class), client.dispatcher().executorService());

    }
}

如果需要中转站代理的话,该类里面的方法如下

public OpenAiService createService() {

        ObjectMapper mapper = OpenAiService.defaultObjectMapper();
        // 设置代理
//        Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(proxyHost, proxyPort));
//        Proxy proxy = new Proxy(Proxy.Type.HTTP, new InetSocketAddress(proxyIp, 8080));
        OkHttpClient client = OpenAiService.defaultClient(token, DEFAULT_TIMEOUT).newBuilder()
//                .proxy(proxy)
                .build();
        Retrofit retrofit = OpenAiService.defaultRetrofit(client, mapper).newBuilder().baseUrl(proxyIp).build();
        //代理服务器,中转站
        return new OpenAiService(retrofit.create(OpenAiApi.class), client.dispatcher().executorService());

    }

五、测试控制器,当然也可以写进service层

package com.example.springbootest3_2.controller;

import com.example.springbootest3_2.config.AiServiceFactory;
import com.theokanning.openai.completion.chat.*;
import com.theokanning.openai.service.OpenAiService;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@RestController
public class OpenAiController {

    @Resource
    private AiServiceFactory aiServiceFactory;

    @PostMapping("/testChat")
    public String testChat(@RequestBody Map<String,String> params) throws UnsupportedEncodingException {

        OpenAiService service = aiServiceFactory.createService();
        final List<ChatMessage> messages = new ArrayList<>();
        final ChatMessage systemMessage = new ChatMessage(ChatMessageRole.USER.value(),
                URLDecoder.decode(params.get("contents"),
                "UTF-8"));

        messages.add(systemMessage);
        ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest
                .builder()
                .model("gpt-3.5-turbo")
                .messages(messages)
                .temperature(0.5)
//                .n(1)
//                .maxTokens(50)
//                .logitBias(new HashMap<>())
                .build();


        ChatCompletionResult chatCompletionResult=service.createChatCompletion(chatCompletionRequest);
        List<ChatCompletionChoice> compList=chatCompletionResult.getChoices();
        StringBuilder sb = new StringBuilder();
        for (ChatCompletionChoice comp : compList) {
            sb.append(comp.getMessage().getContent());
        }
        return sb.toString();
    }

}
相关推荐
乌啼霜满天2497 分钟前
Spring 与 Spring MVC 与 Spring Boot三者之间的区别与联系
java·spring boot·spring·mvc
tangliang_cn12 分钟前
java入门 自定义springboot starter
java·开发语言·spring boot
Grey_fantasy23 分钟前
高级编程之结构化代码
java·spring boot·spring cloud
新知图书24 分钟前
Rust编程与项目实战-模块std::thread(之一)
开发语言·后端·rust
盛夏绽放1 小时前
Node.js 和 Socket.IO 实现实时通信
前端·后端·websocket·node.js
Ares-Wang1 小时前
Asp.net Core Hosted Service(托管服务) Timer (定时任务)
后端·asp.net
苹果酱05671 小时前
前端面试vue篇:Vue2 和 Vue3 在设计和性能上有显著区别
java·spring boot·毕业设计·layui·课程设计
刘大浪2 小时前
后端数据增删改查基于Springboot+mybatis mysql 时间根据当时时间自动填充,数据库连接查询不一致,mysql数据库连接不好用
数据库·spring boot·mybatis
一只爱撸猫的程序猿2 小时前
简单实现一个系统升级过程中的数据平滑迁移的场景实例
数据库·spring boot·程序员
Rverdoser2 小时前
RabbitMQ的基本概念和入门
开发语言·后端·ruby