chore: bump version to 0.0.7 & change default chat model to ernie-lite (

#620) Update ModelEndpointRetriever.java
baidubce · Jun 27, 2024 · a5496bf · a5496bf
1 parent 02f458c
commit a5496bf
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/java/pom.xml b/java/pom.xml
@@ -4,7 +4,7 @@
 
     <groupId>com.baidubce</groupId>
     <artifactId>qianfan</artifactId>
-    <version>0.0.6</version>
+    <version>0.0.7</version>
     <packaging>jar</packaging>
 
     <name>qianfan</name>

diff --git a/java/src/main/java/com/baidubce/qianfan/QianfanClient.java b/java/src/main/java/com/baidubce/qianfan/QianfanClient.java
@@ -36,7 +36,7 @@
 import java.util.Map;
 
 class QianfanClient {
-    private static final String SDK_VERSION = "0.0.6";
+    private static final String SDK_VERSION = "0.0.7";
     private static final String QIANFAN_URL_TEMPLATE = "%s/rpc/2.0/ai_custom/v1/wenxinworkshop%s";
     private static final String EXTRA_PARAM_REQUEST_SOURCE = "request_source";
     private static final String REQUEST_SOURCE_PREFIX = "qianfan_java_sdk_v";

diff --git a/java/src/main/java/com/baidubce/qianfan/core/ModelEndpointRetriever.java b/java/src/main/java/com/baidubce/qianfan/core/ModelEndpointRetriever.java
@@ -34,7 +34,7 @@
 import java.util.Map;
 
 public class ModelEndpointRetriever {
-    private static final String DEFAULT_CHAT_MODEL = "ernie-bot-turbo";
+    private static final String DEFAULT_CHAT_MODEL = "ernie-lite-8k";
     private static final String DEFAULT_COMPLETION_MODEL = "codellama-7b-instruct";
     private static final String DEFAULT_EMBEDDING_MODEL = "embedding-v1";
     private static final String DEFAULT_TEXT_2_IMAGE_MODEL = "stable-diffusion-xl";
@@ -96,6 +96,7 @@ public ModelEndpointRetriever(IAuth auth) {
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie-character-8k-0321", "ernie-char-8k");
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie-lite-8k-0922", "eb-instant");
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie-lite-8k-0308", "ernie-lite-8k");
+        typeModelEndpointMap.get(ModelType.CHAT).put("ernie-lite-8k", "ernie-lite-8k");
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie-functions-8k", "ernie-func-8k");
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie-tiny-8k", "ernie-tiny-8k");
         typeModelEndpointMap.get(ModelType.CHAT).put("ernie speed-appbuilder", "ai_apaas");