apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-21 13:55:15 +02:00
Code Issues Projects Releases Packages Wiki Activity

more cleanups

Browse source
This commit is contained in:
Adil Hafeez 2026-02-10 00:54:10 -08:00
parent 3fdd8a3a35
commit ebe0e18e3b
1 changed files with 11 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

33
crates/brightstaff/src/main.rs
View file

@ -17,12 +17,13 @@ use common::consts::{
use common::llm_providers::LlmProviders; use common::llm_providers::LlmProviders;
use http_body_util::{combinators::BoxBody, BodyExt, Empty}; use http_body_util::{combinators::BoxBody, BodyExt, Empty};
use hyper::body::Incoming; use hyper::body::Incoming;
use hyper::header::HeaderValue;
use hyper::server::conn::http1; use hyper::server::conn::http1;
use hyper::service::service_fn; use hyper::service::service_fn;
use hyper::{Method, Request, Response, StatusCode}; use hyper::{Method, Request, Response, StatusCode};
use hyper_util::rt::TokioIo; use hyper_util::rt::TokioIo;
use opentelemetry::global;
use opentelemetry::trace::FutureExt; use opentelemetry::trace::FutureExt;
use opentelemetry::{global, Context};
use opentelemetry_http::HeaderExtractor; use opentelemetry_http::HeaderExtractor;
use std::sync::Arc; use std::sync::Arc;
use std::{env, fs}; use std::{env, fs};
@ -38,13 +39,6 @@ const DEFAULT_ROUTING_MODEL_NAME: &str = "Arch-Router";
// Helpers // Helpers
// --------------------------------------------------------------------------- // ---------------------------------------------------------------------------
/// Extract the OpenTelemetry context propagated via HTTP headers.
fn extract_context_from_request(req: &Request<Incoming>) -> Context {
global::get_text_map_propagator(|propagator| {
propagator.extract(&HeaderExtractor(req.headers()))
})
}
/// An empty HTTP body (used for 404 / OPTIONS responses). /// An empty HTTP body (used for 404 / OPTIONS responses).
fn empty() -> BoxBody<Bytes, hyper::Error> { fn empty() -> BoxBody<Bytes, hyper::Error> {
Empty::<Bytes>::new() Empty::<Bytes>::new()
@ -56,18 +50,18 @@ fn empty() -> BoxBody<Bytes, hyper::Error> {
fn cors_preflight() -> Result<Response<BoxBody<Bytes, hyper::Error>>, hyper::Error> { fn cors_preflight() -> Result<Response<BoxBody<Bytes, hyper::Error>>, hyper::Error> {
let mut response = Response::new(empty()); let mut response = Response::new(empty());
*response.status_mut() = StatusCode::NO_CONTENT; *response.status_mut() = StatusCode::NO_CONTENT;
let headers = response.headers_mut(); let h = response.headers_mut();
headers.insert("Allow", "GET, OPTIONS".parse().unwrap()); h.insert("Allow", HeaderValue::from_static("GET, OPTIONS"));
headers.insert("Access-Control-Allow-Origin", "*".parse().unwrap()); h.insert("Access-Control-Allow-Origin", HeaderValue::from_static("*"));
headers.insert( h.insert(
"Access-Control-Allow-Headers", "Access-Control-Allow-Headers",
"Authorization, Content-Type".parse().unwrap(), HeaderValue::from_static("Authorization, Content-Type"),
); );
headers.insert( h.insert(
"Access-Control-Allow-Methods", "Access-Control-Allow-Methods",
"GET, POST, OPTIONS".parse().unwrap(), HeaderValue::from_static("GET, POST, OPTIONS"),
); );
headers.insert("Content-Type", "application/json".parse().unwrap()); h.insert("Content-Type", HeaderValue::from_static("application/json"));
Ok(response) Ok(response)
} }
@ -203,7 +197,7 @@ async fn route(
req: Request<Incoming>, req: Request<Incoming>,
state: Arc<AppState>, state: Arc<AppState>,
) -> Result<Response<BoxBody<Bytes, hyper::Error>>, hyper::Error> { ) -> Result<Response<BoxBody<Bytes, hyper::Error>>, hyper::Error> {
let parent_cx = extract_context_from_request(&req); let parent_cx = global::get_text_map_propagator(|p| p.extract(&HeaderExtractor(req.headers())));
let path = req.uri().path().to_string(); let path = req.uri().path().to_string();
// --- Agent routes (/agents/...) --- // --- Agent routes (/agents/...) ---
@ -312,11 +306,6 @@ async fn run_server(state: Arc<AppState>) -> Result<(), Box<dyn std::error::Erro
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> { async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
let config = load_config()?; let config = load_config()?;
let _tracer_provider = init_tracer(config.tracing.as_ref()); let _tracer_provider = init_tracer(config.tracing.as_ref());
let arch_config_path = env::var("ARCH_CONFIG_PATH_RENDERED")
.unwrap_or_else(|_| "./arch_config_rendered.yaml".to_string());
info!(path = %arch_config_path, "loaded arch_config.yaml");
let state = Arc::new(init_app_state(&config).await?); let state = Arc::new(init_app_state(&config).await?);
run_server(state).await run_server(state).await
} }