scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 109

fix(rust): stop allocating covered_indices on the common success path

Browse source 
Signed-off-by: Jacinta Ferrant <jacinta.ferrant@gmail.com>
This commit is contained in:
Jacinta Ferrant 2026-06-18 21:44:26 -07:00
parent 2282ec12a4
commit dd79a0d991
No known key found for this signature in database
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
rust/src/pattern_prefilter.rs
View file

@ -1,3 +1,5 @@
use std::collections::HashSet;
use pyo3::pyclass;
use pyo3::pymethods;
use regex::{RegexBuilder, RegexSet, RegexSetBuilder};
@ -28,15 +30,11 @@ impl PatternPrefilter {
#[pyo3::pyfunction]
pub fn build_pattern_prefilter(patterns: Vec<String>) -> (PatternPrefilter, Vec<usize>) {
let mut covered_sources: Vec<&str> = Vec::new();
let mut covered_indices: Vec<usize> = Vec::new();
let mut rejected_indices: Vec<usize> = Vec::new();
for (index, source) in patterns.iter().enumerate() {
match RegexBuilder::new(source).case_insensitive(true).build() {
Ok(_) => {
covered_sources.push(source);
covered_indices.push(index);
}
Ok(_) => covered_sources.push(source),
Err(_) => rejected_indices.push(index),
}
}
@ -56,7 +54,8 @@ pub fn build_pattern_prefilter(patterns: Vec<String>) -> (PatternPrefilter, Vec<
rejected_indices,
),
Err(_) => {
rejected_indices.extend(covered_indices);
let already_rejected: HashSet<usize> = rejected_indices.iter().copied().collect();
rejected_indices.extend((0..patterns.len()).filter(|i| !already_rejected.contains(i)));
(PatternPrefilter { regex_set: None }, rejected_indices)
}
}