Merge remote-tracking branch 'origin/master'

This commit is contained in:
limin 2024-12-31 11:49:40 +08:00
commit c25a0556bf
3 changed files with 11 additions and 3 deletions

View File

@ -4,19 +4,20 @@ import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
import javax.validation.constraints.NotEmpty;
import javax.validation.constraints.NotNull;
@Schema(description = "管理后台 - 大模型对话推理聊天 Request VO")
@Data
public class ChatReqVO {
@Schema(description = "模型类型", requiredMode = Schema.RequiredMode.REQUIRED)
@NotEmpty(message = "模型类型不能为空")
@NotNull(message = "模型类型不能为空")
private Integer modelType;
@Schema(description = "模型ID", requiredMode = Schema.RequiredMode.REQUIRED)
@NotEmpty(message = "模型ID不能为空")
@NotNull(message = "模型ID不能为空")
private Long modelId;
@Schema(description = "应用ID", requiredMode = Schema.RequiredMode.REQUIRED)
@NotEmpty(message = "应用ID不能为空")
@NotNull(message = "应用ID不能为空")
private Long applicationId;
@Schema(description = "对话的内容", requiredMode = Schema.RequiredMode.REQUIRED)
@NotEmpty(message = "对话的内容不能为空")

View File

@ -5,11 +5,13 @@ import cn.iocoder.yudao.framework.common.util.object.BeanUtils;
import cn.iocoder.yudao.module.llm.framework.backend.config.LLMBackendProperties;
import cn.iocoder.yudao.module.llm.service.http.vo.*;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
@Slf4j
@Service
public class ModelService {
@ -44,7 +46,10 @@ public class ModelService {
if (StringUtils.isBlank(req.getModel())) {
req.setModel(DEFAULT_MODEL_ID);
}
log.info("url: {}", llmBackendProperties.getModelCompletions());
log.info("request: {}", req);
String result = HttpUtils.post(llmBackendProperties.getModelCompletions(), null, JSON.toJSONString(req));
log.info("response: {}", result);
if (StringUtils.isBlank(result)) {
return null;
}

View File

@ -3,12 +3,14 @@ package cn.iocoder.yudao.module.llm.service.http.vo;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.ToString;
import java.util.List;
@Data
@NoArgsConstructor
@AllArgsConstructor
@ToString
public class ModelCompletionsReqVO {
private String model;