Junjie
2 天以前 63b01db83d9aad8a15276b4236a9a22e4aeef065
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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package com.zy.ai.controller;
 
import com.zy.ai.entity.ResponsesApiRequest;
import com.zy.ai.service.OpenAiResponsesService;
import lombok.RequiredArgsConstructor;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;
 
import java.util.LinkedHashMap;
import java.util.Map;
 
@RestController
@RequiredArgsConstructor
public class OpenAiCompatibleController {
 
    private final OpenAiResponsesService openAiResponsesService;
 
    @PostMapping(value = "/v1/responses",
            consumes = MediaType.APPLICATION_JSON_VALUE,
            produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<?> createResponse(@RequestBody ResponsesApiRequest request) {
        if (request == null) {
            return openAiError(HttpStatus.BAD_REQUEST, "请求体不能为空", "input");
        }
        if (Boolean.TRUE.equals(request.getStream())) {
            return openAiError(HttpStatus.BAD_REQUEST, "暂不支持 stream=true", "stream");
        }
 
        try {
            return ResponseEntity.ok(openAiResponsesService.create(request));
        } catch (IllegalArgumentException ex) {
            return openAiError(HttpStatus.BAD_REQUEST, ex.getMessage(), "input");
        } catch (IllegalStateException ex) {
            return openAiError(HttpStatus.BAD_GATEWAY, ex.getMessage(), null);
        }
    }
 
    private ResponseEntity<Map<String, Object>> openAiError(HttpStatus status, String message, String param) {
        Map<String, Object> error = new LinkedHashMap<>();
        error.put("message", message);
        error.put("type", "invalid_request_error");
        error.put("param", param);
        error.put("code", null);
 
        Map<String, Object> body = new LinkedHashMap<>();
        body.put("error", error);
        return ResponseEntity.status(status).body(body);
    }
}