chore: refactor read-write lock on agent (#2225)

Co-authored-by: Alice Hau <ahau@squareup.com>
2025-04-23 23:46:22 -03:00
parent 85e2ee3984
commit 199fa6adbc
24 changed files with 409 additions and 237 deletions
@@ -110,7 +110,8 @@ async fn run_truncate_test(
        .with_temperature(Some(0.0));
    let provider = provider_type.create_provider(model_config)?;

-    let agent = Agent::new(provider);
+    let agent = Agent::new();
+    agent.update_provider(provider).await?;
    let repeat_count = context_window + 10_000;
    let large_message_content = "hello ".repeat(repeat_count);
    let messages = vec![