diff --git a/agent.py b/agent.py
index 2df602e..f650381 100644
--- a/agent.py
+++ b/agent.py
@@ -113,6 +113,9 @@ def __init__(
                 ),
                 types.Tool(function_declarations=custom_functions),
             ],
+            thinking_config=types.ThinkingConfig(
+                include_thoughts=True
+            ),
         )
 
     def handle_action(self, action: types.FunctionCall) -> FunctionResponseT: