Adding support for pheye medical conditions filter. (#287)

Author: jzonthemtn

src/main/java/ai/philterd/phileas/model/filtering/FilterType.java

@@ -52,7 +52,11 @@ public enum FilterType {
     URL("url", true),
     VIN("vin", true),
     ZIP_CODE("zip-code", true),
-    CUSTOM_DICTIONARY("custom-dictionary", false);
+    CUSTOM_DICTIONARY("custom-dictionary", false),
+
+    // These are not natively identified by Phileas.
+    MEDICAL_CONDITION("medical-condition", false),
+    OTHER("other", false);
 
     private final String type;
     private final boolean deterministic;

src/main/java/ai/philterd/phileas/policy/Identifiers.java

@@ -34,6 +34,7 @@
 import ai.philterd.phileas.policy.filters.Identifier;
 import ai.philterd.phileas.policy.filters.IpAddress;
 import ai.philterd.phileas.policy.filters.MacAddress;
+import ai.philterd.phileas.policy.filters.MedicalCondition;
 import ai.philterd.phileas.policy.filters.PassportNumber;
 import ai.philterd.phileas.policy.filters.PhEye;
 import ai.philterd.phileas.policy.filters.PhoneNumber;
@@ -161,6 +162,10 @@ public class Identifiers {
     @Expose
     private ZipCode zipCode;
 
+    @SerializedName("medicalCondition")
+    @Expose
+    private MedicalCondition medicalCondition;
+
     @SerializedName("city")
     @Expose
     private City city;
@@ -239,6 +244,8 @@ public boolean hasFilter(FilterType filterType) {
                 if(this.getIpAddress() != null) { return true; } break;
             case MAC_ADDRESS:
                 if(this.getMacAddress() != null) { return true; } break;
+            case MEDICAL_CONDITION:
+                if(this.getMedicalCondition() != null) {return true; } break;
             case PERSON:
                 if(this.getPhEye() != null) { return true; } break;
             case PASSPORT_NUMBER:
@@ -725,5 +732,13 @@ public Currency getCurrency() {
     public void setCurrency(Currency currency) {
         this.currency = currency;
     }
-    
+
+    public MedicalCondition getMedicalCondition() {
+        return medicalCondition;
+    }
+
+    public void setMedicalCondition(MedicalCondition medicalCondition) {
+        this.medicalCondition = medicalCondition;
+    }
+
 }

src/main/java/ai/philterd/phileas/policy/filters/MedicalCondition.java

@@ -0,0 +1,38 @@
+/*
+ *     Copyright 2025 Philterd, LLC @ https://www.philterd.ai
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *          http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package ai.philterd.phileas.policy.filters;
+
+import ai.philterd.phileas.services.strategies.rules.VinFilterStrategy;
+import com.google.gson.annotations.Expose;
+import com.google.gson.annotations.SerializedName;
+
+import java.util.List;
+
+public class MedicalCondition extends AbstractFilter {
+
+    @SerializedName("medicalConditionFilterStrategies")
+    @Expose
+    private List<VinFilterStrategy> medicalConditionFilterStrategies;
+
+    public List<VinFilterStrategy> getMedicalConditionFilterStrategies() {
+        return medicalConditionFilterStrategies;
+    }
+
+    public void setMedicalConditionFilterStrategies(List<VinFilterStrategy> medicalConditionFilterStrategies) {
+        this.medicalConditionFilterStrategies = medicalConditionFilterStrategies;
+    }
+
+}

src/main/java/ai/philterd/phileas/services/FilterPolicyLoader.java

@@ -1235,6 +1235,46 @@ public List<Filter> getFiltersForPolicy(final Policy policy, final Map<String, M
 
         }
 
+        if(policy.getIdentifiers().hasFilter(FilterType.MEDICAL_CONDITION) && policy.getIdentifiers().getPhEye().isEnabled()) {
+
+            if(cache.containsKey(FilterType.MEDICAL_CONDITION)) {
+                enabledFilters.add(cache.get(FilterType.MEDICAL_CONDITION));
+            } else {
+
+                final int windowSize = policy.getIdentifiers().getPhEye().getWindowSizeOrDefault(phileasConfiguration.spanWindowSize());
+
+                final FilterConfiguration filterConfiguration = new FilterConfiguration.FilterConfigurationBuilder()
+                        .withStrategies(policy.getIdentifiers().getPhEye().getPhEyeFilterStrategies())
+                        .withAnonymizationService(new AlphanumericAnonymizationService(contextService, random))
+                        .withIgnored(policy.getIdentifiers().getPhEye().getIgnored())
+                        .withIgnoredFiles(policy.getIdentifiers().getPhEye().getIgnoredFiles())
+                        .withIgnoredPatterns(policy.getIdentifiers().getPhEye().getIgnoredPatterns())
+                        .withCrypto(policy.getCrypto())
+                        .withFPE(policy.getFpe())
+                        .withWindowSize(windowSize)
+                        .withPriority(policy.getIdentifiers().getPhEye().getPriority())
+                        .build();
+
+                final PhEyeConfiguration phEyeConfiguration = new PhEyeConfiguration(policy.getIdentifiers().getPhEye().getPhEyeConfiguration().getEndpoint());
+                phEyeConfiguration.setTimeout(policy.getIdentifiers().getPhEye().getPhEyeConfiguration().getTimeout());
+                phEyeConfiguration.setMaxIdleConnections(policy.getIdentifiers().getPhEye().getPhEyeConfiguration().getMaxIdleConnections());
+                phEyeConfiguration.setBearerToken(policy.getIdentifiers().getPhEye().getPhEyeConfiguration().getBearerToken());
+
+                final Filter filter = new PhEyeFilter(
+                        filterConfiguration,
+                        phEyeConfiguration,
+                        policy.getIdentifiers().getPhEye().isRemovePunctuation(),
+                        policy.getIdentifiers().getPhEye().getThresholds(),
+                        httpClient
+                );
+
+                enabledFilters.add(filter);
+                filterCache.get(policy.getName()).put(FilterType.MEDICAL_CONDITION, filter);
+
+            }
+
+        }
+
         return enabledFilters;
 
     }

src/main/java/ai/philterd/phileas/services/filters/ai/pheye/PhEyeFilter.java

@@ -157,8 +157,15 @@ public Filtered filter(final Policy policy, final String context, final int piec
                             // Get the window of text surrounding the token.
                             final String[] window = getWindow(formattedInput, phEyeSpan.getStart(), phEyeSpan.getEnd());
 
-                            // Currently only PERSON type is supported.
-                            final FilterType filterType = FilterType.PERSON;
+                            // Set the filter type based on the entity's type that's returned from pheye.
+                            final FilterType filterType;
+                            if(phEyeSpan.getLabel().equalsIgnoreCase("PERSON")) {
+                                filterType = FilterType.PERSON;
+                            } else if(phEyeSpan.getLabel().equalsIgnoreCase("DISEASE_DISORDER")) {
+                                filterType = FilterType.MEDICAL_CONDITION;
+                            } else {
+                                filterType = FilterType.OTHER;
+                            }
 
                             final Span span = createSpan(policy, context, filterType, phEyeSpan.getText(),
                                     window, phEyeSpan.getLabel(), phEyeSpan.getStart(), phEyeSpan.getEnd(),