apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-04-30 11:26:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: add missing ResponseHandler methods and use BrightStaffError

Browse source
This commit is contained in:
Adil Hafeez 2026-03-06 14:10:35 +00:00
parent 66e55e1621
commit 96a2ebf37a
2 changed files with 35 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
crates/brightstaff/src/handlers/agents/orchestrator.rs
View file

@ -29,7 +29,7 @@ pub enum AgentFilterChainError {
#[error("Pipeline processing error: {0}")]
Pipeline(#[from] PipelineError),
#[error("Response handling error: {0}")]
Response(#[from] crate::handlers::response::ResponseError),
Response(#[from] common::errors::BrightStaffError),
#[error("Request parsing error: {0}")]
RequestParsing(#[from] serde_json::Error),
#[error("HTTP error: {0}")]

35
crates/brightstaff/src/handlers/response.rs
View file

@ -27,6 +27,39 @@ impl ResponseHandler {
.boxed()
}
/// Create a JSON error response with BAD_REQUEST status
pub fn create_json_error_response(
json: &serde_json::Value,
) -> Response<BoxBody<Bytes, hyper::Error>> {
let body = Self::create_full_body(json.to_string());
let mut response = Response::new(body);
*response.status_mut() = hyper::StatusCode::BAD_REQUEST;
response.headers_mut().insert(
hyper::header::CONTENT_TYPE,
"application/json".parse().unwrap(),
);
response
}
/// Create a BAD_REQUEST error response with a message
pub fn create_bad_request(message: &str) -> Response<BoxBody<Bytes, hyper::Error>> {
let json = serde_json::json!({"error": message});
Self::create_json_error_response(&json)
}
/// Create an INTERNAL_SERVER_ERROR response with a message
pub fn create_internal_error(message: &str) -> Response<BoxBody<Bytes, hyper::Error>> {
let json = serde_json::json!({"error": message});
let body = Self::create_full_body(json.to_string());
let mut response = Response::new(body);
*response.status_mut() = hyper::StatusCode::INTERNAL_SERVER_ERROR;
response.headers_mut().insert(
hyper::header::CONTENT_TYPE,
"application/json".parse().unwrap(),
);
response
}
/// Create a streaming response from a reqwest response.
/// The spawned streaming task is instrumented with both `agent_span` and `orchestrator_span`
/// so their durations reflect the actual time spent streaming to the client.
@ -113,7 +146,7 @@ impl ResponseHandler {
SupportedUpstreamAPIs::OpenAIChatCompletions(OpenAIApi::ChatCompletions);
let sse_iter = SseStreamIter::try_from(response_bytes.as_ref()).map_err(|e| {
ResponseError::StreamError(format!("Failed to parse SSE stream: {}", e))
BrightStaffError::StreamError(format!("Failed to parse SSE stream: {}", e))
})?;
let mut accumulated_text = String::new();