nvidia-nat-adk 1.3.0rc4__py3-none-any.whl → 1.3.0rc5__py3-none-any.whl

nat/plugins/adk/adk_callback_handler.py

@@ -214,21 +214,23 @@ class ADKProfilerHandler(BaseProfilerCallback):
                     model_name = first
             model_name = model_name or ""
 
-            model_input = ""
+            model_input = []
             try:
                 for message in kwargs.get("messages", []):
                     content = message.get("content", "")
                     if isinstance(content, list):
                         for part in content:
                             if isinstance(part, dict):
-                                model_input += str(part.get("text", ""))  # text parts
+                                model_input.append(str(part.get("text", "")))  # text parts
                             else:
-                                model_input += str(part)
+                                model_input.append(str(part))
                     else:
-                        model_input += content or ""
+                        model_input.append("" if content is None else str(content))
             except Exception as _e:
                 logger.exception("Error getting model input")
 
+            model_input = "".join(model_input)
+
             # Record the start event
             input_stats = IntermediateStepPayload(
                 event_type=IntermediateStepType.LLM_START,
@@ -252,14 +254,16 @@ class ADKProfilerHandler(BaseProfilerCallback):
                 raise RuntimeError("Original LLM function is None - instrumentation may not have been set up correctly")
             output = await original_func(*args, **kwargs)
 
-            model_output = ""
+            model_output = []
             try:
                 for choice in output.choices:
                     msg = choice.message
-                    model_output += msg.content or ""
+                    model_output.append(msg.content or "")
             except Exception as _e:
                 logger.exception("Error getting model output")
 
+            model_output = "".join(model_output)
+
             now = time.time()
             # Record the end event
             # Prepare safe metadata and usage

{nvidia_nat_adk-1.3.0rc4.dist-info → nvidia_nat_adk-1.3.0rc5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-adk
-Version: 1.3.0rc4
+Version: 1.3.0rc5
 Summary: Subpackage for Google ADK integration in NeMo Agent Toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,7 +16,7 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat[litellm]==v1.3.0-rc4
+Requires-Dist: nvidia-nat[litellm]==v1.3.0-rc5
 Requires-Dist: google-adk~=1.14.1
 Dynamic: license-file
 

{nvidia_nat_adk-1.3.0rc4.dist-info → nvidia_nat_adk-1.3.0rc5.dist-info}/RECORD

@@ -1,13 +1,13 @@
 nat/meta/pypi.md,sha256=AOSYTEk4JQQeaQQ1fA7v__j9-5rchBPvvSAICORgttE,953
 nat/plugins/adk/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W-izgx9aMEQg,680
-nat/plugins/adk/adk_callback_handler.py,sha256=MMjMp6WN1zCVOFqoJWC90H0GGCHs49qMgoHfMHFitsU,12562
+nat/plugins/adk/adk_callback_handler.py,sha256=JtyqakzOVGhnGQOZY7RRrWIO6rFNgsrC2j5Dv42HlkQ,12706
 nat/plugins/adk/llm.py,sha256=glwlO9UQhfGagVDNbtw-bNQ7FVdJfs31v6rqd-MBJLY,4120
 nat/plugins/adk/register.py,sha256=q6ZOmldtbr_7Gr8iUS8xrhR8_FhSubwGqr7VJNrZg1A,809
 nat/plugins/adk/tool_wrapper.py,sha256=TroOo7tM6I3gOqMxOL7fQmPsyVKlntPPRNbYQMk6JHs,5779
-nvidia_nat_adk-1.3.0rc4.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_adk-1.3.0rc4.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_adk-1.3.0rc4.dist-info/METADATA,sha256=Pc7kGOtjRWXiKaAwSD1Rh67Yxtf8wfqBJo-PyHGFCwk,1761
-nvidia_nat_adk-1.3.0rc4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_adk-1.3.0rc4.dist-info/entry_points.txt,sha256=K1K0clA1uqXk1Q0dh8ZaWNQyveaFx_XvgBWzCyGMYts,52
-nvidia_nat_adk-1.3.0rc4.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_adk-1.3.0rc4.dist-info/RECORD,,
+nvidia_nat_adk-1.3.0rc5.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_adk-1.3.0rc5.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_adk-1.3.0rc5.dist-info/METADATA,sha256=oF7-bWVnjJ3z54UoQgcyuh9FhpwlXXY8A7MVUFjbKdg,1761
+nvidia_nat_adk-1.3.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_adk-1.3.0rc5.dist-info/entry_points.txt,sha256=K1K0clA1uqXk1Q0dh8ZaWNQyveaFx_XvgBWzCyGMYts,52
+nvidia_nat_adk-1.3.0rc5.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_adk-1.3.0rc5.dist-info/RECORD,,