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__":