added vertex_ai to strip prefix

robusta-dev · Dec 22, 2024 · c47b8b7 · c47b8b7
1 parent 54aa2ba
commit c47b8b7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/holmes/core/llm.py b/holmes/core/llm.py
@@ -121,6 +121,8 @@ def _strip_model_prefix(self) -> str:
             model_name = model_name[len('openai/'):]  # Strip the 'openai/' prefix
         elif model_name.startswith('bedrock/'):
             model_name = model_name[len('bedrock/'):]  # Strip the 'bedrock/' prefix
+        elif model_name.startswith('vertex_ai/'):
+            model_name = model_name[len('vertex_ai/'):]  # Strip the 'vertex_ai/' prefix
 
         return model_name