From fa80caf0ffc3a9d252ac8e789dbfbb4ca7817362 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@home-assistant.io>
Date: Fri, 3 Apr 2026 12:31:44 -1000
Subject: [PATCH] [api] Fall back to RAW_ENCODE_MAP when SMALL_MAP has no entry

---
 script/api_protobuf/api_protobuf.py | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/script/api_protobuf/api_protobuf.py b/script/api_protobuf/api_protobuf.py
index 83aee17cf9..cef4067ea3 100755
--- a/script/api_protobuf/api_protobuf.py
+++ b/script/api_protobuf/api_protobuf.py
@@ -259,12 +259,11 @@ class TypeInfo(ABC):
         if tag >= 128:
             return None
         max_val = self.max_value
-        encode_map = (
-            self.RAW_ENCODE_SMALL_MAP
-            if max_val is not None and max_val < 128
-            else self.RAW_ENCODE_MAP
-        )
-        raw_expr = encode_map.get(self.encode_func)
+        raw_expr = None
+        if max_val is not None and max_val < 128:
+            raw_expr = self.RAW_ENCODE_SMALL_MAP.get(self.encode_func)
+        if raw_expr is None:
+            raw_expr = self.RAW_ENCODE_MAP.get(self.encode_func)
         if raw_expr is None:
             return None
         return f"buffer.write_raw_byte({tag});\n{raw_expr.format(value=value_expr)}"