remove unused queue.py module and refactor chain_id handling in CLI

Deleted aitbc/queue.py containing TaskQueue, JobScheduler, BackgroundTaskManager, and WorkerPool classes that were not being used in the codebase.

Refactored chain_id handling in CLI to use centralized get_chain_id utility function instead of duplicating chain_id detection logic in send_transaction, get_balance, and agent_operations functions.

cli/aitbc_cli.py

@@ -214,18 +214,9 @@ def send_transaction(from_wallet: str, to_address: str, amount: float, fee: floa
         print(f"Error decrypting wallet: {e}")
         return None
     
-    # Get chain_id from RPC health endpoint
-    chain_id = "ait-testnet"  # Default
-    try:
-        http_client = AITBCHTTPClient(base_url=rpc_url, timeout=5)
-        health_data = http_client.get("/health")
-        supported_chains = health_data.get("supported_chains", [])
-        if supported_chains:
-            chain_id = supported_chains[0]
-    except NetworkError:
-        pass
-    except Exception:
-        pass
+    # Get chain_id from RPC health endpoint or use override
+    from aitbc_cli.utils.chain_id import get_chain_id, get_default_chain_id
+    chain_id = get_chain_id(rpc_url, override=None, timeout=5)
     
     # Get actual nonce from blockchain
     actual_nonce = 0
@@ -747,9 +738,13 @@ def get_transactions(wallet_name: str, keystore_dir: Path = DEFAULT_KEYSTORE_DIR
         return []
 
 
-def get_balance(wallet_name: str, rpc_url: str = DEFAULT_RPC_URL) -> Optional[Dict]:
+def get_balance(wallet_name: str, rpc_url: str = DEFAULT_RPC_URL, chain_id_override: str = None) -> Optional[Dict]:
     """Get wallet balance"""
     try:
+        # Get chain_id from RPC health endpoint or use override
+        from aitbc_cli.utils.chain_id import get_chain_id
+        chain_id = get_chain_id(rpc_url, override=chain_id_override, timeout=5)
+        
         # Get wallet address
         wallet_path = DEFAULT_KEYSTORE_DIR / f"{wallet_name}.json"
         if not wallet_path.exists():
@@ -763,7 +758,7 @@ def get_balance(wallet_name: str, rpc_url: str = DEFAULT_RPC_URL) -> Optional[Di
         # Get account info from RPC
         try:
             http_client = AITBCHTTPClient(base_url=rpc_url, timeout=30)
-            account_info = http_client.get(f"/rpc/account/{address}?chain_id=ait-testnet")
+            account_info = http_client.get(f"/rpc/account/{address}?chain_id={chain_id}")
             return {
                 "wallet_name": wallet_name,
                 "address": address,
@@ -1118,13 +1113,18 @@ def agent_operations(action: str, **kwargs) -> Optional[Dict]:
                     format=serialization.PublicFormat.Raw
                 ).hex()
                 
-                # Get chain_id from RPC health endpoint
-                chain_id = "ait-testnet"  # Default
+                # Get chain_id from RPC health endpoint or use provided chain_id
+                chain_id_from_rpc = kwargs.get('chain_id', 'ait-mainnet')
+                # Auto-detect if not provided
+                if not kwargs.get('chain_id'):
+                    from aitbc_cli.utils.chain_id import get_chain_id
+                    chain_id_from_rpc = get_chain_id(rpc_url)
                 try:
                     http_client = AITBCHTTPClient(base_url=rpc_url, timeout=5)
                     health_data = http_client.get("/health")
                     supported_chains = health_data.get("supported_chains", [])
                     if supported_chains:
+                        chain_id_from_rpc = supported_chains[0]
                         chain_id = supported_chains[0]
                 except Exception:
                     pass
@@ -1811,6 +1811,7 @@ def simulate_ai_jobs(jobs: int, models: str, duration_range: str) -> Dict:
 
 def legacy_main():
     parser = argparse.ArgumentParser(description="AITBC CLI - Comprehensive Blockchain Management Tool")
+    parser.add_argument("--chain-id", default=None, help="Chain ID (auto-detected from blockchain node if not provided)")
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
     
     # Create wallet command
@@ -2016,6 +2017,7 @@ def legacy_main():
     wallet_balance_parser = wallet_subparsers.add_parser("balance", help="Wallet balance")
     wallet_balance_parser.add_argument("--name", help="Wallet name")
     wallet_balance_parser.add_argument("--all", action="store_true", help="Show all balances")
+    wallet_balance_parser.add_argument("--chain-id", help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)")
     
     # All balances command (keep for backward compatibility)
     all_balances_parser = subparsers.add_parser("all-balances", help="Show all wallet balances")
@@ -2140,6 +2142,10 @@ def legacy_main():
     
     args = parser.parse_args()
     
+    # Handle chain_id with auto-detection
+    from aitbc_cli.utils.chain_id import get_chain_id
+    chain_id = get_chain_id(DEFAULT_RPC_URL, override=args.chain_id)
+    
     if args.command == "create":
         # Get password
         password = None
@@ -2312,6 +2318,7 @@ def legacy_main():
             kwargs['execution_id'] = args.execution_id
         if hasattr(args, 'status') and args.status:
             kwargs['status'] = args.status
+        kwargs['chain_id'] = chain_id
         
         result = agent_operations(args.agent_action, **kwargs)
         if result:

cli/aitbc_cli/commands/marketplace_cmd.py

@@ -14,9 +14,17 @@ from ..core.marketplace import (
 from ..utils import output, error, success
 
 @click.group()
-def marketplace():
+@click.option("--chain-id", help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)")
+@click.pass_context
+def marketplace(ctx, chain_id: Optional[str]):
     """Global chain marketplace commands"""
-    pass
+    ctx.ensure_object(dict)
+    
+    # Handle chain_id with auto-detection
+    from ..utils.chain_id import get_chain_id
+    config = load_multichain_config()
+    default_rpc_url = config.blockchain_rpc_url if hasattr(config, 'blockchain_rpc_url') else 'http://localhost:8006'
+    ctx.obj['chain_id'] = get_chain_id(default_rpc_url, override=chain_id)
 
 @marketplace.command()
 @click.argument('chain_id')

cli/aitbc_cli/commands/wallet.py

@@ -102,8 +102,9 @@ def _load_wallet(wallet_path: Path, wallet_name: str) -> Dict[str, Any]:
     "--wallet-path", help="Direct path to wallet file (overrides --wallet-name)"
 )
 @click.option("--use-daemon", is_flag=True, default=True, help="Use wallet daemon for operations")
+@click.option("--chain-id", help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)")
 @click.pass_context
-def wallet(ctx, wallet_name: Optional[str], wallet_path: Optional[str], use_daemon: bool):
+def wallet(ctx, wallet_name: Optional[str], wallet_path: Optional[str], use_daemon: bool, chain_id: Optional[str]):
     """Manage your AITBC wallets and transactions"""
     # Ensure wallet object exists
     ctx.ensure_object(dict)
@@ -111,6 +112,12 @@ def wallet(ctx, wallet_name: Optional[str], wallet_path: Optional[str], use_daem
     # Set daemon mode
     ctx.obj["use_daemon"] = use_daemon
     
+    # Handle chain_id with auto-detection
+    from ..utils.chain_id import get_chain_id
+    config = get_config()
+    default_rpc_url = config.blockchain_rpc_url if hasattr(config, 'blockchain_rpc_url') else 'http://localhost:8006'
+    ctx.obj["chain_id"] = get_chain_id(default_rpc_url, override=chain_id)
+    
     # Initialize dual-mode adapter
     from ..config import get_config
     import sys
@@ -532,7 +539,8 @@ def balance(ctx):
                 base_url=config.coordinator_url.replace('/api', ''),
                 timeout=5
             )
-            blockchain_balance = http_client.get(f"/rpc/balance/{wallet_data['address']}")
+            chain_id = ctx.obj.get("chain_id", "ait-mainnet")
+            blockchain_balance = http_client.get(f"/rpc/balance/{wallet_data['address']}?chain_id={chain_id}")
             output(
                 {
                             "wallet": wallet_name,

cli/aitbc_cli/config.py

@@ -30,6 +30,9 @@ class CLIConfig(BaseAITBCConfig):
     wallet_url: str = Field(default="http://localhost:8003", description="Wallet daemon URL (alias for compatibility)")
     blockchain_rpc_url: str = Field(default=f"http://localhost:{BLOCKCHAIN_RPC_PORT}", description="Blockchain RPC URL")
     
+    # Chain configuration
+    chain_id: str = Field(default="ait-mainnet", description="Default chain ID for multichain operations")
+    
     # Authentication
     api_key: Optional[str] = Field(default=None, description="API key for authentication")
     

cli/aitbc_cli/utils/chain_id.py

@@ -0,0 +1,78 @@
+"""Chain ID utilities for AITBC CLI
+
+This module provides functions for auto-detecting and validating chain IDs
+from blockchain nodes, supporting multichain operations.
+"""
+
+from typing import Optional
+from aitbc import AITBCHTTPClient, NetworkError
+
+
+# Known chain IDs
+KNOWN_CHAINS = ["ait-mainnet", "ait-devnet", "ait-testnet", "ait-healthchain"]
+
+
+def get_default_chain_id() -> str:
+    """Return the default chain ID (ait-mainnet for production)."""
+    return "ait-mainnet"
+
+
+def validate_chain_id(chain_id: str) -> bool:
+    """Validate a chain ID against known chains.
+    
+    Args:
+        chain_id: The chain ID to validate
+        
+    Returns:
+        True if the chain ID is known, False otherwise
+    """
+    return chain_id in KNOWN_CHAINS
+
+
+def get_chain_id_from_health(rpc_url: str, timeout: int = 5) -> str:
+    """Auto-detect chain ID from blockchain node's /health endpoint.
+    
+    Args:
+        rpc_url: The blockchain node RPC URL (e.g., http://localhost:8006)
+        timeout: Request timeout in seconds
+        
+    Returns:
+        The detected chain ID, or default if detection fails
+    """
+    try:
+        http_client = AITBCHTTPClient(base_url=rpc_url, timeout=timeout)
+        health_data = http_client.get("/health")
+        supported_chains = health_data.get("supported_chains", [])
+        
+        if supported_chains:
+            # Return the first supported chain (typically the primary chain)
+            return supported_chains[0]
+    except NetworkError:
+        pass
+    except Exception:
+        pass
+    
+    # Fallback to default if detection fails
+    return get_default_chain_id()
+
+
+def get_chain_id(rpc_url: str, override: Optional[str] = None, timeout: int = 5) -> str:
+    """Get chain ID with override support and auto-detection fallback.
+    
+    Args:
+        rpc_url: The blockchain node RPC URL
+        override: Optional chain ID override (e.g., from --chain-id flag)
+        timeout: Request timeout in seconds
+        
+    Returns:
+        The chain ID to use (override takes precedence, then auto-detection, then default)
+    """
+    # If override is provided, validate and use it
+    if override:
+        if validate_chain_id(override):
+            return override
+        # If unknown, still use it (user may be testing new chains)
+        return override
+    
+    # Otherwise, auto-detect from health endpoint
+    return get_chain_id_from_health(rpc_url, timeout)

cli/commands/blockchain.py

@@ -17,12 +17,19 @@ import os
 
 
 @click.group()
+@click.option("--chain-id", help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)")
 @click.pass_context
-def blockchain(ctx):
+def blockchain(ctx, chain_id: Optional[str]):
     """Query blockchain information and status"""
     # Set role for blockchain commands
     ctx.ensure_object(dict)
     ctx.parent.detected_role = 'blockchain'
+    
+    # Handle chain_id with auto-detection
+    from aitbc_cli.utils.chain_id import get_chain_id
+    config = ctx.obj.get('config')
+    default_rpc_url = _get_node_endpoint(ctx)
+    ctx.obj['chain_id'] = get_chain_id(default_rpc_url, override=chain_id)
 
 
 @blockchain.command()

cli/commands/marketplace.py

@@ -266,8 +266,10 @@ def pay(ctx, booking_id: str, amount: float, from_wallet: str, to_wallet: str, t
         address = wallet_data["address"]
         
         # Get wallet balance from blockchain
+        from aitbc_cli.utils.chain_id import get_chain_id
         rpc_url = config.get('rpc_url', 'http://localhost:8006')
-        balance_response = httpx.Client().get(f"{rpc_url}/rpc/account/{address}?chain_id=ait-testnet", timeout=5)
+        chain_id = get_chain_id(rpc_url)
+        balance_response = httpx.Client().get(f"{rpc_url}/rpc/account/{address}?chain_id={chain_id}", timeout=5)
         if balance_response.status_code != 200:
             error(f"Failed to get wallet balance")
             return
@@ -285,7 +287,7 @@ def pay(ctx, booking_id: str, amount: float, from_wallet: str, to_wallet: str, t
             "value": amount,
             "fee": 1,
             "nonce": balance_data["nonce"],
-            "chain_id": "ait-testnet",
+            "chain_id": chain_id,
             "payload": {
                 "type": "marketplace_payment",
                 "booking_id": booking_id,

cli/core/main.py

@@ -101,6 +101,11 @@ def version():
     default=None,
     help="API key for authentication"
 )
+@click.option(
+    "--chain-id",
+    default=None,
+    help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)"
+)
 @click.option(
     "--output",
     default="table",
@@ -119,7 +124,7 @@ def version():
     help="Enable debug mode"
 )
 @click.pass_context
-def cli(ctx, url, api_key, output, verbose, debug):
+def cli(ctx, url, api_key, chain_id, output, verbose, debug):
     """AITBC CLI - Command Line Interface for AITBC Network
     
     Manage jobs, mining, wallets, blockchain operations, marketplaces, and AI
@@ -142,6 +147,11 @@ def cli(ctx, url, api_key, output, verbose, debug):
     ctx.obj['output'] = output
     ctx.obj['verbose'] = verbose
     ctx.obj['debug'] = debug
+    
+    # Handle chain_id with auto-detection
+    from aitbc_cli.utils.chain_id import get_chain_id, get_default_chain_id
+    default_rpc_url = url.replace('/api', '') if url else 'http://localhost:8006'
+    ctx.obj['chain_id'] = get_chain_id(default_rpc_url, override=chain_id)
 
 # Add commands to CLI
 cli.add_command(system)

cli/unified_cli.py

@@ -536,6 +536,7 @@ def run_cli(argv, core):
     wallet_balance_parser.add_argument("--name", dest="wallet_name_opt", help=argparse.SUPPRESS)
     wallet_balance_parser.add_argument("--all", action="store_true")
     wallet_balance_parser.add_argument("--rpc-url", default=default_rpc_url)
+    wallet_balance_parser.add_argument("--chain-id", help="Chain ID for multichain operations (e.g., ait-mainnet, ait-devnet)")
     wallet_balance_parser.set_defaults(handler=handle_wallet_balance)
 
     wallet_transactions_parser = wallet_subparsers.add_parser("transactions", help="Show wallet transactions")

cli/utils/dual_mode_wallet_adapter.py

@@ -18,12 +18,19 @@ from utils import error, success, output
 class DualModeWalletAdapter:
     """Adapter supporting both file-based and daemon-based wallet operations"""
     
-    def __init__(self, config: Config, use_daemon: bool = False):
+    def __init__(self, config: Config, use_daemon: bool = False, chain_id: Optional[str] = None):
         self.config = config
         self.use_daemon = use_daemon
+        self.chain_id = chain_id
         self.wallet_dir = Path.home() / ".aitbc" / "wallets"
         self.wallet_dir.mkdir(parents=True, exist_ok=True)
         
+        # Auto-detect chain_id if not provided
+        if not self.chain_id:
+            from aitbc_cli.utils.chain_id import get_chain_id
+            default_rpc_url = config.blockchain_rpc_url if hasattr(config, 'blockchain_rpc_url') else 'http://localhost:8006'
+            self.chain_id = get_chain_id(default_rpc_url)
+        
         if use_daemon:
             self.daemon_client = WalletDaemonClient(config)
         else:
@@ -311,7 +318,7 @@ class DualModeWalletAdapter:
             
         rpc_url = self.config.blockchain_rpc_url
         try:
-            resp = httpx.get(f"{rpc_url}/rpc/account/{from_address}?chain_id=ait-testnet", timeout=5)
+            resp = httpx.get(f"{rpc_url}/rpc/account/{from_address}?chain_id={self.chain_id}", timeout=5)
             if resp.status_code == 200:
                 data = resp.json()
                 chain_balance = data.get("balance", 0)