diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/MessageDto.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/MessageDto.cs
index 28327a89..bdbca927 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/MessageDto.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Application.Contracts/Dtos/MessageDto.cs
@@ -8,8 +8,6 @@ public class MessageDto : FullAuditedEntityDto<Guid>
     public Guid SessionId { get; set; }
     public string Content { get; set; }
     public string Role { get; set; }
-    public decimal DeductCost { get; set; }
-    public decimal TotalTokens { get; set; }
     public string ModelId { get; set; }
     public string Remark { get; set; }
 }
\ No newline at end of file
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
index d9d5fb58..2307053d 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Entities/Chat/MessageAggregateRoot.cs
@@ -39,8 +39,6 @@ public class MessageAggregateRoot : FullAuditedAggregateRoot<Guid>
     public string Content { get; set; }
 
     public string Role { get; set; }
-    public decimal DeductCost { get; set; }
-    public decimal TotalTokens { get; set; }
     public string ModelId { get; set; }
     public string? Remark { get; set; }
 
diff --git a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
index 40077531..2d294ff3 100644
--- a/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
+++ b/Yi.Abp.Net8/module/ai-hub/Yi.Framework.AiHub.Domain/Managers/UsageStatisticsManager.cs
@@ -7,6 +7,11 @@ public class UsageStatisticsManager : DomainService
 {
     private readonly ISqlSugarRepository<UsageStatisticsAggregateRoot> _repository;
 
+    public UsageStatisticsManager(ISqlSugarRepository<UsageStatisticsAggregateRoot> repository)
+    {
+        _repository = repository;
+    }
+
     private IDistributedLockProvider DistributedLock =>
         LazyServiceProvider.LazyGetRequiredService<IDistributedLockProvider>();