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
89
90
91
|
import AbstractContinueGUIClientProtocol from "./ContinueGUIClientProtocol";
// import { Messenger, WebsocketMessenger } from "../../../src/util/messenger";
import { Messenger, WebsocketMessenger } from "./messenger";
import { VscodeMessenger } from "./vscodeMessenger";
class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
messenger: Messenger;
// Server URL must contain the session ID param
serverUrlWithSessionId: string;
constructor(
serverUrlWithSessionId: string,
useVscodeMessagePassing: boolean
) {
super();
this.serverUrlWithSessionId = serverUrlWithSessionId;
if (useVscodeMessagePassing) {
this.messenger = new VscodeMessenger(serverUrlWithSessionId);
} else {
this.messenger = new WebsocketMessenger(serverUrlWithSessionId);
}
}
sendMainInput(input: string) {
this.messenger.send("main_input", { input });
}
reverseToIndex(index: number) {
this.messenger.send("reverse_to_index", { index });
}
sendRefinementInput(input: string, index: number) {
this.messenger.send("refinement_input", { input, index });
}
sendStepUserInput(input: string, index: number) {
this.messenger.send("step_user_input", { input, index });
}
onStateUpdate(callback: (state: any) => void) {
this.messenger.onMessageType("state_update", (data: any) => {
if (data.state) {
callback(data.state);
}
});
}
onAvailableSlashCommands(
callback: (commands: { name: string; description: string }[]) => void
) {
this.messenger.onMessageType("available_slash_commands", (data: any) => {
if (data.commands) {
callback(data.commands);
}
});
}
changeDefaultModel(model: string) {
this.messenger.send("change_default_model", { model });
}
sendClear() {
this.messenger.send("clear_history", {});
}
retryAtIndex(index: number) {
this.messenger.send("retry_at_index", { index });
}
deleteAtIndex(index: number) {
this.messenger.send("delete_at_index", { index });
}
deleteContextAtIndices(indices: number[]) {
this.messenger.send("delete_context_at_indices", { indices });
}
setEditingAtIndices(indices: number[]) {
this.messenger.send("set_editing_at_indices", { indices });
}
setPinnedAtIndices(indices: number[]) {
this.messenger.send("set_pinned_at_indices", { indices });
}
toggleAddingHighlightedCode(): void {
this.messenger.send("toggle_adding_highlighted_code", {});
}
}
export default ContinueGUIClientProtocol;
|
