diff --git a/src/google/adk/flows/llm_flows/base_llm_flow.py b/src/google/adk/flows/llm_flows/base_llm_flow.py index 52680346..ef5ff1f7 100644 --- a/src/google/adk/flows/llm_flows/base_llm_flow.py +++ b/src/google/adk/flows/llm_flows/base_llm_flow.py @@ -36,9 +36,9 @@ from ...models.base_llm_connection import BaseLlmConnection from ...models.llm_request import LlmRequest from ...models.llm_response import LlmResponse -from ...telemetry import trace_call_llm -from ...telemetry import trace_send_data -from ...telemetry import tracer +from ...telemetry.recording import trace_call_llm +from ...telemetry.recording import trace_send_data +from ...telemetry.recording import tracer from ...tools.tool_context import ToolContext from . import functions diff --git a/src/google/adk/flows/llm_flows/functions.py b/src/google/adk/flows/llm_flows/functions.py index 25a2ab01..6782b9ec 100644 --- a/src/google/adk/flows/llm_flows/functions.py +++ b/src/google/adk/flows/llm_flows/functions.py @@ -32,9 +32,9 @@ from ...auth.auth_tool import AuthToolArguments from ...events.event import Event from ...events.event_actions import EventActions -from ...telemetry import trace_tool_call -from ...telemetry import trace_tool_response -from ...telemetry import tracer +from ...telemetry.recording import trace_tool_call +from ...telemetry.recording import trace_tool_response +from ...telemetry.recording import tracer from ...tools.base_tool import BaseTool from ...tools.tool_context import ToolContext diff --git a/src/google/adk/runners.py b/src/google/adk/runners.py index 529c4675..03c85b0b 100644 --- a/src/google/adk/runners.py +++ b/src/google/adk/runners.py @@ -39,7 +39,7 @@ from .sessions.base_session_service import BaseSessionService from .sessions.in_memory_session_service import InMemorySessionService from .sessions.session import Session -from .telemetry import tracer +from .telemetry.recording import tracer from .tools.built_in_code_execution_tool import built_in_code_execution logger = logging.getLogger(__name__) diff --git a/src/google/adk/telemetry/__init__.py b/src/google/adk/telemetry/__init__.py new file mode 100644 index 00000000..0a2669d7 --- /dev/null +++ b/src/google/adk/telemetry/__init__.py @@ -0,0 +1,13 @@ +# Copyright 2025 Google LLC +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/src/google/adk/telemetry.py b/src/google/adk/telemetry/recording/__init__.py similarity index 97% rename from src/google/adk/telemetry.py rename to src/google/adk/telemetry/recording/__init__.py index 0ee6cf8e..0f167b94 100644 --- a/src/google/adk/telemetry.py +++ b/src/google/adk/telemetry/recording/__init__.py @@ -27,10 +27,10 @@ from google.genai import types from opentelemetry import trace -from .agents.invocation_context import InvocationContext -from .events.event import Event -from .models.llm_request import LlmRequest -from .models.llm_response import LlmResponse +from ...agents.invocation_context import InvocationContext +from ...events.event import Event +from ...models.llm_request import LlmRequest +from ...models.llm_response import LlmResponse tracer = trace.get_tracer('gcp.vertex.agent')