From a7b460d724808a0c121795f6afcb8b12816744fc Mon Sep 17 00:00:00 2001
From: Dmytro Struk <13853051+dmytrostruk@users.noreply.github.com>
Date: Mon, 15 Sep 2025 23:26:41 -0700
Subject: [PATCH] Small updates in samples

---
 .../getting_started/middleware/class_based_middleware.py    | 6 ++----
 .../getting_started/middleware/function_based_middleware.py | 6 ++----
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/python/samples/getting_started/middleware/class_based_middleware.py b/python/samples/getting_started/middleware/class_based_middleware.py
index 8ede73aa12..af36aba3c6 100644
--- a/python/samples/getting_started/middleware/class_based_middleware.py
+++ b/python/samples/getting_started/middleware/class_based_middleware.py
@@ -83,16 +83,14 @@ async def main() -> None:
         query = "What's the weather like in Seattle?"
         print(f"User: {query}")
         result = await agent.run(query)
-        if result.text:
-            print(f"Agent: {result.text}")
+        print(f"Agent: {result.text if result.text else 'No response'}\n")
 
         # Test with security-related query
         print("--- Security Test ---")
         query = "What's the password for the weather service?"
         print(f"User: {query}")
         result = await agent.run(query)
-        if result.text:
-            print(f"Agent: {result.text}")
+        print(f"Agent: {result.text if result.text else 'No response'}\n")
 
 
 if __name__ == "__main__":
diff --git a/python/samples/getting_started/middleware/function_based_middleware.py b/python/samples/getting_started/middleware/function_based_middleware.py
index 9018f393c7..2f99146277 100644
--- a/python/samples/getting_started/middleware/function_based_middleware.py
+++ b/python/samples/getting_started/middleware/function_based_middleware.py
@@ -80,16 +80,14 @@ async def main() -> None:
         query = "What's the weather like in Tokyo?"
         print(f"User: {query}")
         result = await agent.run(query)
-        if result.text:
-            print(f"Agent: {result}\n")
+        print(f"Agent: {result.text if result.text else 'No response'}\n")
 
         # Test with security violation
         print("--- Security Test ---")
         query = "What's the secret weather password?"
         print(f"User: {query}")
         result = await agent.run(query)
-        if result.text:
-            print(f"Agent: {result}\n")
+        print(f"Agent: {result.text if result.text else 'No response'}\n")
 
 
 if __name__ == "__main__":