Update config.py

config.py

@@ -1,6 +1,6 @@
 import json
 import os
-from kucoin.client import User as KuUser
+import ccxt
 import openai
 
 SESSION_FILE = ".session.json"
@@ -22,12 +22,15 @@ class SessionManager:
                 data = json.load(f)
                 self.__dict__.update(data)
 
-    def set_keys(self, openai_key, kucoin_key, kucoin_secret, kucoin_pass, provider):
+    def set_keys(
+        self, openai_key, kucoin_key, kucoin_secret, kucoin_pass, provider, hf_token
+    ):
         self.openai_key = openai_key
         self.kucoin_key = kucoin_key
         self.kucoin_secret = kucoin_secret
         self.kucoin_passphrase = kucoin_pass
         self.provider = provider
+        self.hf_token = hf_token
         with open(SESSION_FILE, "w") as f:
             json.dump(self.__dict__, f)
 
@@ -44,11 +47,14 @@ class SessionManager:
 
 def validate_all_keys(session: SessionManager):
     try:
-        KuUser(
-            key=session.kucoin_key,
-            secret=session.kucoin_secret,
-            passphrase=session.kucoin_passphrase,
-        ).get_account_overview("USDT")
+        kucoin = ccxt.kucoinfutures(
+            {
+                "apiKey": session.kucoin_key,
+                "secret": session.kucoin_secret,
+                "password": session.kucoin_passphrase,
+            }
+        )
+        _ = kucoin.fetch_balance()
     except Exception as e:
         return False, f"❌ KuCoin error: {e}"