mirror of
https://github.com/tencentmusic/supersonic.git
synced 2025-12-10 02:46:56 +00:00
43 lines
1.9 KiB
Java
43 lines
1.9 KiB
Java
package dev.langchain4j.provider;
|
|
|
|
import com.tencent.supersonic.common.pojo.ChatModelConfig;
|
|
import com.tencent.supersonic.common.pojo.EmbeddingModelConfig;
|
|
import com.tencent.supersonic.common.util.AESEncryptionUtil;
|
|
import dev.langchain4j.model.chat.ChatLanguageModel;
|
|
import dev.langchain4j.model.dify.DifyAiChatModel;
|
|
import dev.langchain4j.model.embedding.EmbeddingModel;
|
|
import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
|
|
import org.springframework.beans.factory.InitializingBean;
|
|
import org.springframework.stereotype.Service;
|
|
|
|
@Service
|
|
public class DifyModelFactory implements ModelFactory, InitializingBean {
|
|
public static final String PROVIDER = "DIFY";
|
|
|
|
public static final String DEFAULT_BASE_URL = "https://dify.com/v1/chat-messages";
|
|
public static final String DEFAULT_MODEL_NAME = "demo-预留-可不填写";
|
|
public static final String DEFAULT_EMBEDDING_MODEL_NAME = "all-minilm";
|
|
|
|
@Override
|
|
public ChatLanguageModel createChatModel(ChatModelConfig modelConfig) {
|
|
return DifyAiChatModel.builder().baseUrl(modelConfig.getBaseUrl())
|
|
.apiKey(AESEncryptionUtil.aesDecryptECB(modelConfig.getApiKey()))
|
|
.modelName(modelConfig.getModelName()).timeOut(modelConfig.getTimeOut()).build();
|
|
}
|
|
|
|
@Override
|
|
public EmbeddingModel createEmbeddingModel(EmbeddingModelConfig embeddingModelConfig) {
|
|
return OpenAiEmbeddingModel.builder().baseUrl(embeddingModelConfig.getBaseUrl())
|
|
.apiKey(embeddingModelConfig.getApiKey())
|
|
.modelName(embeddingModelConfig.getModelName())
|
|
.maxRetries(embeddingModelConfig.getMaxRetries())
|
|
.logRequests(embeddingModelConfig.getLogRequests())
|
|
.logResponses(embeddingModelConfig.getLogResponses()).build();
|
|
}
|
|
|
|
@Override
|
|
public void afterPropertiesSet() {
|
|
ModelProvider.add(PROVIDER, this);
|
|
}
|
|
}
|