diff --git a/packages/agents-a365-observability/docs/design.md b/packages/agents-a365-observability/docs/design.md index 7d947eb3..d4cdf37a 100644 --- a/packages/agents-a365-observability/docs/design.md +++ b/packages/agents-a365-observability/docs/design.md @@ -451,9 +451,7 @@ enum InvocationRole { } enum InferenceOperationType { - CHAT = 'Chat', - TEXT_COMPLETION = 'TextCompletion', - GENERATE_CONTENT = 'GenerateContent' + CHAT = 'Chat' } ``` diff --git a/packages/agents-a365-observability/src/tracing/contracts.ts b/packages/agents-a365-observability/src/tracing/contracts.ts index 0ed18d8b..b154b1de 100644 --- a/packages/agents-a365-observability/src/tracing/contracts.ts +++ b/packages/agents-a365-observability/src/tracing/contracts.ts @@ -26,9 +26,7 @@ export enum InvocationRole { * Represents different operation for types for model inference */ export enum InferenceOperationType { - CHAT = 'Chat', - TEXT_COMPLETION = 'TextCompletion', - GENERATE_CONTENT = 'GenerateContent' + CHAT = 'Chat' } /** diff --git a/packages/agents-a365-observability/src/tracing/exporter/utils.ts b/packages/agents-a365-observability/src/tracing/exporter/utils.ts index 2c2f813e..9910ffdd 100644 --- a/packages/agents-a365-observability/src/tracing/exporter/utils.ts +++ b/packages/agents-a365-observability/src/tracing/exporter/utils.ts @@ -97,8 +97,6 @@ const GEN_AI_OPERATION_NAMES: ReadonlySet = new Set([ OpenTelemetryConstants.OUTPUT_MESSAGES_OPERATION_NAME, // 'output_messages' OpenTelemetryConstants.CHAT_OPERATION_NAME, // 'chat' 'Chat', // InferenceOperationType.CHAT - 'TextCompletion', // InferenceOperationType.TEXT_COMPLETION - 'GenerateContent', // InferenceOperationType.GENERATE_CONTENT ]); /** diff --git a/tests/observability/core/scopes.test.ts b/tests/observability/core/scopes.test.ts index 1fec6ff9..f96f46b3 100644 --- a/tests/observability/core/scopes.test.ts +++ b/tests/observability/core/scopes.test.ts @@ -558,7 +558,7 @@ describe('Scopes', () => { it('should create scope with minimal details', () => { const spy = jest.spyOn(OpenTelemetryScope.prototype as any, 'setTagMaybe'); const inferenceDetails: InferenceDetails = { - operationName: InferenceOperationType.TEXT_COMPLETION, + operationName: InferenceOperationType.CHAT, model: 'gpt-3.5-turbo' };