Do not override the container if it's missing from policy config

PiperOrigin-RevId: 653392239
google · Jul 17, 2024 · 995243c · 995243c
1 parent d82820c
commit 995243c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java b/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java
@@ -127,6 +127,10 @@ public Cel extend(Cel cel, CelOptions celOptions) throws CelPolicyValidationExce
                       .map(f -> f.toCelFunctionDecl(celTypeProvider))
                       .collect(toImmutableList()));
 
+      if (!container().isEmpty()) {
+        celBuilder.setContainer(container());
+      }
+
       addAllExtensions(celBuilder, celOptions);
 
       return celBuilder.build();