veadk-python 0.2.19__tar.gz → 0.2.20__tar.gz

{veadk_python-0.2.19 → veadk_python-0.2.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.2.19
+Version: 0.2.20
 Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
 Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
 License:                                  Apache License

{veadk_python-0.2.19 → veadk_python-0.2.20}/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_python-0.2.19 → veadk_python-0.2.20}/veadk/agent.py

@@ -40,9 +40,10 @@ from veadk.processors import BaseRunProcessor, NoOpRunProcessor
 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_python-0.2.19 → veadk_python-0.2.20}/veadk/tools/builtin_tools/run_code.py

@@ -43,7 +43,7 @@ def run_code(code: str, language: str, tool_context: ToolContext) -> str:
     )  # temporary service for code run tool
     region = getenv("AGENTKIT_TOOL_REGION", "cn-beijing")
     host = getenv(
-        "AGENTKIT_TOOL_HOST", service + "." + region + ".volces.com "
+        "AGENTKIT_TOOL_HOST", service + "." + region + ".volces.com"
     )  # temporary host for code run tool
     logger.debug(f"tools endpoint: {host}")
 

{veadk_python-0.2.19 → veadk_python-0.2.20}/veadk/utils/patches.py

@@ -22,6 +22,7 @@ from veadk.tracing.telemetry.telemetry import (
     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_python-0.2.19 → veadk_python-0.2.20}/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"

{veadk_python-0.2.19 → veadk_python-0.2.20}/veadk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: veadk-python
-Version: 0.2.19
+Version: 0.2.20
 Summary: Volcengine agent development kit, integrations with Volcengine cloud services.
 Author-email: Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>, Zhi Han <sliverydayday@gmail.com>, Meng Wang <mengwangwm@gmail.com>
 License:                                  Apache License