hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-07-14 11:05:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【调整】根据接口设计调整，ai model

Browse Source
This commit is contained in:
cherishsince
2024-05-06 16:32:47 +08:00
parent 118158424b
commit c21b5f7679
16 changed files with 392 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/AiPlatformEnum.java
View File

@ -1,8 +1,11 @@
package cn.iocoder.yudao.framework.ai;
import com.google.common.collect.Lists;
import lombok.AllArgsConstructor;
import lombok.Getter;
import java.util.List;
/**
* ai 模型平台
*
@ -17,12 +20,27 @@ public enum AiPlatformEnum {
YI_YAN("yiyan", "一言"),
QIAN_WEN("qianwen", "千问"),
XING_HUO("xinghuo", "星火"),
OPEN_AI("openai", "openAi"),
OPEN_AI_DALL("dall", "dall"),
MIDJOURNEY("midjourney", "midjourney"),
;
private String platform;
private String name;
public static List<AiPlatformEnum> CHAT_PLATFORM_LIST = Lists.newArrayList(
AiPlatformEnum.YI_YAN,
AiPlatformEnum.QIAN_WEN,
AiPlatformEnum.XING_HUO,
AiPlatformEnum.OPEN_AI
);
public static List<AiPlatformEnum> IMAGE_PLATFORM_LIST = Lists.newArrayList(
AiPlatformEnum.OPEN_AI_DALL,
AiPlatformEnum.MIDJOURNEY
);
public static AiPlatformEnum valueOfPlatform(String platform) {
for (AiPlatformEnum itemEnum : AiPlatformEnum.values()) {
if (itemEnum.getPlatform().equals(platform)) {

4
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiProperties.java
View File

@ -3,8 +3,8 @@ package cn.iocoder.yudao.framework.ai.config;
import cn.iocoder.yudao.framework.ai.AiPlatformEnum;
import cn.iocoder.yudao.framework.ai.chatxinghuo.XingHuoChatModel;
import cn.iocoder.yudao.framework.ai.chatyiyan.YiYanChatModel;
import cn.iocoder.yudao.framework.ai.imageopenai.OpenAiImageModelEnum;
import cn.iocoder.yudao.framework.ai.imageopenai.OpenAiImageStyleEnum;
import cn.iocoder.yudao.framework.ai.imageopenai.enums.OpenAiImageModelEnum;
import cn.iocoder.yudao.framework.ai.imageopenai.enums.OpenAiImageStyleEnum;
import lombok.Data;
import lombok.experimental.Accessors;
import org.springframework.boot.context.properties.ConfigurationProperties;