github-copilot-sdk 1.0.0-beta.4

Rust SDK for programmatic control of the GitHub Copilot CLI via JSON-RPC. Technical preview, pre-1.0.
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

use std::sync::Arc;

use github_copilot_sdk::ResumeSessionConfig;

use super::support::{DEFAULT_TEST_TOKEN, assistant_message_content, with_e2e_context};

#[tokio::test]
async fn should_suspend_idle_session_without_throwing() {
    with_e2e_context(
        "suspend",
        "should_suspend_idle_session_without_throwing",
        |ctx| {
            Box::pin(async move {
                ctx.set_default_copilot_user();
                let client = ctx.start_client().await;
                let session = client
                    .create_session(ctx.approve_all_session_config())
                    .await
                    .expect("create session");

                session
                    .send_and_wait("Reply with: SUSPEND_IDLE_OK")
                    .await
                    .expect("send");
                session.rpc().suspend().await.expect("suspend session");

                session.disconnect().await.expect("disconnect session");
                client.stop().await.expect("stop client");
            })
        },
    )
    .await;
}

#[tokio::test]
async fn should_allow_resume_and_continue_conversation_after_suspend() {
    with_e2e_context(
        "suspend",
        "should_allow_resume_and_continue_conversation_after_suspend",
        |ctx| {
            Box::pin(async move {
                ctx.set_default_copilot_user();
                let client = ctx.start_client().await;
                let session = client
                    .create_session(ctx.approve_all_session_config())
                    .await
                    .expect("create session");

                session
                    .send_and_wait(
                        "Remember the magic word: SUSPENSE. Reply with: SUSPEND_TURN_ONE",
                    )
                    .await
                    .expect("first send");
                let session_id = session.id().clone();
                session.rpc().suspend().await.expect("suspend session");
                session.disconnect().await.expect("disconnect first session");
                client.stop().await.expect("stop first client");

                let second_client = ctx.start_client().await;
                let resumed = second_client
                    .resume_session(
                        ResumeSessionConfig::new(session_id)
                            .with_github_token(DEFAULT_TEST_TOKEN)
                            .with_handler(Arc::new(
                                github_copilot_sdk::handler::ApproveAllHandler,
                            )),
                    )
                    .await
                    .expect("resume session");
                let answer = resumed
                    .send_and_wait(
                        "What was the magic word I asked you to remember? Reply with just the word.",
                    )
                    .await
                    .expect("follow-up send")
                    .expect("assistant message");
                assert!(assistant_message_content(&answer)
                    .to_lowercase()
                    .contains("suspense"));

                resumed.disconnect().await.expect("disconnect resumed");
                second_client.stop().await.expect("stop second client");
            })
        },
    )
    .await;
}