@stigmer/react 0.0.56 → 0.0.58
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/execution/ApprovalCard.d.ts.map +1 -1
- package/execution/ApprovalCard.js +1 -1
- package/execution/ApprovalCard.js.map +1 -1
- package/execution/ArtifactsWidget.d.ts +1 -1
- package/execution/ArtifactsWidget.js +1 -1
- package/execution/ExecutionProgress.d.ts.map +1 -1
- package/execution/ExecutionProgress.js +2 -59
- package/execution/ExecutionProgress.js.map +1 -1
- package/execution/MessageThread.d.ts.map +1 -1
- package/execution/MessageThread.js +31 -6
- package/execution/MessageThread.js.map +1 -1
- package/execution/SubAgentSection.d.ts +25 -4
- package/execution/SubAgentSection.d.ts.map +1 -1
- package/execution/SubAgentSection.js +70 -11
- package/execution/SubAgentSection.js.map +1 -1
- package/execution/TodoList.d.ts +42 -0
- package/execution/TodoList.d.ts.map +1 -0
- package/execution/TodoList.js +108 -0
- package/execution/TodoList.js.map +1 -0
- package/execution/ToolCallItem.js +1 -1
- package/execution/ToolCallItem.js.map +1 -1
- package/execution/UsageWidget.d.ts +57 -0
- package/execution/UsageWidget.d.ts.map +1 -0
- package/execution/UsageWidget.js +72 -0
- package/execution/UsageWidget.js.map +1 -0
- package/execution/index.d.ts +4 -4
- package/execution/index.d.ts.map +1 -1
- package/execution/index.js +2 -2
- package/execution/index.js.map +1 -1
- package/execution/useExecutionArtifacts.d.ts +1 -1
- package/execution/useExecutionArtifacts.js +1 -1
- package/index.d.ts +4 -4
- package/index.d.ts.map +1 -1
- package/index.js +2 -2
- package/index.js.map +1 -1
- package/package.json +4 -4
- package/session/index.d.ts +2 -0
- package/session/index.d.ts.map +1 -1
- package/session/index.js +1 -0
- package/session/index.js.map +1 -1
- package/session/useSessionUsage.d.ts +65 -0
- package/session/useSessionUsage.d.ts.map +1 -0
- package/session/useSessionUsage.js +107 -0
- package/session/useSessionUsage.js.map +1 -0
- package/src/execution/ApprovalCard.tsx +7 -13
- package/src/execution/ArtifactsWidget.tsx +1 -1
- package/src/execution/ExecutionProgress.tsx +2 -134
- package/src/execution/MessageThread.tsx +39 -6
- package/src/execution/SubAgentSection.tsx +323 -16
- package/src/execution/TodoList.tsx +202 -0
- package/src/execution/ToolCallItem.tsx +1 -1
- package/src/execution/{ExecutionCostSummary.tsx → UsageWidget.tsx} +43 -50
- package/src/execution/index.ts +10 -4
- package/src/execution/useExecutionArtifacts.ts +1 -1
- package/src/index.ts +12 -5
- package/src/session/index.ts +6 -0
- package/src/session/useSessionUsage.ts +159 -0
- package/styles.css +1 -1
- package/execution/ExecutionCostSummary.d.ts +0 -47
- package/execution/ExecutionCostSummary.d.ts.map +0 -1
- package/execution/ExecutionCostSummary.js +0 -77
- package/execution/ExecutionCostSummary.js.map +0 -1
- package/execution/__tests__/ExecutionCostSummary.test.d.ts +0 -2
- package/execution/__tests__/ExecutionCostSummary.test.d.ts.map +0 -1
- package/execution/__tests__/ExecutionCostSummary.test.js +0 -255
- package/execution/__tests__/ExecutionCostSummary.test.js.map +0 -1
- package/execution/__tests__/useExecutionUsage.test.d.ts +0 -2
- package/execution/__tests__/useExecutionUsage.test.d.ts.map +0 -1
- package/execution/__tests__/useExecutionUsage.test.js +0 -303
- package/execution/__tests__/useExecutionUsage.test.js.map +0 -1
- package/execution/useExecutionUsage.d.ts +0 -45
- package/execution/useExecutionUsage.d.ts.map +0 -1
- package/execution/useExecutionUsage.js +0 -157
- package/execution/useExecutionUsage.js.map +0 -1
- package/src/execution/__tests__/ExecutionCostSummary.test.tsx +0 -416
- package/src/execution/__tests__/useExecutionUsage.test.tsx +0 -408
- package/src/execution/useExecutionUsage.ts +0 -213
|
@@ -1,255 +0,0 @@
|
|
|
1
|
-
import { jsx as _jsx } from "react/jsx-runtime";
|
|
2
|
-
import { describe, it, expect, afterEach } from "vitest";
|
|
3
|
-
import { render, screen, cleanup } from "@testing-library/react";
|
|
4
|
-
import { create } from "@bufbuild/protobuf";
|
|
5
|
-
afterEach(cleanup);
|
|
6
|
-
import { AgentExecutionSchema, AgentExecutionStatusSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
|
|
7
|
-
import { SubAgentExecutionSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/subagent_pb";
|
|
8
|
-
import { UsageMetricsSchema, ModelUsageSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/usage_pb";
|
|
9
|
-
import { ExecutionCostSummary, formatCost, formatTokenCount, } from "../ExecutionCostSummary";
|
|
10
|
-
// ---------------------------------------------------------------------------
|
|
11
|
-
// Test helpers
|
|
12
|
-
// ---------------------------------------------------------------------------
|
|
13
|
-
function makeUsage(overrides = {}) {
|
|
14
|
-
return create(UsageMetricsSchema, overrides);
|
|
15
|
-
}
|
|
16
|
-
function makeUsageWithModels(overrides = {}) {
|
|
17
|
-
return create(UsageMetricsSchema, overrides);
|
|
18
|
-
}
|
|
19
|
-
function makeModelUsage(model, provider, overrides = {}) {
|
|
20
|
-
return create(ModelUsageSchema, { model, provider, ...overrides });
|
|
21
|
-
}
|
|
22
|
-
function makeExecution(mainUsage, subAgents = []) {
|
|
23
|
-
const status = create(AgentExecutionStatusSchema, {
|
|
24
|
-
usage: mainUsage,
|
|
25
|
-
subAgentExecutions: subAgents.map((s) => create(SubAgentExecutionSchema, { name: s.name, usage: s.usage })),
|
|
26
|
-
});
|
|
27
|
-
return create(AgentExecutionSchema, { status });
|
|
28
|
-
}
|
|
29
|
-
// ---------------------------------------------------------------------------
|
|
30
|
-
// formatCost
|
|
31
|
-
// ---------------------------------------------------------------------------
|
|
32
|
-
describe("formatCost", () => {
|
|
33
|
-
it("formats zero as $0.00", () => {
|
|
34
|
-
expect(formatCost(0)).toBe("$0.00");
|
|
35
|
-
});
|
|
36
|
-
it("formats sub-dollar with 4 decimal places", () => {
|
|
37
|
-
expect(formatCost(0.0042)).toBe("$0.0042");
|
|
38
|
-
});
|
|
39
|
-
it("formats sub-dollar with trailing zeros", () => {
|
|
40
|
-
expect(formatCost(0.15)).toBe("$0.1500");
|
|
41
|
-
});
|
|
42
|
-
it("formats $1+ with 2 decimal places", () => {
|
|
43
|
-
expect(formatCost(1.5)).toBe("$1.50");
|
|
44
|
-
});
|
|
45
|
-
it("rounds $1+ to 2 decimal places", () => {
|
|
46
|
-
expect(formatCost(123.456)).toBe("$123.46");
|
|
47
|
-
});
|
|
48
|
-
it("formats exactly $1 with 2 decimal places", () => {
|
|
49
|
-
expect(formatCost(1)).toBe("$1.00");
|
|
50
|
-
});
|
|
51
|
-
});
|
|
52
|
-
// ---------------------------------------------------------------------------
|
|
53
|
-
// formatTokenCount
|
|
54
|
-
// ---------------------------------------------------------------------------
|
|
55
|
-
describe("formatTokenCount", () => {
|
|
56
|
-
it("formats zero", () => {
|
|
57
|
-
expect(formatTokenCount(0)).toBe("0");
|
|
58
|
-
});
|
|
59
|
-
it("formats small numbers without commas", () => {
|
|
60
|
-
expect(formatTokenCount(999)).toBe("999");
|
|
61
|
-
});
|
|
62
|
-
it("formats thousands with commas", () => {
|
|
63
|
-
expect(formatTokenCount(1234)).toBe("1,234");
|
|
64
|
-
});
|
|
65
|
-
it("formats millions with commas", () => {
|
|
66
|
-
expect(formatTokenCount(1234567)).toBe("1,234,567");
|
|
67
|
-
});
|
|
68
|
-
});
|
|
69
|
-
// ---------------------------------------------------------------------------
|
|
70
|
-
// ExecutionCostSummary — component rendering
|
|
71
|
-
// ---------------------------------------------------------------------------
|
|
72
|
-
describe("ExecutionCostSummary", () => {
|
|
73
|
-
it("returns null when execution is null", () => {
|
|
74
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: null }));
|
|
75
|
-
expect(container.innerHTML).toBe("");
|
|
76
|
-
});
|
|
77
|
-
it("returns null when usage is not yet available", () => {
|
|
78
|
-
const execution = create(AgentExecutionSchema, {
|
|
79
|
-
status: create(AgentExecutionStatusSchema),
|
|
80
|
-
});
|
|
81
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
82
|
-
expect(container.innerHTML).toBe("");
|
|
83
|
-
});
|
|
84
|
-
it("renders cost, tokens, calls, and model for a basic execution", () => {
|
|
85
|
-
const execution = makeExecution(makeUsage({
|
|
86
|
-
estimatedCostUsd: 0.0042,
|
|
87
|
-
totalTokens: 1234,
|
|
88
|
-
promptTokens: 1000,
|
|
89
|
-
completionTokens: 234,
|
|
90
|
-
llmCallCount: 3,
|
|
91
|
-
primaryModel: "claude-sonnet-4",
|
|
92
|
-
primaryProvider: "anthropic",
|
|
93
|
-
}));
|
|
94
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
95
|
-
expect(screen.getByText("$0.0042")).toBeTruthy();
|
|
96
|
-
expect(screen.getByText("claude-sonnet-4 · anthropic")).toBeTruthy();
|
|
97
|
-
expect(screen.getByText(/1,234 tokens/)).toBeTruthy();
|
|
98
|
-
expect(screen.getByText(/3 calls/)).toBeTruthy();
|
|
99
|
-
});
|
|
100
|
-
it("renders token breakdown with prompt and completion", () => {
|
|
101
|
-
const execution = makeExecution(makeUsage({
|
|
102
|
-
totalTokens: 1234,
|
|
103
|
-
promptTokens: 1000,
|
|
104
|
-
completionTokens: 234,
|
|
105
|
-
llmCallCount: 1,
|
|
106
|
-
}));
|
|
107
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
108
|
-
expect(screen.getByText(/prompt 1,000/)).toBeTruthy();
|
|
109
|
-
expect(screen.getByText(/completion 234/)).toBeTruthy();
|
|
110
|
-
});
|
|
111
|
-
it("uses singular 'call' for a single LLM call", () => {
|
|
112
|
-
const execution = makeExecution(makeUsage({
|
|
113
|
-
llmCallCount: 1,
|
|
114
|
-
totalTokens: 100,
|
|
115
|
-
promptTokens: 80,
|
|
116
|
-
completionTokens: 20,
|
|
117
|
-
}));
|
|
118
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
119
|
-
const metricsLine = screen.getByText(/100 tokens/);
|
|
120
|
-
expect(metricsLine.textContent).toContain("1 call");
|
|
121
|
-
expect(metricsLine.textContent).not.toContain("1 calls");
|
|
122
|
-
});
|
|
123
|
-
it("does not show cache line when cache tokens are zero", () => {
|
|
124
|
-
const execution = makeExecution(makeUsage({
|
|
125
|
-
totalTokens: 100,
|
|
126
|
-
promptTokens: 80,
|
|
127
|
-
completionTokens: 20,
|
|
128
|
-
}));
|
|
129
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
130
|
-
expect(container.textContent).not.toContain("cache");
|
|
131
|
-
});
|
|
132
|
-
it("shows cache read tokens when present", () => {
|
|
133
|
-
const execution = makeExecution(makeUsage({
|
|
134
|
-
totalTokens: 100,
|
|
135
|
-
promptTokens: 80,
|
|
136
|
-
completionTokens: 20,
|
|
137
|
-
cacheReadTokens: 500,
|
|
138
|
-
}));
|
|
139
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
140
|
-
expect(container.textContent).toContain("cache");
|
|
141
|
-
expect(container.textContent).toContain("500 read");
|
|
142
|
-
expect(container.textContent).not.toContain("write");
|
|
143
|
-
});
|
|
144
|
-
it("shows cache write tokens when present", () => {
|
|
145
|
-
const execution = makeExecution(makeUsage({
|
|
146
|
-
totalTokens: 100,
|
|
147
|
-
promptTokens: 80,
|
|
148
|
-
completionTokens: 20,
|
|
149
|
-
cacheCreationTokens: 200,
|
|
150
|
-
}));
|
|
151
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
152
|
-
expect(container.textContent).toContain("cache");
|
|
153
|
-
expect(container.textContent).toContain("200 write");
|
|
154
|
-
expect(container.textContent).not.toContain("read");
|
|
155
|
-
});
|
|
156
|
-
it("shows both cache read and write when both present", () => {
|
|
157
|
-
const execution = makeExecution(makeUsage({
|
|
158
|
-
totalTokens: 100,
|
|
159
|
-
promptTokens: 80,
|
|
160
|
-
completionTokens: 20,
|
|
161
|
-
cacheReadTokens: 500,
|
|
162
|
-
cacheCreationTokens: 100,
|
|
163
|
-
}));
|
|
164
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
165
|
-
expect(container.textContent).toContain("500 read");
|
|
166
|
-
expect(container.textContent).toContain("100 write");
|
|
167
|
-
});
|
|
168
|
-
it("shows sub-agent annotation when sub-agents have usage", () => {
|
|
169
|
-
const execution = makeExecution(makeUsage({
|
|
170
|
-
totalTokens: 1000,
|
|
171
|
-
promptTokens: 800,
|
|
172
|
-
completionTokens: 200,
|
|
173
|
-
}), [
|
|
174
|
-
{ name: "researcher", usage: makeUsage({ totalTokens: 500 }) },
|
|
175
|
-
{ name: "writer", usage: makeUsage({ totalTokens: 300 }) },
|
|
176
|
-
]);
|
|
177
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
178
|
-
expect(screen.getByText("Includes 2 sub-agents")).toBeTruthy();
|
|
179
|
-
});
|
|
180
|
-
it("uses singular 'sub-agent' for one sub-agent", () => {
|
|
181
|
-
const execution = makeExecution(makeUsage({
|
|
182
|
-
totalTokens: 1000,
|
|
183
|
-
promptTokens: 800,
|
|
184
|
-
completionTokens: 200,
|
|
185
|
-
}), [{ name: "researcher", usage: makeUsage({ totalTokens: 500 }) }]);
|
|
186
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
187
|
-
expect(screen.getByText("Includes 1 sub-agent")).toBeTruthy();
|
|
188
|
-
});
|
|
189
|
-
it("does not show sub-agent annotation when no sub-agents have usage", () => {
|
|
190
|
-
const execution = makeExecution(makeUsage({
|
|
191
|
-
totalTokens: 1000,
|
|
192
|
-
promptTokens: 800,
|
|
193
|
-
completionTokens: 200,
|
|
194
|
-
}), [{ name: "idle-agent" }]);
|
|
195
|
-
const { container } = render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
196
|
-
expect(container.textContent).not.toContain("sub-agent");
|
|
197
|
-
});
|
|
198
|
-
it("shows per-model breakdown when multiple models exist", () => {
|
|
199
|
-
const execution = makeExecution(makeUsageWithModels({
|
|
200
|
-
totalTokens: 2000,
|
|
201
|
-
promptTokens: 1500,
|
|
202
|
-
completionTokens: 500,
|
|
203
|
-
estimatedCostUsd: 0.0142,
|
|
204
|
-
primaryModel: "claude-sonnet-4",
|
|
205
|
-
primaryProvider: "anthropic",
|
|
206
|
-
modelBreakdown: [
|
|
207
|
-
makeModelUsage("claude-sonnet-4", "anthropic", {
|
|
208
|
-
estimatedCostUsd: 0.012,
|
|
209
|
-
}),
|
|
210
|
-
makeModelUsage("gpt-4o", "openai", {
|
|
211
|
-
estimatedCostUsd: 0.0022,
|
|
212
|
-
}),
|
|
213
|
-
],
|
|
214
|
-
}));
|
|
215
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
216
|
-
expect(screen.getByText("claude-sonnet-4")).toBeTruthy();
|
|
217
|
-
expect(screen.getByText("$0.0120")).toBeTruthy();
|
|
218
|
-
expect(screen.getByText("gpt-4o")).toBeTruthy();
|
|
219
|
-
expect(screen.getByText("$0.0022")).toBeTruthy();
|
|
220
|
-
expect(screen.queryByText("claude-sonnet-4 · anthropic")).toBeNull();
|
|
221
|
-
});
|
|
222
|
-
it("shows model breakdown list with proper accessibility", () => {
|
|
223
|
-
const execution = makeExecution(makeUsageWithModels({
|
|
224
|
-
totalTokens: 2000,
|
|
225
|
-
promptTokens: 1500,
|
|
226
|
-
completionTokens: 500,
|
|
227
|
-
modelBreakdown: [
|
|
228
|
-
makeModelUsage("model-a", "provider-a"),
|
|
229
|
-
makeModelUsage("model-b", "provider-b"),
|
|
230
|
-
],
|
|
231
|
-
}));
|
|
232
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
233
|
-
expect(screen.getByRole("list", { name: "Model cost breakdown" })).toBeTruthy();
|
|
234
|
-
});
|
|
235
|
-
it("has proper region role and aria-label", () => {
|
|
236
|
-
const execution = makeExecution(makeUsage({
|
|
237
|
-
totalTokens: 100,
|
|
238
|
-
promptTokens: 80,
|
|
239
|
-
completionTokens: 20,
|
|
240
|
-
}));
|
|
241
|
-
render(_jsx(ExecutionCostSummary, { execution: execution }));
|
|
242
|
-
expect(screen.getByRole("region", { name: "Execution cost summary" })).toBeTruthy();
|
|
243
|
-
});
|
|
244
|
-
it("accepts and applies className", () => {
|
|
245
|
-
const execution = makeExecution(makeUsage({
|
|
246
|
-
totalTokens: 100,
|
|
247
|
-
promptTokens: 80,
|
|
248
|
-
completionTokens: 20,
|
|
249
|
-
}));
|
|
250
|
-
render(_jsx(ExecutionCostSummary, { execution: execution, className: "my-custom" }));
|
|
251
|
-
const region = screen.getByRole("region");
|
|
252
|
-
expect(region.className).toContain("my-custom");
|
|
253
|
-
});
|
|
254
|
-
});
|
|
255
|
-
//# sourceMappingURL=ExecutionCostSummary.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"ExecutionCostSummary.test.js","sourceRoot":"","sources":["../../../src/execution/__tests__/ExecutionCostSummary.test.tsx"],"names":[],"mappings":";AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzD,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,wBAAwB,CAAC;AACjE,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAE5C,SAAS,CAAC,OAAO,CAAC,CAAC;AACnB,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kEAAkE,CAAC;AAC3G,OAAO,EACL,kBAAkB,EAClB,gBAAgB,GAEjB,MAAM,+DAA+D,CAAC;AACvE,OAAO,EACL,oBAAoB,EACpB,UAAU,EACV,gBAAgB,GACjB,MAAM,yBAAyB,CAAC;AAEjC,8EAA8E;AAC9E,eAAe;AACf,8EAA8E;AAE9E,SAAS,SAAS,CAChB,YAUK,EAAE;IAEP,OAAO,MAAM,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,mBAAmB,CAC1B,YAWI,EAAE;IAEN,OAAO,MAAM,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CACrB,KAAa,EACb,QAAgB,EAChB,YAKK,EAAE;IAEP,OAAO,MAAM,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,aAAa,CACpB,SAAwC,EACxC,YAGK,EAAE;IAEP,MAAM,MAAM,GAAG,MAAM,CAAC,0BAA0B,EAAE;QAChD,KAAK,EAAE,SAAS;QAChB,kBAAkB,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACtC,MAAM,CAAC,uBAAuB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAClE;KACF,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAClD,CAAC;AAED,8EAA8E;AAC9E,aAAa;AACb,8EAA8E;AAE9E,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC7C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;QAChD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,cAAc,EAAE,GAAG,EAAE;QACtB,MAAM,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,6CAA6C;AAC7C,8EAA8E;AAE9E,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;IACpC,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,IAAI,GAAI,CAC1C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,SAAS,GAAG,MAAM,CAAC,oBAAoB,EAAE;YAC7C,MAAM,EAAE,MAAM,CAAC,0BAA0B,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;QACtE,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,gBAAgB,EAAE,MAAM;YACxB,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,gBAAgB,EAAE,GAAG;YACrB,YAAY,EAAE,CAAC;YACf,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;SAC7B,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACrE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC5D,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,gBAAgB,EAAE,GAAG;YACrB,YAAY,EAAE,CAAC;SAChB,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACtD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;SACrB,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;QACnD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACpD,MAAM,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;SACrB,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;YACpB,eAAe,EAAE,GAAG;SACrB,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;YACpB,mBAAmB,EAAE,GAAG;SACzB,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACjD,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;QAC3D,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;YACpB,eAAe,EAAE,GAAG;YACpB,mBAAmB,EAAE,GAAG;SACzB,CAAC,CACH,CAAC;QAEF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACpD,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uDAAuD,EAAE,GAAG,EAAE;QAC/D,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;SACtB,CAAC,EACF;YACE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE;YAC9D,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE;SAC3D,CACF,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;SACtB,CAAC,EACF,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,CACjE,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;IAChE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC1E,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,GAAG;SACtB,CAAC,EACF,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CACzB,CAAC;QAEF,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,CAC1B,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAC/C,CAAC;QACF,MAAM,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,aAAa,CAC7B,mBAAmB,CAAC;YAClB,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,gBAAgB,EAAE,GAAG;YACrB,gBAAgB,EAAE,MAAM;YACxB,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;YAC5B,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,gBAAgB,EAAE,KAAK;iBACxB,CAAC;gBACF,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE;oBACjC,gBAAgB,EAAE,MAAM;iBACzB,CAAC;aACH;SACF,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACzD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACjD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QAChD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC,UAAU,EAAE,CAAC;QACjD,MAAM,CACJ,MAAM,CAAC,WAAW,CAAC,6BAA6B,CAAC,CAClD,CAAC,QAAQ,EAAE,CAAC;IACf,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,SAAS,GAAG,aAAa,CAC7B,mBAAmB,CAAC;YAClB,WAAW,EAAE,IAAI;YACjB,YAAY,EAAE,IAAI;YAClB,gBAAgB,EAAE,GAAG;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,SAAS,EAAE,YAAY,CAAC;gBACvC,cAAc,CAAC,SAAS,EAAE,YAAY,CAAC;aACxC;SACF,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,CACJ,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,sBAAsB,EAAE,CAAC,CAC3D,CAAC,UAAU,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;SACrB,CAAC,CACH,CAAC;QAEF,MAAM,CAAC,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,GAAI,CAAC,CAAC;QAEvD,MAAM,CACJ,MAAM,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,wBAAwB,EAAE,CAAC,CAC/D,CAAC,UAAU,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,SAAS,GAAG,aAAa,CAC7B,SAAS,CAAC;YACR,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;SACrB,CAAC,CACH,CAAC;QAEF,MAAM,CACJ,KAAC,oBAAoB,IAAC,SAAS,EAAE,SAAS,EAAE,SAAS,EAAC,WAAW,GAAG,CACrE,CAAC;QAEF,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"useExecutionUsage.test.d.ts","sourceRoot":"","sources":["../../../src/execution/__tests__/useExecutionUsage.test.tsx"],"names":[],"mappings":""}
|
|
@@ -1,303 +0,0 @@
|
|
|
1
|
-
import { describe, it, expect } from "vitest";
|
|
2
|
-
import { renderHook } from "@testing-library/react";
|
|
3
|
-
import { create } from "@bufbuild/protobuf";
|
|
4
|
-
import { AgentExecutionSchema, AgentExecutionStatusSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
|
|
5
|
-
import { SubAgentExecutionSchema } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/subagent_pb";
|
|
6
|
-
import { UsageMetricsSchema, ModelUsageSchema, LlmCallMetricsSchema, } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/usage_pb";
|
|
7
|
-
import { aggregateUsage, useExecutionUsage } from "../useExecutionUsage";
|
|
8
|
-
// ---------------------------------------------------------------------------
|
|
9
|
-
// Test helpers
|
|
10
|
-
// ---------------------------------------------------------------------------
|
|
11
|
-
function makeUsage(overrides = {}) {
|
|
12
|
-
return create(UsageMetricsSchema, overrides);
|
|
13
|
-
}
|
|
14
|
-
function makeModelUsage(model, provider, overrides = {}) {
|
|
15
|
-
return create(ModelUsageSchema, { model, provider, ...overrides });
|
|
16
|
-
}
|
|
17
|
-
function makeLlmCall(sequence, timestamp, overrides = {}) {
|
|
18
|
-
return create(LlmCallMetricsSchema, { sequence, timestamp, ...overrides });
|
|
19
|
-
}
|
|
20
|
-
function makeExecution(mainUsage, subAgents = []) {
|
|
21
|
-
const status = create(AgentExecutionStatusSchema, {
|
|
22
|
-
usage: mainUsage,
|
|
23
|
-
subAgentExecutions: subAgents.map((s) => create(SubAgentExecutionSchema, { name: s.name, usage: s.usage })),
|
|
24
|
-
});
|
|
25
|
-
return create(AgentExecutionSchema, { status });
|
|
26
|
-
}
|
|
27
|
-
// ---------------------------------------------------------------------------
|
|
28
|
-
// aggregateUsage — pure function tests
|
|
29
|
-
// ---------------------------------------------------------------------------
|
|
30
|
-
describe("aggregateUsage", () => {
|
|
31
|
-
it("returns null when execution is null", () => {
|
|
32
|
-
expect(aggregateUsage(null)).toBeNull();
|
|
33
|
-
});
|
|
34
|
-
it("returns null when status is undefined", () => {
|
|
35
|
-
const exec = create(AgentExecutionSchema);
|
|
36
|
-
expect(aggregateUsage(exec)).toBeNull();
|
|
37
|
-
});
|
|
38
|
-
it("returns null when status.usage is undefined", () => {
|
|
39
|
-
const exec = create(AgentExecutionSchema, {
|
|
40
|
-
status: create(AgentExecutionStatusSchema),
|
|
41
|
-
});
|
|
42
|
-
expect(aggregateUsage(exec)).toBeNull();
|
|
43
|
-
});
|
|
44
|
-
it("returns main-only usage when no sub-agents exist", () => {
|
|
45
|
-
const usage = makeUsage({
|
|
46
|
-
promptTokens: 100,
|
|
47
|
-
completionTokens: 50,
|
|
48
|
-
totalTokens: 150,
|
|
49
|
-
llmCallCount: 3,
|
|
50
|
-
estimatedCostUsd: 0.005,
|
|
51
|
-
primaryModel: "claude-sonnet-4",
|
|
52
|
-
primaryProvider: "anthropic",
|
|
53
|
-
});
|
|
54
|
-
const exec = makeExecution(usage);
|
|
55
|
-
const result = aggregateUsage(exec);
|
|
56
|
-
expect(result).toBe(usage);
|
|
57
|
-
});
|
|
58
|
-
it("returns main-only usage when sub-agents exist but none have usage", () => {
|
|
59
|
-
const usage = makeUsage({ promptTokens: 100, totalTokens: 100 });
|
|
60
|
-
const exec = makeExecution(usage, [
|
|
61
|
-
{ name: "researcher" },
|
|
62
|
-
{ name: "coder" },
|
|
63
|
-
]);
|
|
64
|
-
const result = aggregateUsage(exec);
|
|
65
|
-
expect(result).toBe(usage);
|
|
66
|
-
});
|
|
67
|
-
it("sums token counts across main and sub-agents", () => {
|
|
68
|
-
const main = makeUsage({
|
|
69
|
-
promptTokens: 100,
|
|
70
|
-
completionTokens: 50,
|
|
71
|
-
totalTokens: 150,
|
|
72
|
-
cacheCreationTokens: 10,
|
|
73
|
-
cacheReadTokens: 20,
|
|
74
|
-
});
|
|
75
|
-
const sub1 = makeUsage({
|
|
76
|
-
promptTokens: 200,
|
|
77
|
-
completionTokens: 80,
|
|
78
|
-
totalTokens: 280,
|
|
79
|
-
cacheCreationTokens: 5,
|
|
80
|
-
cacheReadTokens: 30,
|
|
81
|
-
});
|
|
82
|
-
const sub2 = makeUsage({
|
|
83
|
-
promptTokens: 50,
|
|
84
|
-
completionTokens: 20,
|
|
85
|
-
totalTokens: 70,
|
|
86
|
-
});
|
|
87
|
-
const exec = makeExecution(main, [
|
|
88
|
-
{ name: "sub1", usage: sub1 },
|
|
89
|
-
{ name: "sub2", usage: sub2 },
|
|
90
|
-
]);
|
|
91
|
-
const result = aggregateUsage(exec);
|
|
92
|
-
expect(result.promptTokens).toBe(350);
|
|
93
|
-
expect(result.completionTokens).toBe(150);
|
|
94
|
-
expect(result.totalTokens).toBe(500);
|
|
95
|
-
expect(result.cacheCreationTokens).toBe(15);
|
|
96
|
-
expect(result.cacheReadTokens).toBe(50);
|
|
97
|
-
});
|
|
98
|
-
it("sums cost across main and sub-agents", () => {
|
|
99
|
-
const main = makeUsage({ estimatedCostUsd: 0.01 });
|
|
100
|
-
const sub = makeUsage({ estimatedCostUsd: 0.005 });
|
|
101
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
102
|
-
const result = aggregateUsage(exec);
|
|
103
|
-
expect(result.estimatedCostUsd).toBeCloseTo(0.015);
|
|
104
|
-
});
|
|
105
|
-
it("sums durations across main and sub-agents", () => {
|
|
106
|
-
const main = makeUsage({
|
|
107
|
-
totalDurationMs: 5000,
|
|
108
|
-
llmDurationMs: 3000,
|
|
109
|
-
toolDurationMs: 1500,
|
|
110
|
-
approvalWaitDurationMs: 500,
|
|
111
|
-
});
|
|
112
|
-
const sub = makeUsage({
|
|
113
|
-
totalDurationMs: 2000,
|
|
114
|
-
llmDurationMs: 1000,
|
|
115
|
-
toolDurationMs: 800,
|
|
116
|
-
approvalWaitDurationMs: 200,
|
|
117
|
-
});
|
|
118
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
119
|
-
const result = aggregateUsage(exec);
|
|
120
|
-
expect(result.totalDurationMs).toBe(7000);
|
|
121
|
-
expect(result.llmDurationMs).toBe(4000);
|
|
122
|
-
expect(result.toolDurationMs).toBe(2300);
|
|
123
|
-
expect(result.approvalWaitDurationMs).toBe(700);
|
|
124
|
-
});
|
|
125
|
-
it("uses main agent primaryModel and primaryProvider", () => {
|
|
126
|
-
const main = makeUsage({
|
|
127
|
-
primaryModel: "claude-sonnet-4",
|
|
128
|
-
primaryProvider: "anthropic",
|
|
129
|
-
});
|
|
130
|
-
const sub = makeUsage({
|
|
131
|
-
primaryModel: "gpt-4o",
|
|
132
|
-
primaryProvider: "openai",
|
|
133
|
-
});
|
|
134
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
135
|
-
const result = aggregateUsage(exec);
|
|
136
|
-
expect(result.primaryModel).toBe("claude-sonnet-4");
|
|
137
|
-
expect(result.primaryProvider).toBe("anthropic");
|
|
138
|
-
});
|
|
139
|
-
it("merges modelBreakdown entries with same model+provider key", () => {
|
|
140
|
-
const main = makeUsage({
|
|
141
|
-
modelBreakdown: [
|
|
142
|
-
makeModelUsage("claude-sonnet-4", "anthropic", {
|
|
143
|
-
inputTokens: 100,
|
|
144
|
-
outputTokens: 50,
|
|
145
|
-
callCount: 2,
|
|
146
|
-
estimatedCostUsd: 0.005,
|
|
147
|
-
}),
|
|
148
|
-
],
|
|
149
|
-
});
|
|
150
|
-
const sub = makeUsage({
|
|
151
|
-
modelBreakdown: [
|
|
152
|
-
makeModelUsage("claude-sonnet-4", "anthropic", {
|
|
153
|
-
inputTokens: 200,
|
|
154
|
-
outputTokens: 80,
|
|
155
|
-
callCount: 3,
|
|
156
|
-
estimatedCostUsd: 0.008,
|
|
157
|
-
}),
|
|
158
|
-
],
|
|
159
|
-
});
|
|
160
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
161
|
-
const result = aggregateUsage(exec);
|
|
162
|
-
expect(result.modelBreakdown).toHaveLength(1);
|
|
163
|
-
expect(result.modelBreakdown[0].model).toBe("claude-sonnet-4");
|
|
164
|
-
expect(result.modelBreakdown[0].provider).toBe("anthropic");
|
|
165
|
-
expect(result.modelBreakdown[0].inputTokens).toBe(300);
|
|
166
|
-
expect(result.modelBreakdown[0].outputTokens).toBe(130);
|
|
167
|
-
expect(result.modelBreakdown[0].callCount).toBe(5);
|
|
168
|
-
expect(result.modelBreakdown[0].estimatedCostUsd).toBeCloseTo(0.013);
|
|
169
|
-
});
|
|
170
|
-
it("keeps modelBreakdown entries with different models separate", () => {
|
|
171
|
-
const main = makeUsage({
|
|
172
|
-
modelBreakdown: [
|
|
173
|
-
makeModelUsage("claude-sonnet-4", "anthropic", { callCount: 2 }),
|
|
174
|
-
],
|
|
175
|
-
});
|
|
176
|
-
const sub = makeUsage({
|
|
177
|
-
modelBreakdown: [
|
|
178
|
-
makeModelUsage("gpt-4o", "openai", { callCount: 1 }),
|
|
179
|
-
],
|
|
180
|
-
});
|
|
181
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
182
|
-
const result = aggregateUsage(exec);
|
|
183
|
-
expect(result.modelBreakdown).toHaveLength(2);
|
|
184
|
-
const models = result.modelBreakdown.map((m) => m.model).sort();
|
|
185
|
-
expect(models).toEqual(["claude-sonnet-4", "gpt-4o"]);
|
|
186
|
-
});
|
|
187
|
-
it("concatenates llmCalls from all agents sorted by timestamp", () => {
|
|
188
|
-
const main = makeUsage({
|
|
189
|
-
llmCalls: [
|
|
190
|
-
makeLlmCall(1, "2026-03-19T10:00:00Z"),
|
|
191
|
-
makeLlmCall(2, "2026-03-19T10:01:00Z"),
|
|
192
|
-
],
|
|
193
|
-
});
|
|
194
|
-
const sub = makeUsage({
|
|
195
|
-
llmCalls: [
|
|
196
|
-
makeLlmCall(1, "2026-03-19T10:00:30Z"),
|
|
197
|
-
],
|
|
198
|
-
});
|
|
199
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
200
|
-
const result = aggregateUsage(exec);
|
|
201
|
-
expect(result.llmCalls).toHaveLength(3);
|
|
202
|
-
expect(result.llmCalls[0].timestamp).toBe("2026-03-19T10:00:00Z");
|
|
203
|
-
expect(result.llmCalls[1].timestamp).toBe("2026-03-19T10:00:30Z");
|
|
204
|
-
expect(result.llmCalls[2].timestamp).toBe("2026-03-19T10:01:00Z");
|
|
205
|
-
});
|
|
206
|
-
it("skips sub-agents with undefined usage gracefully", () => {
|
|
207
|
-
const main = makeUsage({
|
|
208
|
-
promptTokens: 100,
|
|
209
|
-
llmCallCount: 2,
|
|
210
|
-
});
|
|
211
|
-
const subWithUsage = makeUsage({
|
|
212
|
-
promptTokens: 50,
|
|
213
|
-
llmCallCount: 1,
|
|
214
|
-
});
|
|
215
|
-
const exec = makeExecution(main, [
|
|
216
|
-
{ name: "has-usage", usage: subWithUsage },
|
|
217
|
-
{ name: "no-usage" },
|
|
218
|
-
]);
|
|
219
|
-
const result = aggregateUsage(exec);
|
|
220
|
-
expect(result.promptTokens).toBe(150);
|
|
221
|
-
expect(result.llmCallCount).toBe(3);
|
|
222
|
-
});
|
|
223
|
-
it("sums toolResultCharsTruncated as bigint", () => {
|
|
224
|
-
const main = makeUsage({ toolResultCharsTruncated: 1000n });
|
|
225
|
-
const sub = makeUsage({ toolResultCharsTruncated: 500n });
|
|
226
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
227
|
-
const result = aggregateUsage(exec);
|
|
228
|
-
expect(result.toolResultCharsTruncated).toBe(1500n);
|
|
229
|
-
});
|
|
230
|
-
it("preserves pricing rates from first model entry", () => {
|
|
231
|
-
const main = makeUsage({
|
|
232
|
-
modelBreakdown: [
|
|
233
|
-
makeModelUsage("claude-sonnet-4", "anthropic", {
|
|
234
|
-
inputPricePerMillion: 3.0,
|
|
235
|
-
outputPricePerMillion: 15.0,
|
|
236
|
-
callCount: 1,
|
|
237
|
-
}),
|
|
238
|
-
],
|
|
239
|
-
});
|
|
240
|
-
const sub = makeUsage({
|
|
241
|
-
modelBreakdown: [
|
|
242
|
-
makeModelUsage("claude-sonnet-4", "anthropic", {
|
|
243
|
-
inputPricePerMillion: 3.0,
|
|
244
|
-
outputPricePerMillion: 15.0,
|
|
245
|
-
callCount: 1,
|
|
246
|
-
}),
|
|
247
|
-
],
|
|
248
|
-
});
|
|
249
|
-
const exec = makeExecution(main, [{ name: "sub", usage: sub }]);
|
|
250
|
-
const result = aggregateUsage(exec);
|
|
251
|
-
expect(result.modelBreakdown[0].inputPricePerMillion).toBe(3.0);
|
|
252
|
-
expect(result.modelBreakdown[0].outputPricePerMillion).toBe(15.0);
|
|
253
|
-
});
|
|
254
|
-
});
|
|
255
|
-
// ---------------------------------------------------------------------------
|
|
256
|
-
// useExecutionUsage — hook tests
|
|
257
|
-
// ---------------------------------------------------------------------------
|
|
258
|
-
describe("useExecutionUsage", () => {
|
|
259
|
-
it("returns null usage and zero metadata for null execution", () => {
|
|
260
|
-
const { result } = renderHook(() => useExecutionUsage(null));
|
|
261
|
-
expect(result.current.usage).toBeNull();
|
|
262
|
-
expect(result.current.hasSubAgentUsage).toBe(false);
|
|
263
|
-
expect(result.current.subAgentUsageCount).toBe(0);
|
|
264
|
-
});
|
|
265
|
-
it("returns aggregated UsageMetrics for valid execution", () => {
|
|
266
|
-
const main = makeUsage({
|
|
267
|
-
promptTokens: 100,
|
|
268
|
-
estimatedCostUsd: 0.01,
|
|
269
|
-
primaryModel: "claude-sonnet-4",
|
|
270
|
-
});
|
|
271
|
-
const exec = makeExecution(main);
|
|
272
|
-
const { result } = renderHook(() => useExecutionUsage(exec));
|
|
273
|
-
expect(result.current.usage).not.toBeNull();
|
|
274
|
-
expect(result.current.usage.promptTokens).toBe(100);
|
|
275
|
-
expect(result.current.usage.estimatedCostUsd).toBeCloseTo(0.01);
|
|
276
|
-
expect(result.current.usage.primaryModel).toBe("claude-sonnet-4");
|
|
277
|
-
expect(result.current.hasSubAgentUsage).toBe(false);
|
|
278
|
-
expect(result.current.subAgentUsageCount).toBe(0);
|
|
279
|
-
});
|
|
280
|
-
it("returns stable reference when execution has not changed", () => {
|
|
281
|
-
const exec = makeExecution(makeUsage({ promptTokens: 100 }));
|
|
282
|
-
const { result, rerender } = renderHook(() => useExecutionUsage(exec));
|
|
283
|
-
const first = result.current;
|
|
284
|
-
rerender();
|
|
285
|
-
const second = result.current;
|
|
286
|
-
expect(first).toBe(second);
|
|
287
|
-
});
|
|
288
|
-
it("sets hasSubAgentUsage and subAgentUsageCount correctly", () => {
|
|
289
|
-
const main = makeUsage({ promptTokens: 100 });
|
|
290
|
-
const sub1 = makeUsage({ promptTokens: 50 });
|
|
291
|
-
const sub2 = makeUsage({ promptTokens: 30 });
|
|
292
|
-
const exec = makeExecution(main, [
|
|
293
|
-
{ name: "sub1", usage: sub1 },
|
|
294
|
-
{ name: "sub2", usage: sub2 },
|
|
295
|
-
{ name: "sub3-no-usage" },
|
|
296
|
-
]);
|
|
297
|
-
const { result } = renderHook(() => useExecutionUsage(exec));
|
|
298
|
-
expect(result.current.hasSubAgentUsage).toBe(true);
|
|
299
|
-
expect(result.current.subAgentUsageCount).toBe(2);
|
|
300
|
-
expect(result.current.usage.promptTokens).toBe(180);
|
|
301
|
-
});
|
|
302
|
-
});
|
|
303
|
-
//# sourceMappingURL=useExecutionUsage.test.js.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"useExecutionUsage.test.js","sourceRoot":"","sources":["../../../src/execution/__tests__/useExecutionUsage.test.tsx"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EACL,oBAAoB,EACpB,0BAA0B,GAE3B,MAAM,6DAA6D,CAAC;AACrE,OAAO,EAAE,uBAAuB,EAAE,MAAM,kEAAkE,CAAC;AAC3G,OAAO,EACL,kBAAkB,EAClB,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,+DAA+D,CAAC;AACvE,OAAO,EAAE,cAAc,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAEzE,8EAA8E;AAC9E,eAAe;AACf,8EAA8E;AAE9E,SAAS,SAAS,CAChB,YAiBK,EAAE;IAEP,OAAO,MAAM,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,cAAc,CACrB,KAAa,EACb,QAAgB,EAChB,YASK,EAAE;IAEP,OAAO,MAAM,CAAC,gBAAgB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;AACrE,CAAC;AAED,SAAS,WAAW,CAClB,QAAgB,EAChB,SAAiB,EACjB,YAOK,EAAE;IAEP,OAAO,MAAM,CAAC,oBAAoB,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,GAAG,SAAS,EAAE,CAAC,CAAC;AAC7E,CAAC;AAED,SAAS,aAAa,CACpB,SAAwC,EACxC,YAGK,EAAE;IAEP,MAAM,MAAM,GAAG,MAAM,CAAC,0BAA0B,EAAE;QAChD,KAAK,EAAE,SAAS;QAChB,kBAAkB,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CACtC,MAAM,CAAC,uBAAuB,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAClE;KACF,CAAC,CAAC;IACH,OAAO,MAAM,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;AAClD,CAAC;AAED,8EAA8E;AAC9E,uCAAuC;AACvC,8EAA8E;AAE9E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;QAC7C,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,IAAI,GAAG,MAAM,CAAC,oBAAoB,CAAC,CAAC;QAC1C,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,IAAI,GAAG,MAAM,CAAC,oBAAoB,EAAE;YACxC,MAAM,EAAE,MAAM,CAAC,0BAA0B,CAAC;SAC3C,CAAC,CAAC;QACH,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK,GAAG,SAAS,CAAC;YACtB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,YAAY,EAAE,CAAC;YACf,gBAAgB,EAAE,KAAK;YACvB,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;SAC7B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QAElC,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mEAAmE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QACjE,MAAM,IAAI,GAAG,aAAa,CAAC,KAAK,EAAE;YAChC,EAAE,IAAI,EAAE,YAAY,EAAE;YACtB,EAAE,IAAI,EAAE,OAAO,EAAE;SAClB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;QACpC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,mBAAmB,EAAE,EAAE;YACvB,eAAe,EAAE,EAAE;SACpB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,GAAG;YAChB,mBAAmB,EAAE,CAAC;YACtB,eAAe,EAAE,EAAE;SACpB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,EAAE;YAChB,gBAAgB,EAAE,EAAE;YACpB,WAAW,EAAE,EAAE;SAChB,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;SAC9B,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC;QACnD,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACnD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,eAAe,EAAE,IAAI;YACrB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,IAAI;YACpB,sBAAsB,EAAE,GAAG;SAC5B,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,eAAe,EAAE,IAAI;YACrB,aAAa,EAAE,IAAI;YACnB,cAAc,EAAE,GAAG;YACnB,sBAAsB,EAAE,GAAG;SAC5B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,MAAM,CAAC,sBAAsB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAClD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,iBAAiB;YAC/B,eAAe,EAAE,WAAW;SAC7B,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,YAAY,EAAE,QAAQ;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACpE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,EAAE;oBAChB,SAAS,EAAE,CAAC;oBACZ,gBAAgB,EAAE,KAAK;iBACxB,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,WAAW,EAAE,GAAG;oBAChB,YAAY,EAAE,EAAE;oBAChB,SAAS,EAAE,CAAC;oBACZ,gBAAgB,EAAE,KAAK;iBACxB,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACvD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aACjE;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC;aACrD;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAE9C,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,iBAAiB,EAAE,QAAQ,CAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;QACnE,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,QAAQ,EAAE;gBACR,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;gBACtC,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;aACvC;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,QAAQ,EAAE;gBACR,WAAW,CAAC,CAAC,EAAE,sBAAsB,CAAC;aACvC;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,SAAS,CAAC;YAC7B,YAAY,EAAE,EAAE;YAChB,YAAY,EAAE,CAAC;SAChB,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,WAAW,EAAE,KAAK,EAAE,YAAY,EAAE;YAC1C,EAAE,IAAI,EAAE,UAAU,EAAE;SACrB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,wBAAwB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC5D,MAAM,GAAG,GAAG,SAAS,CAAC,EAAE,wBAAwB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC1D,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;QACxD,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,oBAAoB,EAAE,GAAG;oBACzB,qBAAqB,EAAE,IAAI;oBAC3B,SAAS,EAAE,CAAC;iBACb,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC;YACpB,cAAc,EAAE;gBACd,cAAc,CAAC,iBAAiB,EAAE,WAAW,EAAE;oBAC7C,oBAAoB,EAAE,GAAG;oBACzB,qBAAqB,EAAE,IAAI;oBAC3B,SAAS,EAAE,CAAC;iBACb,CAAC;aACH;SACF,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAEhE,MAAM,MAAM,GAAG,cAAc,CAAC,IAAI,CAAE,CAAC;QACrC,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,8EAA8E;AAC9E,iCAAiC;AACjC,8EAA8E;AAE9E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,QAAQ,EAAE,CAAC;QACxC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;QAC7D,MAAM,IAAI,GAAG,SAAS,CAAC;YACrB,YAAY,EAAE,GAAG;YACjB,gBAAgB,EAAE,IAAI;YACtB,YAAY,EAAE,iBAAiB;SAChC,CAAC,CAAC;QACH,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;QAEjC,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;QAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,gBAAgB,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACnE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;QACjE,MAAM,IAAI,GAAG,aAAa,CAAC,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QAC7D,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAEvE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC;QAC7B,QAAQ,EAAE,CAAC;QACX,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;QAE9B,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;QAChE,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,SAAS,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,EAAE;YAC/B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE;YAC7B,EAAE,IAAI,EAAE,eAAe,EAAE;SAC1B,CAAC,CAAC;QAEH,MAAM,EAAE,MAAM,EAAE,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,KAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IACvD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -1,45 +0,0 @@
|
|
|
1
|
-
import type { AgentExecution } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/api_pb";
|
|
2
|
-
import type { UsageMetrics } from "@stigmer/protos/ai/stigmer/agentic/agentexecution/v1/usage_pb";
|
|
3
|
-
export interface UseExecutionUsageReturn {
|
|
4
|
-
/** Aggregated usage (main agent + sub-agents). Null before usage data arrives. */
|
|
5
|
-
readonly usage: UsageMetrics | null;
|
|
6
|
-
/** Whether any sub-agent contributed usage to the aggregated total. */
|
|
7
|
-
readonly hasSubAgentUsage: boolean;
|
|
8
|
-
/** Count of sub-agents that have non-null usage data. */
|
|
9
|
-
readonly subAgentUsageCount: number;
|
|
10
|
-
}
|
|
11
|
-
/**
|
|
12
|
-
* Pure derivation hook that aggregates {@link UsageMetrics} across the
|
|
13
|
-
* main agent and all sub-agents into a single {@link UsageMetrics} total.
|
|
14
|
-
*
|
|
15
|
-
* The proto scoping rule is:
|
|
16
|
-
* > `status.usage` = main agent's direct LLM usage (excludes sub-agents)
|
|
17
|
-
* > `subAgentExecutions[].usage` = each sub-agent's LLM usage
|
|
18
|
-
* > Total cost = `status.usage` + sum(`subAgentExecutions[].usage`)
|
|
19
|
-
*
|
|
20
|
-
* This hook performs that summation, merges `modelBreakdown` entries by
|
|
21
|
-
* `model+provider` key, and concatenates `llmCalls` sorted by timestamp.
|
|
22
|
-
*
|
|
23
|
-
* Returns `null` when the execution is absent or usage data has not yet
|
|
24
|
-
* arrived from the agent runner.
|
|
25
|
-
*
|
|
26
|
-
* @example
|
|
27
|
-
* ```tsx
|
|
28
|
-
* const { execution } = useExecutionStream(executionId);
|
|
29
|
-
* const { usage, hasSubAgentUsage } = useExecutionUsage(execution);
|
|
30
|
-
*
|
|
31
|
-
* if (usage) {
|
|
32
|
-
* console.log(`Cost: $${usage.estimatedCostUsd}`);
|
|
33
|
-
* console.log(`Tokens: ${usage.totalTokens}`);
|
|
34
|
-
* }
|
|
35
|
-
* ```
|
|
36
|
-
*/
|
|
37
|
-
export declare function useExecutionUsage(execution: AgentExecution | null): UseExecutionUsageReturn;
|
|
38
|
-
/**
|
|
39
|
-
* Aggregates usage metrics from the main agent and all sub-agents into
|
|
40
|
-
* a single {@link UsageMetrics} proto object.
|
|
41
|
-
*
|
|
42
|
-
* Returns `null` when the execution or its usage data is not yet available.
|
|
43
|
-
*/
|
|
44
|
-
export declare function aggregateUsage(execution: AgentExecution | null): UsageMetrics | null;
|
|
45
|
-
//# sourceMappingURL=useExecutionUsage.d.ts.map
|
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"useExecutionUsage.d.ts","sourceRoot":"","sources":["../../src/execution/useExecutionUsage.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,6DAA6D,CAAC;AAClG,OAAO,KAAK,EACV,YAAY,EAEb,MAAM,+DAA+D,CAAC;AAMvE,MAAM,WAAW,uBAAuB;IACtC,kFAAkF;IAClF,QAAQ,CAAC,KAAK,EAAE,YAAY,GAAG,IAAI,CAAC;IACpC,uEAAuE;IACvE,QAAQ,CAAC,gBAAgB,EAAE,OAAO,CAAC;IACnC,yDAAyD;IACzD,QAAQ,CAAC,kBAAkB,EAAE,MAAM,CAAC;CACrC;AAED;;;;;;;;;;;;;;;;;;;;;;;;;GAyBG;AACH,wBAAgB,iBAAiB,CAC/B,SAAS,EAAE,cAAc,GAAG,IAAI,GAC/B,uBAAuB,CAoBzB;AAMD;;;;;GAKG;AACH,wBAAgB,cAAc,CAC5B,SAAS,EAAE,cAAc,GAAG,IAAI,GAC/B,YAAY,GAAG,IAAI,CAgCrB"}
|