From 8ddda7fbbb9022c990f2edf1855a25b4da6f596b Mon Sep 17 00:00:00 2001
From: Hamish Fagg <hamish@mindsdb.com>
Date: Thu, 4 Sep 2025 16:17:10 +1200
Subject: [PATCH] support new token auth

---
 mindsdb_sdk/connectors/rest_api.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mindsdb_sdk/connectors/rest_api.py b/mindsdb_sdk/connectors/rest_api.py
index 4b37efd..98fa405 100644
--- a/mindsdb_sdk/connectors/rest_api.py
+++ b/mindsdb_sdk/connectors/rest_api.py
@@ -82,6 +82,12 @@ def login(self):
 
         _raise_for_status(r)
 
+        # Use newer MindsDB auth that uses a token
+        if 'application/json' in r.headers.get('Content-Type', ''):
+            resp_json = r.json()
+            if isinstance(resp_json, dict) and "token" in resp_json:
+                self.session.headers['Authorization'] = f'Bearer {resp_json["token"]}'
+
     @_try_relogin
     def sql_query(self, sql, database=None, lowercase_columns=False):