diff --git a/Partner-Core/src/main/java/work/slhaf/partner/runtime/interaction/data/context/PartnerRunningFlowContext.java b/Partner-Core/src/main/java/work/slhaf/partner/runtime/interaction/data/context/PartnerRunningFlowContext.java
index fafc434f..f88b5824 100644
--- a/Partner-Core/src/main/java/work/slhaf/partner/runtime/interaction/data/context/PartnerRunningFlowContext.java
+++ b/Partner-Core/src/main/java/work/slhaf/partner/runtime/interaction/data/context/PartnerRunningFlowContext.java
@@ -3,7 +3,7 @@ package work.slhaf.partner.runtime.interaction.data.context;
 import com.alibaba.fastjson2.JSONObject;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
-import work.slhaf.partner.api.agent.runtime.interaction.flow.entity.RunningFlowContext;
+import work.slhaf.partner.api.agent.runtime.interaction.flow.RunningFlowContext;
 import work.slhaf.partner.module.common.entity.AppendPromptData;
 import work.slhaf.partner.runtime.interaction.data.context.subcontext.CoreContext;
 import work.slhaf.partner.runtime.interaction.data.context.subcontext.ModuleContext;
diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/component/abstracts/AgentModule.kt b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/component/abstracts/AgentModule.kt
index b6b7143a..1b28ebf4 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/component/abstracts/AgentModule.kt
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/component/abstracts/AgentModule.kt
@@ -5,7 +5,7 @@ import org.slf4j.LoggerFactory
 import work.slhaf.partner.api.agent.factory.component.annotation.AgentComponent
 import work.slhaf.partner.api.agent.factory.component.annotation.Init
 import work.slhaf.partner.api.agent.runtime.config.AgentConfigLoader
-import work.slhaf.partner.api.agent.runtime.interaction.flow.entity.RunningFlowContext
+import work.slhaf.partner.api.agent.runtime.interaction.flow.RunningFlowContext
 import work.slhaf.partner.api.chat.ChatClient
 import work.slhaf.partner.api.chat.constant.ChatConstant
 import work.slhaf.partner.api.chat.pojo.ChatResponse
diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
index 38d81bea..bf864a2e 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/factory/context/AgentContext.kt
@@ -7,7 +7,7 @@ import work.slhaf.partner.api.agent.factory.capability.annotation.CapabilityCore
 import work.slhaf.partner.api.agent.factory.component.abstracts.AbstractAgentModule
 import work.slhaf.partner.api.agent.factory.component.annotation.AgentComponent
 import work.slhaf.partner.api.agent.runtime.exception.AgentRunningFailedException
-import work.slhaf.partner.api.agent.runtime.interaction.flow.entity.RunningFlowContext
+import work.slhaf.partner.api.agent.runtime.interaction.flow.RunningFlowContext
 import java.lang.reflect.Method
 import java.time.ZonedDateTime
 
diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/AgentGateway.java b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/AgentGateway.java
index 1a192a5d..cfdbf660 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/AgentGateway.java
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/AgentGateway.java
@@ -2,7 +2,7 @@ package work.slhaf.partner.api.agent.runtime.interaction;
 
 import work.slhaf.partner.api.agent.runtime.interaction.data.AgentInputData;
 import work.slhaf.partner.api.agent.runtime.interaction.data.AgentOutputData;
-import work.slhaf.partner.api.agent.runtime.interaction.flow.entity.RunningFlowContext;
+import work.slhaf.partner.api.agent.runtime.interaction.flow.RunningFlowContext;
 
 public interface AgentGateway<I extends AgentInputData, O extends AgentOutputData, C extends RunningFlowContext> {
 
diff --git a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/entity/RunningFlowContext.java b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/RunningFlowContext.java
similarity index 84%
rename from Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/entity/RunningFlowContext.java
rename to Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/RunningFlowContext.java
index b05cd63a..02eecd78 100644
--- a/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/entity/RunningFlowContext.java
+++ b/Partner-Framework/src/main/java/work/slhaf/partner/api/agent/runtime/interaction/flow/RunningFlowContext.java
@@ -1,4 +1,4 @@
-package work.slhaf.partner.api.agent.runtime.interaction.flow.entity;
+package work.slhaf.partner.api.agent.runtime.interaction.flow;
 
 import lombok.Data;
 import lombok.EqualsAndHashCode;
diff --git a/Partner-Test-Demo/src/main/java/work/slhaf/demo/flow/AgentDemoFlowContext.java b/Partner-Test-Demo/src/main/java/work/slhaf/demo/flow/AgentDemoFlowContext.java
index 27b395a5..4239d2b8 100644
--- a/Partner-Test-Demo/src/main/java/work/slhaf/demo/flow/AgentDemoFlowContext.java
+++ b/Partner-Test-Demo/src/main/java/work/slhaf/demo/flow/AgentDemoFlowContext.java
@@ -1,6 +1,6 @@
 package work.slhaf.demo.flow;
 
-import work.slhaf.partner.api.agent.runtime.interaction.flow.entity.RunningFlowContext;
+import work.slhaf.partner.api.agent.runtime.interaction.flow.RunningFlowContext;
 
 public class AgentDemoFlowContext extends RunningFlowContext {
 }