From 6eb11f76ede7bedbc5415aee39778f3e51a56b34 Mon Sep 17 00:00:00 2001
From: Sokwhan Huh <sokwhan@google.com>
Date: Mon, 22 Jun 2026 20:20:47 -0700
Subject: [PATCH] Internal Changes

PiperOrigin-RevId: 936393853
---
 .../src/main/java/dev/cel/common/CelAbstractSyntaxTree.java  | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/common/src/main/java/dev/cel/common/CelAbstractSyntaxTree.java b/common/src/main/java/dev/cel/common/CelAbstractSyntaxTree.java
index 6b3b6a74f..b79c67e79 100644
--- a/common/src/main/java/dev/cel/common/CelAbstractSyntaxTree.java
+++ b/common/src/main/java/dev/cel/common/CelAbstractSyntaxTree.java
@@ -103,6 +103,11 @@ public Optional<CelType> getType(long exprId) {
     return Optional.ofNullable(types().get(exprId));
   }
 
+  public CelType getTypeOrThrow(long exprId) {
+    return getType(exprId)
+        .orElseThrow(() -> new NoSuchElementException("Type not found for expr id: " + exprId));
+  }
+
   public ImmutableMap<Long, CelType> getTypeMap() {
     return types();
   }