fix tracer name to veadk (#288)

Author: yaozheng-fang

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.2.19"
+version = "0.2.20"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

veadk/agent.py

@@ -40,9 +40,10 @@
 from veadk.prompts.agent_default_prompt import DEFAULT_DESCRIPTION, DEFAULT_INSTRUCTION
 from veadk.tracing.base_tracer import BaseTracer
 from veadk.utils.logger import get_logger
-from veadk.utils.patches import patch_asyncio
+from veadk.utils.patches import patch_asyncio, patch_tracer
 from veadk.version import VERSION
 
+patch_tracer()
 patch_asyncio()
 logger = get_logger(__name__)
 

veadk/utils/patches.py

@@ -22,6 +22,7 @@
     trace_tool_call,
 )
 from veadk.utils.logger import get_logger
+from veadk.version import VERSION
 
 logger = get_logger(__name__)
 
@@ -78,3 +79,23 @@ def patch_google_adk_telemetry() -> None:
                     logger.debug(
                         f"Patch {mod_name} {var_name} with {trace_functions[var_name]}"
                     )
+
+
+def patch_tracer() -> None:
+    for mod_name, mod in sys.modules.items():
+        from opentelemetry import trace
+
+        if mod_name.startswith("google.adk"):
+            for var_name in dir(mod):
+                var = getattr(mod, var_name, None)
+                if var_name == "tracer" and isinstance(var, trace.Tracer):
+                    setattr(
+                        mod,
+                        var_name,
+                        trace.get_tracer(
+                            instrumenting_module_name="veadk",
+                            instrumenting_library_version=VERSION,
+                            schema_url="https://opentelemetry.io/schemas/1.37.0",
+                        ),
+                    )
+                    logger.debug(f"Patch {mod_name} {var_name} with VeADK tracer.")

veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.2.19"
+VERSION = "0.2.20"