fix

Author: JqRrt

cozeloop/internal/trace/span.py

@@ -533,7 +533,8 @@ def set_stat_info(self):
 
         input_tokens = tag_map.get(INPUT_TOKENS, 0)
         output_tokens = tag_map.get(OUTPUT_TOKENS, 0)
-        self.set_tags({TOKENS: int(input_tokens) + int(output_tokens)})
+        if input_tokens > 0 or output_tokens > 0:
+            self.set_tags({TOKENS: int(input_tokens) + int(output_tokens)})
 
         duration = int((datetime.now().timestamp() - self.start_time.timestamp()) * 1000000)
         with self.lock: