@claude-studio/server 0.1.3 → 0.3.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/dist/claude-session.d.ts
CHANGED
|
@@ -1,12 +1,34 @@
|
|
|
1
1
|
import type { ElementSelection } from "@claude-studio/shared";
|
|
2
|
-
interface
|
|
2
|
+
export interface CompletionData {
|
|
3
|
+
result: string;
|
|
4
|
+
sessionId: string;
|
|
5
|
+
cost: number;
|
|
6
|
+
turns: number;
|
|
7
|
+
usage: {
|
|
8
|
+
input_tokens: number;
|
|
9
|
+
output_tokens: number;
|
|
10
|
+
cache_read_input_tokens: number;
|
|
11
|
+
cache_creation_input_tokens: number;
|
|
12
|
+
};
|
|
13
|
+
duration_ms: number;
|
|
14
|
+
model: string;
|
|
15
|
+
}
|
|
16
|
+
export interface StreamCallbacks {
|
|
3
17
|
onStreaming: (chunk: string) => void;
|
|
4
18
|
onToolUse: (tool: string, input: Record<string, unknown>) => void;
|
|
5
|
-
onComplete: (
|
|
19
|
+
onComplete: (data: CompletionData) => void;
|
|
6
20
|
onError: (error: string) => void;
|
|
7
21
|
}
|
|
22
|
+
interface SessionStats {
|
|
23
|
+
totalCost: number;
|
|
24
|
+
totalInputTokens: number;
|
|
25
|
+
totalOutputTokens: number;
|
|
26
|
+
turnCount: number;
|
|
27
|
+
}
|
|
8
28
|
export declare class ClaudeSessionManager {
|
|
9
29
|
private sessions;
|
|
30
|
+
private sessionStats;
|
|
31
|
+
private cachedCommands;
|
|
10
32
|
executePrompt(clientId: string, params: {
|
|
11
33
|
route: string;
|
|
12
34
|
element: ElementSelection;
|
|
@@ -14,6 +36,14 @@ export declare class ClaudeSessionManager {
|
|
|
14
36
|
}, callbacks: StreamCallbacks): Promise<void>;
|
|
15
37
|
executeRawPrompt(clientId: string, prompt: string, callbacks: StreamCallbacks): Promise<void>;
|
|
16
38
|
resetSession(clientId: string): void;
|
|
39
|
+
getSessionStats(clientId: string): SessionStats | null;
|
|
40
|
+
getCapabilities(): {
|
|
41
|
+
commands: Array<{
|
|
42
|
+
name: string;
|
|
43
|
+
description: string;
|
|
44
|
+
argumentHint: string;
|
|
45
|
+
}>;
|
|
46
|
+
};
|
|
17
47
|
private runQuery;
|
|
18
48
|
}
|
|
19
49
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"claude-session.d.ts","sourceRoot":"","sources":["../src/claude-session.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAK7D,
|
|
1
|
+
{"version":3,"file":"claude-session.d.ts","sourceRoot":"","sources":["../src/claude-session.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAA;AAK7D,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAA;IACd,SAAS,EAAE,MAAM,CAAA;IACjB,IAAI,EAAE,MAAM,CAAA;IACZ,KAAK,EAAE,MAAM,CAAA;IACb,KAAK,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,uBAAuB,EAAE,MAAM,CAAC;QAAC,2BAA2B,EAAE,MAAM,CAAA;KAAE,CAAA;IAC5H,WAAW,EAAE,MAAM,CAAA;IACnB,KAAK,EAAE,MAAM,CAAA;CACd;AAED,MAAM,WAAW,eAAe;IAC9B,WAAW,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAA;IACpC,SAAS,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAA;IACjE,UAAU,EAAE,CAAC,IAAI,EAAE,cAAc,KAAK,IAAI,CAAA;IAC1C,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAA;CACjC;AAED,UAAU,YAAY;IACpB,SAAS,EAAE,MAAM,CAAA;IACjB,gBAAgB,EAAE,MAAM,CAAA;IACxB,iBAAiB,EAAE,MAAM,CAAA;IACzB,SAAS,EAAE,MAAM,CAAA;CAClB;AAED,qBAAa,oBAAoB;IAC/B,OAAO,CAAC,QAAQ,CAA4B;IAC5C,OAAO,CAAC,YAAY,CAAkC;IACtD,OAAO,CAAC,cAAc,CAAyE;IAEzF,aAAa,CACjB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,gBAAgB,CAAC;QAAC,MAAM,EAAE,MAAM,CAAA;KAAE,EACpE,SAAS,EAAE,eAAe,GACzB,OAAO,CAAC,IAAI,CAAC;IAKV,gBAAgB,CACpB,QAAQ,EAAE,MAAM,EAChB,MAAM,EAAE,MAAM,EACd,SAAS,EAAE,eAAe,GACzB,OAAO,CAAC,IAAI,CAAC;IAIhB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAKpC,eAAe,CAAC,QAAQ,EAAE,MAAM,GAAG,YAAY,GAAG,IAAI;IAItD,eAAe,IAAI;QAAE,QAAQ,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,WAAW,EAAE,MAAM,CAAC;YAAC,YAAY,EAAE,MAAM,CAAA;SAAE,CAAC,CAAA;KAAE;YAIrF,QAAQ;CA6FvB"}
|
package/dist/claude-session.js
CHANGED
|
@@ -4,6 +4,8 @@ import { config } from "./config.js";
|
|
|
4
4
|
import { log } from "./logger.js";
|
|
5
5
|
export class ClaudeSessionManager {
|
|
6
6
|
sessions = new Map(); // clientId → sessionId
|
|
7
|
+
sessionStats = new Map();
|
|
8
|
+
cachedCommands = [];
|
|
7
9
|
async executePrompt(clientId, params, callbacks) {
|
|
8
10
|
const fullPrompt = buildPrompt(params);
|
|
9
11
|
await this.runQuery(clientId, fullPrompt, callbacks);
|
|
@@ -13,6 +15,13 @@ export class ClaudeSessionManager {
|
|
|
13
15
|
}
|
|
14
16
|
resetSession(clientId) {
|
|
15
17
|
this.sessions.delete(clientId);
|
|
18
|
+
this.sessionStats.delete(clientId);
|
|
19
|
+
}
|
|
20
|
+
getSessionStats(clientId) {
|
|
21
|
+
return this.sessionStats.get(clientId) || null;
|
|
22
|
+
}
|
|
23
|
+
getCapabilities() {
|
|
24
|
+
return { commands: this.cachedCommands };
|
|
16
25
|
}
|
|
17
26
|
async runQuery(clientId, prompt, callbacks) {
|
|
18
27
|
const options = {
|
|
@@ -37,6 +46,18 @@ export class ClaudeSessionManager {
|
|
|
37
46
|
callbacks.onError("Session timed out after 5 minutes");
|
|
38
47
|
break;
|
|
39
48
|
}
|
|
49
|
+
// Capture slash commands from SDK system init message
|
|
50
|
+
if (msg.type === "system" && msg.subtype === "init") {
|
|
51
|
+
const init = msg;
|
|
52
|
+
if (this.cachedCommands.length === 0 && Array.isArray(init.slash_commands)) {
|
|
53
|
+
this.cachedCommands = init.slash_commands.map((name) => ({
|
|
54
|
+
name,
|
|
55
|
+
description: "",
|
|
56
|
+
argumentHint: "",
|
|
57
|
+
}));
|
|
58
|
+
log.dim("AI", `Cached ${this.cachedCommands.length} slash commands`);
|
|
59
|
+
}
|
|
60
|
+
}
|
|
40
61
|
if (msg.type === "assistant") {
|
|
41
62
|
for (const block of msg.message?.content || []) {
|
|
42
63
|
if (block.type === "text") {
|
|
@@ -50,7 +71,29 @@ export class ClaudeSessionManager {
|
|
|
50
71
|
if (msg.type === "result") {
|
|
51
72
|
const result = msg;
|
|
52
73
|
this.sessions.set(clientId, result.session_id);
|
|
53
|
-
|
|
74
|
+
// Update cumulative stats
|
|
75
|
+
const prev = this.sessionStats.get(clientId) || { totalCost: 0, totalInputTokens: 0, totalOutputTokens: 0, turnCount: 0 };
|
|
76
|
+
const usage = result.usage || { input_tokens: 0, output_tokens: 0, cache_read_input_tokens: 0, cache_creation_input_tokens: 0 };
|
|
77
|
+
this.sessionStats.set(clientId, {
|
|
78
|
+
totalCost: prev.totalCost + (result.total_cost_usd || 0),
|
|
79
|
+
totalInputTokens: prev.totalInputTokens + (usage.input_tokens || 0),
|
|
80
|
+
totalOutputTokens: prev.totalOutputTokens + (usage.output_tokens || 0),
|
|
81
|
+
turnCount: prev.turnCount + (result.num_turns || 0),
|
|
82
|
+
});
|
|
83
|
+
callbacks.onComplete({
|
|
84
|
+
result: result.result_text || result.result || "",
|
|
85
|
+
sessionId: result.session_id,
|
|
86
|
+
cost: result.total_cost_usd || 0,
|
|
87
|
+
turns: result.num_turns || 0,
|
|
88
|
+
usage: {
|
|
89
|
+
input_tokens: usage.input_tokens || 0,
|
|
90
|
+
output_tokens: usage.output_tokens || 0,
|
|
91
|
+
cache_read_input_tokens: usage.cache_read_input_tokens || 0,
|
|
92
|
+
cache_creation_input_tokens: usage.cache_creation_input_tokens || 0,
|
|
93
|
+
},
|
|
94
|
+
duration_ms: result.duration_ms || 0,
|
|
95
|
+
model: config.model,
|
|
96
|
+
});
|
|
54
97
|
}
|
|
55
98
|
}
|
|
56
99
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"claude-session.js","sourceRoot":"","sources":["../src/claude-session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,gCAAgC,CAAA;AAEtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;
|
|
1
|
+
{"version":3,"file":"claude-session.js","sourceRoot":"","sources":["../src/claude-session.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,EAAE,MAAM,gCAAgC,CAAA;AAEtD,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAA;AACjD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAA;AACpC,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AA0BjC,MAAM,OAAO,oBAAoB;IACvB,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAA,CAAC,uBAAuB;IAC5D,YAAY,GAAG,IAAI,GAAG,EAAwB,CAAA;IAC9C,cAAc,GAAuE,EAAE,CAAA;IAE/F,KAAK,CAAC,aAAa,CACjB,QAAgB,EAChB,MAAoE,EACpE,SAA0B;QAE1B,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,CAAC,CAAA;QACtC,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,UAAU,EAAE,SAAS,CAAC,CAAA;IACtD,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,QAAgB,EAChB,MAAc,EACd,SAA0B;QAE1B,MAAM,IAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,SAAS,CAAC,CAAA;IAClD,CAAC;IAED,YAAY,CAAC,QAAgB;QAC3B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;QAC9B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IACpC,CAAC;IAED,eAAe,CAAC,QAAgB;QAC9B,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAA;IAChD,CAAC;IAED,eAAe;QACb,OAAO,EAAE,QAAQ,EAAE,IAAI,CAAC,cAAc,EAAE,CAAA;IAC1C,CAAC;IAEO,KAAK,CAAC,QAAQ,CACpB,QAAgB,EAChB,MAAc,EACd,SAA0B;QAE1B,MAAM,OAAO,GAA4B;YACvC,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YACnE,cAAc,EAAE,aAAa;YAC7B,GAAG,EAAE,MAAM,CAAC,UAAU;YACtB,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;SAC1B,CAAA;QAED,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAA;QACnD,IAAI,eAAe,EAAE,CAAC;YACpB,OAAO,CAAC,MAAM,GAAG,eAAe,CAAA;QAClC,CAAC;QAED,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,kBAAkB,eAAe,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;QAE/F,MAAM,UAAU,GAAG,OAAO,CAAA,CAAC,YAAY;QACvC,IAAI,QAAQ,GAAG,KAAK,CAAA;QACpB,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE,GAAG,QAAQ,GAAG,IAAI,CAAA,CAAC,CAAC,EAAE,UAAU,CAAC,CAAA;QAE/D,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,KAAK,CAAC,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;gBACnD,IAAI,QAAQ,EAAE,CAAC;oBACb,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,6BAA6B,CAAC,CAAA;oBACtE,SAAS,CAAC,OAAO,CAAC,mCAAmC,CAAC,CAAA;oBACtD,MAAK;gBACP,CAAC;gBAED,sDAAsD;gBACtD,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,IAAK,GAAW,CAAC,OAAO,KAAK,MAAM,EAAE,CAAC;oBAC7D,MAAM,IAAI,GAAG,GAAU,CAAA;oBACvB,IAAI,IAAI,CAAC,cAAc,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,CAAC;wBAC3E,IAAI,CAAC,cAAc,GAAI,IAAI,CAAC,cAA2B,CAAC,GAAG,CAAC,CAAC,IAAY,EAAE,EAAE,CAAC,CAAC;4BAC7E,IAAI;4BACJ,WAAW,EAAE,EAAE;4BACf,YAAY,EAAE,EAAE;yBACjB,CAAC,CAAC,CAAA;wBACH,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,UAAU,IAAI,CAAC,cAAc,CAAC,MAAM,iBAAiB,CAAC,CAAA;oBACtE,CAAC;gBACH,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;oBAC7B,KAAK,MAAM,KAAK,IAAK,GAAW,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC;wBACxD,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;4BAC1B,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;wBACnC,CAAC;wBACD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BAC9B,SAAS,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAAgC,CAAC,CAAA;wBACzE,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;oBAC1B,MAAM,MAAM,GAAG,GAAU,CAAA;oBACzB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC,UAAU,CAAC,CAAA;oBAE9C,0BAA0B;oBAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,EAAE,CAAC,EAAE,gBAAgB,EAAE,CAAC,EAAE,iBAAiB,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAA;oBACzH,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,IAAI,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,uBAAuB,EAAE,CAAC,EAAE,2BAA2B,EAAE,CAAC,EAAE,CAAA;oBAC/H,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE;wBAC9B,SAAS,EAAE,IAAI,CAAC,SAAS,GAAG,CAAC,MAAM,CAAC,cAAc,IAAI,CAAC,CAAC;wBACxD,gBAAgB,EAAE,IAAI,CAAC,gBAAgB,GAAG,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,CAAC;wBACnE,iBAAiB,EAAE,IAAI,CAAC,iBAAiB,GAAG,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,CAAC;wBACtE,SAAS,EAAE,IAAI,CAAC,SAAS,GAAG,CAAC,MAAM,CAAC,SAAS,IAAI,CAAC,CAAC;qBACpD,CAAC,CAAA;oBAEF,SAAS,CAAC,UAAU,CAAC;wBACnB,MAAM,EAAE,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,MAAM,IAAI,EAAE;wBACjD,SAAS,EAAE,MAAM,CAAC,UAAU;wBAC5B,IAAI,EAAE,MAAM,CAAC,cAAc,IAAI,CAAC;wBAChC,KAAK,EAAE,MAAM,CAAC,SAAS,IAAI,CAAC;wBAC5B,KAAK,EAAE;4BACL,YAAY,EAAE,KAAK,CAAC,YAAY,IAAI,CAAC;4BACrC,aAAa,EAAE,KAAK,CAAC,aAAa,IAAI,CAAC;4BACvC,uBAAuB,EAAE,KAAK,CAAC,uBAAuB,IAAI,CAAC;4BAC3D,2BAA2B,EAAE,KAAK,CAAC,2BAA2B,IAAI,CAAC;yBACpE;wBACD,WAAW,EAAE,MAAM,CAAC,WAAW,IAAI,CAAC;wBACpC,KAAK,EAAE,MAAM,CAAC,KAAK;qBACpB,CAAC,CAAA;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,mBAAmB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;YAC9G,SAAS,CAAC,OAAO,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAA;QACrE,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,KAAK,CAAC,CAAA;QACrB,CAAC;IACH,CAAC;CACF"}
|
package/dist/ws-handler.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ws-handler.d.ts","sourceRoot":"","sources":["../src/ws-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAA;AAEnC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAA;AAChE,OAAO,KAAK,EAAE,oBAAoB,
|
|
1
|
+
{"version":3,"file":"ws-handler.d.ts","sourceRoot":"","sources":["../src/ws-handler.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAA;AAEnC,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAA;AAChE,OAAO,KAAK,EAAE,oBAAoB,EAAmC,MAAM,qBAAqB,CAAA;AAwDhG,wBAAgB,gBAAgB,CAAC,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,MAAM,EAAE,oBAAoB,GAAG,IAAI,CAmElH"}
|
package/dist/ws-handler.js
CHANGED
|
@@ -14,9 +14,9 @@ function withLogging(clientId, callbacks) {
|
|
|
14
14
|
log.event("AI", `[${clientId.slice(0, 8)}] Tool: ${tool} ${JSON.stringify(input).slice(0, 100)}`);
|
|
15
15
|
callbacks.onToolUse(tool, input);
|
|
16
16
|
},
|
|
17
|
-
onComplete(
|
|
18
|
-
log.success("AI", `[${clientId.slice(0, 8)}] Complete (${turns} turns, $${cost.toFixed(4)})`);
|
|
19
|
-
callbacks.onComplete(
|
|
17
|
+
onComplete(data) {
|
|
18
|
+
log.success("AI", `[${clientId.slice(0, 8)}] Complete (${data.turns} turns, $${data.cost.toFixed(4)}, ${data.usage.input_tokens}in/${data.usage.output_tokens}out)`);
|
|
19
|
+
callbacks.onComplete(data);
|
|
20
20
|
},
|
|
21
21
|
onError(error) {
|
|
22
22
|
log.error("AI", `[${clientId.slice(0, 8)}] Error: ${error}`);
|
|
@@ -24,6 +24,32 @@ function withLogging(clientId, callbacks) {
|
|
|
24
24
|
},
|
|
25
25
|
};
|
|
26
26
|
}
|
|
27
|
+
function makeCallbacks(clientId, connections, claude) {
|
|
28
|
+
return {
|
|
29
|
+
onStreaming: (chunk) => connections.send(clientId, { type: "ai_streaming", chunk }),
|
|
30
|
+
onToolUse: (tool, input) => connections.send(clientId, { type: "tool_use", tool, input }),
|
|
31
|
+
onComplete: (data) => {
|
|
32
|
+
connections.send(clientId, {
|
|
33
|
+
type: "ai_complete",
|
|
34
|
+
result: data.result, sessionId: data.sessionId,
|
|
35
|
+
cost: data.cost, turns: data.turns,
|
|
36
|
+
usage: data.usage, duration_ms: data.duration_ms, model: data.model,
|
|
37
|
+
});
|
|
38
|
+
const stats = claude.getSessionStats(clientId);
|
|
39
|
+
if (stats) {
|
|
40
|
+
connections.send(clientId, {
|
|
41
|
+
type: "session_info",
|
|
42
|
+
model: data.model,
|
|
43
|
+
cumulativeCost: stats.totalCost,
|
|
44
|
+
cumulativeInputTokens: stats.totalInputTokens,
|
|
45
|
+
cumulativeOutputTokens: stats.totalOutputTokens,
|
|
46
|
+
turnCount: stats.turnCount,
|
|
47
|
+
});
|
|
48
|
+
}
|
|
49
|
+
},
|
|
50
|
+
onError: (error) => connections.send(clientId, { type: "ai_error", error }),
|
|
51
|
+
};
|
|
52
|
+
}
|
|
27
53
|
export function handleConnection(ws, connections, claude) {
|
|
28
54
|
const clientId = connections.add(ws);
|
|
29
55
|
const short = clientId.slice(0, 8);
|
|
@@ -53,22 +79,20 @@ export function handleConnection(ws, connections, claude) {
|
|
|
53
79
|
route: msg.route,
|
|
54
80
|
element: msg.element,
|
|
55
81
|
prompt: msg.prompt,
|
|
56
|
-
}, withLogging(clientId,
|
|
57
|
-
onStreaming: (chunk) => connections.send(clientId, { type: "ai_streaming", chunk }),
|
|
58
|
-
onToolUse: (tool, input) => connections.send(clientId, { type: "tool_use", tool, input }),
|
|
59
|
-
onComplete: (result, sessionId, cost, turns) => connections.send(clientId, { type: "ai_complete", result, sessionId, cost, turns }),
|
|
60
|
-
onError: (error) => connections.send(clientId, { type: "ai_error", error }),
|
|
61
|
-
}));
|
|
82
|
+
}, withLogging(clientId, makeCallbacks(clientId, connections, claude)));
|
|
62
83
|
break;
|
|
63
84
|
case "raw_prompt":
|
|
64
85
|
log.event("AI", `[${short}] Raw prompt: "${msg.prompt}"`);
|
|
65
|
-
claude.executeRawPrompt(clientId, msg.prompt, withLogging(clientId,
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
86
|
+
claude.executeRawPrompt(clientId, msg.prompt, withLogging(clientId, makeCallbacks(clientId, connections, claude)));
|
|
87
|
+
break;
|
|
88
|
+
case "query_capabilities": {
|
|
89
|
+
const caps = claude.getCapabilities();
|
|
90
|
+
connections.send(clientId, {
|
|
91
|
+
type: "capabilities",
|
|
92
|
+
commands: caps.commands,
|
|
93
|
+
});
|
|
71
94
|
break;
|
|
95
|
+
}
|
|
72
96
|
case "reset_session":
|
|
73
97
|
log.info("WS", `[${short}] Session reset`);
|
|
74
98
|
claude.resetSession(clientId);
|
package/dist/ws-handler.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ws-handler.js","sourceRoot":"","sources":["../src/ws-handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAG1D,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;
|
|
1
|
+
{"version":3,"file":"ws-handler.js","sourceRoot":"","sources":["../src/ws-handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAA;AAG1D,OAAO,EAAE,GAAG,EAAE,MAAM,aAAa,CAAA;AAGjC,SAAS,WAAW,CAAC,QAAgB,EAAE,SAA0B;IAC/D,IAAI,aAAa,GAAG,KAAK,CAAA;IACzB,OAAO;QACL,WAAW,CAAC,KAAK;YACf,IAAI,CAAC,aAAa,EAAE,CAAC;gBACnB,aAAa,GAAG,IAAI,CAAA;gBACpB,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAA;YAC9D,CAAC;YACD,SAAS,CAAC,WAAW,CAAC,KAAK,CAAC,CAAA;QAC9B,CAAC;QACD,SAAS,CAAC,IAAI,EAAE,KAAK;YACnB,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,WAAW,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAA;YACjG,SAAS,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAA;QAClC,CAAC;QACD,UAAU,CAAC,IAAI;YACb,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,eAAe,IAAI,CAAC,KAAK,YAAY,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,YAAY,MAAM,IAAI,CAAC,KAAK,CAAC,aAAa,MAAM,CAAC,CAAA;YACpK,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAA;QAC5B,CAAC;QACD,OAAO,CAAC,KAAK;YACX,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,YAAY,KAAK,EAAE,CAAC,CAAA;YAC5D,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,CAAA;QAC1B,CAAC;KACF,CAAA;AACH,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB,EAAE,WAA8B,EAAE,MAA4B;IACnG,OAAO;QACL,WAAW,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;QACnF,SAAS,EAAE,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;QACzF,UAAU,EAAE,CAAC,IAAoB,EAAE,EAAE;YACnC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE;gBACzB,IAAI,EAAE,aAAa;gBACnB,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS;gBAC9C,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK;gBAClC,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK;aACpE,CAAC,CAAA;YACF,MAAM,KAAK,GAAG,MAAM,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAA;YAC9C,IAAI,KAAK,EAAE,CAAC;gBACV,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE;oBACzB,IAAI,EAAE,cAAc;oBACpB,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,cAAc,EAAE,KAAK,CAAC,SAAS;oBAC/B,qBAAqB,EAAE,KAAK,CAAC,gBAAgB;oBAC7C,sBAAsB,EAAE,KAAK,CAAC,iBAAiB;oBAC/C,SAAS,EAAE,KAAK,CAAC,SAAS;iBAC3B,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;QACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;KAC5E,CAAA;AACH,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,EAAa,EAAE,WAA8B,EAAE,MAA4B;IAC1G,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;IACpC,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAA;IAClC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,qBAAqB,KAAK,EAAE,CAAC,CAAA;IAC5C,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAA;IAE3D,IAAI,KAAK,GAAG,IAAI,CAAA;IAChB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,KAAK,GAAG,IAAI,CAAA,CAAC,CAAC,CAAC,CAAA;IACrC,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE;QACjC,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,mCAAmC,KAAK,EAAE,CAAC,CAAA;YACzD,EAAE,CAAC,SAAS,EAAE,CAAA;YACd,OAAM;QACR,CAAC;QACD,KAAK,GAAG,KAAK,CAAA;QACb,EAAE,CAAC,IAAI,EAAE,CAAA;IACX,CAAC,EAAE,MAAM,CAAC,CAAA;IAEV,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;QACxB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,kBAAkB,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAA;YAC/C,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;gBACjB,KAAK,MAAM;oBACT,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAA;oBAC5C,MAAK;gBAEP,KAAK,QAAQ;oBACX,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,cAAc,GAAG,CAAC,MAAM,OAAO,GAAG,CAAC,OAAO,CAAC,WAAW,OAAO,GAAG,CAAC,KAAK,EAAE,CAAC,CAAA;oBAClG,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE;wBAC7B,KAAK,EAAE,GAAG,CAAC,KAAK;wBAChB,OAAO,EAAE,GAAG,CAAC,OAAO;wBACpB,MAAM,EAAE,GAAG,CAAC,MAAM;qBACnB,EAAE,WAAW,CAAC,QAAQ,EAAE,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,CAAA;oBACvE,MAAK;gBAEP,KAAK,YAAY;oBACf,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,kBAAkB,GAAG,CAAC,MAAM,GAAG,CAAC,CAAA;oBACzD,MAAM,CAAC,gBAAgB,CAAC,QAAQ,EAAE,GAAG,CAAC,MAAM,EAC1C,WAAW,CAAC,QAAQ,EAAE,aAAa,CAAC,QAAQ,EAAE,WAAW,EAAE,MAAM,CAAC,CAAC,CAAC,CAAA;oBACtE,MAAK;gBAEP,KAAK,oBAAoB,CAAC,CAAC,CAAC;oBAC1B,MAAM,IAAI,GAAG,MAAM,CAAC,eAAe,EAAE,CAAA;oBACrC,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE;wBACzB,IAAI,EAAE,cAAc;wBACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;qBACxB,CAAC,CAAA;oBACF,MAAK;gBACP,CAAC;gBAED,KAAK,eAAe;oBAClB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,iBAAiB,CAAC,CAAA;oBAC1C,MAAM,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAA;oBAC7B,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,eAAe,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAA;oBAC5E,MAAK;YACT,CAAC;QACH,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,KAAK,oBAAoB,GAAG,EAAE,CAAC,CAAA;YACnD,WAAW,CAAC,IAAI,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAA;QACtE,CAAC;IACH,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;QAClB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,wBAAwB,KAAK,EAAE,CAAC,CAAA;QAC/C,aAAa,CAAC,SAAS,CAAC,CAAA;QACxB,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAA;IAC9B,CAAC,CAAC,CAAA;AACJ,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@claude-studio/server",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.3.0",
|
|
4
4
|
"type": "module",
|
|
5
5
|
"publishConfig": {
|
|
6
6
|
"access": "public"
|
|
@@ -11,7 +11,7 @@
|
|
|
11
11
|
},
|
|
12
12
|
"dependencies": {
|
|
13
13
|
"@anthropic-ai/claude-agent-sdk": "^0.2.90",
|
|
14
|
-
"@claude-studio/shared": "^0.
|
|
14
|
+
"@claude-studio/shared": "^0.3.0",
|
|
15
15
|
"dotenv": "^16.0.0",
|
|
16
16
|
"express": "^4.0.0",
|
|
17
17
|
"ws": "^8.0.0"
|