refactor(libsy): share capability input and separate judge settings

crates/libsy-llm-client/tests/observability.rs

@@ -34,9 +34,10 @@ use tracing_subscriber::layer::{Context as LayerContext, SubscriberExt};
 use tracing_subscriber::registry::LookupSpan;
 
 use switchyard_libsy::{
-    Algorithm, ClassifierContractConfig, ClassifyTrigger, DeescalationConfig, Driver,
-    EscalationJudgeConfig, LibsyError, LlmClassifierConfig, LlmTaskClassifier, PickerMode,
-    RoutingOutcome, RuntimeModels, StageRouter, StageRouterConfig, Step, TaskClassifierConfig,
+    Algorithm, CapabilityJudgeConfig, ClassifierContractConfig, ClassifyTrigger,
+    DeescalationConfig, Driver, EscalationJudgeConfig, LibsyError, LlmCapabilityConfig,
+    LlmClassifierConfig, LlmTaskClassifier, PickerMode, RoutingOutcome, RuntimeModels, StageRouter,
+    StageRouterConfig, Step, TaskClassifierConfig,
 };
 use switchyard_llm_client::{ClientRouter, RunObservation, RunObserver};
 use switchyard_protocol::{Category, ModelId};
@@ -606,7 +607,10 @@ fn classifier_router() -> switchyard_libsy::Result<Arc<dyn Algorithm>> {
     Ok(Arc::new(LlmTaskClassifier::new(
         LlmClassifierConfig::Capability {
             config: TaskClassifierConfig {
-                base_threshold: 0.5,
+                judge: CapabilityJudgeConfig::Llm(LlmCapabilityConfig {
+                    base_threshold: 0.5,
+                    ..LlmCapabilityConfig::default()
+                }),
                 fail_open: false,
                 ..TaskClassifierConfig::default()
             },
@@ -869,7 +873,10 @@ async fn affinity_keeps_the_algorithm_selection_after_client_fallback()
     });
     let router = Arc::new(LlmTaskClassifier::new(LlmClassifierConfig::Capability {
         config: TaskClassifierConfig {
-            base_threshold: 0.5,
+            judge: CapabilityJudgeConfig::Llm(LlmCapabilityConfig {
+                base_threshold: 0.5,
+                ..LlmCapabilityConfig::default()
+            }),
             classify_trigger: ClassifyTrigger::NewSession,
             ..TaskClassifierConfig::default()
         },

crates/libsy/src/algorithms/composite.rs

@@ -172,6 +172,7 @@ impl Algorithm for CompositeRouter {
 
 #[cfg(test)]
 mod tests {
+    use crate::{CapabilityJudgeConfig, LlmCapabilityConfig};
     use std::collections::HashMap;
     use std::sync::Arc;
 
@@ -215,7 +216,10 @@ mod tests {
     fn hash_keyed_router() -> Result<Arc<CompositeRouter>> {
         Ok(Arc::new(CompositeRouter::new(CompositeRouterConfig {
             judge: TaskClassifierConfig {
-                base_threshold: 0.5,
+                judge: CapabilityJudgeConfig::Llm(LlmCapabilityConfig {
+                    base_threshold: 0.5,
+                    ..LlmCapabilityConfig::default()
+                }),
                 classify_trigger: ClassifyTrigger::UserTurn,
                 message_hash_fallback: true,
                 ..Default::default()
@@ -227,7 +231,10 @@ mod tests {
     fn router() -> Result<Arc<CompositeRouter>> {
         Ok(Arc::new(CompositeRouter::new(CompositeRouterConfig {
             judge: TaskClassifierConfig {
-                base_threshold: 0.5,
+                judge: CapabilityJudgeConfig::Llm(LlmCapabilityConfig {
+                    base_threshold: 0.5,
+                    ..LlmCapabilityConfig::default()
+                }),
                 classify_trigger: ClassifyTrigger::UserTurn,
                 ..Default::default()
             },