From 80ce2152933dc8b2992c1caa96808b5e84f696a0 Mon Sep 17 00:00:00 2001 From: Adil Hafeez Date: Tue, 24 Jun 2025 13:40:01 -0700 Subject: [PATCH] add comments --- arch/envoy.template.yaml | 2 +- arch/tools/cli/docker_cli.py | 1 - crates/brightstaff/src/handlers/preferences.rs | 4 +++- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/arch/envoy.template.yaml b/arch/envoy.template.yaml index ae41f6c7..53044770 100644 --- a/arch/envoy.template.yaml +++ b/arch/envoy.template.yaml @@ -802,7 +802,7 @@ static_resources: - endpoint: address: socket_address: - address: host.docker.internal + address: 0.0.0.0 port_value: 9091 hostname: localhost diff --git a/arch/tools/cli/docker_cli.py b/arch/tools/cli/docker_cli.py index 1d87fc67..e8a12a13 100644 --- a/arch/tools/cli/docker_cli.py +++ b/arch/tools/cli/docker_cli.py @@ -59,7 +59,6 @@ def docker_start_archgw_detached( volume_mappings = [ f"{arch_config_file}:/app/arch_config.yaml:ro", - # "/Users/adilhafeez/src/intelligent-prompt-gateway/crates/target/wasm32-wasip1/release:/etc/envoy/proxy-wasm-plugins:ro", ] volume_mappings_args = [ item for volume in volume_mappings for item in ("-v", volume) diff --git a/crates/brightstaff/src/handlers/preferences.rs b/crates/brightstaff/src/handlers/preferences.rs index 4a478ad9..8e6843c0 100644 --- a/crates/brightstaff/src/handlers/preferences.rs +++ b/crates/brightstaff/src/handlers/preferences.rs @@ -17,6 +17,7 @@ pub async fn list_preferences( llm_providers: Arc>>, ) -> Response> { let prov = llm_providers.read().await; + // select providers that have usage information let providers_with_usage = prov .iter() .filter(|provider| provider.usage.is_some()) @@ -56,7 +57,6 @@ pub async fn update_preferences( request: Request, llm_providers: Arc>>, ) -> Result>, hyper::Error> { - info!("Updating preferences..."); let request_body = request.collect().await?.to_bytes(); let usage: Vec = match serde_json::from_slice(&request_body) { @@ -76,6 +76,8 @@ pub async fn update_preferences( let usage_model_map: HashMap = usage.into_iter().map(|u| (u.model.clone(), u)).collect(); + info!("Updating usage preferences for models: {:?}", usage_model_map.keys()); + let mut llm_providers = llm_providers.write().await; // ensure that models coming in the request are valid