File tree Expand file tree Collapse file tree 3 files changed +6
-7
lines changed
instrumentation-genai/opentelemetry-instrumentation-vertexai
src/opentelemetry/instrumentation/vertexai Expand file tree Collapse file tree 3 files changed +6
-7
lines changed Original file line number Diff line number Diff line change @@ -228,7 +228,7 @@ def handle_response(
228
228
}
229
229
)
230
230
event = LogRecord (
231
- name = "gen_ai.client.inference.operation.details" ,
231
+ event_name = "gen_ai.client.inference.operation.details" ,
232
232
)
233
233
event .attributes = attributes
234
234
if capture_content in (
Original file line number Diff line number Diff line change @@ -267,8 +267,8 @@ def test_generate_content_invalid_role(
267
267
log = logs [0 ].log_record
268
268
assert log .attributes == {
269
269
"gen_ai.system" : "vertex_ai" ,
270
- "event.name" : "gen_ai.user.message" ,
271
270
}
271
+ assert log .event_name == "gen_ai.user.message"
272
272
assert log .body == {
273
273
"content" : [{"text" : "Say this is a test" }],
274
274
"role" : "invalid_role" ,
Original file line number Diff line number Diff line change @@ -265,22 +265,21 @@ def test_tool_events_no_content(
265
265
266
266
assert tool_log1 .attributes == {
267
267
"gen_ai.system" : "vertex_ai" ,
268
- "event.name" : "gen_ai.tool.message" ,
269
268
}
269
+ assert tool_log1 .event_name == "gen_ai.tool.message"
270
270
assert tool_log1 .body == {
271
271
"role" : "user" ,
272
272
"id" : "get_current_weather_0" ,
273
273
}
274
274
assert tool_log1 .event_name == "gen_ai.tool.message"
275
275
276
- assert tool_log2 .attributes == {
277
- "gen_ai.system" : "vertex_ai" ,
278
- "event.name" : "gen_ai.tool.message" ,
279
- }
276
+ assert tool_log2 .attributes == {"gen_ai.system" : "vertex_ai" }
277
+
280
278
assert tool_log2 .body == {
281
279
"role" : "user" ,
282
280
"id" : "get_current_weather_1" ,
283
281
}
282
+ assert tool_log2 .event_name == "gen_ai.tool.message"
284
283
285
284
assert choice_log .attributes == {"gen_ai.system" : "vertex_ai" }
286
285
assert choice_log .event_name == "gen_ai.choice"
You can’t perform that action at this time.
0 commit comments