Skip to content

Commit d9fb1ec

Browse files
committed
fix ai-sdk test
1 parent 999b323 commit d9fb1ec

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

packages/traceloop-sdk/src/lib/tracing/span-processor.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ export const traceloopInstrumentationLibraries = [
127127
"@traceloop/instrumentation-pinecone",
128128
"@traceloop/instrumentation-qdrant",
129129
"@traceloop/instrumentation-together",
130+
"@traceloop/instrumentation-mcp",
130131
];
131132

132133
/**

packages/traceloop-sdk/test/decorators.test.ts

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -644,10 +644,13 @@ describe("Test SDK Decorators", () => {
644644
experimental_telemetry: { isEnabled: true },
645645
});
646646

647+
// Force flush to ensure all spans are exported
648+
await traceloop.forceFlush();
649+
647650
const spans = memoryExporter.getFinishedSpans();
648651

649-
const generateTextSpan = spans.find(
650-
(span) => span.name === "text.generate",
652+
const generateTextSpan = spans.find((span) =>
653+
span.name.startsWith("text.generate"),
651654
);
652655

653656
assert.ok(result);

0 commit comments

Comments
 (0)