diff --git a/agents-flex-llm/agents-flex-llm-qianfan/src/main/java/com/agentsflex/llm/qianfan/QianFanLlmConfig.java b/agents-flex-llm/agents-flex-llm-qianfan/src/main/java/com/agentsflex/llm/qianfan/QianFanLlmConfig.java new file mode 100644 index 0000000..759e5a6 --- /dev/null +++ b/agents-flex-llm/agents-flex-llm-qianfan/src/main/java/com/agentsflex/llm/qianfan/QianFanLlmConfig.java @@ -0,0 +1,32 @@ +package com.agentsflex.llm.qianfan; + +import com.agentsflex.core.llm.LlmConfig; + +public class QianFanLlmConfig extends LlmConfig { + private static final String DEFAULT_MODEL = "ernie-3.5-8k"; + private static final String DEFAULT_EMBEDDING_MODEL = "embedding-v1"; + private static final String DEFAULT_ENDPOINT = "https://qianfan.baidubce.com/v2"; + private String embeddingModel = DEFAULT_EMBEDDING_MODEL; + + public QianFanLlmConfig() { + setEndpoint(DEFAULT_ENDPOINT); + setModel(DEFAULT_MODEL); + } + + public QianFanLlmConfig(String apikey) { + setEndpoint(DEFAULT_ENDPOINT); + setModel(DEFAULT_MODEL); + super.setApiKey(apikey); + } + + + + public String getEmbeddingModel() { + return embeddingModel; + } + + public void setEmbeddingModel(String embeddingModel) { + this.embeddingModel = embeddingModel; + } + +}