Isolate standalone key wallet handling

crates/aether-usage-runtime/src/settlement.rs

@@ -31,6 +31,7 @@ pub async fn settle_usage_if_needed(
         request_id: usage.request_id.clone(),
         user_id: usage.user_id.clone(),
         api_key_id: usage.api_key_id.clone(),
+        api_key_is_standalone: usage_api_key_is_standalone(usage),
         provider_id: usage.provider_id.clone(),
         status: usage.status.clone(),
         billing_status: usage.billing_status.clone(),
@@ -42,6 +43,15 @@ pub async fn settle_usage_if_needed(
     Ok(())
 }
 
+fn usage_api_key_is_standalone(usage: &StoredRequestUsageAudit) -> bool {
+    usage
+        .request_metadata
+        .as_ref()
+        .and_then(|metadata| metadata.get("api_key_is_standalone"))
+        .and_then(serde_json::Value::as_bool)
+        .unwrap_or(false)
+}
+
 fn finite_cost(value: f64) -> Result<f64, DataLayerError> {
     if value.is_finite() {
         Ok(value)
@@ -60,6 +70,7 @@ mod tests {
     use aether_data_contracts::repository::settlement::UsageSettlementInput;
     use aether_data_contracts::repository::usage::StoredRequestUsageAudit;
     use async_trait::async_trait;
+    use serde_json::json;
 
     #[derive(Default)]
     struct TestSettlementWriter {
@@ -150,6 +161,25 @@ mod tests {
         assert_eq!(inputs[0].finalized_at_unix_secs, Some(200));
         assert_eq!(inputs[0].total_cost_usd, 1.25);
         assert_eq!(inputs[0].actual_total_cost_usd, 0.75);
+        assert!(!inputs[0].api_key_is_standalone);
+    }
+
+    #[tokio::test]
+    async fn propagates_standalone_key_flag_from_usage_metadata() {
+        let writer = TestSettlementWriter {
+            has_writer: true,
+            ..Default::default()
+        };
+        let mut usage = sample_usage();
+        usage.request_metadata = Some(json!({ "api_key_is_standalone": true }));
+
+        settle_usage_if_needed(&writer, &usage)
+            .await
+            .expect("settlement should succeed");
+
+        let inputs = writer.inputs.lock().expect("settlement inputs lock");
+        assert_eq!(inputs.len(), 1);
+        assert!(inputs[0].api_key_is_standalone);
     }
 
     #[tokio::test]