From 516d9a7c4ab2ede0260e17a6aeccb83b1177216c Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Fri, 10 Jan 2025 15:28:33 -0800 Subject: [PATCH] update prompt target name after arch_fc has identified tool --- crates/prompt_gateway/src/stream_context.rs | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/crates/prompt_gateway/src/stream_context.rs b/crates/prompt_gateway/src/stream_context.rs index b8a5007f..9782698e 100644 --- a/crates/prompt_gateway/src/stream_context.rs +++ b/crates/prompt_gateway/src/stream_context.rs @@ -263,6 +263,10 @@ impl StreamContext { ); } + // update prompt target name from the tool call + callout_context.prompt_target_name = + Some(self.tool_calls.as_ref().unwrap()[0].function.name.clone()); + self.schedule_api_call_request(callout_context); } @@ -359,8 +363,8 @@ impl StreamContext { let http_status = self .get_http_call_response_header(":status") .unwrap_or(StatusCode::OK.as_str().to_string()); - debug!("api_call_response_handler: http_status: {}", http_status); - if http_status != StatusCode::OK.as_str() { + debug!("api_call_response_handler: http_status: {}", http_status); + if http_status != StatusCode::OK.as_str() { warn!( "api server responded with non 2xx status code: {}", http_status