feat: Support 'or' logic in context filtering

Author: ayushjain17

crates/superposition_types/src/logic.rs

@@ -10,6 +10,7 @@ use crate::{
     DimensionInfo,
 };
 
+#[inline]
 fn apply_logic(
     condition: &Map<String, Value>,
     context: &Map<String, Value>,
@@ -37,15 +38,42 @@ fn apply_logic(
     true
 }
 
+/// Core context application logic - checks if all dimensions in condition are satisfied by context
+/// Only exact matches are considered valid, except for "variantIds" dimension where containment is checked
+/// Returns true if condition is satisfied by context, false otherwise
 pub fn apply(condition: &Map<String, Value>, context: &Map<String, Value>) -> bool {
     apply_logic(condition, context, false)
 }
 
+/// Filtering logic that allows partial matching of context
+/// For dimensions present in context, performs the same checks as `apply`
+/// For dimensions absent in context, skips the check (allows partial matching)
+/// This is useful for matching contexts that may not have all dimensions defined
+/// For array context values, checks for containment - added behaviour over `apply`
+/// Returns false if there is a mismatch, true otherwise
 pub fn partial_apply(
     condition: &Map<String, Value>,
     context: &Map<String, Value>,
 ) -> bool {
-    apply_logic(condition, context, true)
+    for (dimension, value) in condition {
+        let Some(context_value) = context.get(dimension) else {
+            continue; // Skip dimensions not in context (partial matching)
+        };
+
+        // For array context values, check containment
+        if let Value::Array(context_values) = context_value {
+            if !context_values.contains(value) {
+                return false;
+            }
+        } else if dimension == "variantIds" {
+            // variantIds must always be an array - fail if scalar
+            return false;
+        } else if *context_value != *value {
+            // For non-array values, check equality
+            return false;
+        }
+    }
+    true
 }
 
 fn _evaluate_local_cohort_dimension(