weflayr 0.21.0 → 0.22.0

package/tests/instrumentation/openai/get_embeddings_telemetry.test.js

@@ -79,12 +79,31 @@ test('embeddings emits genai span with prompt tokens', async () => {
     assert.equal(span.attributes['gen_ai.request.model'], 'text-embedding-3-small');
     assert.equal(span.attributes['gen_ai.response.model'], 'text-embedding-3-small');
     assert.equal(span.attributes['gen_ai.usage.input_tokens'], 8);
+    assert.deepEqual(JSON.parse(span.attributes['gen_ai.input.messages']), [
+      { role: 'user', parts: [{ type: 'text', content: 'hello' }] },
+    ]);
     assert.equal(span.status.code, SpanStatusCode.UNSET);
   } finally {
     teardown();
   }
 });
 
+test('embeddings emits one input message per element of a list input', async () => {
+  const teardown = instrumented();
+  try {
+    const client = new StubEmbeddings();
+    await client.create({ input: ['a', 'b'], model: 'text-embedding-3-small' });
+
+    const [span] = exporter.getFinishedSpans();
+    assert.deepEqual(JSON.parse(span.attributes['gen_ai.input.messages']), [
+      { role: 'user', parts: [{ type: 'text', content: 'a' }] },
+      { role: 'user', parts: [{ type: 'text', content: 'b' }] },
+    ]);
+  } finally {
+    teardown();
+  }
+});
+
 test('embeddings records error status on failure', async () => {
   const teardown = instrumented();
   try {

package/tests/instrumentation/openai/get_streaming_usage.test.js

@@ -69,6 +69,35 @@ test('backfills cached-prompt tokens from the streamed usage chunk', async () =>
   assert.equal(span.attrs['gen_ai.usage.cache_read_input_tokens'], 1536);
 });
 
+test('re-yields every stream chunk unchanged — no telemetry collision with the payload', async () => {
+  const emitted = [
+    { choices: [{ delta: { content: 'hel' } }] },
+    { choices: [{ delta: { content: 'lo' } }] },
+    { usage: { prompt_tokens: 12, completion_tokens: 3, total_tokens: 15 } },
+  ];
+  const instrumentation = {
+    _streamingWrapPromise() {
+      return (async function* () {
+        for (const chunk of emitted) yield chunk;
+      })();
+    },
+  };
+  new OpenAIStreamingUsageInstrumentor().instrument(instrumentation);
+
+  const span = fakeSpan();
+  const collected = [];
+  for await (const chunk of instrumentation._streamingWrapPromise({ span })) {
+    collected.push(chunk);
+  }
+
+  // Same count, order, and object identity: backfillUsage re-yields each chunk by
+  // reference without copying or mutating it.
+  assert.equal(collected.length, emitted.length);
+  collected.forEach((chunk, index) => assert.equal(chunk, emitted[index]));
+  // …while still backfilling usage off the trailing chunk.
+  assert.equal(span.attrs['gen_ai.usage.output_tokens'], 3);
+});
+
 test('is a no-op when the instrumentor has no _streamingWrapPromise', () => {
   const instrumentation = {};
   new OpenAIStreamingUsageInstrumentor().instrument(instrumentation);