diff --git a/python/core/components/v2/tree_mind/model.py b/python/core/components/v2/tree_mind/model.py
index d0fa5498..9b396835 100644
--- a/python/core/components/v2/tree_mind/model.py
+++ b/python/core/components/v2/tree_mind/model.py
@@ -16,7 +16,7 @@
 """
 
 import proto
-from typing import List
+from typing import List, Optional
 from pydantic import BaseModel
 
 
@@ -38,8 +38,8 @@ class Info(BaseModel):
 class TreeMindResponse(BaseModel):
     errCode: str
     errMsg: str
-    time_diff: int
-    info: Info
+    time_diff: Optional[int] = None
+    info: Optional[Info] = None