diff --git a/proto/phase_grpc.py b/proto/phase_grpc.py
index 0945028..3dadef8 100644
--- a/proto/phase_grpc.py
+++ b/proto/phase_grpc.py
@@ -150,7 +150,11 @@ def QueryCrossPhaseDiagnosis(citycode: int, crossid: str, date_list: List, tp: s
         response = stub.CrossPhaseDiagnosis(request_params, timeout=30)
         if response.code != 0:
             raise Exception(response.msg)
-        return MessageToDict(response.data), None
+        
+        data = MessageToDict(response.data)
+        if len(data) > 0:
+            data['total'] = len(data['values'])
+        return data, None
     except Exception as e:
         return None, e
     finally: