codetether-agent 4.7.0-a-002.4

A2A-native AI coding agent for the CodeTether ecosystem
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

//! Tests for streaming prompt event behavior.

use super::prompt_events::run_prompt_with_events;
use super::prompt_events_test_provider::StreamContextErrorProvider;
use crate::provider::{ContentPart, Message, ProviderRegistry, Role};
use crate::session::Session;
use std::sync::Arc;
use tokio::sync::mpsc;

#[tokio::test]
async fn streaming_context_errors_retry_with_forced_compaction() {
    let provider = Arc::new(StreamContextErrorProvider::new());
    let mut registry = ProviderRegistry::new();
    registry.register(provider.clone());
    let registry = Arc::new(registry);
    let (tx, mut rx) = mpsc::channel(64);
    tokio::spawn(async move { while rx.recv().await.is_some() {} });
    let mut session = Session::new().await.expect("session");
    session.metadata.model = Some("mock/test".into());
    for _ in 0..7 {
        session.add_message(text_message());
    }

    let result = run_prompt_with_events(&mut session, "hello", Vec::new(), tx, registry).await;

    assert!(result.is_ok());
    assert_eq!(provider.calls(), 2);
}

fn text_message() -> Message {
    Message {
        role: Role::User,
        content: vec![ContentPart::Text { text: "old".into() }],
    }
}