我的小屋

1
key = ***
2
url = https://api.deepseek.com/chat/completions

1
public class TypewriterEffect {
2
    public static void printWord(String text,int delay){
3
        for(int i = 0; i

1
    private static String API_KEY;
2
    private static String API_URL;
3
    //从配置文件中获取key和url
4
    static {
5
        Properties properties = new Properties();
6
        try{
7
            InputStream is = DeepSeekClient.class.getClassLoader().getResourceAsStream("config.properties");
8
            properties.load(is);
9
            API_KEY = properties.getProperty("key");
10
            API_URL = properties.getProperty("url");
11
        }catch(Exception e){
12
            throw new RuntimeException(e);
13
        }
14
    }

1
public static void main(String[] args) {
2
        Scanner scanner = new Scanner(System.in);
3
        String input = "";
4
        System.out.println("请提问");
5
        while(true){
6
            String question = scanner.next();
7
            if(question.equals("exit")){
8
                break;
9
            }
10
            ask(question);//询问
11
            System.out.println();
12
        }
13
    }

1
 //定义内部类请求响应结构
2
    static class Message{
3
        private String content;
4
        private String role;
5
        public Message(String content, String role) {
6
            this.content = content;
7
            this.role = role;
8
        }
9
        public String getContent() {
10
            return content;
11
        }
12
    }

1
//请求体
2
    static class ChatRequest{
3
        private String model;
4
        private List messages;
5
        private double temperature;
6
        private int max_tokens;
7
        public ChatRequest(String model, List messages, double temperature, int max_tokens) {
8
            this.model = model;
9
            this.messages = messages;
10
            this.temperature = temperature;
11
            this.max_tokens = max_tokens;
12
        }
13
    }

1
 //响应体
2
    static class ChatResponse{
3
        private List choices;
4
        public List getChoices() {
5
            return choices;
6
        }
7
        static class Choice{
8
            private Message message;
9
            public Message getMessage() {
10
                return message;
11
            }
12
        }
13
    }

1
public static void ask(String content) {
2
        //创建消息列表
3
        List messages = new ArrayList<>();
4
        messages.add(new Message(content,"user"));
5
        ChatRequest requestBody = new ChatRequest(
6
                "deepseek-chat",//配置模型
7
                messages,
8
                0.7,
9
                1000
10
        );
11
        System.out.println("正在提交问题");
12
        long start = System.currentTimeMillis();
13
        //发送请求
14
        String response = sendRequest(requestBody);
15
        long endTime = System.currentTimeMillis();
16
        System.out.println("思考时间: "+ (endTime - start) + "毫秒");//计算响应时间
17
        TypewriterEffect.printWord(response,0);//在控制栏打印结果
18
    }

1
private static String sendRequest(ChatRequest requestBody){
2
        HttpClient client = HttpClient.newHttpClient();
3
        Gson gson = new Gson();
4
        //将ChatRequest 对象中封装的数据转为JSON格式
5
        String requestBodyJson = gson.toJson(requestBody);
6
        try{
7
            HttpRequest request = HttpRequest.newBuilder()
8
                    .uri(URI.create(API_URL))//配置url
9
                    .header("Content-Type","application/json")
10
                    .header("Authorization","Bearer "+API_KEY)//配置key
11
                    .POST(HttpRequest.BodyPublishers.ofString(requestBodyJson))//将返回数据转成json格式
12
                    .build();
13
            System.out.println("问题以提交，正在思考中");
14
            HttpResponse response = client.send(request,HttpResponse.BodyHandlers.ofString());
15
            if(response.statusCode() == 200){
16
                ChatResponse chatResponse = gson.fromJson(response.body(),ChatResponse.class);
17
                return chatResponse.getChoices().get(0).getMessage().getContent();
18
            }else{
19
                return "请求失败，状态码： " + response.statusCode()+" ，响应： "+response.body();
20
            }
21
        } catch (Exception e) {
22
            e.printStackTrace();
23
            return "请求异常：" +  e.getMessage();
24
        }
25
    }