1001. 1001 assert_eq!(classify_cache_rung(0, 0, &usage), CacheBehaviour::None);
1002. 1002 }
1003. 1003
1004. 1004 #[test]
1005. 1005 fn followed_detects_probe_ack_tool_use_only() {
1006. 1006 let with_ack = vak_llm::AssistantMessage {
1007. 1007 content: vec![vak_llm::ContentBlock::ToolUse {
1008. 1008 id: "1".into(),
1009. 1009 name: "probe_ack".into(),
1010. 1010 input: serde_json::json!({"ok": true}),
1011. 1011 }],
1012. 1012 stop_reason: vak_llm::StopReason::ToolUse,
1013. 1013 usage: vak_llm::Usage::default(),
1014. 1014 model: "m".into(),
1015. 1015 response_id: None,
1016. 1016 };
1017. 1017 assert!(followed(&with_ack));
1018. 1018
1019. 1019 let prose_only = vak_llm::AssistantMessage {
1020. 1020 content: vec![vak_llm::ContentBlock::text("sure, here you go")],
1021. 1021 stop_reason: vak_llm::StopReason::EndTurn,
1022. 1022 usage: vak_llm::Usage::default(),
1023. 1023 model: "m".into(),
1024. 1024 response_id: None,
1025. 1025 };
1026. 1026 assert!(!followed(&prose_only));
1027. 1027 }
1028. 1028
1029. 1029 fn flat_profile(horizon_tokens: u64) -> CapacityProfile {
1030. 1030 flat_profile_with_reserve(horizon_tokens, 1_024)
1031. 1031 }
1032. 1032
1033. 1033 fn flat_profile_with_reserve(horizon_tokens: u64, output_reserve: u64) -> CapacityProfile {
1034. 1034 CapacityProfile::from_probe(
1035. 1035 horizon_tokens,
1036. 1036 None,
1037. 1037 Horizon {
1038. 1038 tokens: horizon_tokens,
1039. 1039 confidence: PROBED_CONFIDENCE,
1040. 1040 last_confirmed: SystemTime::now(),
1041. 1041 },
1042. 1042 CacheBehaviour::Unknown,
1043. 1043 output_reserve,
1044. 1044 ProbeProvenance {
1045. 1045 probed_at: SystemTime::now(),
1046. 1046 rungs: Vec::new(),
1047. 1047 signals: Vec::new(),
1048. 1048 metadata_digest: "digest".into(),
1049. 1049 quantisation: None,
1050. 1050 },
1051. 1051 )
1052. 1052 }
1053. 1053}
1054. 1054