test(server-utils): Cover Flue message content recording

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

Author: RulaKhaled

packages/server-utils/test/ai/lib/tracing/flue.test.ts

@@ -166,6 +166,65 @@ describe('createFlueInstrumentation', () => {
     expect(json?.data['gen_ai.cost.total_tokens']).toBeUndefined();
   });
 
+  describe('content recording', () => {
+    const requestContent = {
+      type: 'turn_request',
+      turnId: 'turn_1',
+      request: {
+        requestedModel: 'claude-haiku-4.5',
+        input: {
+          systemPrompt: 'You are helpful.',
+          messages: [{ role: 'user', content: 'hi' }],
+          tools: [{ name: 'get_weather', description: 'weather', parameters: {} }],
+        },
+      },
+    } satisfies FlueObservation;
+
+    async function record(instr: FlueInstrumentation): Promise<void> {
+      await instr.interceptor({ type: 'agent' }, AGENT_CTX, async () => {
+        instr.observe({ type: 'turn_start', turnId: 'turn_1' }, {});
+        instr.observe(requestContent, {});
+        instr.observe(turn({ response: { ...turn().response, output: { role: 'assistant' } } }), {});
+        instr.observe({ type: 'tool_start', toolCallId: 'c1', toolName: 'get_weather', args: { city: 'Berlin' } }, {});
+        instr.observe({ type: 'tool', toolCallId: 'c1', toolName: 'get_weather', result: 'sunny' }, {});
+      });
+    }
+
+    it('records messages, instructions, tool definitions, arguments and results by default', async () => {
+      await record(instrumentation);
+
+      const chat = findSpan('chat claude-haiku-4.5');
+      expect(chat?.data['gen_ai.system_instructions']).toBe('You are helpful.');
+      expect(chat?.data['gen_ai.input.messages']).toContain('"role":"user"');
+      expect(chat?.data['gen_ai.output.messages']).toContain('"role":"assistant"');
+      expect(chat?.data['gen_ai.tool.definitions']).toContain('get_weather');
+
+      const tool = findSpan('execute_tool get_weather');
+      expect(tool?.data['gen_ai.tool.call.arguments']).toBe('{"city":"Berlin"}');
+      expect(tool?.data['gen_ai.tool.call.result']).toBe('sunny');
+    });
+
+    it('omits inputs when recordInputs is false but keeps outputs', async () => {
+      await record(createFlueInstrumentation({ recordInputs: false }));
+
+      const chat = findSpan('chat claude-haiku-4.5');
+      expect(chat?.data['gen_ai.input.messages']).toBeUndefined();
+      expect(chat?.data['gen_ai.system_instructions']).toBeUndefined();
+      expect(chat?.data['gen_ai.tool.definitions']).toBeUndefined();
+      expect(chat?.data['gen_ai.output.messages']).toBeDefined();
+      expect(findSpan('execute_tool get_weather')?.data['gen_ai.tool.call.arguments']).toBeUndefined();
+    });
+
+    it('omits outputs when recordOutputs is false but keeps inputs', async () => {
+      await record(createFlueInstrumentation({ recordOutputs: false }));
+
+      const chat = findSpan('chat claude-haiku-4.5');
+      expect(chat?.data['gen_ai.output.messages']).toBeUndefined();
+      expect(chat?.data['gen_ai.input.messages']).toBeDefined();
+      expect(findSpan('execute_tool get_weather')?.data['gen_ai.tool.call.result']).toBeUndefined();
+    });
+  });
+
   it('emits execute_tool spans keyed by tool call id', async () => {
     await withAgent(() => {
       instrumentation.observe({ type: 'tool_start', toolCallId: 'call_1', toolName: 'get_weather' }, {});