diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/mapper/ChatContextMapper.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/mapper/ChatContextMapper.java
index 9d73285ce..4326176e9 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/mapper/ChatContextMapper.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/mapper/ChatContextMapper.java
@@ -6,7 +6,7 @@ import org.apache.ibatis.annotations.Mapper;
 @Mapper
 public interface ChatContextMapper {
 
-    ChatContextDO getContextByChatId(int chatId);
+    ChatContextDO getContextByChatId(Integer chatId);
 
     int updateContext(ChatContextDO contextDO);
 
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/ChatContextRepository.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/ChatContextRepository.java
index ea73202e1..7e50a45af 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/ChatContextRepository.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/ChatContextRepository.java
@@ -4,7 +4,7 @@ import com.tencent.supersonic.headless.core.pojo.ChatContext;
 
 public interface ChatContextRepository {
 
-    ChatContext getOrCreateContext(int chatId);
+    ChatContext getOrCreateContext(Integer chatId);
 
     void updateContext(ChatContext chatCtx);
 
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/ChatContextRepositoryImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/ChatContextRepositoryImpl.java
index 50a1f8d89..f687cecea 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/ChatContextRepositoryImpl.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/persistence/repository/impl/ChatContextRepositoryImpl.java
@@ -24,7 +24,7 @@ public class ChatContextRepositoryImpl implements ChatContextRepository {
     }
 
     @Override
-    public ChatContext getOrCreateContext(int chatId) {
+    public ChatContext getOrCreateContext(Integer chatId) {
         ChatContextDO context = chatContextMapper.getContextByChatId(chatId);
         if (context == null) {
             ChatContext chatContext = new ChatContext();
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/ChatContextService.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/ChatContextService.java
index e2b22153e..fc450e492 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/ChatContextService.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/ChatContextService.java
@@ -11,7 +11,7 @@ public interface ChatContextService {
      */
     Long getContextModel(Integer chatId);
 
-    ChatContext getOrCreateContext(int chatId);
+    ChatContext getOrCreateContext(Integer chatId);
 
     void updateContext(ChatContext chatCtx);
 
diff --git a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatContextServiceImpl.java b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatContextServiceImpl.java
index d5c70b545..5e412040e 100644
--- a/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatContextServiceImpl.java
+++ b/headless/server/src/main/java/com/tencent/supersonic/headless/server/service/impl/ChatContextServiceImpl.java
@@ -37,7 +37,7 @@ public class ChatContextServiceImpl implements ChatContextService {
     }
 
     @Override
-    public ChatContext getOrCreateContext(int chatId) {
+    public ChatContext getOrCreateContext(Integer chatId) {
         return chatContextRepository.getOrCreateContext(chatId);
     }