diff --git a/README.md b/README.md
index 1e2e29bd49d64fda123072f11a03b54165064039..26c329755ec634a2d5c49cc988040b8ea37fc206 100644
--- a/README.md
+++ b/README.md
@@ -39,6 +39,23 @@
 ### 2.1 代码执行
 
 ```java
+import cn.bugstack.chatglm.model.*;
+import cn.bugstack.chatglm.session.Configuration;
+import cn.bugstack.chatglm.session.OpenAiSession;
+import cn.bugstack.chatglm.session.OpenAiSessionFactory;
+import cn.bugstack.chatglm.session.defaults.DefaultOpenAiSessionFactory;
+import com.alibaba.fastjson.JSON;
+import com.fasterxml.jackson.core.JsonProcessingException;
+import lombok.extern.slf4j.Slf4j;
+import okhttp3.sse.EventSource;
+import okhttp3.sse.EventSourceListener;
+import org.junit.Before;
+import org.junit.Test;
+
+import javax.annotation.Nullable;
+import java.util.ArrayList;
+import java.util.concurrent.CountDownLatch;
+
 @Slf4j
 public class ApiTest {
 
@@ -97,7 +114,6 @@ public class ApiTest {
         // 等待
         new CountDownLatch(1).await();
     }
-
 }
 ```
 
@@ -215,4 +231,4 @@ chatglm:
 - 📝 博客：[bugstack.cn](https://bugstack.cn/) - 足够硬核，内容老狠了！
 - 📺 视频：[B站 小傅哥の码场](https://space.bilibili.com/15637440)
 - 💌 微信：[fustack](https://bugstack.cn/images/personal/fustack.png) - 备注来意
-- 🐾 我的编程知识星球：[实战生产级项目、手写框架级源码，可以向我 1对1 提问，解答技术/职场/规划问题](https://bugstack.cn/md/zsxq/introduce.html)
\ No newline at end of file
+- 🐾 我的编程知识星球：[实战生产级项目、手写框架级源码，可以向我 1对1 提问，解答技术/职场/规划问题](https://bugstack.cn/md/zsxq/introduce.html)