fix(telemetery): Rolling back change to fix issue affecting LlmAgent creation due to missing version field

Co-authored-by: Achuth Narayan Rajagopal <achuthr@google.com>
PiperOrigin-RevId: 883336463

Author: Achuth17

src/google/adk/agents/base_agent.py

@@ -136,12 +136,6 @@ class MyAgent(BaseAgent):
   sub_agents: list[BaseAgent] = Field(default_factory=list)
   """The sub-agents of this agent."""
 
-  version: str = ''
-  """The agent's version.
-
-  Version of the agent being invoked. Used to identify the Agent involved in telemetry.
-  """
-
   before_agent_callback: Optional[BeforeAgentCallback] = None
   """Callback or list of callbacks to be invoked before the agent run.
 
@@ -686,7 +680,6 @@ def __create_kwargs(
 
     kwargs: Dict[str, Any] = {
         'name': config.name,
-        'version': config.version,
         'description': config.description,
     }
     if config.sub_agents:

src/google/adk/agents/base_agent_config.py

@@ -55,10 +55,6 @@ class BaseAgentConfig(BaseModel):
 
   name: str = Field(description='Required. The name of the agent.')
 
-  version: str = Field(
-      default='', description='Optional. The version of the agent.'
-  )
-
   description: str = Field(
       default='', description='Optional. The description of the agent.'
   )

src/google/adk/telemetry/tracing.py

@@ -83,8 +83,6 @@
 
 USER_CONTENT_ELIDED = '<elided>'
 
-GEN_AI_AGENT_VERSION = 'gen_ai.agent.version'
-
 # Needed to avoid circular imports
 if TYPE_CHECKING:
   from ..agents.base_agent import BaseAgent
@@ -159,7 +157,6 @@ def trace_agent_invocation(
   span.set_attribute(GEN_AI_AGENT_DESCRIPTION, agent.description)
 
   span.set_attribute(GEN_AI_AGENT_NAME, agent.name)
-  span.set_attribute(GEN_AI_AGENT_VERSION, agent.version)
   span.set_attribute(GEN_AI_CONVERSATION_ID, ctx.session.id)
 
 
@@ -495,7 +492,6 @@ def use_generate_content_span(
       USER_ID: invocation_context.session.user_id,
       'gcp.vertex.agent.event_id': model_response_event.id,
       'gcp.vertex.agent.invocation_id': invocation_context.invocation_id,
-      GEN_AI_AGENT_VERSION: invocation_context.agent.version,
   }
   if (
       _is_gemini_agent(invocation_context.agent)
@@ -530,7 +526,6 @@ async def use_inference_span(
       USER_ID: invocation_context.session.user_id,
       'gcp.vertex.agent.event_id': model_response_event.id,
       'gcp.vertex.agent.invocation_id': invocation_context.invocation_id,
-      GEN_AI_AGENT_VERSION: invocation_context.agent.version,
   }
   if (
       _is_gemini_agent(invocation_context.agent)

tests/unittests/telemetry/test_spans.py

@@ -26,7 +26,6 @@
 from google.adk.models.llm_response import LlmResponse
 from google.adk.sessions.in_memory_session_service import InMemorySessionService
 from google.adk.telemetry.tracing import ADK_CAPTURE_MESSAGE_CONTENT_IN_SPANS
-from google.adk.telemetry.tracing import GEN_AI_AGENT_VERSION
 from google.adk.telemetry.tracing import trace_agent_invocation
 from google.adk.telemetry.tracing import trace_call_llm
 from google.adk.telemetry.tracing import trace_inference_result
@@ -122,33 +121,6 @@ async def test_trace_agent_invocation(mock_span_fixture):
       mock.call('gen_ai.operation.name', 'invoke_agent'),
       mock.call('gen_ai.agent.description', agent.description),
       mock.call('gen_ai.agent.name', agent.name),
-      mock.call(GEN_AI_AGENT_VERSION, ''),
-      mock.call(
-          'gen_ai.conversation.id',
-          invocation_context.session.id,
-      ),
-  ]
-  mock_span_fixture.set_attribute.assert_has_calls(
-      expected_calls, any_order=True
-  )
-  assert mock_span_fixture.set_attribute.call_count == len(expected_calls)
-
-
-@pytest.mark.asyncio
-async def test_trace_agent_invocation_with_version(mock_span_fixture):
-  """Test trace_agent_invocation sets span attributes correctly when version is provided."""
-  agent = LlmAgent(name='test_llm_agent', model='gemini-pro')
-  agent.description = 'Test agent description'
-  agent.version = '1.0.0'
-  invocation_context = await _create_invocation_context(agent)
-
-  trace_agent_invocation(mock_span_fixture, agent, invocation_context)
-
-  expected_calls = [
-      mock.call('gen_ai.operation.name', 'invoke_agent'),
-      mock.call('gen_ai.agent.description', agent.description),
-      mock.call('gen_ai.agent.name', agent.name),
-      mock.call(GEN_AI_AGENT_VERSION, agent.version),
       mock.call(
           'gen_ai.conversation.id',
           invocation_context.session.id,
@@ -815,7 +787,6 @@ async def test_generate_content_span(
 
   mock_span.set_attributes.assert_called_once_with({
       GEN_AI_AGENT_NAME: invocation_context.agent.name,
-      GEN_AI_AGENT_VERSION: '',
       GEN_AI_CONVERSATION_ID: invocation_context.session.id,
       USER_ID: invocation_context.session.user_id,
       'gcp.vertex.agent.event_id': 'event-123',
@@ -1136,7 +1107,6 @@ async def test_generate_content_span_with_experimental_semconv(
 
   mock_span.set_attributes.assert_called_once_with({
       GEN_AI_AGENT_NAME: invocation_context.agent.name,
-      GEN_AI_AGENT_VERSION: '',
       GEN_AI_CONVERSATION_ID: invocation_context.session.id,
       USER_ID: invocation_context.session.user_id,
       'gcp.vertex.agent.event_id': 'event-123',