@arizeai/openinference-instrumentation-beeai 1.5.2 → 1.5.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/openinference-instrumentation-beeai",
-  "version": "1.5.2",
+  "version": "1.5.4",
   "license": "Apache-2.0",
   "author": "gallas.milan@gmail.com",
   "description": "OpenInference Instrumentation for BeeAI framework",
@@ -33,14 +33,13 @@
     "@opentelemetry/instrumentation": "^0.57.1",
     "remeda": "^2.20.2",
     "semver": "^7.7.0",
-    "@arizeai/openinference-core": "1.0.6",
-    "@arizeai/openinference-semantic-conventions": "2.1.1"
+    "@arizeai/openinference-core": "1.0.8",
+    "@arizeai/openinference-semantic-conventions": "2.1.2"
   },
   "peerDependencies": {
     "beeai-framework": "^0.1.13"
   },
   "devDependencies": {
-    "@types/jest": "^29.5.12",
     "@types/node": "^20.14.11",
     "@types/semver": "7.7.0",
     "beeai-framework": "^0.1.13",
@@ -50,7 +49,7 @@
     "@opentelemetry/sdk-node": "^0.57.1",
     "@opentelemetry/sdk-trace-node": "^1.30.1",
     "@opentelemetry/semantic-conventions": "^1.28.0",
-    "jest": "^29.7.0",
+    "vitest": "^4.0.3",
     "ts-node": "^10.9.2"
   },
   "optionalDependencies": {
@@ -61,6 +60,6 @@
     "build": "tsc --build tsconfig.json tsconfig.esm.json tsconfig.esnext.json && tsc-alias -p tsconfig.esm.json",
     "postbuild": "echo '{\"type\": \"module\"}' > ./dist/esm/package.json && rimraf dist/test",
     "type:check": "tsc --noEmit",
-    "test": "jest"
+    "test": "vitest --typecheck"
   }
 }

package/src/helpers/buildTraceTree.ts

@@ -1,11 +1,13 @@
 import { OITracer } from "@arizeai/openinference-core";
-import { FrameworkSpan, GeneratedResponse } from "../types";
-import { SpanStatusCode, TimeInput } from "@opentelemetry/api";
 import {
   OpenInferenceSpanKind,
   SemanticConventions,
 } from "@arizeai/openinference-semantic-conventions";
+
+import { SpanStatusCode, TimeInput } from "@opentelemetry/api";
+
 import { startEventName, successEventName } from "../config";
