diff --git a/src/databricks_ai_bridge/genie.py b/src/databricks_ai_bridge/genie.py
index 7bbbdc5..c10aa10 100644
--- a/src/databricks_ai_bridge/genie.py
+++ b/src/databricks_ai_bridge/genie.py
@@ -157,6 +157,7 @@ def poll_result():
 
         return poll_result()
 
+    @mlflow.trace()
     def ask_question(self, question):
         resp = self.start_conversation(question)
         return self.poll_for_result(resp["conversation_id"], resp["message_id"])