@arizeai/openinference-instrumentation-beeai 1.0.0 → 1.1.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/esm/config.d.ts +10 -10
- package/dist/esm/config.d.ts.map +1 -1
- package/dist/esm/config.js.map +1 -1
- package/dist/esm/helpers/buildTraceTree.d.ts +3 -1
- package/dist/esm/helpers/buildTraceTree.d.ts.map +1 -1
- package/dist/esm/helpers/buildTraceTree.js +55 -7
- package/dist/esm/helpers/buildTraceTree.js.map +1 -1
- package/dist/esm/helpers/getSerializedObjectSafe.d.ts.map +1 -1
- package/dist/esm/helpers/getSerializedObjectSafe.js +2 -2
- package/dist/esm/helpers/getSerializedObjectSafe.js.map +1 -1
- package/dist/esm/instrumentation.d.ts.map +1 -1
- package/dist/esm/instrumentation.js +24 -7
- package/dist/esm/instrumentation.js.map +1 -1
- package/dist/esm/middleware.d.ts +2 -1
- package/dist/esm/middleware.d.ts.map +1 -1
- package/dist/esm/middleware.js +3 -2
- package/dist/esm/middleware.js.map +1 -1
- package/dist/esm/tsconfig.esm.tsbuildinfo +1 -1
- package/dist/esnext/config.d.ts +10 -10
- package/dist/esnext/config.d.ts.map +1 -1
- package/dist/esnext/config.js.map +1 -1
- package/dist/esnext/helpers/buildTraceTree.d.ts +3 -1
- package/dist/esnext/helpers/buildTraceTree.d.ts.map +1 -1
- package/dist/esnext/helpers/buildTraceTree.js +44 -4
- package/dist/esnext/helpers/buildTraceTree.js.map +1 -1
- package/dist/esnext/helpers/getSerializedObjectSafe.d.ts.map +1 -1
- package/dist/esnext/helpers/getSerializedObjectSafe.js +2 -2
- package/dist/esnext/helpers/getSerializedObjectSafe.js.map +1 -1
- package/dist/esnext/instrumentation.d.ts.map +1 -1
- package/dist/esnext/instrumentation.js +24 -7
- package/dist/esnext/instrumentation.js.map +1 -1
- package/dist/esnext/middleware.d.ts +2 -1
- package/dist/esnext/middleware.d.ts.map +1 -1
- package/dist/esnext/middleware.js +3 -2
- package/dist/esnext/middleware.js.map +1 -1
- package/dist/esnext/tsconfig.esnext.tsbuildinfo +1 -1
- package/dist/examples/run-agent.d.ts +2 -0
- package/dist/examples/run-agent.d.ts.map +1 -0
- package/dist/examples/{bee.js → run-agent.js} +3 -3
- package/dist/examples/run-agent.js.map +1 -0
- package/dist/examples/run-llm.d.ts +2 -0
- package/dist/examples/run-llm.d.ts.map +1 -0
- package/dist/examples/run-llm.js +30 -0
- package/dist/examples/run-llm.js.map +1 -0
- package/dist/examples/run-tool.d.ts +2 -0
- package/dist/examples/run-tool.d.ts.map +1 -0
- package/dist/examples/run-tool.js +28 -0
- package/dist/examples/run-tool.js.map +1 -0
- package/dist/src/config.d.ts +10 -10
- package/dist/src/config.d.ts.map +1 -1
- package/dist/src/config.js.map +1 -1
- package/dist/src/helpers/buildTraceTree.d.ts +3 -1
- package/dist/src/helpers/buildTraceTree.d.ts.map +1 -1
- package/dist/src/helpers/buildTraceTree.js +44 -4
- package/dist/src/helpers/buildTraceTree.js.map +1 -1
- package/dist/src/helpers/getSerializedObjectSafe.d.ts.map +1 -1
- package/dist/src/helpers/getSerializedObjectSafe.js +2 -2
- package/dist/src/helpers/getSerializedObjectSafe.js.map +1 -1
- package/dist/src/instrumentation.d.ts.map +1 -1
- package/dist/src/instrumentation.js +24 -7
- package/dist/src/instrumentation.js.map +1 -1
- package/dist/src/middleware.d.ts +2 -1
- package/dist/src/middleware.d.ts.map +1 -1
- package/dist/src/middleware.js +2 -1
- package/dist/src/middleware.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +5 -5
- package/src/config.ts +11 -10
- package/src/helpers/buildTraceTree.ts +88 -10
- package/src/helpers/getSerializedObjectSafe.ts +14 -17
- package/src/instrumentation.ts +60 -9
- package/src/middleware.ts +13 -6
- package/dist/examples/bee.d.ts +0 -2
- package/dist/examples/bee.d.ts.map +0 -1
- package/dist/examples/bee.js.map +0 -1
package/dist/esm/config.d.ts
CHANGED
|
@@ -13,19 +13,19 @@
|
|
|
13
13
|
* See the License for the specific language governing permissions and
|
|
14
14
|
* limitations under the License.
|
|
15
15
|
*/
|
|
16
|
-
import {
|
|
16
|
+
import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
|
|
17
17
|
import { ChatModelEvents } from "beeai-framework/backend/chat";
|
|
18
18
|
import { ToolEvents } from "beeai-framework/tools/base";
|
|
19
19
|
export declare const INSTRUMENTATION_IGNORED_KEYS: string[];
|
|
20
|
-
export declare const partialUpdateEventName: keyof
|
|
21
|
-
export declare const updateEventName: keyof
|
|
22
|
-
export declare const toolErrorEventName: keyof
|
|
23
|
-
export declare const toolStartEventName: keyof
|
|
24
|
-
export declare const toolSuccessEventName: keyof
|
|
25
|
-
export declare const startEventName: keyof
|
|
26
|
-
export declare const successEventName: keyof
|
|
27
|
-
export declare const errorEventName: keyof
|
|
28
|
-
export declare const retryEventName: keyof
|
|
20
|
+
export declare const partialUpdateEventName: keyof ReActAgentCallbacks;
|
|
21
|
+
export declare const updateEventName: keyof ReActAgentCallbacks;
|
|
22
|
+
export declare const toolErrorEventName: keyof ReActAgentCallbacks;
|
|
23
|
+
export declare const toolStartEventName: keyof ReActAgentCallbacks;
|
|
24
|
+
export declare const toolSuccessEventName: keyof ReActAgentCallbacks;
|
|
25
|
+
export declare const startEventName: keyof ReActAgentCallbacks;
|
|
26
|
+
export declare const successEventName: keyof ReActAgentCallbacks;
|
|
27
|
+
export declare const errorEventName: keyof ReActAgentCallbacks;
|
|
28
|
+
export declare const retryEventName: keyof ReActAgentCallbacks;
|
|
29
29
|
export declare const newTokenLLMEventName: keyof ChatModelEvents;
|
|
30
30
|
export declare const successLLMEventName: keyof ChatModelEvents;
|
|
31
31
|
export declare const startLLMEventName: keyof ChatModelEvents;
|
package/dist/esm/config.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,oCAAoC,CAAC;AAC9E,OAAO,EAAE,eAAe,EAAE,MAAM,8BAA8B,CAAC;AAC/D,OAAO,EAAE,UAAU,EAAE,MAAM,4BAA4B,CAAC;AAExD,eAAO,MAAM,4BAA4B,UAKnC,CAAC;AAEP,eAAO,MAAM,sBAAsB,EAAE,MAAM,mBAC1B,CAAC;AAClB,eAAO,MAAM,eAAe,EAAE,MAAM,mBAA8B,CAAC;AACnE,eAAO,MAAM,kBAAkB,EAAE,MAAM,mBAAiC,CAAC;AACzE,eAAO,MAAM,kBAAkB,EAAE,MAAM,mBAAiC,CAAC;AACzE,eAAO,MAAM,oBAAoB,EAAE,MAAM,mBAAmC,CAAC;AAC7E,eAAO,MAAM,cAAc,EAAE,MAAM,mBAA6B,CAAC;AACjE,eAAO,MAAM,gBAAgB,EAAE,MAAM,mBAA+B,CAAC;AACrE,eAAO,MAAM,cAAc,EAAE,MAAM,mBAA6B,CAAC;AACjE,eAAO,MAAM,cAAc,EAAE,MAAM,mBAA6B,CAAC;AAEjE,eAAO,MAAM,oBAAoB,EAAE,MAAM,eAA4B,CAAC;AACtE,eAAO,MAAM,mBAAmB,EAAE,MAAM,eAA2B,CAAC;AACpE,eAAO,MAAM,iBAAiB,EAAE,MAAM,eAAyB,CAAC;AAChE,eAAO,MAAM,iBAAiB,EAAE,MAAM,eAAyB,CAAC;AAChE,eAAO,MAAM,kBAAkB,EAAE,MAAM,eAA0B,CAAC;AAElE,eAAO,MAAM,kBAAkB,EAAE,MAAM,UAAoB,CAAC;AAC5D,eAAO,MAAM,oBAAoB,EAAE,MAAM,UAAsB,CAAC;AAChE,eAAO,MAAM,mBAAmB,EAAE,MAAM,UAAqB,CAAC;AAC9D,eAAO,MAAM,kBAAkB,EAAE,MAAM,UAAoB,CAAC;AAC5D,eAAO,MAAM,kBAAkB,EAAE,MAAM,UAAoB,CAAC"}
|
package/dist/esm/config.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAMH,MAAM,CAAC,MAAM,4BAA4B,GAAG,OAAO,CAAC,GAAG;KACpD,0CAA0C;IAC3C,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CACtE,OAAO,CACR;IACH,CAAC,CAAC,EAAE,CAAC;AAEP,MAAM,CAAC,MAAM,sBAAsB,
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/config.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAMH,MAAM,CAAC,MAAM,4BAA4B,GAAG,OAAO,CAAC,GAAG;KACpD,0CAA0C;IAC3C,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,0CAA0C,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CACtE,OAAO,CACR;IACH,CAAC,CAAC,EAAE,CAAC;AAEP,MAAM,CAAC,MAAM,sBAAsB,GACjC,eAAe,CAAC;AAClB,MAAM,CAAC,MAAM,eAAe,GAA8B,QAAQ,CAAC;AACnE,MAAM,CAAC,MAAM,kBAAkB,GAA8B,WAAW,CAAC;AACzE,MAAM,CAAC,MAAM,kBAAkB,GAA8B,WAAW,CAAC;AACzE,MAAM,CAAC,MAAM,oBAAoB,GAA8B,aAAa,CAAC;AAC7E,MAAM,CAAC,MAAM,cAAc,GAA8B,OAAO,CAAC;AACjE,MAAM,CAAC,MAAM,gBAAgB,GAA8B,SAAS,CAAC;AACrE,MAAM,CAAC,MAAM,cAAc,GAA8B,OAAO,CAAC;AACjE,MAAM,CAAC,MAAM,cAAc,GAA8B,OAAO,CAAC;AAEjE,MAAM,CAAC,MAAM,oBAAoB,GAA0B,UAAU,CAAC;AACtE,MAAM,CAAC,MAAM,mBAAmB,GAA0B,SAAS,CAAC;AACpE,MAAM,CAAC,MAAM,iBAAiB,GAA0B,OAAO,CAAC;AAChE,MAAM,CAAC,MAAM,iBAAiB,GAA0B,OAAO,CAAC;AAChE,MAAM,CAAC,MAAM,kBAAkB,GAA0B,QAAQ,CAAC;AAElE,MAAM,CAAC,MAAM,kBAAkB,GAAqB,OAAO,CAAC;AAC5D,MAAM,CAAC,MAAM,oBAAoB,GAAqB,SAAS,CAAC;AAChE,MAAM,CAAC,MAAM,mBAAmB,GAAqB,QAAQ,CAAC;AAC9D,MAAM,CAAC,MAAM,kBAAkB,GAAqB,OAAO,CAAC;AAC5D,MAAM,CAAC,MAAM,kBAAkB,GAAqB,OAAO,CAAC"}
|
|
@@ -1,8 +1,10 @@
|
|
|
1
1
|
import { OITracer } from "@arizeai/openinference-core";
|
|
2
2
|
import { FrameworkSpan, GeneratedResponse } from "../types.js";
|
|
3
3
|
import { TimeInput } from "@opentelemetry/api";
|
|
4
|
+
import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
|
|
4
5
|
interface BuiltTraceTreeProps {
|
|
5
6
|
tracer: OITracer;
|
|
7
|
+
mainSpanKind: OpenInferenceSpanKind;
|
|
6
8
|
data: {
|
|
7
9
|
prompt?: string | null;
|
|
8
10
|
history: GeneratedResponse[] | undefined;
|
|
@@ -16,6 +18,6 @@ interface BuiltTraceTreeProps {
|
|
|
16
18
|
source: string;
|
|
17
19
|
};
|
|
18
20
|
}
|
|
19
|
-
export declare function buildTraceTree({ tracer, data }: BuiltTraceTreeProps): void;
|
|
21
|
+
export declare function buildTraceTree({ tracer, data, mainSpanKind, }: BuiltTraceTreeProps): void;
|
|
20
22
|
export {};
|
|
21
23
|
//# sourceMappingURL=buildTraceTree.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"buildTraceTree.d.ts","sourceRoot":"","sources":["../../../src/helpers/buildTraceTree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAkB,SAAS,EAAE,MAAM,oBAAoB,CAAC;
|
|
1
|
+
{"version":3,"file":"buildTraceTree.d.ts","sourceRoot":"","sources":["../../../src/helpers/buildTraceTree.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAC5D,OAAO,EAAkB,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACL,qBAAqB,EAEtB,MAAM,6CAA6C,CAAC;AAGrD,UAAU,mBAAmB;IAC3B,MAAM,EAAE,QAAQ,CAAC;IACjB,YAAY,EAAE,qBAAqB,CAAC;IACpC,IAAI,EAAE;QACJ,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;QACvB,OAAO,EAAE,iBAAiB,EAAE,GAAG,SAAS,CAAC;QACzC,gBAAgB,EAAE,iBAAiB,GAAG,SAAS,CAAC;QAChD,KAAK,EAAE,aAAa,EAAE,CAAC;QACvB,OAAO,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,CAAC;QAChB,eAAe,EAAE,MAAM,CAAC;QACxB,SAAS,EAAE,SAAS,CAAC;QACrB,OAAO,EAAE,SAAS,CAAC;QACnB,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;CACH;AA0HD,wBAAgB,cAAc,CAAC,EAC7B,MAAM,EACN,IAAI,EACJ,YAAY,GACb,EAAE,mBAAmB,QA6CrB"}
|
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
import { SpanStatusCode } from "@opentelemetry/api";
|
|
2
2
|
import { OpenInferenceSpanKind, SemanticConventions, } from "@arizeai/openinference-semantic-conventions";
|
|
3
|
+
import { startEventName, successEventName } from "../config.js";
|
|
3
4
|
function buildSpansForParent({ tracer, data }) {
|
|
4
5
|
data.spans
|
|
5
6
|
.filter((beeaiSpan) => beeaiSpan.parent_id === data.parentId)
|
|
@@ -34,7 +35,58 @@ function buildSpansForParent({ tracer, data }) {
|
|
|
34
35
|
});
|
|
35
36
|
});
|
|
36
37
|
}
|
|
37
|
-
|
|
38
|
+
function buildAgentMainSpanData(data) {
|
|
39
|
+
return {
|
|
40
|
+
...(data.prompt && { [SemanticConventions.INPUT_VALUE]: data.prompt }),
|
|
41
|
+
...(data.generatedMessage !== undefined && {
|
|
42
|
+
[SemanticConventions.OUTPUT_VALUE]: JSON.stringify(data.generatedMessage),
|
|
43
|
+
}),
|
|
44
|
+
...(data.history && { history: JSON.stringify(data.history) }),
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
function buildToolMainSpanData(data) {
|
|
48
|
+
const startBeeaiSpan = data.spans.find((span) => span.name === startEventName);
|
|
49
|
+
const successBeeeaiSpan = data.spans.find((span) => span.name === successEventName);
|
|
50
|
+
return {
|
|
51
|
+
...(startBeeaiSpan && {
|
|
52
|
+
[SemanticConventions.INPUT_VALUE]: startBeeaiSpan.attributes.data?.[SemanticConventions.TOOL_PARAMETERS],
|
|
53
|
+
[SemanticConventions.TOOL_PARAMETERS]: startBeeaiSpan.attributes.data?.[SemanticConventions.TOOL_PARAMETERS],
|
|
54
|
+
}),
|
|
55
|
+
...(successBeeeaiSpan && {
|
|
56
|
+
[SemanticConventions.OUTPUT_VALUE]: successBeeeaiSpan.attributes.data?.[SemanticConventions.OUTPUT_VALUE],
|
|
57
|
+
}),
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
function buildLLMMainSpanData(data) {
|
|
61
|
+
const startBeeaiSpan = data.spans.find((span) => span.name === startEventName);
|
|
62
|
+
const successBeeeaiSpan = data.spans.find((span) => span.name === successEventName);
|
|
63
|
+
if (!startBeeaiSpan && !successBeeeaiSpan)
|
|
64
|
+
return {};
|
|
65
|
+
const provider = startBeeaiSpan?.attributes.data?.[SemanticConventions.LLM_PROVIDER] ||
|
|
66
|
+
successBeeeaiSpan?.attributes.data?.[SemanticConventions.LLM_PROVIDER];
|
|
67
|
+
const modelName = startBeeaiSpan?.attributes.data?.[SemanticConventions.LLM_MODEL_NAME] ||
|
|
68
|
+
successBeeeaiSpan?.attributes.data?.[SemanticConventions.LLM_MODEL_NAME];
|
|
69
|
+
return {
|
|
70
|
+
...(startBeeaiSpan && {
|
|
71
|
+
[SemanticConventions.INPUT_VALUE]: startBeeaiSpan?.attributes.data?.[SemanticConventions.INPUT_VALUE],
|
|
72
|
+
[SemanticConventions.INPUT_MIME_TYPE]: startBeeaiSpan?.attributes.data?.[SemanticConventions.INPUT_MIME_TYPE],
|
|
73
|
+
}),
|
|
74
|
+
...(successBeeeaiSpan && {
|
|
75
|
+
[SemanticConventions.OUTPUT_MIME_TYPE]: successBeeeaiSpan.attributes.data?.[SemanticConventions.OUTPUT_MIME_TYPE],
|
|
76
|
+
[SemanticConventions.OUTPUT_VALUE]: successBeeeaiSpan.attributes.data?.[SemanticConventions.OUTPUT_VALUE],
|
|
77
|
+
}),
|
|
78
|
+
...(provider && { [SemanticConventions.LLM_PROVIDER]: provider }),
|
|
79
|
+
...(modelName && { [SemanticConventions.LLM_MODEL_NAME]: modelName }),
|
|
80
|
+
};
|
|
81
|
+
}
|
|
82
|
+
export function buildTraceTree({ tracer, data, mainSpanKind, }) {
|
|
83
|
+
const computedData = mainSpanKind === OpenInferenceSpanKind.AGENT
|
|
84
|
+
? buildAgentMainSpanData(data)
|
|
85
|
+
: mainSpanKind === OpenInferenceSpanKind.TOOL
|
|
86
|
+
? buildToolMainSpanData(data)
|
|
87
|
+
: mainSpanKind === OpenInferenceSpanKind.LLM
|
|
88
|
+
? buildLLMMainSpanData(data)
|
|
89
|
+
: {};
|
|
38
90
|
tracer.startActiveSpan(`beeai-framework-main`, {
|
|
39
91
|
// custom start time
|
|
40
92
|
startTime: data.startTime,
|
|
@@ -42,13 +94,9 @@ export function buildTraceTree({ tracer, data }) {
|
|
|
42
94
|
attributes: {
|
|
43
95
|
source: data.source,
|
|
44
96
|
traceId: data.traceId,
|
|
45
|
-
[SemanticConventions.OPENINFERENCE_SPAN_KIND]:
|
|
97
|
+
[SemanticConventions.OPENINFERENCE_SPAN_KIND]: mainSpanKind,
|
|
46
98
|
["beeai.version"]: data.version,
|
|
47
|
-
...
|
|
48
|
-
...(data.generatedMessage !== undefined && {
|
|
49
|
-
[SemanticConventions.OUTPUT_VALUE]: JSON.stringify(data.generatedMessage),
|
|
50
|
-
}),
|
|
51
|
-
...(data.history && { history: JSON.stringify(data.history) }),
|
|
99
|
+
...computedData,
|
|
52
100
|
},
|
|
53
101
|
}, (activeSpan) => {
|
|
54
102
|
// set status
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"buildTraceTree.js","sourceRoot":"","sources":["../../../src/helpers/buildTraceTree.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAa,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,6CAA6C,CAAC;
|
|
1
|
+
{"version":3,"file":"buildTraceTree.js","sourceRoot":"","sources":["../../../src/helpers/buildTraceTree.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAa,MAAM,oBAAoB,CAAC;AAC/D,OAAO,EACL,qBAAqB,EACrB,mBAAmB,GACpB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AA4B7D,SAAS,mBAAmB,CAAC,EAAE,MAAM,EAAE,IAAI,EAA4B;IACrE,IAAI,CAAC,KAAK;SACP,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,SAAS,KAAK,IAAI,CAAC,QAAQ,CAAC;SAC5D,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;QACrB,MAAM,CAAC,eAAe,CACpB,SAAS,CAAC,OAAO,CAAC,OAAO,EACzB;YACE,oBAAoB;YACpB,SAAS,EAAE,SAAS,CAAC,UAAU;YAC/B,gCAAgC;YAChC,UAAU,EAAE;gBACV,MAAM,EAAE,SAAS,CAAC,UAAU,CAAC,MAAM;gBACnC,IAAI,EAAE,SAAS,CAAC,IAAI;gBACpB,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,IAAI;oBACnC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC;iBACxD,CAAC;gBACF,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,IAAI,EAAE,GAAG,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;aACnE;SACF,EACD,CAAC,UAAU,EAAE,EAAE;YACb,aAAa;YACb,UAAU,CAAC,SAAS,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAEvC,mBAAmB;YACnB,mBAAmB,CAAC;gBAClB,MAAM;gBACN,IAAI,EAAE;oBACJ,KAAK,EAAE,IAAI,CAAC,KAAK;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO;oBACrB,QAAQ,EAAE,SAAS,CAAC,OAAO,CAAC,OAAO;iBACpC;aACF,CAAC,CAAC;YAEH,kBAAkB;YAClB,UAAU,CAAC,GAAG,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QACrC,CAAC,CACF,CAAC;IACJ,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,sBAAsB,CAAC,IAAiC;IAC/D,OAAO;QACL,GAAG,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC;QACtE,GAAG,CAAC,IAAI,CAAC,gBAAgB,KAAK,SAAS,IAAI;YACzC,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC;SAC1E,CAAC;QACF,GAAG,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;KAC/D,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAiC;IAC9D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CACpC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CACvC,CAAC;IACF,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CACzC,CAAC;IAEF,OAAO;QACL,GAAG,CAAC,cAAc,IAAI;YACpB,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAC/B,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC;YACvE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EACnC,cAAc,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC;SACxE,CAAC;QACF,GAAG,CAAC,iBAAiB,IAAI;YACvB,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAChC,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,YAAY,CAAC;SACxE,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,oBAAoB,CAAC,IAAiC;IAC7D,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CACpC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,cAAc,CACvC,CAAC;IACF,MAAM,iBAAiB,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CACvC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,gBAAgB,CACzC,CAAC;IAEF,IAAI,CAAC,cAAc,IAAI,CAAC,iBAAiB;QAAE,OAAO,EAAE,CAAC;IAErD,MAAM,QAAQ,GACZ,cAAc,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,YAAY,CAAC;QACnE,iBAAiB,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,YAAY,CAAC,CAAC;IAEzE,MAAM,SAAS,GACb,cAAc,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,cAAc,CAAC;QACrE,iBAAiB,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;IAE3E,OAAO;QACL,GAAG,CAAC,cAAc,IAAI;YACpB,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAC/B,cAAc,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,WAAW,CAAC;YACpE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EACnC,cAAc,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC;SACzE,CAAC;QACF,GAAG,CAAC,iBAAiB,IAAI;YACvB,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EACpC,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,CACjC,mBAAmB,CAAC,gBAAgB,CACrC;YACH,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAChC,iBAAiB,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,mBAAmB,CAAC,YAAY,CAAC;SACxE,CAAC;QACF,GAAG,CAAC,QAAQ,IAAI,EAAE,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,QAAQ,EAAE,CAAC;QACjE,GAAG,CAAC,SAAS,IAAI,EAAE,CAAC,mBAAmB,CAAC,cAAc,CAAC,EAAE,SAAS,EAAE,CAAC;KACtE,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,EAC7B,MAAM,EACN,IAAI,EACJ,YAAY,GACQ;IACpB,MAAM,YAAY,GAChB,YAAY,KAAK,qBAAqB,CAAC,KAAK;QAC1C,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC;QAC9B,CAAC,CAAC,YAAY,KAAK,qBAAqB,CAAC,IAAI;YAC3C,CAAC,CAAC,qBAAqB,CAAC,IAAI,CAAC;YAC7B,CAAC,CAAC,YAAY,KAAK,qBAAqB,CAAC,GAAG;gBAC1C,CAAC,CAAC,oBAAoB,CAAC,IAAI,CAAC;gBAC5B,CAAC,CAAC,EAAE,CAAC;IAEb,MAAM,CAAC,eAAe,CACpB,sBAAsB,EACtB;QACE,oBAAoB;QACpB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,qCAAqC;QACrC,UAAU,EAAE;YACV,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,YAAY;YAC3D,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,OAAO;YAC/B,GAAG,YAAY;SAChB;KACF,EACD,CAAC,UAAU,EAAE,EAAE;QACb,aAAa;QACb,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAClC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,IAAI,CAAC,eAAe,CAC1D,CAAC;QACF,IAAI,YAAY,EAAE,CAAC;YACjB,UAAU,CAAC,SAAS,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5C,CAAC;aAAM,CAAC;YACN,UAAU,CAAC,SAAS,CAAC,EAAE,IAAI,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QACpD,CAAC;QAED,mBAAmB;QACnB,mBAAmB,CAAC;YAClB,MAAM;YACN,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE;SACxE,CAAC,CAAC;QAEH,4CAA4C;QAC5C,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC,CACF,CAAC;AACJ,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"getSerializedObjectSafe.d.ts","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;
|
|
1
|
+
{"version":3,"file":"getSerializedObjectSafe.d.ts","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAIH,OAAO,EAAE,SAAS,EAAsB,MAAM,+BAA+B,CAAC;AAyE9E,wBAAgB,uBAAuB,CACrC,UAAU,EAAE,OAAO,EACnB,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,OAoOzB"}
|
|
@@ -14,13 +14,13 @@
|
|
|
14
14
|
* limitations under the License.
|
|
15
15
|
*/
|
|
16
16
|
import { Serializable } from "beeai-framework/internals/serializable";
|
|
17
|
-
import { instrumentationLogger } from "beeai-framework/instrumentation/logger";
|
|
18
17
|
import { getProp } from "beeai-framework/internals/helpers/object";
|
|
19
18
|
import { ChatModel } from "beeai-framework/backend/chat";
|
|
20
19
|
import { BaseAgent } from "beeai-framework/agents/base";
|
|
21
20
|
import { errorEventName, errorLLMEventName, errorToolEventName, finishLLMEventName, finishToolEventName, newTokenLLMEventName, partialUpdateEventName, retryEventName, retryToolEventName, startEventName, startLLMEventName, startToolEventName, successEventName, successLLMEventName, successToolEventName, toolErrorEventName, toolStartEventName, toolSuccessEventName, updateEventName, } from "../config.js";
|
|
22
21
|
import { LLMAttributePostfixes, MessageAttributePostfixes, MimeType, OpenInferenceSpanKind, SemanticAttributePrefixes, SemanticConventions, } from "@arizeai/openinference-semantic-conventions";
|
|
23
22
|
import { Tool } from "beeai-framework/tools/base";
|
|
23
|
+
import { diag } from "@opentelemetry/api";
|
|
24
24
|
function parserLLMInputMessages(messages) {
|
|
25
25
|
return messages.reduce((acc, item, key) => ({
|
|
26
26
|
...acc,
|
|
@@ -208,7 +208,7 @@ export function getSerializedObjectSafe(dataObject, meta) {
|
|
|
208
208
|
return { data: JSON.stringify(data.createSnapshot()), test: "hallo" };
|
|
209
209
|
}
|
|
210
210
|
catch (e) {
|
|
211
|
-
|
|
211
|
+
diag.warn("Invalid createSnapshot method in the Serializable class", e);
|
|
212
212
|
return null;
|
|
213
213
|
}
|
|
214
214
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"getSerializedObjectSafe.js","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AACtE,OAAO,EAAE,
|
|
1
|
+
{"version":3,"file":"getSerializedObjectSafe.js","sourceRoot":"","sources":["../../../src/helpers/getSerializedObjectSafe.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,wCAAwC,CAAC;AACtE,OAAO,EAAE,OAAO,EAAE,MAAM,0CAA0C,CAAC;AAGnE,OAAO,EAAE,SAAS,EAAmB,MAAM,8BAA8B,CAAC;AAC1E,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AAExD,OAAO,EACL,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,kBAAkB,EAClB,mBAAmB,EACnB,oBAAoB,EACpB,sBAAsB,EACtB,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,mBAAmB,EACnB,oBAAoB,EACpB,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,eAAe,GAChB,MAAM,WAAW,CAAC;AACnB,OAAO,EACL,qBAAqB,EACrB,yBAAyB,EACzB,QAAQ,EACR,qBAAqB,EACrB,yBAAyB,EACzB,mBAAmB,GACpB,MAAM,6CAA6C,CAAC;AACrD,OAAO,EAAE,IAAI,EAAc,MAAM,4BAA4B,CAAC;AAC9D,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAE1C,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnB,GAAG,GAAG;QACN,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,IAAI,EAAE,CAAC,EACxJ,IAAI,CAAC,IAAI;QACX,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,cAAc,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC,EAC3J,IAAI,CAAC,OAAO;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,EAAE,CAAC;KACd,CAAC,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,SAAS,sBAAsB,CAC7B,QAAwD;IAExD,OAAO,QAAQ,CAAC,MAAM,CACpB,CAAC,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;QACnB,GAAG,GAAG;QACN,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,IAAI,EAAE,CAAC,EACzJ,IAAI,CAAC,IAAI;QACX,CAAC,GAAG,yBAAyB,CAAC,GAAG,IAAI,qBAAqB,CAAC,eAAe,IAAI,GAAG,IAAI,yBAAyB,CAAC,OAAO,IAAI,yBAAyB,CAAC,OAAO,EAAE,CAAC,EAC5J,IAAI,CAAC,OAAO;aACT,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;aAClB,IAAI,CAAC,EAAE,CAAC;KACd,CAAC,EACF,EAAE,CACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,UAAmB,EACnB,IAAwB;IAExB,eAAe;IACf,IACE,CAAC,cAAc,EAAE,gBAAgB,EAAE,cAAc,EAAE,cAAc,CAAC,CAAC,QAAQ,CACzE,IAAI,CAAC,IAAiC,CACvC;QACD,IAAI,CAAC,OAAO,YAAY,SAAS,EACjC,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,GACxC,UAKC,CAAC;QACJ,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,KAAK;YAC7B,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,GAAG,CAAC,KAAK,EAAE,MAAM,GAAG,CAAC,IAAI;gBACvB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBACpD,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI;oBAC1C,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,WAAW;oBACxD,cAAc,EAAE,IAAI,CAAC,OAAO;iBAC7B,CAAC,CAAC;aACJ,CAAC;YACF,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACpD,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,QAAQ,CAAC;aACnE,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;YACF,GAAG,CAAC,IAAI,IAAI;gBACV,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;aACzD,CAAC;SACH,CAAC;IACJ,CAAC;IAED,gBAAgB;IAChB,IACE,CAAC,eAAe,EAAE,sBAAsB,CAAC,CAAC,QAAQ,CAChD,IAAI,CAAC,IAAiC,CACvC,EACD,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAEhB,CAAC;QAEF,MAAM,MAAM,GAAG,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,WAAW,CAAC;QACrD,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,KAAK;YAC7B,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,GAAG,CAAC,IAAI,CAAC,SAAS,IAAI;gBACpB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,SAAS;aAChD,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI;gBACrB,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,CAAC;aACvE,CAAC;YACF,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,MAAM;aAC3C,CAAC;SACH,CAAC;IACJ,CAAC;IAED,2BAA2B;IAC3B,IACE,CAAC,kBAAkB,EAAE,kBAAkB,EAAE,oBAAoB,CAAC,CAAC,QAAQ,CACrE,IAAI,CAAC,IAAiC,CACvC,EACD,CAAC;QACD,MAAM,EAAE,IAAI,EAAE,GAAG,UAIhB,CAAC;QACF,8DAA8D;QAC9D,MAAM,MAAM,GAAQ,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;QAEhE,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,IAAI;YACzE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO;YAC/B,GAAG,CAAC,IAAI,CAAC,KAAK;gBACZ,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE;gBACvE,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,WAAW,IAAI;gBAC3B,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW;aAC9D,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,IAAI;gBACpB,CAAC,mBAAmB,CAAC,SAAS,CAAC,EAAE,IAAI,CAAC,IAAI,CAAC,IAAI;aAChD,CAAC;YACF,GAAG,CAAC,IAAI,CAAC,KAAK,IAAI;gBAChB,mBAAmB,EAAE,IAAI,CAAC,KAAK,CAAC,OAAO;gBACvC,sBAAsB,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;gBACxC,gBAAgB,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI;aAClC,CAAC;YACF,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC;SACH,CAAC;IACJ,CAAC;IACD,qBAAqB;IACrB,IACE;QACE,kBAAkB;QAClB,oBAAoB;QACpB,mBAAmB;QACnB,kBAAkB;QAClB,kBAAkB;KACnB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAwB,CAAC;QACzC,IAAI,CAAC,OAAO,YAAY,IAAI,EAC5B,CAAC;QACD,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,OAAO;gBACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAC3C,qBAAqB,CAAC,IAAI;aAC7B,CAAC;QACJ,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,UAKhC,CAAC;QAEF,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,IAAI;YACzE,GAAG,CAAC,KAAK;gBACP,CAAC,CAAC,EAAE,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,EAAE;gBAClE,CAAC,CAAC,EAAE,CAAC;YACP,GAAG,CAAC,MAAM,IAAI;gBACZ,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;aAC3D,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;SACH,CAAC;IACJ,CAAC;IAED,aAAa;IACb,IACE;QACE,mBAAmB;QACnB,iBAAiB;QACjB,iBAAiB;QACjB,oBAAoB;KACrB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAA6B,CAAC;QAC9C,IAAI,CAAC,OAAO,YAAY,SAAS,EACjC,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,UAK/B,CAAC;QAEF,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC;QAC9C,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,GAAG;YACxE,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,gBAAgB,IAAI;gBACpC,CAAC,mBAAmB,CAAC,0BAA0B,CAAC,EAC9C,KAAK,EAAE,KAAK,EAAE,gBAAgB;aACjC,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,YAAY,IAAI;gBAChC,CAAC,mBAAmB,CAAC,sBAAsB,CAAC,EAC1C,KAAK,EAAE,KAAK,EAAE,YAAY;aAC7B,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,IAAI;gBAC/B,CAAC,mBAAmB,CAAC,qBAAqB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,WAAW;aACvE,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,eAAe,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACpD,CAAC,mBAAmB,CAAC,WAAW,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACjE,GAAG,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC;aAC1C,CAAC;YACF,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI;gBAChC,CAAC,mBAAmB,CAAC,gBAAgB,CAAC,EAAE,QAAQ,CAAC,IAAI;gBACrD,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,CAAC;gBAClE,GAAG,sBAAsB,CAAC,KAAK,CAAC,QAAQ,CAAC;aAC1C,CAAC;YACF,GAAG,CAAC,KAAK,IAAI;gBACX,mBAAmB,EAAE,KAAK,CAAC,OAAO;gBAClC,sBAAsB,EAAE,KAAK,CAAC,KAAK;gBACnC,gBAAgB,EAAE,KAAK,CAAC,IAAI;aAC7B,CAAC;YACF,GAAG,CAAC,YAAY,IAAI,OAAO,IAAI;gBAC7B,CAAC,mBAAmB,CAAC,YAAY,CAAC,EAAE,OAAO,CAAC,UAAU;gBACtD,CAAC,GAAG,yBAAyB,CAAC,QAAQ,IAAI,qBAAqB,CAAC,QAAQ,EAAE,CAAC,EACzE,OAAO,CAAC,UAAU;aACrB,CAAC;YACF,GAAG,CAAC,SAAS,IAAI,OAAO,IAAI;gBAC1B,CAAC,mBAAmB,CAAC,cAAc,CAAC,EAAE,OAAO,CAAC,OAAO;gBACrD,CAAC,GAAG,yBAAyB,CAAC,QAAQ,IAAI,qBAAqB,CAAC,UAAU,EAAE,CAAC,EAC3E,OAAO,CAAC,OAAO;aAClB,CAAC;YACF,GAAG,CAAC,OAAO,CAAC,UAAU,IAAI;gBACxB,CAAC,mBAAmB,CAAC,yBAAyB,CAAC,EAAE,IAAI,CAAC,SAAS,CAC7D,OAAO,CAAC,UAAU,CACnB;aACF,CAAC;SACH,CAAC;IACJ,CAAC;IACD,IAAI,IAAI,CAAC,IAAI,KAAK,kBAAkB,IAAI,IAAI,CAAC,OAAO,YAAY,SAAS,EAAE,CAAC;QAC1E,OAAO;YACL,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,qBAAqB,CAAC,GAAG;SACzE,CAAC;IACJ,CAAC;IAED,eAAe;IACf,MAAM,IAAI,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,UAAU,CAAC,CAAC;IACvD,IAAI,IAAI,YAAY,YAAY,EAAE,CAAC;QACjC,IAAI,CAAC;YACH,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;QACxE,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,yDAAyD,EAAE,CAAC,CAAC,CAAC;YACxE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;AACxC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,EACnC,qBAAqB,EACtB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,KAAK,GAAG,MAAM,iBAAiB,CAAC;AAEvC,OAAO,EAAY,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;
|
|
1
|
+
{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,EACnC,qBAAqB,EACtB,MAAM,gCAAgC,CAAC;AAGxC,OAAO,KAAK,GAAG,MAAM,iBAAiB,CAAC;AAEvC,OAAO,EAAY,kBAAkB,EAAE,MAAM,6BAA6B,CAAC;AAY3E;;GAEG;AACH,wBAAgB,SAAS,YAExB;AAED,qBAAa,oBAAqB,SAAQ,mBAAmB;IAC3D,OAAO,CAAC,QAAQ,CAAW;gBAEf,EACV,qBAAqB,EACrB,WAAW,GACZ,GAAE;QACD;;;WAGG;QACH,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;QAC9C;;;WAGG;QACH,WAAW,CAAC,EAAE,kBAAkB,CAAC;KAC7B;IASN,SAAS,CAAC,IAAI;IAWd;;;OAGG;IACH,kBAAkB,CAAC,MAAM,EAAE,OAAO,GAAG;IAKrC,OAAO,CAAC,KAAK;IAiFb;;OAEG;IACH,OAAO,CAAC,OAAO;CAehB"}
|
|
@@ -3,6 +3,7 @@ import { diag } from "@opentelemetry/api";
|
|
|
3
3
|
import { Version } from "beeai-framework";
|
|
4
4
|
import { OITracer } from "@arizeai/openinference-core";
|
|
5
5
|
import { createTelemetryMiddleware } from "./middleware.js";
|
|
6
|
+
import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
|
|
6
7
|
const MODULE_NAME = "beeai-framework";
|
|
7
8
|
/**
|
|
8
9
|
* Flag to check if the openai module has been patched
|
|
@@ -40,16 +41,32 @@ export class BeeAIInstrumentation extends InstrumentationBase {
|
|
|
40
41
|
if (module?.openInferencePatched || _isOpenInferencePatched) {
|
|
41
42
|
return module;
|
|
42
43
|
}
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
this._wrap(module.BaseAgent.prototype, "run", (original) => {
|
|
47
|
-
return function wrappedRun(...args) {
|
|
44
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
45
|
+
const createWrappedMethod = (instrumentation, original, mainSpanKind) => {
|
|
46
|
+
return function wrappedMethod(...args) {
|
|
48
47
|
const returned = original.apply(this, args);
|
|
49
|
-
returned
|
|
48
|
+
if (returned?.middleware) {
|
|
49
|
+
returned.middleware(createTelemetryMiddleware(instrumentation.oiTracer, mainSpanKind));
|
|
50
|
+
}
|
|
50
51
|
return returned;
|
|
51
52
|
};
|
|
52
|
-
}
|
|
53
|
+
};
|
|
54
|
+
diag.info("[BeeaiInstrumentation] Patching BaseAgent.run");
|
|
55
|
+
// eslint-disable-next-line @typescript-eslint/no-this-alias
|
|
56
|
+
const instrumentation = this;
|
|
57
|
+
this._wrap(module.BaseAgent.prototype, "run", (original) => createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.AGENT));
|
|
58
|
+
// tool instrumentation support
|
|
59
|
+
if (module.Tool) {
|
|
60
|
+
this._wrap(module.Tool.prototype, "run", (original) => createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.TOOL));
|
|
61
|
+
}
|
|
62
|
+
// model instrumentation support
|
|
63
|
+
if (module.ChatModel) {
|
|
64
|
+
this._wrap(module.ChatModel.prototype, "create", (original) => createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.LLM));
|
|
65
|
+
this._wrap(module.ChatModel.prototype, "createStructure", (original) => createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.LLM));
|
|
66
|
+
}
|
|
67
|
+
if (module.EmbeddingModel) {
|
|
68
|
+
this._wrap(module.EmbeddingModel.prototype, "create", (original) => createWrappedMethod(instrumentation, original, OpenInferenceSpanKind.EMBEDDING));
|
|
69
|
+
}
|
|
53
70
|
return module;
|
|
54
71
|
}
|
|
55
72
|
/**
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,GAEpC,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAG1C,OAAO,EAAE,QAAQ,EAAsB,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;
|
|
1
|
+
{"version":3,"file":"instrumentation.js","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,mCAAmC,GAEpC,MAAM,gCAAgC,CAAC;AACxC,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAC1C,OAAO,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAG1C,OAAO,EAAE,QAAQ,EAAsB,MAAM,6BAA6B,CAAC;AAC3E,OAAO,EAAE,yBAAyB,EAAE,MAAM,cAAc,CAAC;AACzD,OAAO,EAAE,qBAAqB,EAAE,MAAM,6CAA6C,CAAC;AAEpF,MAAM,WAAW,GAAG,iBAAiB,CAAC;AAEtC;;;GAGG;AACH,IAAI,uBAAuB,GAAG,KAAK,CAAC;AAEpC;;GAEG;AACH,MAAM,UAAU,SAAS;IACvB,OAAO,uBAAuB,CAAC;AACjC,CAAC;AAED,MAAM,OAAO,oBAAqB,SAAQ,mBAAmB;IAG3D,YAAY,EACV,qBAAqB,EACrB,WAAW,MAYT,EAAE;QACJ,KAAK,CACH,8CAA8C,EAC9C,OAAO,EACP,MAAM,CAAC,MAAM,CAAC,EAAE,EAAE,qBAAqB,CAAC,CACzC,CAAC;QACF,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,WAAW,EAAE,CAAC,CAAC;IACrE,CAAC;IAES,IAAI;QACZ,MAAM,MAAM,GAAG,IAAI,mCAAmC,CACpD,WAAW,EACX,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EACrB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CACxB,CAAC;QAEF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;OAGG;IACH,kBAAkB,CAAC,MAAkB;QACnC,IAAI,CAAC,KAAK,CAAC,iDAAiD,WAAW,EAAE,CAAC,CAAC;QAC3E,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,MAAuD;QACnE,IAAI,CAAC,MAAM,EAAE,SAAS,EAAE,CAAC;YACvB,IAAI,CAAC,IAAI,CACP,uEAAuE,CACxE,CAAC;YACF,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,IAAI,MAAM,EAAE,oBAAoB,IAAI,uBAAuB,EAAE,CAAC;YAC5D,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,8DAA8D;QAC9D,MAAM,mBAAmB,GAAG,CAC1B,eAAqC,EACrC,QAAW,EACX,YAAmC,EACnC,EAAE;YACF,OAAO,SAAS,aAAa,CAAgB,GAAG,IAAe;gBAC7D,MAAM,QAAQ,GAAG,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC5C,IAAI,QAAQ,EAAE,UAAU,EAAE,CAAC;oBACzB,QAAQ,CAAC,UAAU,CACjB,yBAAyB,CAAC,eAAe,CAAC,QAAQ,EAAE,YAAY,CAAC,CAClE,CAAC;gBACJ,CAAC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC,CAAC;QACJ,CAAC,CAAC;QACF,IAAI,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAE3D,4DAA4D;QAC5D,MAAM,eAAe,GAAyB,IAAI,CAAC;QAEnD,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,QAAQ,EAAE,EAAE,CACzD,mBAAmB,CACjB,eAAe,EACf,QAAQ,EACR,qBAAqB,CAAC,KAAK,CAC5B,CACF,CAAC;QAEF,+BAA+B;QAC/B,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAChB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC,QAAQ,EAAE,EAAE,CACpD,mBAAmB,CACjB,eAAe,EACf,QAAQ,EACR,qBAAqB,CAAC,IAAI,CAC3B,CACF,CAAC;QACJ,CAAC;QACD,gCAAgC;QAChC,IAAI,MAAM,CAAC,SAAS,EAAE,CAAC;YACrB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CAC5D,mBAAmB,CACjB,eAAe,EACf,QAAQ,EACR,qBAAqB,CAAC,GAAG,CAC1B,CACF,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,SAAS,EAAE,iBAAiB,EAAE,CAAC,QAAQ,EAAE,EAAE,CACrE,mBAAmB,CACjB,eAAe,EACf,QAAQ,EACR,qBAAqB,CAAC,GAAG,CAC1B,CACF,CAAC;QACJ,CAAC;QACD,IAAI,MAAM,CAAC,cAAc,EAAE,CAAC;YAC1B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,CAAC,SAAS,EAAE,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CACjE,mBAAmB,CACjB,eAAe,EACf,QAAQ,EACR,qBAAqB,CAAC,SAAS,CAChC,CACF,CAAC;QACJ,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,OAAO,CACb,aAA8D,EAC9D,aAAsB;QAEtB,IAAI,CAAC,KAAK,CAAC,sBAAsB,WAAW,IAAI,aAAa,EAAE,CAAC,CAAC;QACjE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,SAAS,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;QAEvD,uBAAuB,GAAG,KAAK,CAAC;QAChC,IAAI,CAAC;YACH,2EAA2E;YAC3E,aAAa,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAC7C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,IAAI,CAAC,IAAI,CAAC,mBAAmB,WAAW,6BAA6B,EAAE,CAAC,CAAC,CAAC;QAC5E,CAAC;IACH,CAAC;CACF"}
|
package/dist/esm/middleware.d.ts
CHANGED
|
@@ -15,6 +15,7 @@
|
|
|
15
15
|
*/
|
|
16
16
|
import { OITracer } from "@arizeai/openinference-core";
|
|
17
17
|
import type { GetRunContext, RunInstance } from "beeai-framework/context";
|
|
18
|
+
import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
|
|
18
19
|
export declare const activeTracesMap: Map<string, string>;
|
|
19
20
|
/**
|
|
20
21
|
* First we collect, filter and transform events from the beeai emitter to our internal framework spans in the `emitter.match("*.*"` block
|
|
@@ -23,5 +24,5 @@ export declare const activeTracesMap: Map<string, string>;
|
|
|
23
24
|
* Then we create the open telemetry spans when all data are collected. We are retroactively deleting some unnecessary internal spans
|
|
24
25
|
* see "emitter.match((event) => event.path === `${basePath}.run.${finishEventName}`" section
|
|
25
26
|
*/
|
|
26
|
-
export declare function createTelemetryMiddleware(tracer: OITracer): (context: GetRunContext<RunInstance, unknown>) => void;
|
|
27
|
+
export declare function createTelemetryMiddleware(tracer: OITracer, mainSpanKind: OpenInferenceSpanKind): (context: GetRunContext<RunInstance, unknown>) => void;
|
|
27
28
|
//# sourceMappingURL=middleware.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAWvD,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;
|
|
1
|
+
{"version":3,"file":"middleware.d.ts","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAWvD,OAAO,KAAK,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAgB1E,OAAO,EACL,qBAAqB,EAEtB,MAAM,6CAA6C,CAAC;AAErD,eAAO,MAAM,eAAe,qBAA4B,CAAC;AAEzD;;;;;;GAMG;AACH,wBAAgB,yBAAyB,CACvC,MAAM,EAAE,QAAQ,EAChB,YAAY,EAAE,qBAAqB,aAElB,aAAa,CAAC,WAAW,EAAE,OAAO,CAAC,UAuQrD"}
|
package/dist/esm/middleware.js
CHANGED
|
@@ -27,7 +27,7 @@ import { createFullPath } from "beeai-framework/emitter/utils";
|
|
|
27
27
|
import { BaseAgent } from "beeai-framework/agents/base";
|
|
28
28
|
import { diag } from "@opentelemetry/api";
|
|
29
29
|
import { buildTraceTree } from "./helpers/buildTraceTree.js";
|
|
30
|
-
import { SemanticConventions } from "@arizeai/openinference-semantic-conventions";
|
|
30
|
+
import { SemanticConventions, } from "@arizeai/openinference-semantic-conventions";
|
|
31
31
|
export const activeTracesMap = new Map();
|
|
32
32
|
/**
|
|
33
33
|
* First we collect, filter and transform events from the beeai emitter to our internal framework spans in the `emitter.match("*.*"` block
|
|
@@ -36,7 +36,7 @@ export const activeTracesMap = new Map();
|
|
|
36
36
|
* Then we create the open telemetry spans when all data are collected. We are retroactively deleting some unnecessary internal spans
|
|
37
37
|
* see "emitter.match((event) => event.path === `${basePath}.run.${finishEventName}`" section
|
|
38
38
|
*/
|
|
39
|
-
export function createTelemetryMiddleware(tracer) {
|
|
39
|
+
export function createTelemetryMiddleware(tracer, mainSpanKind) {
|
|
40
40
|
return (context) => {
|
|
41
41
|
if (!context.emitter?.trace?.id) {
|
|
42
42
|
throw new FrameworkError(`Fatal error. Missing traceId`, [], { context });
|
|
@@ -111,6 +111,7 @@ export function createTelemetryMiddleware(tracer) {
|
|
|
111
111
|
// create tracer spans from collected data
|
|
112
112
|
buildTraceTree({
|
|
113
113
|
tracer,
|
|
114
|
+
mainSpanKind,
|
|
114
115
|
data: {
|
|
115
116
|
prompt: prompt,
|
|
116
117
|
history,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC;AAG3C,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,IAAI,EAAE,MAAM,iCAAiC,CAAC;AAEvD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,4BAA4B,EAC5B,oBAAoB,EACpB,sBAAsB,EACtB,mBAAmB,GACpB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAE1C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,
|
|
1
|
+
{"version":3,"file":"middleware.js","sourceRoot":"","sources":["../../src/middleware.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAGH,OAAO,EAAE,uBAAuB,EAAE,MAAM,mCAAmC,CAAC;AAC5E,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AACtD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,QAAQ,CAAC;AAG3C,OAAO,EAAE,cAAc,EAAE,MAAM,wBAAwB,CAAC;AACxD,OAAO,EAAE,OAAO,EAAE,MAAM,yBAAyB,CAAC;AAClD,OAAO,EAAE,IAAI,EAAE,MAAM,iCAAiC,CAAC;AAEvD,OAAO,EAAE,eAAe,EAAE,MAAM,2BAA2B,CAAC;AAC5D,OAAO,EACL,iBAAiB,EACjB,kBAAkB,EAClB,4BAA4B,EAC5B,oBAAoB,EACpB,sBAAsB,EACtB,mBAAmB,GACpB,MAAM,UAAU,CAAC;AAClB,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAE/D,OAAO,EAAE,SAAS,EAAE,MAAM,6BAA6B,CAAC;AACxD,OAAO,EAAE,IAAI,EAAE,MAAM,oBAAoB,CAAC;AAE1C,OAAO,EAAE,cAAc,EAAE,MAAM,0BAA0B,CAAC;AAC1D,OAAO,EAEL,mBAAmB,GACpB,MAAM,6CAA6C,CAAC;AAErD,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,GAAG,EAAkB,CAAC;AAEzD;;;;;;GAMG;AACH,MAAM,UAAU,yBAAyB,CACvC,MAAgB,EAChB,YAAmC;IAEnC,OAAO,CAAC,OAA4C,EAAE,EAAE;QACtD,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;YAChC,MAAM,IAAI,cAAc,CAAC,8BAA8B,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,CAAC;QAC3C,IAAI,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YACjC,OAAO;QACT,CAAC;QACD,eAAe,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;QAEhE,IAAI,CAAC,KAAK,CAAC,2BAA2B,EAAE;YACtC,MAAM,EAAE,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI;YACzC,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC;QACjD,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAEvD,IAAI,MAAM,GAA8B,IAAI,CAAC;QAC7C,IAAI,QAAQ,YAAY,SAAS,EAAE,CAAC;YAClC,MAAM,GAAI,SAA2C,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC;QAClE,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAyB,CAAC;QAClD,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC/C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAqB,CAAC;QAEtD,IAAI,gBAAgB,GAAkC,SAAS,CAAC;QAChE,IAAI,OAAO,GAAoC,SAAS,CAAC;QACzD,MAAM,eAAe,GAAa,EAAE,CAAC;QAErC,MAAM,aAAa,GAAG,IAAI,aAAa,EAAE,CAAC;QAE1C,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA+B,CAAC;QAEnE,MAAM,aAAa,GAAG,IAAI,IAAI,EAAE,CAAC,OAAO,EAAE,CAAC;QAC3C,MAAM,aAAa,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACxC,SAAS,wBAAwB,CAAC,IAAU;YAC1C,OAAO,IAAI,CAAC,OAAO,EAAE,GAAG,aAAa,GAAG,aAAa,CAAC;QACxD,CAAC;QAED,MAAM,UAAU,GAAG,eAAe,CAAC;YACjC,YAAY,EAAE,4BAA4B;SAC3C,CAAC,CAAC;QAEH,SAAS,gBAAgB,CAAC,EAAE,MAAM,EAAsB;YACtD,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,SAAS,CAAC;YACjD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,YAAY,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YAC7C,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,OAAO;YACT,CAAC;YAED,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC;gBAClB,2CAA2C;gBAC3C,YAAY,CAAC,GAAG,CAAC,QAAQ,EAAE,SAAS,GAAG,CAAC,CAAC,CAAC;YAC5C,CAAC;iBAAM,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;gBAC3B,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBAC9B,oFAAoF;gBACpF,IAAI,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;oBACnC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;oBAC1B,gBAAgB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACpC,CAAC;YACH,CAAC;QACH,CAAC;QAED;;WAEG;QACH,OAAO,CAAC,KAAK,CACX,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,QAAQ,QAAQ,kBAAkB,EAAE,EACjE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE;YAChB,IAAI,CAAC;gBACH,IAAI,CAAC,KAAK,CAAC,kBAAkB,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBAEtE,IAAI,CAAC,MAAM,IAAI,QAAQ,YAAY,SAAS,EAAE,CAAC;oBAC7C,MAAM,GAAG,QAAQ,CACf,QAAQ,CAAC,MAAM,CAAC,QAAQ,EACxB,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,KAAK,IAAI,CAAC,IAAI,CACxC,EAAE,IAAI,CAAC;oBAER,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,MAAM,IAAI,cAAc,CACtB,gDAAgD,EAChD,EAAE,EACF;4BACE,OAAO;yBACR,CACF,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,0CAA0C;gBAC1C,cAAc,CAAC;oBACb,MAAM;oBACN,YAAY;oBACZ,IAAI,EAAE;wBACJ,MAAM,EAAE,MAAM;wBACd,OAAO;wBACP,gBAAgB;wBAChB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;wBAC5D,OAAO;wBACP,OAAO,EAAE,OAAO;wBAChB,eAAe,EAAE,GAAG,QAAQ,QAAQ,iBAAiB,EAAE;wBACvD,SAAS,EAAE,aAAa;wBACxB,OAAO,EAAE,WAAW,CAAC,GAAG,EAAE;wBAC1B,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,OAAO,CAAE;qBACtC;iBACF,CAAC,CAAC;YACL,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,IAAI,CAAC,IAAI,CAAC,uBAAuB,EAAE,CAAC,CAAC,CAAC;YACxC,CAAC;oBAAS,CAAC;gBACT,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CACF,CAAC;QAEF;;;;;WAKG;QACH,OAAO,CAAC,KAAK,CAAC,KAAK,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE;YAClC,+DAA+D;YAC/D,IACE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;gBAC3B,IAAI,CAAC,IAAI,KAAK,GAAG,QAAQ,QAAQ,iBAAiB,EAAE,EACpD,CAAC;gBACD,OAAO;YACT,CAAC;YACD,4BAA4B;YAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,EAAE,CAAC;gBACvB,MAAM,IAAI,cAAc,CACtB,yCAAyC,IAAI,CAAC,IAAI,EAAE,EACpD,EAAE,EACF;oBACE,OAAO;iBACR,CACF,CAAC;YACJ,CAAC;YAED;;;eAGG;YACH,IACE,IAAI,CAAC,OAAO;gBACZ,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW;gBACvB,CAAC,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EACvC,CAAC;gBACD,QAAQ,CAAC,GAAG,CACV,IAAI,CAAC,OAAO,EACZ,UAAU,CAAC;oBACT,EAAE,EAAE,IAAI,CAAC,OAAO;oBAChB,IAAI,EAAE,IAAI,CAAC,OAAO;oBAClB,MAAM,EAAE,SAAS;oBACjB,IAAI,EAAE;wBACJ,CAAC,mBAAmB,CAAC,uBAAuB,CAAC,EAAE,OAAO;qBACvD;oBACD,SAAS,EAAE,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC;iBACpD,CAAC,CACH,CAAC;gBACF,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;YAED,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,aAAa,CAAC,MAAM,CAAC;gBACpD,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;gBACvB,WAAW,EAAE,IAAI,CAAC,KAAK,CAAC,WAAW;gBACnC,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,uBAAuB,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YAE3D,yCAAyC;YACzC,IAAI,IAAI,CAAC,IAAI,KAAK,sBAAsB,IAAI,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;gBACpE,OAAO;YACT,CAAC;YAED,MAAM,IAAI,GAAG,UAAU,CAAC;gBACtB,EAAE,EAAE,MAAM;gBACV,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,MAAM,EAAE,IAAI,CAAC,IAAI;gBACjB,GAAG,CAAC,YAAY,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,CAAC;gBACrD,GAAG,EAAE,IAAI,CAAC,OAAO;gBACjB,IAAI,EAAE,cAAc;gBACpB,KAAK,EAAE,YAAY,CAAC,IAAI,CAAC;gBACzB,SAAS,EAAE,wBAAwB,CAAC,IAAI,CAAC,SAAS,CAAC;aACpD,CAAC,CAAC;YAEH,MAAM,aAAa,GAAG,eAAe,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;YAElE,iEAAiE;YACjE,MAAM,wBAAwB,GAAG,mBAAmB;iBACjD,GAAG,CAAC,aAAa,CAAC;gBACnB,EAAE,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnB,IACE,wBAAwB;gBACxB,sBAAsB,KAAK,IAAI,CAAC,IAAI;gBACpC,QAAQ,CAAC,GAAG,CAAC,wBAAwB,CAAC,EACtC,CAAC;gBACD,MAAM,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,GAAG,CAAC,wBAAwB,CAAE,CAAC;gBAC5D,IAAI,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;oBACtC,gBAAgB,CAAC,GAAG,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;gBAC5D,CAAC;qBAAM,CAAC;oBACN,cAAc;oBACd,gBAAgB,CAAC,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAC,CAAC;oBACvD,QAAQ,CAAC,MAAM,CAAC,wBAAwB,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,kBAAkB;YAClB,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;YACzC,wDAAwD;YACxD,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,YAAY,CAAC,GAAG,CACd,IAAI,CAAC,SAAS,EACd,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAC5C,CAAC;YACJ,CAAC;YAED,yCAAyC;YACzC,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC/B,IAAI,mBAAmB,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;oBAC3C,mBAAmB;yBAChB,GAAG,CAAC,aAAa,CAAE;yBACnB,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAC1C,CAAC;qBAAM,CAAC;oBACN,mBAAmB,CAAC,GAAG,CACrB,aAAa,EACb,IAAI,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAC/C,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,4FAA4F;QAC5F,OAAO,CAAC,KAAK,CACX,CAAC,KAAK,EAAE,EAAE,CACR,KAAK,CAAC,IAAI,KAAK,mBAAmB;YAClC,KAAK,CAAC,OAAO,YAAY,SAAS,EACpC,CAAC,IAAwD,EAAE,EAAE;YAC3D,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;YAE1C,gBAAgB,GAAG;gBACjB,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,IAAI,EAAE,UAAU,CAAC,IAAI;aACtB,CAAC;YACF,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACtC,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,IAAI,EAAE,GAAG,CAAC,IAAI;aACf,CAAC,CAAC,CAAC;QACN,CAAC,CACF,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC"}
|