fix

Author: nisha2003

src/databricks_ai_bridge/genie.py

@@ -470,7 +470,7 @@ def ask_question(self, question, conversation_id: Optional[str] = None):
 
             if conversation_id:
                 resp = self.create_message(conversation_id, question)
-                return self.poll_for_result(conversation_id, resp["id"])
+                return self.poll_for_result(conversation_id, resp["message_id"])
             else:
                 resp = self.start_conversation(question)
-                return self.poll_for_result(resp["conversation_id"], resp["id"])
+                return self.poll_for_result(resp["conversation_id"], resp["message_id"])