+import { FrameworkSpan, GeneratedResponse } from "../types";
 
 interface BuiltTraceTreeProps {
   tracer: OITracer;

package/src/helpers/create-span.ts

@@ -15,7 +15,9 @@
  */
 
 import { SpanStatusCode, TimeInput } from "@opentelemetry/api";
+
 import { FrameworkSpan } from "../types";
+
 import { isEmpty } from "remeda";
 
 interface CreateSpanProps {

package/src/helpers/getSerializedObjectSafe.ts

@@ -14,13 +14,17 @@
  * limitations under the License.
  */
 
-import { Serializable } from "beeai-framework/internals/serializable";
-import { getProp } from "beeai-framework/internals/helpers/object";
-import { EventMeta, InferCallbackValue } from "beeai-framework/emitter/types";
-import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
-import { ChatModel, ChatModelEvents } from "beeai-framework/backend/chat";
-import { BaseAgent } from "beeai-framework/agents/base";
-import { Message, MessageContentPart } from "beeai-framework/backend/message";
+import {
+  LLMAttributePostfixes,
+  MessageAttributePostfixes,
+  MimeType,
+  OpenInferenceSpanKind,
+  SemanticAttributePrefixes,
+  SemanticConventions,
+} from "@arizeai/openinference-semantic-conventions";
+
+import { diag } from "@opentelemetry/api";
+
 import {
   errorEventName,
   errorLLMEventName,
@@ -42,16 +46,15 @@ import {
   toolSuccessEventName,
   updateEventName,
 } from "../config";
-import {
-  LLMAttributePostfixes,
-  MessageAttributePostfixes,
-  MimeType,
-  OpenInferenceSpanKind,
-  SemanticAttributePrefixes,
-  SemanticConventions,
-} from "@arizeai/openinference-semantic-conventions";
+
+import { BaseAgent } from "beeai-framework/agents/base";
+import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
+import { ChatModel, ChatModelEvents } from "beeai-framework/backend/chat";
+import { Message, MessageContentPart } from "beeai-framework/backend/message";
+import { EventMeta, InferCallbackValue } from "beeai-framework/emitter/types";
+import { getProp } from "beeai-framework/internals/helpers/object";
+import { Serializable } from "beeai-framework/internals/serializable";
 import { Tool, ToolEvents } from "beeai-framework/tools/base";
-import { diag } from "@opentelemetry/api";
 
 function parserLLMInputMessages(
   messages: readonly Message<MessageContentPart, string>[],

package/src/instrumentation.ts

@@ -1,15 +1,17 @@
+import { OITracer, TraceConfigOptions } from "@arizeai/openinference-core";
+import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
+
+import { diag, Tracer, TracerProvider } from "@opentelemetry/api";
 import {
   InstrumentationBase,
-  InstrumentationNodeModuleDefinition,
   InstrumentationConfig,
+  InstrumentationNodeModuleDefinition,
 } from "@opentelemetry/instrumentation";
-import { diag, Tracer, TracerProvider } from "@opentelemetry/api";
-import { Version } from "beeai-framework";
-import * as bee from "beeai-framework";
 
-import { OITracer, TraceConfigOptions } from "@arizeai/openinference-core";
 import { createTelemetryMiddleware } from "./middleware";
-import { OpenInferenceSpanKind } from "@arizeai/openinference-semantic-conventions";
+
+import { Version } from "beeai-framework";
+import * as bee from "beeai-framework";
 import { satisfies } from "semver";
 
 const MODULE_NAME = "beeai-framework";

package/src/middleware.ts

@@ -15,17 +15,18 @@
  */
 
 import { OITracer } from "@arizeai/openinference-core";
-import { getSerializedObjectSafe } from "./helpers/getSerializedObjectSafe";
+import {
+  OpenInferenceSpanKind,
+  SemanticConventions,
+} from "@arizeai/openinference-semantic-conventions";
+
+import { diag } from "@opentelemetry/api";
+
+import { buildTraceTree } from "./helpers/buildTraceTree";
 import { createSpan } from "./helpers/create-span";
-import { IdNameManager } from "./helpers/idNameManager";
 import { getErrorSafe } from "./helpers/getErrorSafe";
-import { findLast, isEmpty } from "remeda";
-import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
-import type { InferCallbackValue } from "beeai-framework/emitter/types";
-import { FrameworkError } from "beeai-framework/errors";
-import { Version } from "beeai-framework/version";
-import { Role } from "beeai-framework/backend/message";
-import type { GetRunContext, RunInstance } from "beeai-framework/context";
+import { getSerializedObjectSafe } from "./helpers/getSerializedObjectSafe";
+import { IdNameManager } from "./helpers/idNameManager";
 import { traceSerializer } from "./helpers/traceSerializer";
 import {
   errorLLMEventName,
@@ -35,18 +36,20 @@ import {
   partialUpdateEventName,
   successLLMEventName,
 } from "./config";
-import { createFullPath } from "beeai-framework/emitter/utils";
-import { ReActAgent } from "beeai-framework/agents/react/agent";
-import { BaseAgent } from "beeai-framework/agents/base";
-import { diag } from "@opentelemetry/api";
 import { FrameworkSpan, GeneratedResponse } from "./types";
-import { buildTraceTree } from "./helpers/buildTraceTree";
-import {
-  OpenInferenceSpanKind,
-  SemanticConventions,
-} from "@arizeai/openinference-semantic-conventions";
+
+import { BaseAgent } from "beeai-framework/agents/base";
+import { ReActAgent } from "beeai-framework/agents/react/agent";
+import type { ReActAgentCallbacks } from "beeai-framework/agents/react/types";
 import { ToolCallingAgent } from "beeai-framework/agents/toolCalling/agent";
 import type { ToolCallingAgentCallbacks } from "beeai-framework/agents/toolCalling/types";
+import { Role } from "beeai-framework/backend/message";
+import type { GetRunContext, RunInstance } from "beeai-framework/context";
+import type { InferCallbackValue } from "beeai-framework/emitter/types";
+import { createFullPath } from "beeai-framework/emitter/utils";
+import { FrameworkError } from "beeai-framework/errors";
+import { Version } from "beeai-framework/version";
+import { findLast, isEmpty } from "remeda";
 
 export const activeTracesMap = new Map<string, string>();