@@ -324,9 +324,7 @@ public void testTraceToolCall() {
324324 assertEquals ("tool-name" , attrs .get (AttributeKey .stringKey ("gen_ai.tool.name" )));
325325 assertEquals ("tool-description" , attrs .get (AttributeKey .stringKey ("gen_ai.tool.description" )));
326326 assertEquals ("tool-type" , attrs .get (AttributeKey .stringKey ("gen_ai.tool.type" )));
327- assertEquals (
328- "{\" arg1\" :\" value1\" }" ,
329- attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.tool_call_args" )));
327+ assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.tool_call_args" )));
330328 assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_request" )));
331329 assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_response" )));
332330 }
@@ -371,9 +369,7 @@ public void testTraceToolResponse() {
371369 assertEquals ("tool-description" , attrs .get (AttributeKey .stringKey ("gen_ai.tool.description" )));
372370 assertEquals ("tool-type" , attrs .get (AttributeKey .stringKey ("gen_ai.tool.type" )));
373371 assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.tool_call_args" )));
374- assertEquals (
375- "{\" result\" :\" tool-result\" }" ,
376- attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.tool_response" )));
372+ assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.tool_response" )));
377373 }
378374
379375 @ Test
@@ -420,9 +416,8 @@ public void testTraceCallLlm() {
420416 assertEquals (
421417 ImmutableList .of ("stop" ),
422418 attrs .get (AttributeKey .stringArrayKey ("gen_ai.response.finish_reasons" )));
423- assertTrue (
424- attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_request" )).contains ("gemini-pro" ));
425- assertTrue (attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_response" )).contains ("STOP" ));
419+ assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_request" )));
420+ assertEquals ("{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.llm_response" )));
426421 }
427422
428423 @ Test
@@ -618,7 +613,7 @@ public void testTraceSendData() {
618613 "test-invocation-id" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.invocation_id" )));
619614 assertEquals ("event-1" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.event_id" )));
620615 assertEquals ("test-session" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.session_id" )));
621- assertTrue ( attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.data" )). contains ( "hello" ));
616+ assertEquals ( "{}" , attrs .get (AttributeKey .stringKey ("gcp.vertex.agent.data" )));
622617 }
623618
624619 // Agent that emits one event on a computation thread.
0 commit comments