From a5496bf194b119b885313d20a4e977a8b2df1d55 Mon Sep 17 00:00:00 2001
From: Azure99 <i@rainng.com>
Date: Thu, 27 Jun 2024 15:11:59 +0800
Subject: [PATCH] chore: bump version to 0.0.7 & change default chat model to
 ernie-lite (#620)

Update ModelEndpointRetriever.java
---
 java/pom.xml                                                   | 2 +-
 java/src/main/java/com/baidubce/qianfan/QianfanClient.java     | 2 +-
 .../java/com/baidubce/qianfan/core/ModelEndpointRetriever.java | 3 ++-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/java/pom.xml b/java/pom.xml
index d81f2b5a..b9b9f399 100644
--- 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
index 24b67578..2ee3ea64 100644
--- 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
index cb443691..15d7ec56 100644
--- 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");