Add support for gen_ai.response.id (#2831)

Author: Kludex

pydantic_ai_slim/pydantic_ai/models/instrumented.py

@@ -424,13 +424,15 @@ def _record_metrics():
                         cost_attributes = {'operation.cost': float(response.cost().total_price)}
                     except LookupError:
                         cost_attributes = {}
-                    span.set_attributes(
-                        {
-                            **response.usage.opentelemetry_attributes(),
-                            'gen_ai.response.model': response_model,
-                            **cost_attributes,
-                        }
-                    )
+
+                    attributes_to_set = {
+                        **response.usage.opentelemetry_attributes(),
+                        'gen_ai.response.model': response_model,
+                        **cost_attributes,
+                    }
+                    if response.provider_response_id is not None:
+                        attributes_to_set['gen_ai.response.id'] = response.provider_response_id
+                    span.set_attributes(attributes_to_set)
                     span.update_name(f'{operation} {request_model}')
 
                 yield finish

tests/models/test_instrumented.py

@@ -86,6 +86,7 @@ async def request(
             usage=RequestUsage(input_tokens=100, output_tokens=200),
             model_name='gpt-4o-2024-11-20',
             provider_details=dict(finish_reason='stop', foo='bar'),
+            provider_response_id='response_id',
         )
 
     @asynccontextmanager
@@ -183,6 +184,7 @@ async def test_instrumented_model(capfire: CaptureLogfire):
                     'logfire.msg': 'chat gpt-4o',
                     'logfire.span_type': 'span',
                     'gen_ai.response.model': 'gpt-4o-2024-11-20',
+                    'gen_ai.response.id': 'response_id',
                     'gen_ai.usage.input_tokens': 100,
                     'gen_ai.usage.output_tokens': 200,
                     'operation.cost': 0.00225,
@@ -719,6 +721,7 @@ async def test_instrumented_model_attributes_mode(capfire: CaptureLogfire, instr
                             'properties': {'events': {'type': 'array'}, 'model_request_parameters': {'type': 'object'}},
                         },
                         'operation.cost': 0.00225,
+                        'gen_ai.response.id': 'response_id',
                     },
                 },
             ]
@@ -820,6 +823,7 @@ async def test_instrumented_model_attributes_mode(capfire: CaptureLogfire, instr
                             },
                         },
                         'operation.cost': 0.00225,
+                        'gen_ai.response.id': 'response_id',
                     },
                 },
             ]