Additional cleanup of caching

2026-06-01 15:21:03 -05:00
parent 7f039d2c08
commit 54e4e8abf5
12 changed files with 100 additions and 59 deletions
@@ -550,6 +550,7 @@ impl AgentSimulation {
                needs_create_task,
                self.conversation.clone(),
                Some(self.system_prompt.clone()),
+                None,
                self.tools.clone(),
                8192,
                None,
@@ -557,6 +558,7 @@ impl AgentSimulation {
                None,
                None,
                Arc::new(Mutex::new(Vec::new())),
+                false,
            )
            .await
            .expect("converse_stream should succeed");
@@ -1137,6 +1139,7 @@ async fn test_reasoning_model_produces_substantial_output() {
            true,
            messages,
            system,
+            None,
            agent_tools(),
            8192,
            None,
@@ -1144,6 +1147,7 @@ async fn test_reasoning_model_produces_substantial_output() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("converse_stream should succeed");
@@ -1260,6 +1264,7 @@ async fn test_event_sequence_matches_controller_expectations() {
            true,
            messages,
            None,
+            None,
            vec![],
            100,
            None,
@@ -1267,6 +1272,7 @@ async fn test_event_sequence_matches_controller_expectations() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("should connect");
@@ -1375,6 +1381,7 @@ async fn test_followup_turn_does_not_send_create_task() {
            false,
            messages,
            None,
+            None,
            vec![],
            100,
            None,
@@ -1382,6 +1389,7 @@ async fn test_followup_turn_does_not_send_create_task() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("should connect");
@@ -1459,6 +1467,7 @@ async fn run_slash_command_test(
            true,
            messages,
            system_prompt,
+            None,
            tools,
            4096,
            None,
@@ -1466,6 +1475,7 @@ async fn run_slash_command_test(
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("stream should connect");
@@ -1687,6 +1697,7 @@ async fn test_slash_resume_conversation() {
            true,
            messages,
            None,
+            None,
            vec![],
            256,
            None,
@@ -1694,6 +1705,7 @@ async fn test_slash_resume_conversation() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("stream should connect");
@@ -1799,6 +1811,7 @@ async fn test_empty_messages_safety_check() {
            true,
            messages,
            None,
+            None,
            vec![],
            100,
            None,
@@ -1806,6 +1819,7 @@ async fn test_empty_messages_safety_check() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await
        .expect("safety fallback message should work");
@@ -1960,6 +1974,7 @@ async fn test_full_proto_round_trip_with_tool_history() {
            false,
            messages,
            system_prompt,
+            None,
            tools,
            1024,
            None,
@@ -1967,6 +1982,7 @@ async fn test_full_proto_round_trip_with_tool_history() {
            None,
            None,
            Arc::new(Mutex::new(Vec::new())),
+            false,
        )
        .await;