Skip to content

Commit 4cdf4b1

Browse files
committed
refactor(relay): use route error summary methods
Signed-off-by: Bryan Bednarski <bbednarski@nvidia.com>
1 parent 4541efd commit 4cdf4b1

1 file changed

Lines changed: 4 additions & 8 deletions

File tree

crates/switchyard-nemo-relay-plugin/src/runtime.rs

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ use nemo_relay_plugin::{Json, LlmRequest as RelayRequest, PluginRuntime};
99
use serde_json::{Map, json};
1010
use switchyard_llm_client::{LlmCallObservation, RunObservation, RunObserver};
1111
use switchyard_protocol::{LlmResponse, Metadata, Request, Response, WireFormat};
12-
use switchyard_runner::{Route, RouteErrorSummary, Runner, stream_failure_summary};
12+
use switchyard_runner::{Route, RouteErrorSummary, Runner, stream_error_summary};
1313
use switchyard_translation::{TranslationEngine, encode_stream};
1414

1515
use crate::config::SwitchyardConfig;
@@ -170,11 +170,7 @@ impl SwitchyardRuntime {
170170
}
171171
Err(error) => {
172172
self.emit_observations(&mut marks, take_observations(&observations), &metadata);
173-
self.route_execution_error_mark(
174-
&mut marks,
175-
&error.execution_failure_summary(),
176-
None,
177-
);
173+
self.route_execution_error_mark(&mut marks, &error.execution_error_summary(), None);
178174
Execution {
179175
result: Err("Switchyard route execution failed".into()),
180176
marks,
@@ -326,7 +322,7 @@ fn returned_events(
326322
let chunks = Box::pin(chunks.map(move |item| {
327323
if let Err(error) = &item {
328324
emit_mark(route_execution_error_mark(
329-
&stream_failure_summary(error, served_model.as_ref()),
325+
&stream_error_summary(error, served_model.as_ref()),
330326
metadata.clone(),
331327
));
332328
}
@@ -434,7 +430,7 @@ mod tests {
434430
});
435431

436432
let mark = route_execution_error_mark(
437-
&error.execution_failure_summary(),
433+
&error.execution_error_summary(),
438434
json!({"session_id": "session"}),
439435
);
440436

0 commit comments

Comments
 (0)