From 40217572b1fec6cd8c14857c61f21e809adc1217 Mon Sep 17 00:00:00 2001
From: Erik <erik.nihlen@gmail.com>
Date: Fri, 13 Mar 2026 10:55:27 +0100
Subject: [PATCH] fix: accept burden stats from plugin

---
 main.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/main.py b/main.py
index 2fdb8422..4432696d 100644
--- a/main.py
+++ b/main.py
@@ -1103,6 +1103,9 @@ class CharacterStatsMessage(BaseModel):
     birth: Optional[str] = None
     current_title: Optional[int] = None
     skill_credits: Optional[int] = None
+    burden: Optional[int] = None
+    burden_units: Optional[int] = None
+    encumbrance_capacity: Optional[int] = None
     attributes: Optional[dict] = None
     vitals: Optional[dict] = None
     skills: Optional[dict] = None
@@ -2479,6 +2482,9 @@ async def ws_receive_snapshots(
                         "birth",
                         "current_title",
                         "skill_credits",
+                        "burden",
+                        "burden_units",
+                        "encumbrance_capacity",
                         "properties",
                         "titles",
                     ):
@@ -2987,6 +2993,9 @@ async def test_character_stats(name: str):
             "birth",
             "current_title",
             "skill_credits",
+            "burden",
+            "burden_units",
+            "encumbrance_capacity",
         ):
             if stats_dict.get(key) is not None:
                 stats_data[key] = stats_dict[key]