From f3238a857e2d2d356fd387ffa20ac158ed747091 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@home-assistant.io>
Date: Fri, 3 Apr 2026 12:31:01 -1000
Subject: [PATCH] [api] Select encode map before lookup

---
 script/api_protobuf/api_protobuf.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/script/api_protobuf/api_protobuf.py b/script/api_protobuf/api_protobuf.py
index 1381d5586f..83aee17cf9 100755
--- a/script/api_protobuf/api_protobuf.py
+++ b/script/api_protobuf/api_protobuf.py
@@ -259,11 +259,12 @@ class TypeInfo(ABC):
         if tag >= 128:
             return None
         max_val = self.max_value
-        raw_expr = (
-            self.RAW_ENCODE_SMALL_MAP.get(self.encode_func)
+        encode_map = (
+            self.RAW_ENCODE_SMALL_MAP
             if max_val is not None and max_val < 128
-            else self.RAW_ENCODE_MAP.get(self.encode_func)
+            else self.RAW_ENCODE_MAP
         )
+        raw_expr = 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)}"