@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.
Files changed (37) hide show
  1. package/dist/src/fixes/flattened-model-efforts/index.d.ts.map +1 -1
  2. package/dist/src/fixes/flattened-model-efforts/index.js +15 -2
  3. package/dist/src/fixes/flattened-model-efforts/index.js.map +1 -1
  4. package/dist/src/fixes/index.d.ts +1 -0
  5. package/dist/src/fixes/index.d.ts.map +1 -1
  6. package/dist/src/fixes/index.js +5 -0
  7. package/dist/src/fixes/index.js.map +1 -1
  8. package/dist/src/otel/config.d.ts +19 -0
  9. package/dist/src/otel/config.d.ts.map +1 -0
  10. package/dist/src/otel/config.js +121 -0
  11. package/dist/src/otel/config.js.map +1 -0
  12. package/dist/src/otel/exporter.d.ts +25 -0
  13. package/dist/src/otel/exporter.d.ts.map +1 -0
  14. package/dist/src/otel/exporter.js +102 -0
  15. package/dist/src/otel/exporter.js.map +1 -0
  16. package/dist/src/otel/fix.d.ts +24 -0
  17. package/dist/src/otel/fix.d.ts.map +1 -0
  18. package/dist/src/otel/fix.js +170 -0
  19. package/dist/src/otel/fix.js.map +1 -0
  20. package/dist/src/otel/index.d.ts +10 -0
  21. package/dist/src/otel/index.d.ts.map +1 -0
  22. package/dist/src/otel/index.js +10 -0
  23. package/dist/src/otel/index.js.map +1 -0
  24. package/dist/src/otel/tracker.d.ts +73 -0
  25. package/dist/src/otel/tracker.d.ts.map +1 -0
  26. package/dist/src/otel/tracker.js +339 -0
  27. package/dist/src/otel/tracker.js.map +1 -0
  28. package/dist/src/otel/types.d.ts +93 -0
  29. package/dist/src/otel/types.d.ts.map +1 -0
  30. package/dist/src/otel/types.js +49 -0
  31. package/dist/src/otel/types.js.map +1 -0
  32. package/dist/src/otel/utf8.d.ts +15 -0
  33. package/dist/src/otel/utf8.d.ts.map +1 -0
  34. package/dist/src/otel/utf8.js +49 -0
  35. package/dist/src/otel/utf8.js.map +1 -0
  36. package/package.json +1 -1
  37. 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"}