0.2.1 update: add debug log; add stream model finish callback

Author: lrbmike

pom.xml

@@ -6,7 +6,7 @@
 
   <groupId>org.liurb.ai.sdk</groupId>
   <artifactId>java-ai-sdk</artifactId>
-  <version>0.2.0</version>
+  <version>0.2.1</version>
 
   <name>java-ai-sdk</name>
   <description>
@@ -52,6 +52,7 @@
     <lombok.version>1.18.32</lombok.version>
     <fastjson2.version>2.0.52</fastjson2.version>
     <okhttp.version>4.12.0</okhttp.version>
+    <slf4j-api.version>2.0.16</slf4j-api.version>
   </properties>
 
   <dependencies>
@@ -81,6 +82,13 @@
       <scope>provided</scope>
     </dependency>
 
+    <dependency>
+      <groupId>org.slf4j</groupId>
+      <artifactId>slf4j-api</artifactId>
+      <version>${slf4j-api.version}</version>
+    </dependency>
+
+
     <dependency>
       <groupId>junit</groupId>
       <artifactId>junit</artifactId>

src/main/java/org/liurb/ai/sdk/common/AiBaseClient.java

@@ -2,6 +2,7 @@
 
 import com.alibaba.fastjson2.JSON;
 import com.alibaba.fastjson2.JSONObject;
+import lombok.extern.slf4j.Slf4j;
 import okhttp3.*;
 import okhttp3.logging.HttpLoggingInterceptor;
 import org.liurb.ai.sdk.common.bean.*;
@@ -15,6 +16,7 @@
 import java.util.List;
 import java.util.concurrent.TimeUnit;
 
+@Slf4j
 public abstract class AiBaseClient {
 
     private ModelAccount account;
@@ -100,7 +102,7 @@ public void onResponse(Call call, Response response) throws IOException {
                         String line;
                         while ((line = reader.readLine()) != null) {
                             line = line.trim();
-                            System.out.println(line);
+                            log.debug(line);
 
                             AiStreamMessage streamMessage = buildStreamMessage(line);
                             if (streamMessage != null) {
@@ -117,6 +119,9 @@ public void onResponse(Call call, Response response) throws IOException {
                         //handle history
                         buildStreamChatHistory(message, mediaData, textSb.toString(), history);
 
+                    } finally {
+                        //finish callback
+                        responseListener.accept(AiStreamMessage.builder().stop(true).build());
                     }
                 }
             }