@blackbox_ai/blackbox-cli-core 0.0.4

package/dist/src/core/openaiContentGenerator/telemetryService.js

@@ -0,0 +1,146 @@
+/**
+ * @license
+ * Copyright 2025 Blackbox
+ * SPDX-License-Identifier: Apache-2.0
+ */
+import { logApiError, logApiResponse } from '../../telemetry/loggers.js';
+import { ApiErrorEvent, ApiResponseEvent } from '../../telemetry/types.js';
+import { openaiLogger } from '../../utils/openaiLogger.js';
+export class DefaultTelemetryService {
+    config;
+    enableOpenAILogging;
+    constructor(config, enableOpenAILogging = false) {
+        this.config = config;
+        this.enableOpenAILogging = enableOpenAILogging;
+    }
+    async logSuccess(context, response, openaiRequest, openaiResponse) {
+        // Log API response event for UI telemetry
+        const responseEvent = new ApiResponseEvent(response.responseId || 'unknown', context.model, context.duration, context.userPromptId, context.authType, response.usageMetadata);
+        logApiResponse(this.config, responseEvent);
+        // Log interaction if enabled
+        if (this.enableOpenAILogging && openaiRequest && openaiResponse) {
+            await openaiLogger.logInteraction(openaiRequest, openaiResponse);
+        }
+    }
+    async logError(context, error, openaiRequest) {
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        // Log API error event for UI telemetry
+        const errorEvent = new ApiErrorEvent(
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        error?.requestID || 'unknown', context.model, errorMessage, context.duration, context.userPromptId, context.authType, 
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        error?.type, 
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        error?.code);
+        logApiError(this.config, errorEvent);
+        // Log error interaction if enabled
+        if (this.enableOpenAILogging && openaiRequest) {
+            await openaiLogger.logInteraction(openaiRequest, undefined, error);
+        }
+    }
+    async logStreamingSuccess(context, responses, openaiRequest, openaiChunks) {
+        // Get final usage metadata from the last response that has it
+        const finalUsageMetadata = responses
+            .slice()
+            .reverse()
+            .find((r) => r.usageMetadata)?.usageMetadata;
+        // Log API response event for UI telemetry
+        const responseEvent = new ApiResponseEvent(responses[responses.length - 1]?.responseId || 'unknown', context.model, context.duration, context.userPromptId, context.authType, finalUsageMetadata);
+        logApiResponse(this.config, responseEvent);
+        // Log interaction if enabled - combine chunks only when needed
+        if (this.enableOpenAILogging &&
+            openaiRequest &&
+            openaiChunks &&
+            openaiChunks.length > 0) {
+            const combinedResponse = this.combineOpenAIChunksForLogging(openaiChunks);
+            await openaiLogger.logInteraction(openaiRequest, combinedResponse);
+        }
+    }
+    /**
+     * Combine OpenAI chunks for logging purposes
+     * This method consolidates all OpenAI stream chunks into a single ChatCompletion response
+     * for telemetry and logging purposes, avoiding unnecessary format conversions
+     */
+    combineOpenAIChunksForLogging(chunks) {
+        if (chunks.length === 0) {
+            throw new Error('No chunks to combine');
+        }
+        const firstChunk = chunks[0];
+        // Combine all content from chunks
+        let combinedContent = '';
+        const toolCalls = [];
+        let finishReason = null;
+        let usage;
+        for (const chunk of chunks) {
+            const choice = chunk.choices?.[0];
+            if (choice) {
+                // Combine text content
+                if (choice.delta?.content) {
+                    combinedContent += choice.delta.content;
+                }
+                // Collect tool calls
+                if (choice.delta?.tool_calls) {
+                    for (const toolCall of choice.delta.tool_calls) {
+                        if (toolCall.index !== undefined) {
+                            if (!toolCalls[toolCall.index]) {
+                                toolCalls[toolCall.index] = {
+                                    id: toolCall.id || '',
+                                    type: toolCall.type || 'function',
+                                    function: { name: '', arguments: '' },
+                                };
+                            }
+                            if (toolCall.function?.name) {
+                                toolCalls[toolCall.index].function.name +=
+                                    toolCall.function.name;
+                            }
+                            if (toolCall.function?.arguments) {
+                                toolCalls[toolCall.index].function.arguments +=
+                                    toolCall.function.arguments;
+                            }
+                        }
+                    }
+                }
+                // Get finish reason from the last chunk
+                if (choice.finish_reason) {
+                    finishReason = choice.finish_reason;
+                }
+            }
+            // Get usage from the last chunk that has it
+            if (chunk.usage) {
+                usage = chunk.usage;
+            }
+        }
+        // Create the combined ChatCompletion response
+        const message = {
+            role: 'assistant',
+            content: combinedContent || null,
+            refusal: null,
+        };
+        // Add tool calls if any
+        if (toolCalls.length > 0) {
+            message.tool_calls = toolCalls.filter((tc) => tc.id); // Filter out empty tool calls
+        }
+        const combinedResponse = {
+            id: firstChunk.id,
+            object: 'chat.completion',
+            created: firstChunk.created,
+            model: firstChunk.model,
+            choices: [
+                {
+                    index: 0,
+                    message,
+                    finish_reason: finishReason || 'stop',
+                    logprobs: null,
+                },
+            ],
+            usage: usage || {
+                prompt_tokens: 0,
+                completion_tokens: 0,
+                total_tokens: 0,
+            },
+            system_fingerprint: firstChunk.system_fingerprint,
+        };
+        return combinedResponse;
+    }
+}
+//# sourceMappingURL=telemetryService.js.map

