File tree Expand file tree Collapse file tree 3 files changed +4
-5
lines changed
model_explainability/guardrails
model_serving/model_server/llmd Expand file tree Collapse file tree 3 files changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -64,7 +64,7 @@ def guardrails_orchestrator(
6464 "otlpMetricsEndpoint" : metrics_endpoint ,
6565 "otlpTracesEndpoint" : traces_endpoint ,
6666 "enableMetrics" : True ,
67- "enableTracing " : True ,
67+ "enableTraces " : True ,
6868 }
6969
7070 with GuardrailsOrchestrator (** gorch_kwargs , teardown = teardown_resources ) as gorch :
Original file line number Diff line number Diff line change @@ -390,8 +390,7 @@ def test_guardrails_traces_in_tempo(
390390
391391 @retry (wait_timeout = Timeout .TIMEOUT_1MIN , sleep = 5 )
392392 def check_traces ():
393- services = requests .get (f"{ tempo_traces_service_portforward } /api/services" ).json ().get ("data" , [])
394-
393+ services = requests .get (f"{ tempo_traces_service_portforward } /api/services" ).json ().get ("data" ) or []
395394 guardrails_services = [s for s in services if "guardrails" in s ]
396395 if not guardrails_services :
397396 return False
Original file line number Diff line number Diff line change @@ -381,10 +381,10 @@ def get_scheduler_decision_logs(
381381
382382 # Get all logs from the scheduler pod
383383 # Note: The router-scheduler container is the default/main container
384- raw_logs = router_scheduler_pod .log ()
384+ raw_logs = router_scheduler_pod .log (container = "main" )
385385
386386 # Target decision message
387- target_decision_msg = "Selecting pods from candidates sorted by max score"
387+ target_decision_msg = "Selecting endpoints from candidates sorted by max score"
388388
389389 # Filtering logs
390390 filtered_logs = "\n " .join (line for line in raw_logs .splitlines () if target_decision_msg in line )
You can’t perform that action at this time.
0 commit comments