@loongsuite/otel-util-genai 0.1.0-beta.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/LICENSE +201 -0
- package/README.md +262 -0
- package/dist/environment-variables.d.ts +8 -0
- package/dist/environment-variables.d.ts.map +1 -0
- package/dist/environment-variables.js +21 -0
- package/dist/environment-variables.js.map +1 -0
- package/dist/event-log/converter.d.ts +19 -0
- package/dist/event-log/converter.d.ts.map +1 -0
- package/dist/event-log/converter.js +216 -0
- package/dist/event-log/converter.js.map +1 -0
- package/dist/event-log/field-mapping.d.ts +53 -0
- package/dist/event-log/field-mapping.d.ts.map +1 -0
- package/dist/event-log/field-mapping.js +400 -0
- package/dist/event-log/field-mapping.js.map +1 -0
- package/dist/event-log/grouping.d.ts +67 -0
- package/dist/event-log/grouping.d.ts.map +1 -0
- package/dist/event-log/grouping.js +241 -0
- package/dist/event-log/grouping.js.map +1 -0
- package/dist/event-log/index.d.ts +5 -0
- package/dist/event-log/index.d.ts.map +1 -0
- package/dist/event-log/index.js +18 -0
- package/dist/event-log/index.js.map +1 -0
- package/dist/event-log/parent-context.d.ts +14 -0
- package/dist/event-log/parent-context.d.ts.map +1 -0
- package/dist/event-log/parent-context.js +45 -0
- package/dist/event-log/parent-context.js.map +1 -0
- package/dist/event-log/readable-spans.d.ts +36 -0
- package/dist/event-log/readable-spans.d.ts.map +1 -0
- package/dist/event-log/readable-spans.js +73 -0
- package/dist/event-log/readable-spans.js.map +1 -0
- package/dist/event-log/types.d.ts +78 -0
- package/dist/event-log/types.d.ts.map +1 -0
- package/dist/event-log/types.js +35 -0
- package/dist/event-log/types.js.map +1 -0
- package/dist/extended-environment-variables.d.ts +12 -0
- package/dist/extended-environment-variables.d.ts.map +1 -0
- package/dist/extended-environment-variables.js +25 -0
- package/dist/extended-environment-variables.js.map +1 -0
- package/dist/extended-handler.d.ts +50 -0
- package/dist/extended-handler.d.ts.map +1 -0
- package/dist/extended-handler.js +334 -0
- package/dist/extended-handler.js.map +1 -0
- package/dist/extended-metrics.d.ts +13 -0
- package/dist/extended-metrics.d.ts.map +1 -0
- package/dist/extended-metrics.js +27 -0
- package/dist/extended-metrics.js.map +1 -0
- package/dist/extended-span-utils.d.ts +14 -0
- package/dist/extended-span-utils.d.ts.map +1 -0
- package/dist/extended-span-utils.js +427 -0
- package/dist/extended-span-utils.js.map +1 -0
- package/dist/extended-types.d.ts +150 -0
- package/dist/extended-types.d.ts.map +1 -0
- package/dist/extended-types.js +46 -0
- package/dist/extended-types.js.map +1 -0
- package/dist/handler.d.ts +23 -0
- package/dist/handler.d.ts.map +1 -0
- package/dist/handler.js +127 -0
- package/dist/handler.js.map +1 -0
- package/dist/index.d.ts +22 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +43 -0
- package/dist/index.js.map +1 -0
- package/dist/instruments.d.ts +4 -0
- package/dist/instruments.d.ts.map +1 -0
- package/dist/instruments.js +37 -0
- package/dist/instruments.js.map +1 -0
- package/dist/memory/memory-types.d.ts +26 -0
- package/dist/memory/memory-types.d.ts.map +1 -0
- package/dist/memory/memory-types.js +17 -0
- package/dist/memory/memory-types.js.map +1 -0
- package/dist/memory/memory-utils.d.ts +7 -0
- package/dist/memory/memory-utils.d.ts.map +1 -0
- package/dist/memory/memory-utils.js +121 -0
- package/dist/memory/memory-utils.js.map +1 -0
- package/dist/metrics.d.ts +11 -0
- package/dist/metrics.d.ts.map +1 -0
- package/dist/metrics.js +77 -0
- package/dist/metrics.js.map +1 -0
- package/dist/semconv/gen-ai-extended-attributes.d.ts +101 -0
- package/dist/semconv/gen-ai-extended-attributes.d.ts.map +1 -0
- package/dist/semconv/gen-ai-extended-attributes.js +120 -0
- package/dist/semconv/gen-ai-extended-attributes.js.map +1 -0
- package/dist/semconv/gen-ai-memory-attributes.d.ts +28 -0
- package/dist/semconv/gen-ai-memory-attributes.d.ts.map +1 -0
- package/dist/semconv/gen-ai-memory-attributes.js +42 -0
- package/dist/semconv/gen-ai-memory-attributes.js.map +1 -0
- package/dist/span-utils.d.ts +18 -0
- package/dist/span-utils.d.ts.map +1 -0
- package/dist/span-utils.js +236 -0
- package/dist/span-utils.js.map +1 -0
- package/dist/types.d.ts +117 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +33 -0
- package/dist/types.js.map +1 -0
- package/dist/utils.d.ts +8 -0
- package/dist/utils.d.ts.map +1 -0
- package/dist/utils.js +88 -0
- package/dist/utils.js.map +1 -0
- package/dist/version.d.ts +2 -0
- package/dist/version.d.ts.map +1 -0
- package/dist/version.js +2 -0
- package/dist/version.js.map +1 -0
- package/package.json +72 -0
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
// Copyright The OpenTelemetry Authors
|
|
2
|
+
//
|
|
3
|
+
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
// you may not use this file except in compliance with the License.
|
|
5
|
+
// You may obtain a copy of the License at
|
|
6
|
+
//
|
|
7
|
+
// http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
//
|
|
9
|
+
// Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
// See the License for the specific language governing permissions and
|
|
13
|
+
// limitations under the License.
|
|
14
|
+
export const OTEL_INSTRUMENTATION_GENAI_MESSAGE_CONTENT_MAX_LENGTH = "OTEL_INSTRUMENTATION_GENAI_MESSAGE_CONTENT_MAX_LENGTH";
|
|
15
|
+
export const OTEL_INSTRUMENTATION_GENAI_MESSAGE_CONTENT_CAPTURE_STRATEGY = "OTEL_INSTRUMENTATION_GENAI_MESSAGE_CONTENT_CAPTURE_STRATEGY";
|
|
16
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_STORAGE_BASE_PATH = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_STORAGE_BASE_PATH";
|
|
17
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_UPLOAD_MODE = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_UPLOAD_MODE";
|
|
18
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_DOWNLOAD_ENABLED = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_DOWNLOAD_ENABLED";
|
|
19
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_DOWNLOAD_SSL_VERIFY = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_DOWNLOAD_SSL_VERIFY";
|
|
20
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_AUDIO_CONVERSION_ENABLED = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_AUDIO_CONVERSION_ENABLED";
|
|
21
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_LOCAL_FILE_ENABLED = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_LOCAL_FILE_ENABLED";
|
|
22
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_ALLOWED_ROOT_PATHS = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_ALLOWED_ROOT_PATHS";
|
|
23
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_UPLOADER = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_UPLOADER";
|
|
24
|
+
export const OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_PRE_UPLOADER = "OTEL_INSTRUMENTATION_GENAI_MULTIMODAL_PRE_UPLOADER";
|
|
25
|
+
//# sourceMappingURL=extended-environment-variables.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-environment-variables.js","sourceRoot":"","sources":["../src/extended-environment-variables.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,EAAE;AACF,kEAAkE;AAClE,mEAAmE;AACnE,0CAA0C;AAC1C,EAAE;AACF,iDAAiD;AACjD,EAAE;AACF,sEAAsE;AACtE,oEAAoE;AACpE,2EAA2E;AAC3E,sEAAsE;AACtE,iCAAiC;AAEjC,MAAM,CAAC,MAAM,qDAAqD,GAChE,uDAAuD,CAAC;AAC1D,MAAM,CAAC,MAAM,2DAA2D,GACtE,6DAA6D,CAAC;AAChE,MAAM,CAAC,MAAM,uDAAuD,GAClE,yDAAyD,CAAC;AAC5D,MAAM,CAAC,MAAM,iDAAiD,GAC5D,mDAAmD,CAAC;AACtD,MAAM,CAAC,MAAM,sDAAsD,GACjE,wDAAwD,CAAC;AAC3D,MAAM,CAAC,MAAM,yDAAyD,GACpE,2DAA2D,CAAC;AAC9D,MAAM,CAAC,MAAM,8DAA8D,GACzE,gEAAgE,CAAC;AACnE,MAAM,CAAC,MAAM,wDAAwD,GACnE,0DAA0D,CAAC;AAC7D,MAAM,CAAC,MAAM,wDAAwD,GACnE,0DAA0D,CAAC;AAC7D,MAAM,CAAC,MAAM,8CAA8C,GACzD,gDAAgD,CAAC;AACnD,MAAM,CAAC,MAAM,kDAAkD,GAC7D,oDAAoD,CAAC"}
|
|
@@ -0,0 +1,50 @@
|
|
|
1
|
+
import { type Context } from "@opentelemetry/api";
|
|
2
|
+
import { TelemetryHandler, type TelemetryHandlerOptions } from "./handler.js";
|
|
3
|
+
import type { GenAIError } from "./types.js";
|
|
4
|
+
import type { CreateAgentInvocation, EmbeddingInvocation, ExecuteToolInvocation, InvokeAgentInvocation, RetrievalInvocation, RerankInvocation, EntryInvocation, ReactStepInvocation } from "./extended-types.js";
|
|
5
|
+
import type { MemoryInvocation } from "./memory/memory-types.js";
|
|
6
|
+
export declare class ExtendedTelemetryHandler extends TelemetryHandler {
|
|
7
|
+
private _extMetricsRecorder;
|
|
8
|
+
constructor(options?: TelemetryHandlerOptions);
|
|
9
|
+
private _startSpan;
|
|
10
|
+
private _endSpan;
|
|
11
|
+
startCreateAgent(invocation: CreateAgentInvocation, parentContext?: Context, startTime?: number): CreateAgentInvocation;
|
|
12
|
+
stopCreateAgent(invocation: CreateAgentInvocation, endTime?: number): CreateAgentInvocation;
|
|
13
|
+
failCreateAgent(invocation: CreateAgentInvocation, error: GenAIError, endTime?: number): CreateAgentInvocation;
|
|
14
|
+
createAgent<T>(invocationOrFn?: CreateAgentInvocation | null, fn?: (inv: CreateAgentInvocation) => T): T extends Promise<unknown> ? Promise<CreateAgentInvocation> : CreateAgentInvocation;
|
|
15
|
+
startEmbedding(invocation: EmbeddingInvocation, parentContext?: Context, startTime?: number): EmbeddingInvocation;
|
|
16
|
+
stopEmbedding(invocation: EmbeddingInvocation, endTime?: number): EmbeddingInvocation;
|
|
17
|
+
failEmbedding(invocation: EmbeddingInvocation, error: GenAIError, endTime?: number): EmbeddingInvocation;
|
|
18
|
+
embedding<T>(invocationOrFn?: EmbeddingInvocation | null, fn?: (inv: EmbeddingInvocation) => T): T extends Promise<unknown> ? Promise<EmbeddingInvocation> : EmbeddingInvocation;
|
|
19
|
+
startExecuteTool(invocation: ExecuteToolInvocation, parentContext?: Context, startTime?: number): ExecuteToolInvocation;
|
|
20
|
+
stopExecuteTool(invocation: ExecuteToolInvocation, endTime?: number): ExecuteToolInvocation;
|
|
21
|
+
failExecuteTool(invocation: ExecuteToolInvocation, error: GenAIError, endTime?: number): ExecuteToolInvocation;
|
|
22
|
+
executeTool<T>(invocationOrFn?: ExecuteToolInvocation | null, fn?: (inv: ExecuteToolInvocation) => T): T extends Promise<unknown> ? Promise<ExecuteToolInvocation> : ExecuteToolInvocation;
|
|
23
|
+
startInvokeAgent(invocation: InvokeAgentInvocation, parentContext?: Context, startTime?: number): InvokeAgentInvocation;
|
|
24
|
+
stopInvokeAgent(invocation: InvokeAgentInvocation, endTime?: number): InvokeAgentInvocation;
|
|
25
|
+
failInvokeAgent(invocation: InvokeAgentInvocation, error: GenAIError, endTime?: number): InvokeAgentInvocation;
|
|
26
|
+
invokeAgent<T>(invocationOrFn?: InvokeAgentInvocation | null, fn?: (inv: InvokeAgentInvocation) => T): T extends Promise<unknown> ? Promise<InvokeAgentInvocation> : InvokeAgentInvocation;
|
|
27
|
+
startRetrieval(invocation: RetrievalInvocation, parentContext?: Context, startTime?: number): RetrievalInvocation;
|
|
28
|
+
stopRetrieval(invocation: RetrievalInvocation, endTime?: number): RetrievalInvocation;
|
|
29
|
+
failRetrieval(invocation: RetrievalInvocation, error: GenAIError, endTime?: number): RetrievalInvocation;
|
|
30
|
+
retrieval<T>(invocationOrFn?: RetrievalInvocation | null, fn?: (inv: RetrievalInvocation) => T): T extends Promise<unknown> ? Promise<RetrievalInvocation> : RetrievalInvocation;
|
|
31
|
+
startRerank(invocation: RerankInvocation, parentContext?: Context, startTime?: number): RerankInvocation;
|
|
32
|
+
stopRerank(invocation: RerankInvocation, endTime?: number): RerankInvocation;
|
|
33
|
+
failRerank(invocation: RerankInvocation, error: GenAIError, endTime?: number): RerankInvocation;
|
|
34
|
+
rerank<T>(invocationOrFn?: RerankInvocation | null, fn?: (inv: RerankInvocation) => T): T extends Promise<unknown> ? Promise<RerankInvocation> : RerankInvocation;
|
|
35
|
+
startMemory(invocation: MemoryInvocation, parentContext?: Context, startTime?: number): MemoryInvocation;
|
|
36
|
+
stopMemory(invocation: MemoryInvocation, endTime?: number): MemoryInvocation;
|
|
37
|
+
failMemory(invocation: MemoryInvocation, error: GenAIError, endTime?: number): MemoryInvocation;
|
|
38
|
+
memory<T>(invocationOrFn?: MemoryInvocation | null, fn?: (inv: MemoryInvocation) => T): T extends Promise<unknown> ? Promise<MemoryInvocation> : MemoryInvocation;
|
|
39
|
+
startEntry(invocation: EntryInvocation, parentContext?: Context, startTime?: number): EntryInvocation;
|
|
40
|
+
stopEntry(invocation: EntryInvocation, endTime?: number): EntryInvocation;
|
|
41
|
+
failEntry(invocation: EntryInvocation, error: GenAIError, endTime?: number): EntryInvocation;
|
|
42
|
+
entry<T>(invocationOrFn?: EntryInvocation | null, fn?: (inv: EntryInvocation) => T): T extends Promise<unknown> ? Promise<EntryInvocation> : EntryInvocation;
|
|
43
|
+
startReactStep(invocation: ReactStepInvocation, parentContext?: Context, startTime?: number): ReactStepInvocation;
|
|
44
|
+
stopReactStep(invocation: ReactStepInvocation, endTime?: number): ReactStepInvocation;
|
|
45
|
+
failReactStep(invocation: ReactStepInvocation, error: GenAIError, endTime?: number): ReactStepInvocation;
|
|
46
|
+
reactStep<T>(invocationOrFn?: ReactStepInvocation | null, fn?: (inv: ReactStepInvocation) => T): T extends Promise<unknown> ? Promise<ReactStepInvocation> : ReactStepInvocation;
|
|
47
|
+
private _withInvocation;
|
|
48
|
+
}
|
|
49
|
+
export declare function getExtendedTelemetryHandler(options?: TelemetryHandlerOptions): ExtendedTelemetryHandler;
|
|
50
|
+
//# sourceMappingURL=extended-handler.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-handler.d.ts","sourceRoot":"","sources":["../src/extended-handler.ts"],"names":[],"mappings":"AAcA,OAAO,EAKL,KAAK,OAAO,EAEb,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAE,KAAK,uBAAuB,EAAE,MAAM,cAAc,CAAC;AAE9E,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAC7C,OAAO,KAAK,EACV,qBAAqB,EACrB,mBAAmB,EACnB,qBAAqB,EACrB,qBAAqB,EACrB,mBAAmB,EACnB,gBAAgB,EAChB,eAAe,EACf,mBAAmB,EACpB,MAAM,qBAAqB,CAAC;AAW7B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAuCjE,qBAAa,wBAAyB,SAAQ,gBAAgB;IAC5D,OAAO,CAAC,mBAAmB,CAAoC;gBAEnD,OAAO,GAAE,uBAA4B;IAajD,OAAO,CAAC,UAAU;IAqBlB,OAAO,CAAC,QAAQ;IAMhB,gBAAgB,CACd,UAAU,EAAE,qBAAqB,EACjC,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,qBAAqB;IAOxB,eAAe,CAAC,UAAU,EAAE,qBAAqB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,qBAAqB;IAQ3F,eAAe,CACb,UAAU,EAAE,qBAAqB,EACjC,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,qBAAqB;IAWxB,WAAW,CAAC,CAAC,EACX,cAAc,CAAC,EAAE,qBAAqB,GAAG,IAAI,EAC7C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,CAAC,GACrC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,qBAAqB,CAAC,GAC9B,qBAAqB;IAYzB,cAAc,CACZ,UAAU,EAAE,mBAAmB,EAC/B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,mBAAmB;IAUtB,aAAa,CAAC,UAAU,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,mBAAmB;IAQrF,aAAa,CACX,UAAU,EAAE,mBAAmB,EAC/B,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,mBAAmB;IAWtB,SAAS,CAAC,CAAC,EACT,cAAc,CAAC,EAAE,mBAAmB,GAAG,IAAI,EAC3C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,mBAAmB,KAAK,CAAC,GACnC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,mBAAmB,CAAC,GAC5B,mBAAmB;IAYvB,gBAAgB,CACd,UAAU,EAAE,qBAAqB,EACjC,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,qBAAqB;IAUxB,eAAe,CAAC,UAAU,EAAE,qBAAqB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,qBAAqB;IAQ3F,eAAe,CACb,UAAU,EAAE,qBAAqB,EACjC,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,qBAAqB;IAWxB,WAAW,CAAC,CAAC,EACX,cAAc,CAAC,EAAE,qBAAqB,GAAG,IAAI,EAC7C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,CAAC,GACrC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,qBAAqB,CAAC,GAC9B,qBAAqB;IAYzB,gBAAgB,CACd,UAAU,EAAE,qBAAqB,EACjC,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,qBAAqB;IAOxB,eAAe,CAAC,UAAU,EAAE,qBAAqB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,qBAAqB;IAS3F,eAAe,CACb,UAAU,EAAE,qBAAqB,EACjC,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,qBAAqB;IAiBxB,WAAW,CAAC,CAAC,EACX,cAAc,CAAC,EAAE,qBAAqB,GAAG,IAAI,EAC7C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,qBAAqB,KAAK,CAAC,GACrC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,qBAAqB,CAAC,GAC9B,qBAAqB;IAYzB,cAAc,CACZ,UAAU,EAAE,mBAAmB,EAC/B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,mBAAmB;IAWtB,aAAa,CAAC,UAAU,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,mBAAmB;IAQrF,aAAa,CACX,UAAU,EAAE,mBAAmB,EAC/B,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,mBAAmB;IAWtB,SAAS,CAAC,CAAC,EACT,cAAc,CAAC,EAAE,mBAAmB,GAAG,IAAI,EAC3C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,mBAAmB,KAAK,CAAC,GACnC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,mBAAmB,CAAC,GAC5B,mBAAmB;IAYvB,WAAW,CACT,UAAU,EAAE,gBAAgB,EAC5B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,gBAAgB;IAUnB,UAAU,CAAC,UAAU,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,gBAAgB;IAQ5E,UAAU,CACR,UAAU,EAAE,gBAAgB,EAC5B,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,gBAAgB;IAWnB,MAAM,CAAC,CAAC,EACN,cAAc,CAAC,EAAE,gBAAgB,GAAG,IAAI,EACxC,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,gBAAgB,KAAK,CAAC,GAChC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,gBAAgB,CAAC,GACzB,gBAAgB;IAYpB,WAAW,CACT,UAAU,EAAE,gBAAgB,EAC5B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,gBAAgB;IAanB,UAAU,CAAC,UAAU,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,gBAAgB;IAS5E,UAAU,CACR,UAAU,EAAE,gBAAgB,EAC5B,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,gBAAgB;IAYnB,MAAM,CAAC,CAAC,EACN,cAAc,CAAC,EAAE,gBAAgB,GAAG,IAAI,EACxC,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,gBAAgB,KAAK,CAAC,GAChC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,gBAAgB,CAAC,GACzB,gBAAgB;IAYpB,UAAU,CACR,UAAU,EAAE,eAAe,EAC3B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,eAAe;IAUlB,SAAS,CAAC,UAAU,EAAE,eAAe,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,eAAe;IAQzE,SAAS,CAAC,UAAU,EAAE,eAAe,EAAE,KAAK,EAAE,UAAU,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,eAAe;IAW5F,KAAK,CAAC,CAAC,EACL,cAAc,CAAC,EAAE,eAAe,GAAG,IAAI,EACvC,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,eAAe,KAAK,CAAC,GAC/B,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,eAAe,CAAC,GACxB,eAAe;IAYnB,cAAc,CACZ,UAAU,EAAE,mBAAmB,EAC/B,aAAa,CAAC,EAAE,OAAO,EACvB,SAAS,CAAC,EAAE,MAAM,GACjB,mBAAmB;IAUtB,aAAa,CAAC,UAAU,EAAE,mBAAmB,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,mBAAmB;IAQrF,aAAa,CACX,UAAU,EAAE,mBAAmB,EAC/B,KAAK,EAAE,UAAU,EACjB,OAAO,CAAC,EAAE,MAAM,GACf,mBAAmB;IAWtB,SAAS,CAAC,CAAC,EACT,cAAc,CAAC,EAAE,mBAAmB,GAAG,IAAI,EAC3C,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,mBAAmB,KAAK,CAAC,GACnC,CAAC,SAAS,OAAO,CAAC,OAAO,CAAC,GACzB,OAAO,CAAC,mBAAmB,CAAC,GAC5B,mBAAmB;IAYvB,OAAO,CAAC,eAAe;CAkCxB;AAID,wBAAgB,2BAA2B,CACzC,OAAO,CAAC,EAAE,uBAAuB,GAChC,wBAAwB,CAK1B"}
|
|
@@ -0,0 +1,334 @@
|
|
|
1
|
+
// Copyright The OpenTelemetry Authors
|
|
2
|
+
//
|
|
3
|
+
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
// you may not use this file except in compliance with the License.
|
|
5
|
+
// You may obtain a copy of the License at
|
|
6
|
+
//
|
|
7
|
+
// http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
//
|
|
9
|
+
// Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
// See the License for the specific language governing permissions and
|
|
13
|
+
// limitations under the License.
|
|
14
|
+
import { trace, metrics, context, SpanKind, } from "@opentelemetry/api";
|
|
15
|
+
import { TelemetryHandler } from "./handler.js";
|
|
16
|
+
import { applyErrorAttributes } from "./span-utils.js";
|
|
17
|
+
import { createCreateAgentInvocation, createEmbeddingInvocation, createExecuteToolInvocation, createInvokeAgentInvocation, createRetrievalInvocation, createRerankInvocation, createEntryInvocation, createReactStepInvocation, } from "./extended-types.js";
|
|
18
|
+
import { createMemoryInvocation } from "./memory/memory-types.js";
|
|
19
|
+
import { applyEmbeddingFinishAttributes, applyCreateAgentFinishAttributes, applyExecuteToolFinishAttributes, applyInvokeAgentFinishAttributes, maybeEmitInvokeAgentEvent, applyRetrievalFinishAttributes, applyRerankFinishAttributes, applyEntryFinishAttributes, applyReactStepFinishAttributes, } from "./extended-span-utils.js";
|
|
20
|
+
import { applyMemoryFinishAttributes, maybeEmitMemoryEvent, } from "./memory/memory-utils.js";
|
|
21
|
+
import { ExtendedInvocationMetricsRecorder } from "./extended-metrics.js";
|
|
22
|
+
import { GenAiOperationNameValues, GenAiExtendedOperationNameValues, } from "./semconv/gen-ai-extended-attributes.js";
|
|
23
|
+
import { VERSION } from "./version.js";
|
|
24
|
+
const INSTRUMENTATION_NAME = "@loongsuite/otel-util-genai";
|
|
25
|
+
function makeGenAIError(err) {
|
|
26
|
+
return {
|
|
27
|
+
message: String(err),
|
|
28
|
+
type: err instanceof Error ? err.constructor.name : "Error",
|
|
29
|
+
};
|
|
30
|
+
}
|
|
31
|
+
export class ExtendedTelemetryHandler extends TelemetryHandler {
|
|
32
|
+
_extMetricsRecorder;
|
|
33
|
+
constructor(options = {}) {
|
|
34
|
+
super(options);
|
|
35
|
+
const scopeName = options.instrumentationName ?? INSTRUMENTATION_NAME;
|
|
36
|
+
const scopeVersion = options.instrumentationVersion ?? VERSION;
|
|
37
|
+
const meter = (options.meterProvider ?? metrics).getMeter(scopeName, scopeVersion);
|
|
38
|
+
this._extMetricsRecorder = new ExtendedInvocationMetricsRecorder(meter);
|
|
39
|
+
}
|
|
40
|
+
// ==================== Generic Helpers ====================
|
|
41
|
+
_startSpan(invocation, spanName, kind = SpanKind.CLIENT, parentContext, startTime) {
|
|
42
|
+
const span = this._tracer.startSpan(spanName, { kind, startTime }, parentContext);
|
|
43
|
+
invocation.monotonicStartS = performance.now() / 1000;
|
|
44
|
+
invocation.span = span;
|
|
45
|
+
invocation.contextToken = trace.setSpan(parentContext ?? context.active(), span);
|
|
46
|
+
return invocation;
|
|
47
|
+
}
|
|
48
|
+
_endSpan(inv, endTime) {
|
|
49
|
+
inv.span?.end(endTime);
|
|
50
|
+
}
|
|
51
|
+
// ==================== Create Agent ====================
|
|
52
|
+
startCreateAgent(invocation, parentContext, startTime) {
|
|
53
|
+
const name = invocation.agentName
|
|
54
|
+
? `${GenAiOperationNameValues.CREATE_AGENT} ${invocation.agentName}`
|
|
55
|
+
: GenAiOperationNameValues.CREATE_AGENT;
|
|
56
|
+
return this._startSpan(invocation, name, SpanKind.CLIENT, parentContext, startTime);
|
|
57
|
+
}
|
|
58
|
+
stopCreateAgent(invocation, endTime) {
|
|
59
|
+
if (!invocation.span)
|
|
60
|
+
return invocation;
|
|
61
|
+
applyCreateAgentFinishAttributes(invocation.span, invocation);
|
|
62
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
63
|
+
this._endSpan(invocation, endTime);
|
|
64
|
+
return invocation;
|
|
65
|
+
}
|
|
66
|
+
failCreateAgent(invocation, error, endTime) {
|
|
67
|
+
if (!invocation.span)
|
|
68
|
+
return invocation;
|
|
69
|
+
applyCreateAgentFinishAttributes(invocation.span, invocation);
|
|
70
|
+
applyErrorAttributes(invocation.span, error);
|
|
71
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
72
|
+
errorType: error.type,
|
|
73
|
+
});
|
|
74
|
+
this._endSpan(invocation, endTime);
|
|
75
|
+
return invocation;
|
|
76
|
+
}
|
|
77
|
+
createAgent(invocationOrFn, fn) {
|
|
78
|
+
return this._withInvocation(invocationOrFn ?? createCreateAgentInvocation(""), fn, (inv, ctx) => this.startCreateAgent(inv, ctx), (inv) => this.stopCreateAgent(inv), (inv, err) => this.failCreateAgent(inv, err));
|
|
79
|
+
}
|
|
80
|
+
// ==================== Embedding ====================
|
|
81
|
+
startEmbedding(invocation, parentContext, startTime) {
|
|
82
|
+
return this._startSpan(invocation, `${GenAiOperationNameValues.EMBEDDINGS} ${invocation.requestModel}`, SpanKind.CLIENT, parentContext, startTime);
|
|
83
|
+
}
|
|
84
|
+
stopEmbedding(invocation, endTime) {
|
|
85
|
+
if (!invocation.span)
|
|
86
|
+
return invocation;
|
|
87
|
+
applyEmbeddingFinishAttributes(invocation.span, invocation);
|
|
88
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
89
|
+
this._endSpan(invocation, endTime);
|
|
90
|
+
return invocation;
|
|
91
|
+
}
|
|
92
|
+
failEmbedding(invocation, error, endTime) {
|
|
93
|
+
if (!invocation.span)
|
|
94
|
+
return invocation;
|
|
95
|
+
applyEmbeddingFinishAttributes(invocation.span, invocation);
|
|
96
|
+
applyErrorAttributes(invocation.span, error);
|
|
97
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
98
|
+
errorType: error.type,
|
|
99
|
+
});
|
|
100
|
+
this._endSpan(invocation, endTime);
|
|
101
|
+
return invocation;
|
|
102
|
+
}
|
|
103
|
+
embedding(invocationOrFn, fn) {
|
|
104
|
+
return this._withInvocation(invocationOrFn ?? createEmbeddingInvocation(""), fn, (inv, ctx) => this.startEmbedding(inv, ctx), (inv) => this.stopEmbedding(inv), (inv, err) => this.failEmbedding(inv, err));
|
|
105
|
+
}
|
|
106
|
+
// ==================== Execute Tool ====================
|
|
107
|
+
startExecuteTool(invocation, parentContext, startTime) {
|
|
108
|
+
return this._startSpan(invocation, `${GenAiOperationNameValues.EXECUTE_TOOL} ${invocation.toolName}`, SpanKind.INTERNAL, parentContext, startTime);
|
|
109
|
+
}
|
|
110
|
+
stopExecuteTool(invocation, endTime) {
|
|
111
|
+
if (!invocation.span)
|
|
112
|
+
return invocation;
|
|
113
|
+
applyExecuteToolFinishAttributes(invocation.span, invocation);
|
|
114
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
115
|
+
this._endSpan(invocation, endTime);
|
|
116
|
+
return invocation;
|
|
117
|
+
}
|
|
118
|
+
failExecuteTool(invocation, error, endTime) {
|
|
119
|
+
if (!invocation.span)
|
|
120
|
+
return invocation;
|
|
121
|
+
applyExecuteToolFinishAttributes(invocation.span, invocation);
|
|
122
|
+
applyErrorAttributes(invocation.span, error);
|
|
123
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
124
|
+
errorType: error.type,
|
|
125
|
+
});
|
|
126
|
+
this._endSpan(invocation, endTime);
|
|
127
|
+
return invocation;
|
|
128
|
+
}
|
|
129
|
+
executeTool(invocationOrFn, fn) {
|
|
130
|
+
return this._withInvocation(invocationOrFn ?? createExecuteToolInvocation(""), fn, (inv, ctx) => this.startExecuteTool(inv, ctx), (inv) => this.stopExecuteTool(inv), (inv, err) => this.failExecuteTool(inv, err));
|
|
131
|
+
}
|
|
132
|
+
// ==================== Invoke Agent ====================
|
|
133
|
+
startInvokeAgent(invocation, parentContext, startTime) {
|
|
134
|
+
const name = invocation.agentName
|
|
135
|
+
? `${GenAiOperationNameValues.INVOKE_AGENT} ${invocation.agentName}`
|
|
136
|
+
: GenAiOperationNameValues.INVOKE_AGENT;
|
|
137
|
+
return this._startSpan(invocation, name, SpanKind.CLIENT, parentContext, startTime);
|
|
138
|
+
}
|
|
139
|
+
stopInvokeAgent(invocation, endTime) {
|
|
140
|
+
if (!invocation.span)
|
|
141
|
+
return invocation;
|
|
142
|
+
applyInvokeAgentFinishAttributes(invocation.span, invocation);
|
|
143
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
144
|
+
maybeEmitInvokeAgentEvent(this._logger, invocation.span, invocation);
|
|
145
|
+
this._endSpan(invocation, endTime);
|
|
146
|
+
return invocation;
|
|
147
|
+
}
|
|
148
|
+
failInvokeAgent(invocation, error, endTime) {
|
|
149
|
+
if (!invocation.span)
|
|
150
|
+
return invocation;
|
|
151
|
+
applyInvokeAgentFinishAttributes(invocation.span, invocation);
|
|
152
|
+
applyErrorAttributes(invocation.span, error);
|
|
153
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
154
|
+
errorType: error.type,
|
|
155
|
+
});
|
|
156
|
+
maybeEmitInvokeAgentEvent(this._logger, invocation.span, invocation, error);
|
|
157
|
+
this._endSpan(invocation, endTime);
|
|
158
|
+
return invocation;
|
|
159
|
+
}
|
|
160
|
+
invokeAgent(invocationOrFn, fn) {
|
|
161
|
+
return this._withInvocation(invocationOrFn ?? createInvokeAgentInvocation(""), fn, (inv, ctx) => this.startInvokeAgent(inv, ctx), (inv) => this.stopInvokeAgent(inv), (inv, err) => this.failInvokeAgent(inv, err));
|
|
162
|
+
}
|
|
163
|
+
// ==================== Retrieval ====================
|
|
164
|
+
startRetrieval(invocation, parentContext, startTime) {
|
|
165
|
+
const dsId = invocation.dataSourceId ?? "";
|
|
166
|
+
return this._startSpan(invocation, `${GenAiExtendedOperationNameValues.RETRIEVAL} ${dsId}`.trim(), SpanKind.CLIENT, parentContext, startTime);
|
|
167
|
+
}
|
|
168
|
+
stopRetrieval(invocation, endTime) {
|
|
169
|
+
if (!invocation.span)
|
|
170
|
+
return invocation;
|
|
171
|
+
applyRetrievalFinishAttributes(invocation.span, invocation);
|
|
172
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
173
|
+
this._endSpan(invocation, endTime);
|
|
174
|
+
return invocation;
|
|
175
|
+
}
|
|
176
|
+
failRetrieval(invocation, error, endTime) {
|
|
177
|
+
if (!invocation.span)
|
|
178
|
+
return invocation;
|
|
179
|
+
applyRetrievalFinishAttributes(invocation.span, invocation);
|
|
180
|
+
applyErrorAttributes(invocation.span, error);
|
|
181
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
182
|
+
errorType: error.type,
|
|
183
|
+
});
|
|
184
|
+
this._endSpan(invocation, endTime);
|
|
185
|
+
return invocation;
|
|
186
|
+
}
|
|
187
|
+
retrieval(invocationOrFn, fn) {
|
|
188
|
+
return this._withInvocation(invocationOrFn ?? createRetrievalInvocation(), fn, (inv, ctx) => this.startRetrieval(inv, ctx), (inv) => this.stopRetrieval(inv), (inv, err) => this.failRetrieval(inv, err));
|
|
189
|
+
}
|
|
190
|
+
// ==================== Rerank ====================
|
|
191
|
+
startRerank(invocation, parentContext, startTime) {
|
|
192
|
+
return this._startSpan(invocation, `${GenAiExtendedOperationNameValues.RERANK_DOCUMENTS} ${invocation.requestModel ?? ""}`.trim(), SpanKind.CLIENT, parentContext, startTime);
|
|
193
|
+
}
|
|
194
|
+
stopRerank(invocation, endTime) {
|
|
195
|
+
if (!invocation.span)
|
|
196
|
+
return invocation;
|
|
197
|
+
applyRerankFinishAttributes(invocation.span, invocation);
|
|
198
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
199
|
+
this._endSpan(invocation, endTime);
|
|
200
|
+
return invocation;
|
|
201
|
+
}
|
|
202
|
+
failRerank(invocation, error, endTime) {
|
|
203
|
+
if (!invocation.span)
|
|
204
|
+
return invocation;
|
|
205
|
+
applyRerankFinishAttributes(invocation.span, invocation);
|
|
206
|
+
applyErrorAttributes(invocation.span, error);
|
|
207
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
208
|
+
errorType: error.type,
|
|
209
|
+
});
|
|
210
|
+
this._endSpan(invocation, endTime);
|
|
211
|
+
return invocation;
|
|
212
|
+
}
|
|
213
|
+
rerank(invocationOrFn, fn) {
|
|
214
|
+
return this._withInvocation(invocationOrFn ?? createRerankInvocation(""), fn, (inv, ctx) => this.startRerank(inv, ctx), (inv) => this.stopRerank(inv), (inv, err) => this.failRerank(inv, err));
|
|
215
|
+
}
|
|
216
|
+
// ==================== Memory ====================
|
|
217
|
+
startMemory(invocation, parentContext, startTime) {
|
|
218
|
+
const spanName = invocation.operation
|
|
219
|
+
? `memory_operation ${invocation.operation}`
|
|
220
|
+
: "memory_operation";
|
|
221
|
+
return this._startSpan(invocation, spanName, SpanKind.CLIENT, parentContext, startTime);
|
|
222
|
+
}
|
|
223
|
+
stopMemory(invocation, endTime) {
|
|
224
|
+
if (!invocation.span)
|
|
225
|
+
return invocation;
|
|
226
|
+
applyMemoryFinishAttributes(invocation.span, invocation);
|
|
227
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
228
|
+
maybeEmitMemoryEvent(this._logger, invocation.span, invocation);
|
|
229
|
+
this._endSpan(invocation, endTime);
|
|
230
|
+
return invocation;
|
|
231
|
+
}
|
|
232
|
+
failMemory(invocation, error, endTime) {
|
|
233
|
+
if (!invocation.span)
|
|
234
|
+
return invocation;
|
|
235
|
+
applyMemoryFinishAttributes(invocation.span, invocation);
|
|
236
|
+
applyErrorAttributes(invocation.span, error);
|
|
237
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
238
|
+
errorType: error.type,
|
|
239
|
+
});
|
|
240
|
+
maybeEmitMemoryEvent(this._logger, invocation.span, invocation, error);
|
|
241
|
+
this._endSpan(invocation, endTime);
|
|
242
|
+
return invocation;
|
|
243
|
+
}
|
|
244
|
+
memory(invocationOrFn, fn) {
|
|
245
|
+
return this._withInvocation(invocationOrFn ?? createMemoryInvocation(""), fn, (inv, ctx) => this.startMemory(inv, ctx), (inv) => this.stopMemory(inv), (inv, err) => this.failMemory(inv, err));
|
|
246
|
+
}
|
|
247
|
+
// ==================== Entry ====================
|
|
248
|
+
startEntry(invocation, parentContext, startTime) {
|
|
249
|
+
return this._startSpan(invocation, "enter_ai_application_system", SpanKind.SERVER, parentContext, startTime);
|
|
250
|
+
}
|
|
251
|
+
stopEntry(invocation, endTime) {
|
|
252
|
+
if (!invocation.span)
|
|
253
|
+
return invocation;
|
|
254
|
+
applyEntryFinishAttributes(invocation.span, invocation);
|
|
255
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
256
|
+
this._endSpan(invocation, endTime);
|
|
257
|
+
return invocation;
|
|
258
|
+
}
|
|
259
|
+
failEntry(invocation, error, endTime) {
|
|
260
|
+
if (!invocation.span)
|
|
261
|
+
return invocation;
|
|
262
|
+
applyEntryFinishAttributes(invocation.span, invocation);
|
|
263
|
+
applyErrorAttributes(invocation.span, error);
|
|
264
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
265
|
+
errorType: error.type,
|
|
266
|
+
});
|
|
267
|
+
this._endSpan(invocation, endTime);
|
|
268
|
+
return invocation;
|
|
269
|
+
}
|
|
270
|
+
entry(invocationOrFn, fn) {
|
|
271
|
+
return this._withInvocation(invocationOrFn ?? createEntryInvocation(), fn, (inv, ctx) => this.startEntry(inv, ctx), (inv) => this.stopEntry(inv), (inv, err) => this.failEntry(inv, err));
|
|
272
|
+
}
|
|
273
|
+
// ==================== ReAct Step ====================
|
|
274
|
+
startReactStep(invocation, parentContext, startTime) {
|
|
275
|
+
return this._startSpan(invocation, "react step", SpanKind.INTERNAL, parentContext, startTime);
|
|
276
|
+
}
|
|
277
|
+
stopReactStep(invocation, endTime) {
|
|
278
|
+
if (!invocation.span)
|
|
279
|
+
return invocation;
|
|
280
|
+
applyReactStepFinishAttributes(invocation.span, invocation);
|
|
281
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation);
|
|
282
|
+
this._endSpan(invocation, endTime);
|
|
283
|
+
return invocation;
|
|
284
|
+
}
|
|
285
|
+
failReactStep(invocation, error, endTime) {
|
|
286
|
+
if (!invocation.span)
|
|
287
|
+
return invocation;
|
|
288
|
+
applyReactStepFinishAttributes(invocation.span, invocation);
|
|
289
|
+
applyErrorAttributes(invocation.span, error);
|
|
290
|
+
this._extMetricsRecorder.recordExtended(invocation.span, invocation, {
|
|
291
|
+
errorType: error.type,
|
|
292
|
+
});
|
|
293
|
+
this._endSpan(invocation, endTime);
|
|
294
|
+
return invocation;
|
|
295
|
+
}
|
|
296
|
+
reactStep(invocationOrFn, fn) {
|
|
297
|
+
return this._withInvocation(invocationOrFn ?? createReactStepInvocation(), fn, (inv, ctx) => this.startReactStep(inv, ctx), (inv) => this.stopReactStep(inv), (inv, err) => this.failReactStep(inv, err));
|
|
298
|
+
}
|
|
299
|
+
// ==================== Generic Callback Pattern ====================
|
|
300
|
+
_withInvocation(invocation, fn, start, stop, fail) {
|
|
301
|
+
start(invocation);
|
|
302
|
+
if (!fn) {
|
|
303
|
+
return invocation;
|
|
304
|
+
}
|
|
305
|
+
try {
|
|
306
|
+
const result = fn(invocation);
|
|
307
|
+
if (result instanceof Promise) {
|
|
308
|
+
return result
|
|
309
|
+
.then(() => {
|
|
310
|
+
stop(invocation);
|
|
311
|
+
return invocation;
|
|
312
|
+
})
|
|
313
|
+
.catch((err) => {
|
|
314
|
+
fail(invocation, makeGenAIError(err));
|
|
315
|
+
throw err;
|
|
316
|
+
});
|
|
317
|
+
}
|
|
318
|
+
stop(invocation);
|
|
319
|
+
return invocation;
|
|
320
|
+
}
|
|
321
|
+
catch (err) {
|
|
322
|
+
fail(invocation, makeGenAIError(err));
|
|
323
|
+
throw err;
|
|
324
|
+
}
|
|
325
|
+
}
|
|
326
|
+
}
|
|
327
|
+
let _defaultExtendedHandler = null;
|
|
328
|
+
export function getExtendedTelemetryHandler(options) {
|
|
329
|
+
if (!_defaultExtendedHandler) {
|
|
330
|
+
_defaultExtendedHandler = new ExtendedTelemetryHandler(options);
|
|
331
|
+
}
|
|
332
|
+
return _defaultExtendedHandler;
|
|
333
|
+
}
|
|
334
|
+
//# sourceMappingURL=extended-handler.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-handler.js","sourceRoot":"","sources":["../src/extended-handler.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,EAAE;AACF,kEAAkE;AAClE,mEAAmE;AACnE,0CAA0C;AAC1C,EAAE;AACF,iDAAiD;AACjD,EAAE;AACF,sEAAsE;AACtE,oEAAoE;AACpE,2EAA2E;AAC3E,sEAAsE;AACtE,iCAAiC;AAEjC,OAAO,EACL,KAAK,EACL,OAAO,EACP,OAAO,EACP,QAAQ,GAGT,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,gBAAgB,EAAgC,MAAM,cAAc,CAAC;AAC9E,OAAO,EAAE,oBAAoB,EAAE,MAAM,iBAAiB,CAAC;AAYvD,OAAO,EACL,2BAA2B,EAC3B,yBAAyB,EACzB,2BAA2B,EAC3B,2BAA2B,EAC3B,yBAAyB,EACzB,sBAAsB,EACtB,qBAAqB,EACrB,yBAAyB,GAC1B,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EACL,8BAA8B,EAC9B,gCAAgC,EAChC,gCAAgC,EAChC,gCAAgC,EAChC,yBAAyB,EACzB,8BAA8B,EAC9B,2BAA2B,EAC3B,0BAA0B,EAC1B,8BAA8B,GAC/B,MAAM,0BAA0B,CAAC;AAClC,OAAO,EACL,2BAA2B,EAC3B,oBAAoB,GACrB,MAAM,0BAA0B,CAAC;AAClC,OAAO,EAAE,iCAAiC,EAAE,MAAM,uBAAuB,CAAC;AAC1E,OAAO,EACL,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EAAE,OAAO,EAAE,MAAM,cAAc,CAAC;AAEvC,MAAM,oBAAoB,GAAG,6BAA6B,CAAC;AAQ3D,SAAS,cAAc,CAAC,GAAY;IAClC,OAAO;QACL,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC;QACpB,IAAI,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO;KAC5D,CAAC;AACJ,CAAC;AAED,MAAM,OAAO,wBAAyB,SAAQ,gBAAgB;IACpD,mBAAmB,CAAoC;IAE/D,YAAY,UAAmC,EAAE;QAC/C,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,MAAM,SAAS,GAAG,OAAO,CAAC,mBAAmB,IAAI,oBAAoB,CAAC;QACtE,MAAM,YAAY,GAAG,OAAO,CAAC,sBAAsB,IAAI,OAAO,CAAC;QAC/D,MAAM,KAAK,GAAG,CAAC,OAAO,CAAC,aAAa,IAAI,OAAO,CAAC,CAAC,QAAQ,CACvD,SAAS,EACT,YAAY,CACb,CAAC;QACF,IAAI,CAAC,mBAAmB,GAAG,IAAI,iCAAiC,CAAC,KAAK,CAAC,CAAC;IAC1E,CAAC;IAED,4DAA4D;IAEpD,UAAU,CAChB,UAAa,EACb,QAAgB,EAChB,OAAiB,QAAQ,CAAC,MAAM,EAChC,aAAuB,EACvB,SAAkB;QAElB,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CACjC,QAAQ,EACR,EAAE,IAAI,EAAE,SAAS,EAAE,EACnB,aAAa,CACd,CAAC;QACF,UAAU,CAAC,eAAe,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;QACtD,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC;QACvB,UAAU,CAAC,YAAY,GAAG,KAAK,CAAC,OAAO,CACrC,aAAa,IAAI,OAAO,CAAC,MAAM,EAAE,EACjC,IAAI,CACL,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,QAAQ,CAAC,GAA0B,EAAE,OAAgB;QAC3D,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;IACzB,CAAC;IAED,yDAAyD;IAEzD,gBAAgB,CACd,UAAiC,EACjC,aAAuB,EACvB,SAAkB;QAElB,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS;YAC/B,CAAC,CAAC,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,SAAS,EAAE;YACpE,CAAC,CAAC,wBAAwB,CAAC,YAAY,CAAC;QAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;IACtF,CAAC;IAED,eAAe,CAAC,UAAiC,EAAE,OAAgB;QACjE,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,eAAe,CACb,UAAiC,EACjC,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,WAAW,CACT,cAA6C,EAC7C,EAAsC;QAItC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,2BAA2B,CAAC,EAAE,CAAC,EACjD,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,EAC7C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAClC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,sDAAsD;IAEtD,cAAc,CACZ,UAA+B,EAC/B,aAAuB,EACvB,SAAkB;QAElB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,GAAG,wBAAwB,CAAC,UAAU,IAAI,UAAU,CAAC,YAAY,EAAE,EACnE,QAAQ,CAAC,MAAM,EACf,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,UAA+B,EAAE,OAAgB;QAC7D,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,aAAa,CACX,UAA+B,EAC/B,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,SAAS,CACP,cAA2C,EAC3C,EAAoC;QAIpC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,yBAAyB,CAAC,EAAE,CAAC,EAC/C,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,EAC3C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAChC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAC3C,CAAC;IACJ,CAAC;IAED,yDAAyD;IAEzD,gBAAgB,CACd,UAAiC,EACjC,aAAuB,EACvB,SAAkB;QAElB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,QAAQ,EAAE,EACjE,QAAQ,CAAC,QAAQ,EACjB,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,eAAe,CAAC,UAAiC,EAAE,OAAgB;QACjE,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,eAAe,CACb,UAAiC,EACjC,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,WAAW,CACT,cAA6C,EAC7C,EAAsC;QAItC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,2BAA2B,CAAC,EAAE,CAAC,EACjD,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,EAC7C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAClC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,yDAAyD;IAEzD,gBAAgB,CACd,UAAiC,EACjC,aAAuB,EACvB,SAAkB;QAElB,MAAM,IAAI,GAAG,UAAU,CAAC,SAAS;YAC/B,CAAC,CAAC,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,SAAS,EAAE;YACpE,CAAC,CAAC,wBAAwB,CAAC,YAAY,CAAC;QAC1C,OAAO,IAAI,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,EAAE,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,SAAS,CAAC,CAAC;IACtF,CAAC;IAED,eAAe,CAAC,UAAiC,EAAE,OAAgB;QACjE,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,yBAAyB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,eAAe,CACb,UAAiC,EACjC,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,gCAAgC,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC9D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,yBAAyB,CACvB,IAAI,CAAC,OAAO,EACZ,UAAU,CAAC,IAAI,EACf,UAAU,EACV,KAAK,CACN,CAAC;QACF,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,WAAW,CACT,cAA6C,EAC7C,EAAsC;QAItC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,2BAA2B,CAAC,EAAE,CAAC,EACjD,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,GAAG,CAAC,EAC7C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAClC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,sDAAsD;IAEtD,cAAc,CACZ,UAA+B,EAC/B,aAAuB,EACvB,SAAkB;QAElB,MAAM,IAAI,GAAG,UAAU,CAAC,YAAY,IAAI,EAAE,CAAC;QAC3C,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,GAAG,gCAAgC,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC,IAAI,EAAE,EAC9D,QAAQ,CAAC,MAAM,EACf,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,UAA+B,EAAE,OAAgB;QAC7D,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,aAAa,CACX,UAA+B,EAC/B,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,SAAS,CACP,cAA2C,EAC3C,EAAoC;QAIpC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,yBAAyB,EAAE,EAC7C,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,EAC3C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAChC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAC3C,CAAC;IACJ,CAAC;IAED,mDAAmD;IAEnD,WAAW,CACT,UAA4B,EAC5B,aAAuB,EACvB,SAAkB;QAElB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,GAAG,gCAAgC,CAAC,gBAAgB,IAAI,UAAU,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,EAC9F,QAAQ,CAAC,MAAM,EACf,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,UAA4B,EAAE,OAAgB;QACvD,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,2BAA2B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACzD,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,UAAU,CACR,UAA4B,EAC5B,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,2BAA2B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACzD,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,CACJ,cAAwC,EACxC,EAAiC;QAIjC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,sBAAsB,CAAC,EAAE,CAAC,EAC5C,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EACxC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAC7B,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,CACxC,CAAC;IACJ,CAAC;IAED,mDAAmD;IAEnD,WAAW,CACT,UAA4B,EAC5B,aAAuB,EACvB,SAAkB;QAElB,MAAM,QAAQ,GAAG,UAAU,CAAC,SAAS;YACnC,CAAC,CAAC,oBAAoB,UAAU,CAAC,SAAS,EAAE;YAC5C,CAAC,CAAC,kBAAkB,CAAC;QACvB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,QAAQ,EACR,QAAQ,CAAC,MAAM,EACf,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,UAAU,CAAC,UAA4B,EAAE,OAAgB;QACvD,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,2BAA2B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACzD,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAChE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,UAAU,CACR,UAA4B,EAC5B,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,2BAA2B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACzD,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,oBAAoB,CAAC,IAAI,CAAC,OAAO,EAAE,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE,KAAK,CAAC,CAAC;QACvE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,MAAM,CACJ,cAAwC,EACxC,EAAiC;QAIjC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,sBAAsB,CAAC,EAAE,CAAC,EAC5C,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EACxC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAC7B,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,CACxC,CAAC;IACJ,CAAC;IAED,kDAAkD;IAElD,UAAU,CACR,UAA2B,EAC3B,aAAuB,EACvB,SAAkB;QAElB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,6BAA6B,EAC7B,QAAQ,CAAC,MAAM,EACf,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,SAAS,CAAC,UAA2B,EAAE,OAAgB;QACrD,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,0BAA0B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACxD,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,SAAS,CAAC,UAA2B,EAAE,KAAiB,EAAE,OAAgB;QACxE,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,0BAA0B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACxD,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,KAAK,CACH,cAAuC,EACvC,EAAgC;QAIhC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,qBAAqB,EAAE,EACzC,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,EACvC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAC5B,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CACvC,CAAC;IACJ,CAAC;IAED,uDAAuD;IAEvD,cAAc,CACZ,UAA+B,EAC/B,aAAuB,EACvB,SAAkB;QAElB,OAAO,IAAI,CAAC,UAAU,CACpB,UAAU,EACV,YAAY,EACZ,QAAQ,CAAC,QAAQ,EACjB,aAAa,EACb,SAAS,CACV,CAAC;IACJ,CAAC;IAED,aAAa,CAAC,UAA+B,EAAE,OAAgB;QAC7D,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QACrE,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,aAAa,CACX,UAA+B,EAC/B,KAAiB,EACjB,OAAgB;QAEhB,IAAI,CAAC,UAAU,CAAC,IAAI;YAAE,OAAO,UAAU,CAAC;QACxC,8BAA8B,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;QAC5D,oBAAoB,CAAC,UAAU,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC7C,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,UAAU,EAAE;YACnE,SAAS,EAAE,KAAK,CAAC,IAAI;SACtB,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,SAAS,CACP,cAA2C,EAC3C,EAAoC;QAIpC,OAAO,IAAI,CAAC,eAAe,CACzB,cAAc,IAAI,yBAAyB,EAAE,EAC7C,EAAE,EACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,cAAc,CAAC,GAAG,EAAE,GAAG,CAAC,EAC3C,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,EAChC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,aAAa,CAAC,GAAG,EAAE,GAAG,CAAC,CAC3C,CAAC;IACJ,CAAC;IAED,qEAAqE;IAE7D,eAAe,CACrB,UAAa,EACb,EAA+B,EAC/B,KAAmC,EACnC,IAAmB,EACnB,IAAoC;QAGpC,KAAK,CAAC,UAAU,CAAC,CAAC;QAElB,IAAI,CAAC,EAAE,EAAE,CAAC;YACR,OAAO,UAAU,CAAC;QACpB,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,EAAE,CAAC,UAAU,CAAC,CAAC;YAC9B,IAAI,MAAM,YAAY,OAAO,EAAE,CAAC;gBAC9B,OAAO,MAAM;qBACV,IAAI,CAAC,GAAG,EAAE;oBACT,IAAI,CAAC,UAAU,CAAC,CAAC;oBACjB,OAAO,UAAU,CAAC;gBACpB,CAAC,CAAC;qBACD,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;oBACtB,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;oBACtC,MAAM,GAAG,CAAC;gBACZ,CAAC,CAAC,CAAC;YACP,CAAC;YACD,IAAI,CAAC,UAAU,CAAC,CAAC;YACjB,OAAO,UAAU,CAAC;QACpB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC;YACtC,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF;AAED,IAAI,uBAAuB,GAAoC,IAAI,CAAC;AAEpE,MAAM,UAAU,2BAA2B,CACzC,OAAiC;IAEjC,IAAI,CAAC,uBAAuB,EAAE,CAAC;QAC7B,uBAAuB,GAAG,IAAI,wBAAwB,CAAC,OAAO,CAAC,CAAC;IAClE,CAAC;IACD,OAAO,uBAAuB,CAAC;AACjC,CAAC"}
|
|
@@ -0,0 +1,13 @@
|
|
|
1
|
+
import type { Span } from "@opentelemetry/api";
|
|
2
|
+
import { InvocationMetricsRecorder } from "./metrics.js";
|
|
3
|
+
import type { LLMInvocation } from "./types.js";
|
|
4
|
+
import type { EmbeddingInvocation, ExecuteToolInvocation, InvokeAgentInvocation, CreateAgentInvocation, RetrievalInvocation, RerankInvocation, EntryInvocation, ReactStepInvocation } from "./extended-types.js";
|
|
5
|
+
import type { MemoryInvocation } from "./memory/memory-types.js";
|
|
6
|
+
type AnyInvocation = LLMInvocation | EmbeddingInvocation | ExecuteToolInvocation | InvokeAgentInvocation | CreateAgentInvocation | RetrievalInvocation | RerankInvocation | MemoryInvocation | EntryInvocation | ReactStepInvocation;
|
|
7
|
+
export declare class ExtendedInvocationMetricsRecorder extends InvocationMetricsRecorder {
|
|
8
|
+
recordExtended(span: Span | null | undefined, invocation: AnyInvocation, options?: {
|
|
9
|
+
errorType?: string;
|
|
10
|
+
}): void;
|
|
11
|
+
}
|
|
12
|
+
export {};
|
|
13
|
+
//# sourceMappingURL=extended-metrics.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-metrics.d.ts","sourceRoot":"","sources":["../src/extended-metrics.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AACzD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAChD,OAAO,KAAK,EACV,mBAAmB,EACnB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACrB,mBAAmB,EACnB,gBAAgB,EAChB,eAAe,EACf,mBAAmB,EACpB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,0BAA0B,CAAC;AAEjE,KAAK,aAAa,GACd,aAAa,GACb,mBAAmB,GACnB,qBAAqB,GACrB,qBAAqB,GACrB,qBAAqB,GACrB,mBAAmB,GACnB,gBAAgB,GAChB,gBAAgB,GAChB,eAAe,GACf,mBAAmB,CAAC;AAMxB,qBAAa,iCAAkC,SAAQ,yBAAyB;IAC9E,cAAc,CACZ,IAAI,EAAE,IAAI,GAAG,IAAI,GAAG,SAAS,EAC7B,UAAU,EAAE,aAAa,EACzB,OAAO,CAAC,EAAE;QAAE,SAAS,CAAC,EAAE,MAAM,CAAA;KAAE,GAC/B,IAAI;CAOR"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
// Copyright The OpenTelemetry Authors
|
|
2
|
+
//
|
|
3
|
+
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
// you may not use this file except in compliance with the License.
|
|
5
|
+
// You may obtain a copy of the License at
|
|
6
|
+
//
|
|
7
|
+
// http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
//
|
|
9
|
+
// Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
// See the License for the specific language governing permissions and
|
|
13
|
+
// limitations under the License.
|
|
14
|
+
import { InvocationMetricsRecorder } from "./metrics.js";
|
|
15
|
+
function isLLMInvocation(inv) {
|
|
16
|
+
return "operationName" in inv && "inputMessages" in inv;
|
|
17
|
+
}
|
|
18
|
+
export class ExtendedInvocationMetricsRecorder extends InvocationMetricsRecorder {
|
|
19
|
+
recordExtended(span, invocation, options) {
|
|
20
|
+
if (isLLMInvocation(invocation)) {
|
|
21
|
+
this.record(span, invocation, options);
|
|
22
|
+
return;
|
|
23
|
+
}
|
|
24
|
+
// TODO: Implement extended metrics for other invocation types
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=extended-metrics.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-metrics.js","sourceRoot":"","sources":["../src/extended-metrics.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,EAAE;AACF,kEAAkE;AAClE,mEAAmE;AACnE,0CAA0C;AAC1C,EAAE;AACF,iDAAiD;AACjD,EAAE;AACF,sEAAsE;AACtE,oEAAoE;AACpE,2EAA2E;AAC3E,sEAAsE;AACtE,iCAAiC;AAGjC,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AA0BzD,SAAS,eAAe,CAAC,GAAkB;IACzC,OAAO,eAAe,IAAI,GAAG,IAAI,eAAe,IAAI,GAAG,CAAC;AAC1D,CAAC;AAED,MAAM,OAAO,iCAAkC,SAAQ,yBAAyB;IAC9E,cAAc,CACZ,IAA6B,EAC7B,UAAyB,EACzB,OAAgC;QAEhC,IAAI,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;YAChC,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YACvC,OAAO;QACT,CAAC;QACD,8DAA8D;IAChE,CAAC;CACF"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { Span } from "@opentelemetry/api";
|
|
2
|
+
import type { GenAIError } from "./types.js";
|
|
3
|
+
import type { EmbeddingInvocation, CreateAgentInvocation, ExecuteToolInvocation, InvokeAgentInvocation, RetrievalInvocation, RerankInvocation, EntryInvocation, ReactStepInvocation } from "./extended-types.js";
|
|
4
|
+
import { type EventLogger } from "./span-utils.js";
|
|
5
|
+
export declare function applyEmbeddingFinishAttributes(span: Span, invocation: EmbeddingInvocation): void;
|
|
6
|
+
export declare function applyCreateAgentFinishAttributes(span: Span, invocation: CreateAgentInvocation): void;
|
|
7
|
+
export declare function applyExecuteToolFinishAttributes(span: Span, invocation: ExecuteToolInvocation): void;
|
|
8
|
+
export declare function applyInvokeAgentFinishAttributes(span: Span, invocation: InvokeAgentInvocation): void;
|
|
9
|
+
export declare function maybeEmitInvokeAgentEvent(logger: EventLogger | null | undefined, span: Span, invocation: InvokeAgentInvocation, error?: GenAIError | null): void;
|
|
10
|
+
export declare function applyRetrievalFinishAttributes(span: Span, invocation: RetrievalInvocation): void;
|
|
11
|
+
export declare function applyRerankFinishAttributes(span: Span, invocation: RerankInvocation): void;
|
|
12
|
+
export declare function applyEntryFinishAttributes(span: Span, invocation: EntryInvocation): void;
|
|
13
|
+
export declare function applyReactStepFinishAttributes(span: Span, invocation: ReactStepInvocation): void;
|
|
14
|
+
//# sourceMappingURL=extended-span-utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"extended-span-utils.d.ts","sourceRoot":"","sources":["../src/extended-span-utils.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,OAAO,KAAK,EACV,mBAAmB,EACnB,qBAAqB,EACrB,qBAAqB,EACrB,qBAAqB,EACrB,mBAAmB,EAEnB,gBAAgB,EAChB,eAAe,EACf,mBAAmB,EACpB,MAAM,qBAAqB,CAAC;AA6D7B,OAAO,EAGL,KAAK,WAAW,EACjB,MAAM,iBAAiB,CAAC;AAWzB,wBAAgB,8BAA8B,CAC5C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,mBAAmB,GAC9B,IAAI,CAqCN;AAID,wBAAgB,gCAAgC,CAC9C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,qBAAqB,GAChC,IAAI,CAoCN;AA0BD,wBAAgB,gCAAgC,CAC9C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,qBAAqB,GAChC,IAAI,CAmCN;AAWD,wBAAgB,gCAAgC,CAC9C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,qBAAqB,GAChC,IAAI,CA8HN;AAED,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,WAAW,GAAG,IAAI,GAAG,SAAS,EACtC,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,qBAAqB,EACjC,KAAK,CAAC,EAAE,UAAU,GAAG,IAAI,GACxB,IAAI,CAsBN;AAwBD,wBAAgB,8BAA8B,CAC5C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,mBAAmB,GAC9B,IAAI,CA4CN;AA0BD,wBAAgB,2BAA2B,CACzC,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,gBAAgB,GAC3B,IAAI,CA8DN;AAID,wBAAgB,0BAA0B,CACxC,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,eAAe,GAC1B,IAAI,CA+BN;AAID,wBAAgB,8BAA8B,CAC5C,IAAI,EAAE,IAAI,EACV,UAAU,EAAE,mBAAmB,GAC9B,IAAI,CAmBN"}
|