test update

Author: jsl517

packages/agents-a365-observability/src/tracing/middleware/BaggageBuilder.ts

@@ -234,14 +234,14 @@ export class BaggageBuilder {
    * @param pairs Dictionary or iterable of key-value pairs
    * @returns Self for method chaining
    */
-  setPairs(pairs: Record<string, any> | Iterable<[string, any]> | null | undefined): BaggageBuilder {
+  setPairs(pairs: Record<string, unknown> | Iterable<[string, unknown]> | null | undefined): BaggageBuilder {
     if (!pairs) {
       return this;
     }
 
-    let entries: Iterable<[string, any]>;
+    let entries: Iterable<[string, unknown]>;
     if (Symbol.iterator in Object(pairs)) {
-      entries = pairs as Iterable<[string, any]>;
+      entries = pairs as Iterable<[string, unknown]>;
     } else {
       entries = Object.entries(pairs);
     }

tests/observability/core/BaggageBuilder.test.ts

@@ -140,9 +140,11 @@ describe('BaggageBuilder', () => {
         .agentId('agent-456')
         .correlationId('corr-789')
         .sessionId('session-0001')
+        .sessionDescription('My session desc')
         .build();
       const bag = propagation.getBaggage((scope as any).contextWithBaggage);
       expect(bag?.getEntry(OpenTelemetryConstants.SESSION_ID_KEY)?.value).toBe('session-0001');
+      expect(bag?.getEntry(OpenTelemetryConstants.GEN_AI_SESSION_DESCRIPTION_KEY)?.value).toBe('My session desc');
     });
 
     it('should omit empty sessionId value', () => {
@@ -152,26 +154,15 @@ describe('BaggageBuilder', () => {
       const bag = propagation.getBaggage((scope as any).contextWithBaggage);
       expect(bag?.getEntry(OpenTelemetryConstants.SESSION_ID_KEY)).toBeUndefined();
     });
-  });
 
-  describe('sessionDescription support', () => {
-    it('should set sessionDescription via fluent API', () => {
-      const scope = new BaggageBuilder()
-        .sessionDescription('My session desc')
-        .build();
-      const bag = propagation.getBaggage((scope as any).contextWithBaggage);
-      expect(bag?.getEntry(OpenTelemetryConstants.GEN_AI_SESSION_DESCRIPTION_KEY)?.value).toBe('My session desc');
-    });
-
-    it('should omit null sessionDescription value', () => {
+     it('should omit null sessionDescription value', () => {
       const scope = new BaggageBuilder()
         .sessionDescription(null)
         .build();
       const bag = propagation.getBaggage((scope as any).contextWithBaggage);
       expect(bag?.getEntry(OpenTelemetryConstants.GEN_AI_SESSION_DESCRIPTION_KEY)).toBeUndefined();
     });
   });
-
 });
 
 describe('BaggageScope', () => {