@simonepri/refined-antigravity-acp 1.5.0 → 1.6.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/src/fixes/flattened-model-efforts/index.d.ts.map +1 -1
- package/dist/src/fixes/flattened-model-efforts/index.js +15 -2
- package/dist/src/fixes/flattened-model-efforts/index.js.map +1 -1
- package/dist/src/fixes/index.d.ts +1 -0
- package/dist/src/fixes/index.d.ts.map +1 -1
- package/dist/src/fixes/index.js +5 -0
- package/dist/src/fixes/index.js.map +1 -1
- package/dist/src/otel/config.d.ts +19 -0
- package/dist/src/otel/config.d.ts.map +1 -0
- package/dist/src/otel/config.js +121 -0
- package/dist/src/otel/config.js.map +1 -0
- package/dist/src/otel/exporter.d.ts +25 -0
- package/dist/src/otel/exporter.d.ts.map +1 -0
- package/dist/src/otel/exporter.js +102 -0
- package/dist/src/otel/exporter.js.map +1 -0
- package/dist/src/otel/fix.d.ts +24 -0
- package/dist/src/otel/fix.d.ts.map +1 -0
- package/dist/src/otel/fix.js +170 -0
- package/dist/src/otel/fix.js.map +1 -0
- package/dist/src/otel/index.d.ts +10 -0
- package/dist/src/otel/index.d.ts.map +1 -0
- package/dist/src/otel/index.js +10 -0
- package/dist/src/otel/index.js.map +1 -0
- package/dist/src/otel/tracker.d.ts +73 -0
- package/dist/src/otel/tracker.d.ts.map +1 -0
- package/dist/src/otel/tracker.js +339 -0
- package/dist/src/otel/tracker.js.map +1 -0
- package/dist/src/otel/types.d.ts +93 -0
- package/dist/src/otel/types.d.ts.map +1 -0
- package/dist/src/otel/types.js +49 -0
- package/dist/src/otel/types.js.map +1 -0
- package/dist/src/otel/utf8.d.ts +15 -0
- package/dist/src/otel/utf8.d.ts.map +1 -0
- package/dist/src/otel/utf8.js +49 -0
- package/dist/src/otel/utf8.js.map +1 -0
- package/package.json +1 -1
- package/readme.md +51 -33
|
@@ -0,0 +1,170 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* OpenTelemetry ACP Telemetry Fix.
|
|
3
|
+
* Listens to ACP session prompt streams and exports turn spans, tool child spans,
|
|
4
|
+
* and Antigravity CLI log events in the background.
|
|
5
|
+
*/
|
|
6
|
+
import { ACP_METHODS, SESSION_UPDATES, isJsonRpcErrorResponse, isJsonRpcRequest, isJsonRpcResponse, isJsonRpcSuccessResponse, } from "../core/types.js";
|
|
7
|
+
import { extractSessionId } from "../core/session-cache.js";
|
|
8
|
+
import { getOtelConfig } from "./config.js";
|
|
9
|
+
import { OtlpExporter } from "./exporter.js";
|
|
10
|
+
import { SessionTurnTracker } from "./tracker.js";
|
|
11
|
+
function handlePromptResponse(msg, sessionId, tracker, customDbPath) {
|
|
12
|
+
if (isJsonRpcErrorResponse(msg)) {
|
|
13
|
+
tracker.finishTurn(sessionId, {
|
|
14
|
+
promptRequestId: msg.id,
|
|
15
|
+
isError: true,
|
|
16
|
+
errorMessage: msg.error.message,
|
|
17
|
+
customDbPath,
|
|
18
|
+
});
|
|
19
|
+
return;
|
|
20
|
+
}
|
|
21
|
+
const resultObj = msg.result && typeof msg.result === "object" ? msg.result : {};
|
|
22
|
+
const stopReason = typeof resultObj.stopReason === "string" ? resultObj.stopReason : undefined;
|
|
23
|
+
const usage = resultObj.usage;
|
|
24
|
+
if (usage) {
|
|
25
|
+
tracker.recordExplicitUsage(sessionId, usage);
|
|
26
|
+
}
|
|
27
|
+
tracker.finishTurn(sessionId, {
|
|
28
|
+
promptRequestId: msg.id,
|
|
29
|
+
stopReason,
|
|
30
|
+
customDbPath,
|
|
31
|
+
});
|
|
32
|
+
}
|
|
33
|
+
function handleToolUpdate(update, sessionId, tracker) {
|
|
34
|
+
const call = update;
|
|
35
|
+
const callId = call.toolCallId || call.callId || call.id;
|
|
36
|
+
if (!callId)
|
|
37
|
+
return;
|
|
38
|
+
if (update.sessionUpdate === SESSION_UPDATES.TOOL_CALL) {
|
|
39
|
+
tracker.recordToolCall(sessionId, callId, call.title || call.tool || call.name || "tool");
|
|
40
|
+
}
|
|
41
|
+
else if (update.sessionUpdate === SESSION_UPDATES.TOOL_CALL_UPDATE && call.status) {
|
|
42
|
+
tracker.updateToolCall(sessionId, callId, call.status);
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
function handleNonToolUpdate(update, sessionId, tracker) {
|
|
46
|
+
const su = update.sessionUpdate;
|
|
47
|
+
if (su === SESSION_UPDATES.AGENT_MESSAGE_CHUNK) {
|
|
48
|
+
const text = update.content?.text;
|
|
49
|
+
if (typeof text === "string")
|
|
50
|
+
tracker.appendResponseChunk(sessionId, text);
|
|
51
|
+
}
|
|
52
|
+
else if (su === SESSION_UPDATES.USAGE_UPDATE) {
|
|
53
|
+
const usage = update;
|
|
54
|
+
if (typeof usage.used === "number")
|
|
55
|
+
tracker.recordUsageUpdate(sessionId, usage.used, usage.size);
|
|
56
|
+
}
|
|
57
|
+
else if (su === SESSION_UPDATES.CONFIG_OPTION_UPDATE) {
|
|
58
|
+
const configOpt = update;
|
|
59
|
+
if (configOpt.configId === "model" && typeof configOpt.value === "string") {
|
|
60
|
+
tracker.setModel(sessionId, configOpt.value);
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
function handleUpdate(params, tracker, customDbPathGetter) {
|
|
65
|
+
const { sessionId, update } = params;
|
|
66
|
+
if (!sessionId || !update)
|
|
67
|
+
return;
|
|
68
|
+
const su = update.sessionUpdate;
|
|
69
|
+
if (su === SESSION_UPDATES.TOOL_CALL || su === SESSION_UPDATES.TOOL_CALL_UPDATE) {
|
|
70
|
+
handleToolUpdate(update, sessionId, tracker);
|
|
71
|
+
}
|
|
72
|
+
else {
|
|
73
|
+
handleNonToolUpdate(update, sessionId, tracker);
|
|
74
|
+
}
|
|
75
|
+
const rawUpdate = update;
|
|
76
|
+
if (typeof rawUpdate.stopReason === "string") {
|
|
77
|
+
const customDbPath = customDbPathGetter ? customDbPathGetter(sessionId) : undefined;
|
|
78
|
+
tracker.finishTurn(sessionId, { stopReason: rawUpdate.stopReason, customDbPath });
|
|
79
|
+
}
|
|
80
|
+
}
|
|
81
|
+
function handleConfigChange(msg, sessionId, tracker) {
|
|
82
|
+
if (msg.method === ACP_METHODS.SESSION_SET_MODEL) {
|
|
83
|
+
const params = msg.params;
|
|
84
|
+
if (params?.modelId)
|
|
85
|
+
tracker.setModel(sessionId, params.modelId);
|
|
86
|
+
}
|
|
87
|
+
else if (msg.method === ACP_METHODS.SESSION_SET_CONFIG_OPTION) {
|
|
88
|
+
const params = msg.params;
|
|
89
|
+
if (params?.configId === "model" && typeof params.value === "string") {
|
|
90
|
+
tracker.setModel(sessionId, params.value);
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
function handleOutbound(msg, tracker, pendingPromptSessions) {
|
|
95
|
+
const sessionId = extractSessionId(msg);
|
|
96
|
+
if (!sessionId)
|
|
97
|
+
return;
|
|
98
|
+
if (msg.method === ACP_METHODS.SESSION_PROMPT) {
|
|
99
|
+
pendingPromptSessions.set(msg.id, sessionId);
|
|
100
|
+
const params = msg.params;
|
|
101
|
+
if (params?.prompt !== undefined) {
|
|
102
|
+
tracker.startTurn(sessionId, msg.id, params.prompt);
|
|
103
|
+
}
|
|
104
|
+
}
|
|
105
|
+
else if (msg.method === ACP_METHODS.SESSION_CANCEL) {
|
|
106
|
+
tracker.finishTurn(sessionId, { stopReason: "cancelled", isError: true });
|
|
107
|
+
}
|
|
108
|
+
else {
|
|
109
|
+
handleConfigChange(msg, sessionId, tracker);
|
|
110
|
+
}
|
|
111
|
+
}
|
|
112
|
+
function handleModelDiscovery(msg, sessionId, tracker) {
|
|
113
|
+
if (!sessionId || !isJsonRpcSuccessResponse(msg))
|
|
114
|
+
return;
|
|
115
|
+
const res = msg.result && typeof msg.result === "object"
|
|
116
|
+
? msg.result
|
|
117
|
+
: {};
|
|
118
|
+
if (res.models?.currentModelId)
|
|
119
|
+
tracker.setModel(sessionId, res.models.currentModelId);
|
|
120
|
+
}
|
|
121
|
+
function handleResponse(msg, context, tracker, pendingPromptSessions, customDbPathGetter) {
|
|
122
|
+
const pendingSessionId = pendingPromptSessions.get(msg.id);
|
|
123
|
+
if (pendingSessionId) {
|
|
124
|
+
pendingPromptSessions.delete(msg.id);
|
|
125
|
+
const dbPath = customDbPathGetter ? customDbPathGetter(pendingSessionId) : undefined;
|
|
126
|
+
handlePromptResponse(msg, pendingSessionId, tracker, dbPath);
|
|
127
|
+
return;
|
|
128
|
+
}
|
|
129
|
+
const fallbackSessionId = context.session?.sessionId ?? extractSessionId(msg);
|
|
130
|
+
handleModelDiscovery(msg, fallbackSessionId, tracker);
|
|
131
|
+
}
|
|
132
|
+
export function createTelemetryFix(options) {
|
|
133
|
+
const config = options?.config ?? getOtelConfig();
|
|
134
|
+
const exporter = options?.exporter ?? new OtlpExporter(config);
|
|
135
|
+
const tracker = new SessionTurnTracker(config, exporter);
|
|
136
|
+
const pendingPromptSessions = new Map();
|
|
137
|
+
return {
|
|
138
|
+
name: "opentelemetry",
|
|
139
|
+
description: "Exports ACP session prompts, agent responses, tool calls, and model metrics over OpenTelemetry OTLP",
|
|
140
|
+
tracker,
|
|
141
|
+
exporter,
|
|
142
|
+
config,
|
|
143
|
+
onOutbound: (msg, _context) => {
|
|
144
|
+
if (!config.enabled || !isJsonRpcRequest(msg))
|
|
145
|
+
return msg;
|
|
146
|
+
if (!String(msg.id).startsWith("__refined_agy_recycle_")) {
|
|
147
|
+
handleOutbound(msg, tracker, pendingPromptSessions);
|
|
148
|
+
}
|
|
149
|
+
return msg;
|
|
150
|
+
},
|
|
151
|
+
onInbound: async (msg, context) => {
|
|
152
|
+
if (!config.enabled || context.isRecycling)
|
|
153
|
+
return [msg];
|
|
154
|
+
if (isJsonRpcResponse(msg)) {
|
|
155
|
+
handleResponse(msg, context, tracker, pendingPromptSessions, options?.customDbPathGetter);
|
|
156
|
+
}
|
|
157
|
+
else if ("method" in msg && msg.method === ACP_METHODS.SESSION_UPDATE && msg.params) {
|
|
158
|
+
handleUpdate(msg.params, tracker, options?.customDbPathGetter);
|
|
159
|
+
}
|
|
160
|
+
return [msg];
|
|
161
|
+
},
|
|
162
|
+
onRecycle: () => { },
|
|
163
|
+
dispose: () => {
|
|
164
|
+
pendingPromptSessions.clear();
|
|
165
|
+
void exporter.shutdown();
|
|
166
|
+
},
|
|
167
|
+
};
|
|
168
|
+
}
|
|
169
|
+
export const telemetryFix = createTelemetryFix();
|
|
170
|
+
//# sourceMappingURL=fix.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fix.js","sourceRoot":"","sources":["../../../src/otel/fix.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EACL,WAAW,EACX,eAAe,EACf,sBAAsB,EACtB,gBAAgB,EAChB,iBAAiB,EACjB,wBAAwB,GASzB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC5D,OAAO,EAAE,aAAa,EAAmB,MAAM,aAAa,CAAC;AAC7D,OAAO,EAAE,YAAY,EAAsB,MAAM,eAAe,CAAC;AACjE,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAElD,SAAS,oBAAoB,CAC3B,GAAyB,EACzB,SAAiB,EACjB,OAA2B,EAC3B,YAAqB;IAErB,IAAI,sBAAsB,CAAC,GAAG,CAAC,EAAE,CAAC;QAChC,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE;YAC5B,eAAe,EAAE,GAAG,CAAC,EAAE;YACvB,OAAO,EAAE,IAAI;YACb,YAAY,EAAE,GAAG,CAAC,KAAK,CAAC,OAAO;YAC/B,YAAY;SACb,CAAC,CAAC;QACH,OAAO;IACT,CAAC;IAED,MAAM,SAAS,GACb,GAAG,CAAC,MAAM,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAE,GAAG,CAAC,MAAkC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC9F,MAAM,UAAU,GAAG,OAAO,SAAS,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;IAC/F,MAAM,KAAK,GAAG,SAAS,CAAC,KAEX,CAAC;IAEd,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,mBAAmB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAChD,CAAC;IAED,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE;QAC5B,eAAe,EAAE,GAAG,CAAC,EAAE;QACvB,UAAU;QACV,YAAY;KACb,CAAC,CAAC;AACL,CAAC;AAED,SAAS,gBAAgB,CACvB,MAAkD,EAClD,SAAiB,EACjB,OAA2B;IAE3B,MAAM,IAAI,GAAG,MAQZ,CAAC;IACF,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE,CAAC;IACzD,IAAI,CAAC,MAAM;QAAE,OAAO;IAEpB,IAAI,MAAM,CAAC,aAAa,KAAK,eAAe,CAAC,SAAS,EAAE,CAAC;QACvD,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,KAAK,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,CAAC;IAC5F,CAAC;SAAM,IAAI,MAAM,CAAC,aAAa,KAAK,eAAe,CAAC,gBAAgB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QACpF,OAAO,CAAC,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;IACzD,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAC1B,MAAkD,EAClD,SAAiB,EACjB,OAA2B;IAE3B,MAAM,EAAE,GAAG,MAAM,CAAC,aAAa,CAAC;IAChC,IAAI,EAAE,KAAK,eAAe,CAAC,mBAAmB,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAI,MAA0C,CAAC,OAAO,EAAE,IAAI,CAAC;QACvE,IAAI,OAAO,IAAI,KAAK,QAAQ;YAAE,OAAO,CAAC,mBAAmB,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC7E,CAAC;SAAM,IAAI,EAAE,KAAK,eAAe,CAAC,YAAY,EAAE,CAAC;QAC/C,MAAM,KAAK,GAAG,MAA0C,CAAC;QACzD,IAAI,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ;YAChC,OAAO,CAAC,iBAAiB,CAAC,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC;SAAM,IAAI,EAAE,KAAK,eAAe,CAAC,oBAAoB,EAAE,CAAC;QACvD,MAAM,SAAS,GAAG,MAAgD,CAAC;QACnE,IAAI,SAAS,CAAC,QAAQ,KAAK,OAAO,IAAI,OAAO,SAAS,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC1E,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,YAAY,CACnB,MAA2B,EAC3B,OAA2B,EAC3B,kBAAkD;IAElD,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IACrC,IAAI,CAAC,SAAS,IAAI,CAAC,MAAM;QAAE,OAAO;IAElC,MAAM,EAAE,GAAG,MAAM,CAAC,aAAa,CAAC;IAChC,IAAI,EAAE,KAAK,eAAe,CAAC,SAAS,IAAI,EAAE,KAAK,eAAe,CAAC,gBAAgB,EAAE,CAAC;QAChF,gBAAgB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC/C,CAAC;SAAM,CAAC;QACN,mBAAmB,CAAC,MAAM,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,SAAS,GAAG,MAAkC,CAAC;IACrD,IAAI,OAAO,SAAS,CAAC,UAAU,KAAK,QAAQ,EAAE,CAAC;QAC7C,MAAM,YAAY,GAAG,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACpF,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,UAAU,EAAE,SAAS,CAAC,UAAU,EAAE,YAAY,EAAE,CAAC,CAAC;IACpF,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CACzB,GAAmB,EACnB,SAAiB,EACjB,OAA2B;IAE3B,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC,iBAAiB,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,GAAG,CAAC,MAA0C,CAAC;QAC9D,IAAI,MAAM,EAAE,OAAO;YAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IACnE,CAAC;SAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC,yBAAyB,EAAE,CAAC;QAChE,MAAM,MAAM,GAAG,GAAG,CAAC,MAA4D,CAAC;QAChF,IAAI,MAAM,EAAE,QAAQ,KAAK,OAAO,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;YACrE,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;IACH,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CACrB,GAAmB,EACnB,OAA2B,EAC3B,qBAAmD;IAEnD,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;IACxC,IAAI,CAAC,SAAS;QAAE,OAAO;IAEvB,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC,cAAc,EAAE,CAAC;QAC9C,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QAC7C,MAAM,MAAM,GAAG,GAAG,CAAC,MAAyC,CAAC;QAC7D,IAAI,MAAM,EAAE,MAAM,KAAK,SAAS,EAAE,CAAC;YACjC,OAAO,CAAC,SAAS,CAAC,SAAS,EAAE,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;SAAM,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC,cAAc,EAAE,CAAC;QACrD,OAAO,CAAC,UAAU,CAAC,SAAS,EAAE,EAAE,UAAU,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5E,CAAC;SAAM,CAAC;QACN,kBAAkB,CAAC,GAAG,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;AACH,CAAC;AAED,SAAS,oBAAoB,CAC3B,GAAyB,EACzB,SAA6B,EAC7B,OAA2B;IAE3B,IAAI,CAAC,SAAS,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC;QAAE,OAAO;IACzD,MAAM,GAAG,GACP,GAAG,CAAC,MAAM,IAAI,OAAO,GAAG,CAAC,MAAM,KAAK,QAAQ;QAC1C,CAAC,CAAE,GAAG,CAAC,MAAmD;QAC1D,CAAC,CAAC,EAAE,CAAC;IACT,IAAI,GAAG,CAAC,MAAM,EAAE,cAAc;QAAE,OAAO,CAAC,QAAQ,CAAC,SAAS,EAAE,GAAG,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;AACzF,CAAC;AAED,SAAS,cAAc,CACrB,GAAyB,EACzB,OAAuB,EACvB,OAA2B,EAC3B,qBAAmD,EACnD,kBAAkD;IAElD,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC3D,IAAI,gBAAgB,EAAE,CAAC;QACrB,qBAAqB,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,MAAM,MAAM,GAAG,kBAAkB,CAAC,CAAC,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACrF,oBAAoB,CAAC,GAAG,EAAE,gBAAgB,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7D,OAAO;IACT,CAAC;IAED,MAAM,iBAAiB,GAAG,OAAO,CAAC,OAAO,EAAE,SAAS,IAAI,gBAAgB,CAAC,GAAG,CAAC,CAAC;IAC9E,oBAAoB,CAAC,GAAG,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,OAIlC;IACC,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,aAAa,EAAE,CAAC;IAClD,MAAM,QAAQ,GAAG,OAAO,EAAE,QAAQ,IAAI,IAAI,YAAY,CAAC,MAAM,CAAC,CAAC;IAC/D,MAAM,OAAO,GAAG,IAAI,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACzD,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAA2B,CAAC;IAEjE,OAAO;QACL,IAAI,EAAE,eAAe;QACrB,WAAW,EACT,qGAAqG;QACvG,OAAO;QACP,QAAQ;QACR,MAAM;QAEN,UAAU,EAAE,CAAC,GAAqB,EAAE,QAAyB,EAAoB,EAAE;YACjF,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC;gBAAE,OAAO,GAAG,CAAC;YAC1D,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,wBAAwB,CAAC,EAAE,CAAC;gBACzD,cAAc,CAAC,GAAG,EAAE,OAAO,EAAE,qBAAqB,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC;QAED,SAAS,EAAE,KAAK,EACd,GAAqB,EACrB,OAAuB,EACM,EAAE;YAC/B,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,OAAO,CAAC,WAAW;gBAAE,OAAO,CAAC,GAAG,CAAC,CAAC;YAEzD,IAAI,iBAAiB,CAAC,GAAG,CAAC,EAAE,CAAC;gBAC3B,cAAc,CAAC,GAAG,EAAE,OAAO,EAAE,OAAO,EAAE,qBAAqB,EAAE,OAAO,EAAE,kBAAkB,CAAC,CAAC;YAC5F,CAAC;iBAAM,IAAI,QAAQ,IAAI,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,WAAW,CAAC,cAAc,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC;gBACtF,YAAY,CAAC,GAAG,CAAC,MAA6B,EAAE,OAAO,EAAE,OAAO,EAAE,kBAAkB,CAAC,CAAC;YACxF,CAAC;YAED,OAAO,CAAC,GAAG,CAAC,CAAC;QACf,CAAC;QAED,SAAS,EAAE,GAAS,EAAE,GAAE,CAAC;QAEzB,OAAO,EAAE,GAAS,EAAE;YAClB,qBAAqB,CAAC,KAAK,EAAE,CAAC;YAC9B,KAAK,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAC3B,CAAC;KACF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,MAAM,YAAY,GAAG,kBAAkB,EAAE,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* OpenTelemetry OTLP integration for Refined Antigravity ACP.
|
|
3
|
+
*/
|
|
4
|
+
export * from "./types.js";
|
|
5
|
+
export * from "./config.js";
|
|
6
|
+
export * from "./utf8.js";
|
|
7
|
+
export * from "./exporter.js";
|
|
8
|
+
export * from "./tracker.js";
|
|
9
|
+
export * from "./fix.js";
|
|
10
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/otel/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC;AAC1B,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAC7B,cAAc,UAAU,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* OpenTelemetry OTLP integration for Refined Antigravity ACP.
|
|
3
|
+
*/
|
|
4
|
+
export * from "./types.js";
|
|
5
|
+
export * from "./config.js";
|
|
6
|
+
export * from "./utf8.js";
|
|
7
|
+
export * from "./exporter.js";
|
|
8
|
+
export * from "./tracker.js";
|
|
9
|
+
export * from "./fix.js";
|
|
10
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/otel/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,YAAY,CAAC;AAC3B,cAAc,aAAa,CAAC;AAC5B,cAAc,WAAW,CAAC;AAC1B,cAAc,eAAe,CAAC;AAC9B,cAAc,cAAc,CAAC;AAC7B,cAAc,UAAU,CAAC"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Session turn and telemetry tracker.
|
|
3
|
+
* Maps ACP session prompts, response streams, and tool executions to OpenTelemetry
|
|
4
|
+
* GenAI semantic conventions and Antigravity CLI log events.
|
|
5
|
+
*/
|
|
6
|
+
import type { OtelConfig } from "./config.js";
|
|
7
|
+
import type { IOtelExporter } from "./exporter.js";
|
|
8
|
+
import { type OtelSpan } from "./types.js";
|
|
9
|
+
import type { PromptContentItem } from "../core/types.js";
|
|
10
|
+
export declare function generateTraceId(): string;
|
|
11
|
+
export declare function generateSpanId(): string;
|
|
12
|
+
export declare function nowUnixNano(): string;
|
|
13
|
+
export interface InFlightToolCall {
|
|
14
|
+
toolCallId: string;
|
|
15
|
+
toolName: string;
|
|
16
|
+
spanId: string;
|
|
17
|
+
startTimeUnixNano: string;
|
|
18
|
+
endTimeUnixNano?: string;
|
|
19
|
+
status: "in_progress" | "completed" | "failed";
|
|
20
|
+
}
|
|
21
|
+
export interface InFlightTurn {
|
|
22
|
+
promptRequestId: string | number;
|
|
23
|
+
sessionId: string;
|
|
24
|
+
model: string;
|
|
25
|
+
traceId: string;
|
|
26
|
+
spanId: string;
|
|
27
|
+
startTimeUnixNano: string;
|
|
28
|
+
rawPromptText: string;
|
|
29
|
+
responseChunks: string[];
|
|
30
|
+
toolCalls: Map<string, InFlightToolCall>;
|
|
31
|
+
orderedCompletedTools: OtelSpan[];
|
|
32
|
+
inputTokens?: number;
|
|
33
|
+
outputTokens?: number;
|
|
34
|
+
cachedInputTokens?: number;
|
|
35
|
+
stopReason?: string;
|
|
36
|
+
isError?: boolean;
|
|
37
|
+
errorMessage?: string;
|
|
38
|
+
}
|
|
39
|
+
export declare class SessionTurnTracker {
|
|
40
|
+
private readonly config;
|
|
41
|
+
private readonly exporter;
|
|
42
|
+
private readonly sessionModels;
|
|
43
|
+
private readonly activeTurnsBySession;
|
|
44
|
+
private readonly activeTurnsByPromptId;
|
|
45
|
+
constructor(config: OtelConfig, exporter: IOtelExporter);
|
|
46
|
+
getModel(sessionId: string): string;
|
|
47
|
+
setModel(sessionId: string, model: string): void;
|
|
48
|
+
formatPromptText(prompt: string | PromptContentItem[]): string;
|
|
49
|
+
startTurn(sessionId: string, promptRequestId: string | number, rawPrompt: string | PromptContentItem[]): InFlightTurn;
|
|
50
|
+
private emitUserPromptLog;
|
|
51
|
+
appendResponseChunk(sessionId: string, text: string): void;
|
|
52
|
+
recordToolCall(sessionId: string, toolCallId: string, toolName: string): void;
|
|
53
|
+
updateToolCall(sessionId: string, toolCallId: string, status: "completed" | "failed" | string): void;
|
|
54
|
+
recordUsageUpdate(sessionId: string, used: number, _size?: number): void;
|
|
55
|
+
recordExplicitUsage(sessionId: string, usage: {
|
|
56
|
+
inputTokens?: number;
|
|
57
|
+
outputTokens?: number;
|
|
58
|
+
cachedTokens?: number;
|
|
59
|
+
}): void;
|
|
60
|
+
private reconcileTokens;
|
|
61
|
+
private resolveActiveTurn;
|
|
62
|
+
finishTurn(sessionId: string, options?: {
|
|
63
|
+
promptRequestId?: string | number | undefined;
|
|
64
|
+
stopReason?: string | undefined;
|
|
65
|
+
isError?: boolean | undefined;
|
|
66
|
+
errorMessage?: string | undefined;
|
|
67
|
+
customDbPath?: string | undefined;
|
|
68
|
+
}): void;
|
|
69
|
+
private emitAgentResponseLog;
|
|
70
|
+
private extractDatabaseUsage;
|
|
71
|
+
hasActiveTurn(sessionId: string): boolean;
|
|
72
|
+
}
|
|
73
|
+
//# sourceMappingURL=tracker.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracker.d.ts","sourceRoot":"","sources":["../../../src/otel/tracker.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAKH,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAC9C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAEnD,OAAO,EAKL,KAAK,QAAQ,EACd,MAAM,YAAY,CAAC;AACpB,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAM1D,wBAAgB,eAAe,IAAI,MAAM,CAExC;AAED,wBAAgB,cAAc,IAAI,MAAM,CAEvC;AAED,wBAAgB,WAAW,IAAI,MAAM,CAEpC;AAED,MAAM,WAAW,gBAAgB;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,MAAM,EAAE,aAAa,GAAG,WAAW,GAAG,QAAQ,CAAC;CAChD;AAED,MAAM,WAAW,YAAY;IAC3B,eAAe,EAAE,MAAM,GAAG,MAAM,CAAC;IACjC,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,cAAc,EAAE,MAAM,EAAE,CAAC;IACzB,SAAS,EAAE,GAAG,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACzC,qBAAqB,EAAE,QAAQ,EAAE,CAAC;IAClC,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAkGD,qBAAa,kBAAkB;IAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAa;IACpC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAgB;IACzC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAA6B;IAC3D,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAmC;IACxE,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAA4C;IAElF,YAAY,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,aAAa,EAGtD;IAED,QAAQ,CAAC,SAAS,EAAE,MAAM,GAAG,MAAM,CAElC;IAED,QAAQ,CAAC,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,IAAI,CAI/C;IAED,gBAAgB,CAAC,MAAM,EAAE,MAAM,GAAG,iBAAiB,EAAE,GAAG,MAAM,CAc7D;IAED,SAAS,CACP,SAAS,EAAE,MAAM,EACjB,eAAe,EAAE,MAAM,GAAG,MAAM,EAChC,SAAS,EAAE,MAAM,GAAG,iBAAiB,EAAE,GACtC,YAAY,CAyBd;IAED,OAAO,CAAC,iBAAiB;IA8BzB,mBAAmB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI,CAIzD;IAED,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,IAAI,CAa5E;IAED,cAAc,CACZ,SAAS,EAAE,MAAM,EACjB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,WAAW,GAAG,QAAQ,GAAG,MAAM,GACtC,IAAI,CAgCN;IAED,iBAAiB,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAMvE;IAED,mBAAmB,CACjB,SAAS,EAAE,MAAM,EACjB,KAAK,EAAE;QAAE,WAAW,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,GAC5E,IAAI,CAMN;IAED,OAAO,CAAC,eAAe;IAWvB,OAAO,CAAC,iBAAiB;IAWzB,UAAU,CACR,SAAS,EAAE,MAAM,EACjB,OAAO,CAAC,EAAE;QACR,eAAe,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,SAAS,CAAC;QAC9C,UAAU,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;QAChC,OAAO,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;QAC9B,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;QAClC,YAAY,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;KACnC,GACA,IAAI,CAwBN;IAED,OAAO,CAAC,oBAAoB;IAgC5B,OAAO,CAAC,oBAAoB;IAkB5B,aAAa,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAExC;CACF"}
|
|
@@ -0,0 +1,339 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Session turn and telemetry tracker.
|
|
3
|
+
* Maps ACP session prompts, response streams, and tool executions to OpenTelemetry
|
|
4
|
+
* GenAI semantic conventions and Antigravity CLI log events.
|
|
5
|
+
*/
|
|
6
|
+
import { randomBytes } from "node:crypto";
|
|
7
|
+
import { existsSync } from "node:fs";
|
|
8
|
+
import { DatabaseSync } from "node:sqlite";
|
|
9
|
+
import { truncateUtf8 } from "./utf8.js";
|
|
10
|
+
import { SPAN_KIND, STATUS_CODE, toOtelKeyValues, } from "./types.js";
|
|
11
|
+
import { getLatestSessionUsage } from "../fixes/missing-usage-metrics/index.js";
|
|
12
|
+
import { getConversationDbPath } from "../fixes/orphaned-checkpoints/index.js";
|
|
13
|
+
import { getField, getVarintField } from "../lib/protobuf.js";
|
|
14
|
+
export function generateTraceId() {
|
|
15
|
+
return randomBytes(16).toString("hex");
|
|
16
|
+
}
|
|
17
|
+
export function generateSpanId() {
|
|
18
|
+
return randomBytes(8).toString("hex");
|
|
19
|
+
}
|
|
20
|
+
export function nowUnixNano() {
|
|
21
|
+
return (BigInt(Date.now()) * 1000000n).toString();
|
|
22
|
+
}
|
|
23
|
+
function readCachedTokensFromDb(dbPath) {
|
|
24
|
+
if (!existsSync(dbPath))
|
|
25
|
+
return undefined;
|
|
26
|
+
let db = null;
|
|
27
|
+
try {
|
|
28
|
+
db = new DatabaseSync(dbPath, { timeout: 1000, readOnly: true });
|
|
29
|
+
const row = db
|
|
30
|
+
.prepare("SELECT metadata FROM steps WHERE step_type = 15 AND metadata IS NOT NULL ORDER BY idx DESC LIMIT 1;")
|
|
31
|
+
.get();
|
|
32
|
+
if (!row?.metadata)
|
|
33
|
+
return undefined;
|
|
34
|
+
const f9 = getField(row.metadata, 9);
|
|
35
|
+
if (!f9)
|
|
36
|
+
return undefined;
|
|
37
|
+
const f5 = getVarintField(f9, 5);
|
|
38
|
+
return typeof f5 === "number" && f5 > 0 ? f5 : undefined;
|
|
39
|
+
}
|
|
40
|
+
catch {
|
|
41
|
+
return undefined;
|
|
42
|
+
}
|
|
43
|
+
finally {
|
|
44
|
+
db?.close();
|
|
45
|
+
}
|
|
46
|
+
}
|
|
47
|
+
function closeRemainingTools(turn, endTimeUnixNano, isError) {
|
|
48
|
+
for (const tool of turn.toolCalls.values()) {
|
|
49
|
+
const toolSpan = {
|
|
50
|
+
traceId: turn.traceId,
|
|
51
|
+
spanId: tool.spanId,
|
|
52
|
+
parentSpanId: turn.spanId,
|
|
53
|
+
name: `execute_tool ${tool.toolName}`,
|
|
54
|
+
kind: SPAN_KIND.INTERNAL,
|
|
55
|
+
startTimeUnixNano: tool.startTimeUnixNano,
|
|
56
|
+
endTimeUnixNano,
|
|
57
|
+
attributes: toOtelKeyValues({
|
|
58
|
+
"gen_ai.tool.name": tool.toolName,
|
|
59
|
+
}),
|
|
60
|
+
status: {
|
|
61
|
+
code: isError ? STATUS_CODE.ERROR : STATUS_CODE.OK,
|
|
62
|
+
},
|
|
63
|
+
};
|
|
64
|
+
turn.orderedCompletedTools.push(toolSpan);
|
|
65
|
+
}
|
|
66
|
+
turn.toolCalls.clear();
|
|
67
|
+
}
|
|
68
|
+
function determineTurnStatus(options) {
|
|
69
|
+
const isErr = Boolean(options?.isError || options?.stopReason === "cancelled");
|
|
70
|
+
const stopReason = options?.stopReason ?? (isErr ? "error" : "end_turn");
|
|
71
|
+
return { stopReason, isError: isErr };
|
|
72
|
+
}
|
|
73
|
+
function buildTurnSpan(turn, endTimeUnixNano, stopReason, isError, errorMessage) {
|
|
74
|
+
const spanAttributes = {
|
|
75
|
+
"gen_ai.operation.name": "chat",
|
|
76
|
+
"gen_ai.request.model": turn.model,
|
|
77
|
+
"gen_ai.provider.name": "gcp.gemini",
|
|
78
|
+
"gen_ai.conversation.id": turn.sessionId,
|
|
79
|
+
"gen_ai.response.finish_reasons": [stopReason],
|
|
80
|
+
};
|
|
81
|
+
if (turn.inputTokens !== undefined) {
|
|
82
|
+
spanAttributes["gen_ai.usage.input_tokens"] = turn.inputTokens;
|
|
83
|
+
}
|
|
84
|
+
if (turn.outputTokens !== undefined) {
|
|
85
|
+
spanAttributes["gen_ai.usage.output_tokens"] = turn.outputTokens;
|
|
86
|
+
}
|
|
87
|
+
if (turn.cachedInputTokens !== undefined) {
|
|
88
|
+
spanAttributes["gen_ai.usage.cache_read.input_tokens"] = turn.cachedInputTokens;
|
|
89
|
+
}
|
|
90
|
+
return {
|
|
91
|
+
traceId: turn.traceId,
|
|
92
|
+
spanId: turn.spanId,
|
|
93
|
+
name: `chat ${turn.model}`,
|
|
94
|
+
kind: SPAN_KIND.CLIENT,
|
|
95
|
+
startTimeUnixNano: turn.startTimeUnixNano,
|
|
96
|
+
endTimeUnixNano,
|
|
97
|
+
attributes: toOtelKeyValues(spanAttributes),
|
|
98
|
+
status: {
|
|
99
|
+
code: isError ? STATUS_CODE.ERROR : STATUS_CODE.OK,
|
|
100
|
+
...(isError ? { message: errorMessage || stopReason } : {}),
|
|
101
|
+
},
|
|
102
|
+
};
|
|
103
|
+
}
|
|
104
|
+
export class SessionTurnTracker {
|
|
105
|
+
config;
|
|
106
|
+
exporter;
|
|
107
|
+
sessionModels = new Map();
|
|
108
|
+
activeTurnsBySession = new Map();
|
|
109
|
+
activeTurnsByPromptId = new Map();
|
|
110
|
+
constructor(config, exporter) {
|
|
111
|
+
this.config = config;
|
|
112
|
+
this.exporter = exporter;
|
|
113
|
+
}
|
|
114
|
+
getModel(sessionId) {
|
|
115
|
+
return this.sessionModels.get(sessionId) || "gemini-3.8-flash";
|
|
116
|
+
}
|
|
117
|
+
setModel(sessionId, model) {
|
|
118
|
+
if (model && typeof model === "string") {
|
|
119
|
+
this.sessionModels.set(sessionId, model);
|
|
120
|
+
}
|
|
121
|
+
}
|
|
122
|
+
formatPromptText(prompt) {
|
|
123
|
+
if (typeof prompt === "string")
|
|
124
|
+
return prompt;
|
|
125
|
+
if (!Array.isArray(prompt))
|
|
126
|
+
return String(prompt ?? "");
|
|
127
|
+
return prompt
|
|
128
|
+
.map((item) => {
|
|
129
|
+
if (!item || typeof item !== "object")
|
|
130
|
+
return "";
|
|
131
|
+
if (item.type === "text" && typeof item.text === "string") {
|
|
132
|
+
return item.text;
|
|
133
|
+
}
|
|
134
|
+
return `[${item.type || "content"}]`;
|
|
135
|
+
})
|
|
136
|
+
.filter(Boolean)
|
|
137
|
+
.join("\n");
|
|
138
|
+
}
|
|
139
|
+
startTurn(sessionId, promptRequestId, rawPrompt) {
|
|
140
|
+
const model = this.getModel(sessionId);
|
|
141
|
+
const traceId = generateTraceId();
|
|
142
|
+
const spanId = generateSpanId();
|
|
143
|
+
const startTimeUnixNano = nowUnixNano();
|
|
144
|
+
const rawPromptText = this.formatPromptText(rawPrompt);
|
|
145
|
+
const turn = {
|
|
146
|
+
promptRequestId,
|
|
147
|
+
sessionId,
|
|
148
|
+
model,
|
|
149
|
+
traceId,
|
|
150
|
+
spanId,
|
|
151
|
+
startTimeUnixNano,
|
|
152
|
+
rawPromptText,
|
|
153
|
+
responseChunks: [],
|
|
154
|
+
toolCalls: new Map(),
|
|
155
|
+
orderedCompletedTools: [],
|
|
156
|
+
};
|
|
157
|
+
this.activeTurnsBySession.set(sessionId, turn);
|
|
158
|
+
this.activeTurnsByPromptId.set(promptRequestId, turn);
|
|
159
|
+
this.emitUserPromptLog(turn);
|
|
160
|
+
return turn;
|
|
161
|
+
}
|
|
162
|
+
emitUserPromptLog(turn) {
|
|
163
|
+
const rawLen = turn.rawPromptText.length;
|
|
164
|
+
const { text: truncatedPrompt } = truncateUtf8(turn.rawPromptText);
|
|
165
|
+
const attributes = {
|
|
166
|
+
"event.name": "agy.user_prompt",
|
|
167
|
+
prompt_length: rawLen,
|
|
168
|
+
"gen_ai.conversation.id": turn.sessionId,
|
|
169
|
+
"gen_ai.request.model": turn.model,
|
|
170
|
+
"gen_ai.provider.name": "gcp.gemini",
|
|
171
|
+
};
|
|
172
|
+
if (this.config.logUserPrompts) {
|
|
173
|
+
attributes.prompt = truncatedPrompt;
|
|
174
|
+
}
|
|
175
|
+
const logRecord = {
|
|
176
|
+
timeUnixNano: turn.startTimeUnixNano,
|
|
177
|
+
observedTimeUnixNano: turn.startTimeUnixNano,
|
|
178
|
+
severityNumber: 9,
|
|
179
|
+
severityText: "INFO",
|
|
180
|
+
body: { stringValue: "agy.user_prompt" },
|
|
181
|
+
attributes: toOtelKeyValues(attributes),
|
|
182
|
+
traceId: turn.traceId,
|
|
183
|
+
spanId: turn.spanId,
|
|
184
|
+
};
|
|
185
|
+
this.exporter.exportLogs([logRecord]);
|
|
186
|
+
}
|
|
187
|
+
appendResponseChunk(sessionId, text) {
|
|
188
|
+
const turn = this.activeTurnsBySession.get(sessionId);
|
|
189
|
+
if (!turn)
|
|
190
|
+
return;
|
|
191
|
+
turn.responseChunks.push(text);
|
|
192
|
+
}
|
|
193
|
+
recordToolCall(sessionId, toolCallId, toolName) {
|
|
194
|
+
const turn = this.activeTurnsBySession.get(sessionId);
|
|
195
|
+
if (!turn)
|
|
196
|
+
return;
|
|
197
|
+
if (!turn.toolCalls.has(toolCallId)) {
|
|
198
|
+
turn.toolCalls.set(toolCallId, {
|
|
199
|
+
toolCallId,
|
|
200
|
+
toolName: toolName || "tool",
|
|
201
|
+
spanId: generateSpanId(),
|
|
202
|
+
startTimeUnixNano: nowUnixNano(),
|
|
203
|
+
status: "in_progress",
|
|
204
|
+
});
|
|
205
|
+
}
|
|
206
|
+
}
|
|
207
|
+
updateToolCall(sessionId, toolCallId, status) {
|
|
208
|
+
const turn = this.activeTurnsBySession.get(sessionId);
|
|
209
|
+
if (!turn)
|
|
210
|
+
return;
|
|
211
|
+
const tool = turn.toolCalls.get(toolCallId);
|
|
212
|
+
if (!tool)
|
|
213
|
+
return;
|
|
214
|
+
const endNano = nowUnixNano();
|
|
215
|
+
tool.endTimeUnixNano = endNano;
|
|
216
|
+
const isFailure = status === "failed";
|
|
217
|
+
tool.status = isFailure ? "failed" : "completed";
|
|
218
|
+
const toolSpan = {
|
|
219
|
+
traceId: turn.traceId,
|
|
220
|
+
spanId: tool.spanId,
|
|
221
|
+
parentSpanId: turn.spanId,
|
|
222
|
+
name: `execute_tool ${tool.toolName}`,
|
|
223
|
+
kind: SPAN_KIND.INTERNAL,
|
|
224
|
+
startTimeUnixNano: tool.startTimeUnixNano,
|
|
225
|
+
endTimeUnixNano: endNano,
|
|
226
|
+
attributes: toOtelKeyValues({
|
|
227
|
+
"gen_ai.tool.name": tool.toolName,
|
|
228
|
+
}),
|
|
229
|
+
status: {
|
|
230
|
+
code: isFailure ? STATUS_CODE.ERROR : STATUS_CODE.OK,
|
|
231
|
+
...(isFailure ? { message: "tool call failed" } : {}),
|
|
232
|
+
},
|
|
233
|
+
};
|
|
234
|
+
turn.orderedCompletedTools.push(toolSpan);
|
|
235
|
+
turn.toolCalls.delete(toolCallId);
|
|
236
|
+
}
|
|
237
|
+
recordUsageUpdate(sessionId, used, _size) {
|
|
238
|
+
const turn = this.activeTurnsBySession.get(sessionId);
|
|
239
|
+
if (!turn)
|
|
240
|
+
return;
|
|
241
|
+
if (typeof used === "number" && turn.inputTokens === undefined) {
|
|
242
|
+
turn.inputTokens = used;
|
|
243
|
+
}
|
|
244
|
+
}
|
|
245
|
+
recordExplicitUsage(sessionId, usage) {
|
|
246
|
+
const turn = this.activeTurnsBySession.get(sessionId);
|
|
247
|
+
if (!turn)
|
|
248
|
+
return;
|
|
249
|
+
if (typeof usage.inputTokens === "number")
|
|
250
|
+
turn.inputTokens = usage.inputTokens;
|
|
251
|
+
if (typeof usage.outputTokens === "number")
|
|
252
|
+
turn.outputTokens = usage.outputTokens;
|
|
253
|
+
if (typeof usage.cachedTokens === "number")
|
|
254
|
+
turn.cachedInputTokens = usage.cachedTokens;
|
|
255
|
+
}
|
|
256
|
+
reconcileTokens(turn, sessionId, customDbPath) {
|
|
257
|
+
if (turn.inputTokens !== undefined && turn.outputTokens !== undefined)
|
|
258
|
+
return;
|
|
259
|
+
const dbUsage = this.extractDatabaseUsage(sessionId, customDbPath);
|
|
260
|
+
if (!dbUsage)
|
|
261
|
+
return;
|
|
262
|
+
if (turn.inputTokens === undefined)
|
|
263
|
+
turn.inputTokens = dbUsage.promptTokens;
|
|
264
|
+
if (turn.outputTokens === undefined)
|
|
265
|
+
turn.outputTokens = dbUsage.candidateTokens;
|
|
266
|
+
if (turn.cachedInputTokens === undefined && dbUsage.cachedTokens) {
|
|
267
|
+
turn.cachedInputTokens = dbUsage.cachedTokens;
|
|
268
|
+
}
|
|
269
|
+
}
|
|
270
|
+
resolveActiveTurn(sessionId, promptRequestId) {
|
|
271
|
+
if (promptRequestId !== undefined) {
|
|
272
|
+
const turn = this.activeTurnsByPromptId.get(promptRequestId);
|
|
273
|
+
if (turn)
|
|
274
|
+
return turn;
|
|
275
|
+
}
|
|
276
|
+
return this.activeTurnsBySession.get(sessionId);
|
|
277
|
+
}
|
|
278
|
+
finishTurn(sessionId, options) {
|
|
279
|
+
const turn = this.resolveActiveTurn(sessionId, options?.promptRequestId);
|
|
280
|
+
if (!turn)
|
|
281
|
+
return;
|
|
282
|
+
this.activeTurnsBySession.delete(sessionId);
|
|
283
|
+
this.activeTurnsByPromptId.delete(turn.promptRequestId);
|
|
284
|
+
const endTimeUnixNano = nowUnixNano();
|
|
285
|
+
const { stopReason, isError } = determineTurnStatus(options);
|
|
286
|
+
closeRemainingTools(turn, endTimeUnixNano, isError);
|
|
287
|
+
this.reconcileTokens(turn, sessionId, options?.customDbPath);
|
|
288
|
+
this.emitAgentResponseLog(turn, endTimeUnixNano);
|
|
289
|
+
const turnSpan = buildTurnSpan(turn, endTimeUnixNano, stopReason, isError, options?.errorMessage);
|
|
290
|
+
this.exporter.exportSpans([turnSpan, ...turn.orderedCompletedTools]);
|
|
291
|
+
}
|
|
292
|
+
emitAgentResponseLog(turn, timeUnixNano) {
|
|
293
|
+
const rawResponse = turn.responseChunks.join("");
|
|
294
|
+
const rawLen = rawResponse.length;
|
|
295
|
+
const { text: truncatedResponse, truncated } = truncateUtf8(rawResponse);
|
|
296
|
+
const attributes = {
|
|
297
|
+
"event.name": "agy.agent_response",
|
|
298
|
+
response_length: rawLen,
|
|
299
|
+
response_truncated: truncated,
|
|
300
|
+
"gen_ai.conversation.id": turn.sessionId,
|
|
301
|
+
"gen_ai.request.model": turn.model,
|
|
302
|
+
"gen_ai.provider.name": "gcp.gemini",
|
|
303
|
+
};
|
|
304
|
+
if (this.config.logAssistantResponses) {
|
|
305
|
+
attributes.response = truncatedResponse;
|
|
306
|
+
}
|
|
307
|
+
const logRecord = {
|
|
308
|
+
timeUnixNano,
|
|
309
|
+
observedTimeUnixNano: timeUnixNano,
|
|
310
|
+
severityNumber: 9,
|
|
311
|
+
severityText: "INFO",
|
|
312
|
+
body: { stringValue: "agy.agent_response" },
|
|
313
|
+
attributes: toOtelKeyValues(attributes),
|
|
314
|
+
traceId: turn.traceId,
|
|
315
|
+
spanId: turn.spanId,
|
|
316
|
+
};
|
|
317
|
+
this.exporter.exportLogs([logRecord]);
|
|
318
|
+
}
|
|
319
|
+
extractDatabaseUsage(sessionId, customDbPath) {
|
|
320
|
+
try {
|
|
321
|
+
const usage = getLatestSessionUsage(sessionId, customDbPath);
|
|
322
|
+
if (!usage)
|
|
323
|
+
return null;
|
|
324
|
+
const dbPath = customDbPath ?? getConversationDbPath(sessionId);
|
|
325
|
+
const cachedTokens = readCachedTokensFromDb(dbPath);
|
|
326
|
+
return {
|
|
327
|
+
...usage,
|
|
328
|
+
...(cachedTokens !== undefined ? { cachedTokens } : {}),
|
|
329
|
+
};
|
|
330
|
+
}
|
|
331
|
+
catch {
|
|
332
|
+
return null;
|
|
333
|
+
}
|
|
334
|
+
}
|
|
335
|
+
hasActiveTurn(sessionId) {
|
|
336
|
+
return this.activeTurnsBySession.has(sessionId);
|
|
337
|
+
}
|
|
338
|
+
}
|
|
339
|
+
//# sourceMappingURL=tracker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tracker.js","sourceRoot":"","sources":["../../../src/otel/tracker.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EACL,SAAS,EACT,WAAW,EACX,eAAe,GAGhB,MAAM,YAAY,CAAC;AAGpB,OAAO,EAAE,qBAAqB,EAAE,MAAM,yCAAyC,CAAC;AAChF,OAAO,EAAE,qBAAqB,EAAE,MAAM,wCAAwC,CAAC;AAC/E,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAE9D,MAAM,UAAU,eAAe;IAC7B,OAAO,WAAW,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,cAAc;IAC5B,OAAO,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;AACxC,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,GAAG,QAAU,CAAC,CAAC,QAAQ,EAAE,CAAC;AACtD,CAAC;AA8BD,SAAS,sBAAsB,CAAC,MAAc;IAC5C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC;QAAE,OAAO,SAAS,CAAC;IAC1C,IAAI,EAAE,GAAwB,IAAI,CAAC;IACnC,IAAI,CAAC;QACH,EAAE,GAAG,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC;QACjE,MAAM,GAAG,GAAG,EAAE;aACX,OAAO,CACN,qGAAqG,CACtG;aACA,GAAG,EAAiD,CAAC;QACxD,IAAI,CAAC,GAAG,EAAE,QAAQ;YAAE,OAAO,SAAS,CAAC;QACrC,MAAM,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACrC,IAAI,CAAC,EAAE;YAAE,OAAO,SAAS,CAAC;QAC1B,MAAM,EAAE,GAAG,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;QACjC,OAAO,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3D,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;YAAS,CAAC;QACT,EAAE,EAAE,KAAK,EAAE,CAAC;IACd,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAkB,EAAE,eAAuB,EAAE,OAAgB;IACxF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,EAAE,CAAC;QAC3C,MAAM,QAAQ,GAAa;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,MAAM;YACzB,IAAI,EAAE,gBAAgB,IAAI,CAAC,QAAQ,EAAE;YACrC,IAAI,EAAE,SAAS,CAAC,QAAQ;YACxB,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,eAAe;YACf,UAAU,EAAE,eAAe,CAAC;gBAC1B,kBAAkB,EAAE,IAAI,CAAC,QAAQ;aAClC,CAAC;YACF,MAAM,EAAE;gBACN,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE;aACnD;SACF,CAAC;QACF,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC;IACD,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,CAAC;AACzB,CAAC;AAED,SAAS,mBAAmB,CAAC,OAG5B;IAIC,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,EAAE,OAAO,IAAI,OAAO,EAAE,UAAU,KAAK,WAAW,CAAC,CAAC;IAC/E,MAAM,UAAU,GAAG,OAAO,EAAE,UAAU,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC;IACzE,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;AACxC,CAAC;AAED,SAAS,aAAa,CACpB,IAAkB,EAClB,eAAuB,EACvB,UAAkB,EAClB,OAAgB,EAChB,YAAqB;IAErB,MAAM,cAAc,GAA4B;QAC9C,uBAAuB,EAAE,MAAM;QAC/B,sBAAsB,EAAE,IAAI,CAAC,KAAK;QAClC,sBAAsB,EAAE,YAAY;QACpC,wBAAwB,EAAE,IAAI,CAAC,SAAS;QACxC,gCAAgC,EAAE,CAAC,UAAU,CAAC;KAC/C,CAAC;IAEF,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;QACnC,cAAc,CAAC,2BAA2B,CAAC,GAAG,IAAI,CAAC,WAAW,CAAC;IACjE,CAAC;IACD,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS,EAAE,CAAC;QACpC,cAAc,CAAC,4BAA4B,CAAC,GAAG,IAAI,CAAC,YAAY,CAAC;IACnE,CAAC;IACD,IAAI,IAAI,CAAC,iBAAiB,KAAK,SAAS,EAAE,CAAC;QACzC,cAAc,CAAC,sCAAsC,CAAC,GAAG,IAAI,CAAC,iBAAiB,CAAC;IAClF,CAAC;IAED,OAAO;QACL,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,IAAI,EAAE,QAAQ,IAAI,CAAC,KAAK,EAAE;QAC1B,IAAI,EAAE,SAAS,CAAC,MAAM;QACtB,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;QACzC,eAAe;QACf,UAAU,EAAE,eAAe,CAAC,cAAc,CAAC;QAC3C,MAAM,EAAE;YACN,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE;YAClD,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,YAAY,IAAI,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5D;KACF,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,kBAAkB;IACZ,MAAM,CAAa;IACnB,QAAQ,CAAgB;IACxB,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC1C,oBAAoB,GAAG,IAAI,GAAG,EAAwB,CAAC;IACvD,qBAAqB,GAAG,IAAI,GAAG,EAAiC,CAAC;IAElF,YAAY,MAAkB,EAAE,QAAuB;QACrD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,QAAQ,CAAC,SAAiB;QACxB,OAAO,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,kBAAkB,CAAC;IACjE,CAAC;IAED,QAAQ,CAAC,SAAiB,EAAE,KAAa;QACvC,IAAI,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACvC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,MAAoC;QACnD,IAAI,OAAO,MAAM,KAAK,QAAQ;YAAE,OAAO,MAAM,CAAC;QAC9C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,OAAO,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAExD,OAAO,MAAM;aACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACZ,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAE,CAAC;YACjD,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC1D,OAAO,IAAI,CAAC,IAAI,CAAC;YACnB,CAAC;YACD,OAAO,IAAI,IAAI,CAAC,IAAI,IAAI,SAAS,GAAG,CAAC;QACvC,CAAC,CAAC;aACD,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAED,SAAS,CACP,SAAiB,EACjB,eAAgC,EAChC,SAAuC;QAEvC,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACvC,MAAM,OAAO,GAAG,eAAe,EAAE,CAAC;QAClC,MAAM,MAAM,GAAG,cAAc,EAAE,CAAC;QAChC,MAAM,iBAAiB,GAAG,WAAW,EAAE,CAAC;QACxC,MAAM,aAAa,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;QAEvD,MAAM,IAAI,GAAiB;YACzB,eAAe;YACf,SAAS;YACT,KAAK;YACL,OAAO;YACP,MAAM;YACN,iBAAiB;YACjB,aAAa;YACb,cAAc,EAAE,EAAE;YAClB,SAAS,EAAE,IAAI,GAAG,EAAE;YACpB,qBAAqB,EAAE,EAAE;SAC1B,CAAC;QAEF,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAEtD,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,iBAAiB,CAAC,IAAkB;QAC1C,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,CAAC;QACzC,MAAM,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,YAAY,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAEnE,MAAM,UAAU,GAA4B;YAC1C,YAAY,EAAE,iBAAiB;YAC/B,aAAa,EAAE,MAAM;YACrB,wBAAwB,EAAE,IAAI,CAAC,SAAS;YACxC,sBAAsB,EAAE,IAAI,CAAC,KAAK;YAClC,sBAAsB,EAAE,YAAY;SACrC,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YAC/B,UAAU,CAAC,MAAM,GAAG,eAAe,CAAC;QACtC,CAAC;QAED,MAAM,SAAS,GAAkB;YAC/B,YAAY,EAAE,IAAI,CAAC,iBAAiB;YACpC,oBAAoB,EAAE,IAAI,CAAC,iBAAiB;YAC5C,cAAc,EAAE,CAAC;YACjB,YAAY,EAAE,MAAM;YACpB,IAAI,EAAE,EAAE,WAAW,EAAE,iBAAiB,EAAE;YACxC,UAAU,EAAE,eAAe,CAAC,UAAU,CAAC;YACvC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,mBAAmB,CAAC,SAAiB,EAAE,IAAY;QACjD,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO;QAClB,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,SAAiB,EAAE,UAAkB,EAAE,QAAgB;QACpE,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,CAAC;YACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;gBAC7B,UAAU;gBACV,QAAQ,EAAE,QAAQ,IAAI,MAAM;gBAC5B,MAAM,EAAE,cAAc,EAAE;gBACxB,iBAAiB,EAAE,WAAW,EAAE;gBAChC,MAAM,EAAE,aAAa;aACtB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,cAAc,CACZ,SAAiB,EACjB,UAAkB,EAClB,MAAuC;QAEvC,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,MAAM,OAAO,GAAG,WAAW,EAAE,CAAC;QAC9B,IAAI,CAAC,eAAe,GAAG,OAAO,CAAC;QAE/B,MAAM,SAAS,GAAG,MAAM,KAAK,QAAQ,CAAC;QACtC,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,CAAC;QAEjD,MAAM,QAAQ,GAAa;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,YAAY,EAAE,IAAI,CAAC,MAAM;YACzB,IAAI,EAAE,gBAAgB,IAAI,CAAC,QAAQ,EAAE;YACrC,IAAI,EAAE,SAAS,CAAC,QAAQ;YACxB,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,eAAe,EAAE,OAAO;YACxB,UAAU,EAAE,eAAe,CAAC;gBAC1B,kBAAkB,EAAE,IAAI,CAAC,QAAQ;aAClC,CAAC;YACF,MAAM,EAAE;gBACN,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE;gBACpD,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACtD;SACF,CAAC;QAEF,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;IACpC,CAAC;IAED,iBAAiB,CAAC,SAAiB,EAAE,IAAY,EAAE,KAAc;QAC/D,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO;QAClB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,EAAE,CAAC;YAC/D,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,mBAAmB,CACjB,SAAiB,EACjB,KAA6E;QAE7E,MAAM,IAAI,GAAG,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,IAAI;YAAE,OAAO;QAClB,IAAI,OAAO,KAAK,CAAC,WAAW,KAAK,QAAQ;YAAE,IAAI,CAAC,WAAW,GAAG,KAAK,CAAC,WAAW,CAAC;QAChF,IAAI,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ;YAAE,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;QACnF,IAAI,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ;YAAE,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,YAAY,CAAC;IAC1F,CAAC;IAEO,eAAe,CAAC,IAAkB,EAAE,SAAiB,EAAE,YAAqB;QAClF,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;YAAE,OAAO;QAC9E,MAAM,OAAO,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;QACnE,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,IAAI,CAAC,WAAW,KAAK,SAAS;YAAE,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,YAAY,CAAC;QAC5E,IAAI,IAAI,CAAC,YAAY,KAAK,SAAS;YAAE,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,eAAe,CAAC;QACjF,IAAI,IAAI,CAAC,iBAAiB,KAAK,SAAS,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACjE,IAAI,CAAC,iBAAiB,GAAG,OAAO,CAAC,YAAY,CAAC;QAChD,CAAC;IACH,CAAC;IAEO,iBAAiB,CACvB,SAAiB,EACjB,eAAiC;QAEjC,IAAI,eAAe,KAAK,SAAS,EAAE,CAAC;YAClC,MAAM,IAAI,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;YAC7D,IAAI,IAAI;gBAAE,OAAO,IAAI,CAAC;QACxB,CAAC;QACD,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC;IAED,UAAU,CACR,SAAiB,EACjB,OAMC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,EAAE,OAAO,EAAE,eAAe,CAAC,CAAC;QACzE,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC5C,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAExD,MAAM,eAAe,GAAG,WAAW,EAAE,CAAC;QACtC,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,mBAAmB,CAAC,OAAO,CAAC,CAAC;QAE7D,mBAAmB,CAAC,IAAI,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;QACpD,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;QAE7D,IAAI,CAAC,oBAAoB,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;QAEjD,MAAM,QAAQ,GAAG,aAAa,CAC5B,IAAI,EACJ,eAAe,EACf,UAAU,EACV,OAAO,EACP,OAAO,EAAE,YAAY,CACtB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC;IACvE,CAAC;IAEO,oBAAoB,CAAC,IAAkB,EAAE,YAAoB;QACnE,MAAM,WAAW,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QAClC,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,GAAG,YAAY,CAAC,WAAW,CAAC,CAAC;QAEzE,MAAM,UAAU,GAA4B;YAC1C,YAAY,EAAE,oBAAoB;YAClC,eAAe,EAAE,MAAM;YACvB,kBAAkB,EAAE,SAAS;YAC7B,wBAAwB,EAAE,IAAI,CAAC,SAAS;YACxC,sBAAsB,EAAE,IAAI,CAAC,KAAK;YAClC,sBAAsB,EAAE,YAAY;SACrC,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,CAAC,qBAAqB,EAAE,CAAC;YACtC,UAAU,CAAC,QAAQ,GAAG,iBAAiB,CAAC;QAC1C,CAAC;QAED,MAAM,SAAS,GAAkB;YAC/B,YAAY;YACZ,oBAAoB,EAAE,YAAY;YAClC,cAAc,EAAE,CAAC;YACjB,YAAY,EAAE,MAAM;YACpB,IAAI,EAAE,EAAE,WAAW,EAAE,oBAAoB,EAAE;YAC3C,UAAU,EAAE,eAAe,CAAC,UAAU,CAAC;YACvC,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC;IACxC,CAAC;IAEO,oBAAoB,CAC1B,SAAiB,EACjB,YAAqB;QAErB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,qBAAqB,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;YAC7D,IAAI,CAAC,KAAK;gBAAE,OAAO,IAAI,CAAC;YACxB,MAAM,MAAM,GAAG,YAAY,IAAI,qBAAqB,CAAC,SAAS,CAAC,CAAC;YAChE,MAAM,YAAY,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;YACpD,OAAO;gBACL,GAAG,KAAK;gBACR,GAAG,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACxD,CAAC;QACJ,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,aAAa,CAAC,SAAiB;QAC7B,OAAO,IAAI,CAAC,oBAAoB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAClD,CAAC;CACF"}
|