fix: modify AI chat

Author: lu-yg

base/src/main/java/com/tinyengine/it/controller/AiChatController.java

@@ -32,6 +32,7 @@
 import org.springframework.validation.annotation.Validated;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestHeader;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.servlet.mvc.method.annotation.StreamingResponseBody;
@@ -103,7 +104,12 @@ public ResponseEntity<?> aiChat(@RequestBody ChatRequest request) {
     })
     @SystemControllerLog(description = "AI api v1")
     @PostMapping("/chat/completions")
-    public ResponseEntity<?> chat(@RequestBody ChatRequest request) {
+    public ResponseEntity<?> chat(@RequestBody ChatRequest request,
+        @RequestHeader("Authorization") String authorization) {
+        if (authorization != null && authorization.startsWith("Bearer ")) {
+            String token = authorization.replace("Bearer ", "");
+            request.setApiKey(token);
+        }
         try {
             Object response = aiChatV1Service.chatCompletion(request);
 

base/src/main/java/com/tinyengine/it/controller/ResourceController.java

@@ -164,6 +164,27 @@ public Result<Resource> createResource(@Valid @RequestBody Resource resource) th
         return Result.success(result);
     }
 
+    /**
+     * 上传图片
+     *
+     * @param resource the resource
+     * @return Resource信息 result
+     */
+    @Operation(summary = "创建resource", description = "创建resource",
+            parameters = {
+                    @Parameter(name = "resource", description = "Resource入参对象")
+            }, responses = {
+            @ApiResponse(responseCode = "200", description = "返回信息",
+                    content = @Content(mediaType = "application/json", schema = @Schema(implementation = Resource.class))),
+            @ApiResponse(responseCode = "400", description = "请求失败")
+    })
+    @SystemControllerLog(description = "创建resource")
+    @PostMapping("/resource/uoload")
+    public Result<Resource> resourceUoload(@Valid @RequestBody Resource resource) throws Exception {
+        Resource result = resourceService.resourceUpload(resource);
+        return Result.success(result);
+    }
+
     /**
      * 批量创建Resource
      *

base/src/main/java/com/tinyengine/it/service/app/impl/v1/AiChatV1ServiceImpl.java

@@ -319,12 +319,6 @@ private StreamingResponseBody processStreamResponse(HttpRequest.Builder requestB
                     }
                 }
             } catch (Exception e) {
-                // 简单的错误处理：如果是客户端断开连接，忽略错误
-                String errorMsg = e.getMessage();
-                if (errorMsg != null &&
-                    (errorMsg.contains("Broken pipe") || errorMsg.contains("Connection reset"))) {
-                    return;
-                }
 
                 try {
                     String errorEvent = "data: {\"error\": \"" + e.getMessage() + "\"}\n\n";

base/src/main/java/com/tinyengine/it/service/material/ResourceService.java

@@ -84,6 +84,14 @@ public interface ResourceService extends IService<Resource> {
      */
     Resource createResource(Resource resource) throws Exception;
 
+    /**
+     * 图片上传
+     *
+     * @param resource the resource
+     * @return the integer
+     */
+    Resource resourceUpload(Resource resource) throws Exception;
+
     /**
      * 批量新增表t_resource数据
      *

base/src/main/java/com/tinyengine/it/service/material/impl/ResourceServiceImpl.java

@@ -155,6 +155,23 @@ public Result<Resource> updateResourceById(Resource resource) {
     @SystemServiceLog(description = "新增表t_resource数据")
     public Resource createResource(Resource resource) throws Exception {
 
+        Resource res = this.resourceUpload(resource);
+        ResourceGroupResource resourceGroupResource = new ResourceGroupResource();
+        resourceGroupResource.setResourceId(res.getId());
+        resourceGroupResource.setResourceGroupId(resource.getResourceGroupId());
+        resourceGroupResourceMapper.createResourceGroupResource(resourceGroupResource);
+
+        return res;
+    }
+
+    /**
+     * 图片上传
+     *
+     * @param resource the resource
+     * @return the integer
+     */
+    @Override
+    public Resource resourceUpload(Resource resource) throws Exception {
         ResourceRequestDto resourceParam = new ResourceRequestDto();
         resourceParam.setName(resource.getName());
         resourceParam.setCategory(resource.getCategory());
@@ -179,7 +196,7 @@ public Resource createResource(Resource resource) throws Exception {
         QueryWrapper<Resource> queryWrapper = new QueryWrapper<>();
         queryWrapper.eq("name", resource.getName());
         queryWrapper.eq("category", resource.getCategory());
-        // 接入组合系统需添加租户id查询
+        // 接入租户系统需添加租户id查询
         Resource resourceResult = this.baseMapper.selectOne(queryWrapper);
         if (resourceResult != null) {
             throw new ServiceException(ExceptionEnum.CM003.getResultCode(), ExceptionEnum.CM003.getResultMsg());
@@ -188,10 +205,6 @@ public Resource createResource(Resource resource) throws Exception {
         if (createResult != 1) {
             throw new ServiceException(ExceptionEnum.CM002.getResultCode(), ExceptionEnum.CM002.getResultMsg());
         }
-        ResourceGroupResource resourceGroupResource = new ResourceGroupResource();
-        resourceGroupResource.setResourceId(resource.getId());
-        resourceGroupResource.setResourceGroupId(resource.getResourceGroupId());
-        resourceGroupResourceMapper.createResourceGroupResource(resourceGroupResource);
 
         return this.baseMapper.queryResourceById(resource.getId());
     }