package/dist/src/core/openaiContentGenerator/telemetryService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"telemetryService.js","sourceRoot":"","sources":["../../../../src/core/openaiContentGenerator/telemetryService.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,WAAW,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AACzE,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAC3E,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAmC3D,MAAM,OAAO,uBAAuB;IAExB;IACA;IAFV,YACU,MAAc,EACd,sBAA+B,KAAK;QADpC,WAAM,GAAN,MAAM,CAAQ;QACd,wBAAmB,GAAnB,mBAAmB,CAAiB;IAC3C,CAAC;IAEJ,KAAK,CAAC,UAAU,CACd,OAAuB,EACvB,QAAiC,EACjC,aAAsD,EACtD,cAA2C;QAE3C,0CAA0C;QAC1C,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,QAAQ,CAAC,UAAU,IAAI,SAAS,EAChC,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,QAAQ,EAChB,QAAQ,CAAC,aAAa,CACvB,CAAC;QAEF,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAE3C,6BAA6B;QAC7B,IAAI,IAAI,CAAC,mBAAmB,IAAI,aAAa,IAAI,cAAc,EAAE,CAAC;YAChE,MAAM,YAAY,CAAC,cAAc,CAAC,aAAa,EAAE,cAAc,CAAC,CAAC;QACnE,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,OAAuB,EACvB,KAAc,EACd,aAAsD;QAEtD,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAE5E,uCAAuC;QACvC,MAAM,UAAU,GAAG,IAAI,aAAa;QAClC,8DAA8D;QAC7D,KAAa,EAAE,SAAS,IAAI,SAAS,EACtC,OAAO,CAAC,KAAK,EACb,YAAY,EACZ,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,QAAQ;QAChB,8DAA8D;QAC7D,KAAa,EAAE,IAAI;QACpB,8DAA8D;QAC7D,KAAa,EAAE,IAAI,CACrB,CAAC;QACF,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAErC,mCAAmC;QACnC,IAAI,IAAI,CAAC,mBAAmB,IAAI,aAAa,EAAE,CAAC;YAC9C,MAAM,YAAY,CAAC,cAAc,CAC/B,aAAa,EACb,SAAS,EACT,KAAc,CACf,CAAC;QACJ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,mBAAmB,CACvB,OAAuB,EACvB,SAAoC,EACpC,aAAsD,EACtD,YAAgD;QAEhD,8DAA8D;QAC9D,MAAM,kBAAkB,GAAG,SAAS;aACjC,KAAK,EAAE;aACP,OAAO,EAAE;aACT,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,aAAa,CAAC,EAAE,aAAa,CAAC;QAE/C,0CAA0C;QAC1C,MAAM,aAAa,GAAG,IAAI,gBAAgB,CACxC,SAAS,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,EAAE,UAAU,IAAI,SAAS,EACxD,OAAO,CAAC,KAAK,EACb,OAAO,CAAC,QAAQ,EAChB,OAAO,CAAC,YAAY,EACpB,OAAO,CAAC,QAAQ,EAChB,kBAAkB,CACnB,CAAC;QAEF,cAAc,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC;QAE3C,+DAA+D;QAC/D,IACE,IAAI,CAAC,mBAAmB;YACxB,aAAa;YACb,YAAY;YACZ,YAAY,CAAC,MAAM,GAAG,CAAC,EACvB,CAAC;YACD,MAAM,gBAAgB,GAAG,IAAI,CAAC,6BAA6B,CAAC,YAAY,CAAC,CAAC;YAC1E,MAAM,YAAY,CAAC,cAAc,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QACrE,CAAC;IACH,CAAC;IAED;;;;OAIG;IACK,6BAA6B,CACnC,MAAyC;QAEzC,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAC1C,CAAC;QAED,MAAM,UAAU,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QAE7B,kCAAkC;QAClC,IAAI,eAAe,GAAG,EAAE,CAAC;QACzB,MAAM,SAAS,GAAgD,EAAE,CAAC;QAClE,IAAI,YAAY,GAML,IAAI,CAAC;QAChB,IAAI,KAMS,CAAC;QAEd,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;YAClC,IAAI,MAAM,EAAE,CAAC;gBACX,uBAAuB;gBACvB,IAAI,MAAM,CAAC,KAAK,EAAE,OAAO,EAAE,CAAC;oBAC1B,eAAe,IAAI,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC1C,CAAC;gBAED,qBAAqB;gBACrB,IAAI,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,CAAC;oBAC7B,KAAK,MAAM,QAAQ,IAAI,MAAM,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC;wBAC/C,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;4BACjC,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gCAC/B,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG;oCAC1B,EAAE,EAAE,QAAQ,CAAC,EAAE,IAAI,EAAE;oCACrB,IAAI,EAAE,QAAQ,CAAC,IAAI,IAAI,UAAU;oCACjC,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;iCACtC,CAAC;4BACJ,CAAC;4BAED,IAAI,QAAQ,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;gCAC5B,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,IAAI;oCACrC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;4BAC3B,CAAC;4BACD,IAAI,QAAQ,CAAC,QAAQ,EAAE,SAAS,EAAE,CAAC;gCACjC,SAAS,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,SAAS;oCAC1C,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;4BAChC,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,wCAAwC;gBACxC,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;oBACzB,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;gBACtC,CAAC;YACH,CAAC;YAED,4CAA4C;YAC5C,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;gBAChB,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;YACtB,CAAC;QACH,CAAC;QAED,8CAA8C;QAC9C,MAAM,OAAO,GAAsC;YACjD,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,eAAe,IAAI,IAAI;YAChC,OAAO,EAAE,IAAI;SACd,CAAC;QAEF,wBAAwB;QACxB,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACzB,OAAO,CAAC,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,8BAA8B;QACtF,CAAC;QAED,MAAM,gBAAgB,GAA+B;YACnD,EAAE,EAAE,UAAU,CAAC,EAAE;YACjB,MAAM,EAAE,iBAAiB;YACzB,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,KAAK,EAAE,UAAU,CAAC,KAAK;YACvB,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,CAAC;oBACR,OAAO;oBACP,aAAa,EAAE,YAAY,IAAI,MAAM;oBACrC,QAAQ,EAAE,IAAI;iBACf;aACF;YACD,KAAK,EAAE,KAAK,IAAI;gBACd,aAAa,EAAE,CAAC;gBAChB,iBAAiB,EAAE,CAAC;gBACpB,YAAY,EAAE,CAAC;aAChB;YACD,kBAAkB,EAAE,UAAU,CAAC,kBAAkB;SAClD,CAAC;QAEF,OAAO,gBAAgB,CAAC;IAC1B,CAAC;CACF"}

package/dist/src/core/openaiContentGenerator/telemetryService.test.d.ts

@@ -0,0 +1,6 @@
+/**
+ * @license
+ * Copyright 2025 Blackbox
+ * SPDX-License-Identifier: Apache-2.0
+ */
+export {};