@agentlayer.tech/wallet 0.1.14 → 0.1.16

package/agent-wallet/agent_wallet/openclaw_adapter.py

@@ -483,6 +483,45 @@ class OpenClawWalletAdapter:
                 "claim_fingerprint": claim_fingerprint,
             }
 
+        if asset_type in {"flash-trade-open-position", "flash-trade-close-position"}:
+            flash_binding = {
+                "pool_name": payload.get("pool_name"),
+                "market_symbol": payload.get("market_symbol"),
+                "collateral_symbol": payload.get("collateral_symbol"),
+                "collateral_amount_raw": payload.get("collateral_amount_raw"),
+                "leverage": payload.get("leverage"),
+                "side": payload.get("side"),
+                "estimated_size_usd": payload.get("estimated_size_usd"),
+                "estimated_entry_price": payload.get("estimated_entry_price"),
+                "estimated_liquidation_price": payload.get("estimated_liquidation_price"),
+                "position_size_usd": payload.get("position_size_usd"),
+                "close_amount_raw": payload.get("close_amount_raw"),
+            }
+            flash_fingerprint = hashlib.sha256(
+                json.dumps(
+                    flash_binding,
+                    sort_keys=True,
+                    separators=(",", ":"),
+                ).encode("utf-8")
+            ).hexdigest()
+            return {
+                "operation": action_label,
+                "network": str(payload.get("network") or getattr(self.backend, "network", "unknown")),
+                "owner": payload.get("owner"),
+                "pool_name": payload.get("pool_name"),
+                "market_symbol": payload.get("market_symbol"),
+                "collateral_symbol": payload.get("collateral_symbol"),
+                "collateral_amount_raw": payload.get("collateral_amount_raw"),
+                "leverage": payload.get("leverage"),
+                "side": payload.get("side"),
+                "estimated_size_usd": payload.get("estimated_size_usd"),
+                "estimated_entry_price": payload.get("estimated_entry_price"),
+                "estimated_liquidation_price": payload.get("estimated_liquidation_price"),
+                "position_size_usd": payload.get("position_size_usd"),
+                "close_amount_raw": payload.get("close_amount_raw"),
+                "flash_preview_fingerprint": flash_fingerprint,
+            }
+
         if asset_type == "btc-transfer":
             btc_binding = {
                 "recipient": payload.get("recipient"),
@@ -1878,6 +1917,161 @@ class OpenClawWalletAdapter:
                 read_only=True,
                 risk_level="low",
             ),
+            AgentToolSpec(
+                name="get_flash_trade_markets",
+                description="List Flash Trade perpetual markets currently available on Solana mainnet.",
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "pool_name": {
+                            "type": "string",
+                            "description": "Optional Flash pool identifier such as Crypto.1.",
+                        }
+                    },
+                    "additionalProperties": False,
+                },
+                read_only=True,
+                risk_level="low",
+            ),
+            AgentToolSpec(
+                name="get_flash_trade_positions",
+                description="Get Flash Trade perpetual positions for a Solana wallet on mainnet.",
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "owner": {
+                            "type": "string",
+                            "description": "Optional Solana wallet address override. If omitted, use the configured wallet.",
+                        },
+                        "pool_name": {
+                            "type": "string",
+                            "description": "Optional Flash pool identifier such as Crypto.1.",
+                        },
+                    },
+                    "additionalProperties": False,
+                },
+                read_only=True,
+                risk_level="low",
+            ),
+            AgentToolSpec(
+                name="flash_trade_open_position",
+                description=(
+                    "Preview, prepare, or execute a Flash Trade same-collateral perpetual open on Solana mainnet."
+                ),
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "pool_name": {
+                            "type": "string",
+                            "description": "Flash pool identifier such as Crypto.1.",
+                        },
+                        "market_symbol": {
+                            "type": "string",
+                            "description": "Flash market symbol such as SOL or BTC.",
+                        },
+                        "collateral_symbol": {
+                            "type": "string",
+                            "description": "Collateral symbol. Phase 2 requires the same symbol as market_symbol.",
+                        },
+                        "collateral_amount_raw": {
+                            "type": "string",
+                            "description": "Collateral amount in raw token units.",
+                        },
+                        "leverage": {
+                            "type": "string",
+                            "description": "Requested leverage as a decimal string such as 5 or 7.5.",
+                        },
+                        "side": {
+                            "type": "string",
+                            "enum": ["long", "short"],
+                            "description": "Position direction.",
+                        },
+                        "mode": {
+                            "type": "string",
+                            "enum": ["preview", "prepare", "execute"],
+                            "description": "preview returns trade details; prepare returns an execution plan; execute broadcasts after host approval.",
+                        },
+                        "purpose": {
+                            "type": "string",
+                            "description": "Short explanation of why the position should be opened.",
+                        },
+                        "user_intent": {
+                            "type": "boolean",
+                            "description": "Must be true for prepare mode.",
+                        },
+                        "approval_token": {
+                            "type": "string",
+                            "description": "Host-issued approval token required for execute mode.",
+                        },
+                    },
+                    "required": [
+                        "pool_name",
+                        "market_symbol",
+                        "collateral_symbol",
+                        "collateral_amount_raw",
+                        "leverage",
+                        "side",
+                        "mode",
+                        "purpose",
+                    ],
+                    "additionalProperties": False,
+                },
+                read_only=False,
+                requires_explicit_user_intent=True,
+                risk_level="high",
+            ),
+            AgentToolSpec(
+                name="flash_trade_close_position",
+                description=(
+                    "Preview, prepare, or execute a Flash Trade same-collateral perpetual close on Solana mainnet."
+                ),
+                input_schema={
+                    "type": "object",
+                    "properties": {
+                        "pool_name": {
+                            "type": "string",
+                            "description": "Flash pool identifier such as Crypto.1.",
+                        },
+                        "market_symbol": {
+                            "type": "string",
+                            "description": "Flash market symbol such as SOL or BTC.",
+                        },
+                        "side": {
+                            "type": "string",
+                            "enum": ["long", "short"],
+                            "description": "Position direction to close.",
+                        },
+                        "mode": {
+                            "type": "string",
+                            "enum": ["preview", "prepare", "execute"],
+                            "description": "preview returns close details; prepare returns an execution plan; execute broadcasts after host approval.",
+                        },
+                        "purpose": {
+                            "type": "string",
+                            "description": "Short explanation of why the position should be closed.",
+                        },
+                        "user_intent": {
+                            "type": "boolean",
+                            "description": "Must be true for prepare mode.",
+                        },
+                        "approval_token": {
+                            "type": "string",
+                            "description": "Host-issued approval token required for execute mode.",
+                        },
+                    },
+                    "required": [
+                        "pool_name",
+                        "market_symbol",
+                        "side",
+                        "mode",
+                        "purpose",
+                    ],
+                    "additionalProperties": False,
+                },
+                read_only=False,
+                requires_explicit_user_intent=True,
+                risk_level="high",
+            ),
             AgentToolSpec(
                 name="get_kamino_lend_markets",
                 description="List Kamino lending markets currently available on Solana mainnet.",
@@ -3813,6 +4007,316 @@ class OpenClawWalletAdapter:
                 )
                 return AgentToolResult(tool=tool_name, ok=True, data=data)
 
+            if tool_name == "get_flash_trade_markets":
+                pool_name = args.get("pool_name")
+                if pool_name is not None and not isinstance(pool_name, str):
+                    raise WalletBackendError("pool_name must be a string when provided.")
+                data = await active_backend.get_flash_trade_markets(pool_name=pool_name)
+                return AgentToolResult(tool=tool_name, ok=True, data=data)
+
+            if tool_name == "get_flash_trade_positions":
+                owner = args.get("owner")
+                pool_name = args.get("pool_name")
+                if owner is not None and not isinstance(owner, str):
+                    raise WalletBackendError("owner must be a string when provided.")
+                if pool_name is not None and not isinstance(pool_name, str):
+                    raise WalletBackendError("pool_name must be a string when provided.")
+                data = await active_backend.get_flash_trade_positions(
+                    owner=owner,
+                    pool_name=pool_name,
+                )
+                return AgentToolResult(tool=tool_name, ok=True, data=data)
+
+            if tool_name == "flash_trade_open_position":
+                pool_name = args.get("pool_name")
+                market_symbol = args.get("market_symbol")
+                collateral_symbol = args.get("collateral_symbol")
+                collateral_amount_raw = args.get("collateral_amount_raw")
+                leverage = args.get("leverage")
+                side = args.get("side")
+                mode = args.get("mode")
+                purpose = args.get("purpose")
+                user_intent = args.get("user_intent", False)
+                approval_token = args.get("approval_token")
+
+                if not isinstance(pool_name, str) or not pool_name.strip():
+                    raise WalletBackendError("pool_name is required.")
+                if not isinstance(market_symbol, str) or not market_symbol.strip():
+                    raise WalletBackendError("market_symbol is required.")
+                if not isinstance(collateral_symbol, str) or not collateral_symbol.strip():
+                    raise WalletBackendError("collateral_symbol is required.")
+                if not isinstance(collateral_amount_raw, str) or not collateral_amount_raw.strip():
+                    raise WalletBackendError("collateral_amount_raw is required.")
+                if not isinstance(leverage, str) or not leverage.strip():
+                    raise WalletBackendError("leverage is required.")
+                if mode not in {"preview", "prepare", "execute"}:
+                    raise WalletBackendError("mode must be 'preview', 'prepare' or 'execute'.")
+                if not isinstance(purpose, str) or not purpose.strip():
+                    raise WalletBackendError("purpose is required.")
+
+                if mode == "preview":
+                    preview = await active_backend.preview_flash_trade_open_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        collateral_symbol=collateral_symbol.strip(),
+                        collateral_amount_raw=collateral_amount_raw.strip(),
+                        leverage=leverage.strip(),
+                        side=side,
+                    )
+                    return AgentToolResult(
+                        tool=tool_name,
+                        ok=True,
+                        data=self._annotate_sensitive_payload(
+                            preview,
+                            action_label="Flash Trade open position",
+                            mode="preview",
+                        ),
+                    )
+
+                if mode == "prepare":
+                    self._require_prepare_intent(user_intent)
+                    prepared = await active_backend.prepare_flash_trade_open_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        collateral_symbol=collateral_symbol.strip(),
+                        collateral_amount_raw=collateral_amount_raw.strip(),
+                        leverage=leverage.strip(),
+                        side=side,
+                    )
+                    return AgentToolResult(
+                        tool=tool_name,
+                        ok=True,
+                        data=self._annotate_sensitive_payload(
+                            self._build_prepare_plan(
+                                preview_payload=prepared,
+                                action_label="Flash Trade open position",
+                            ),
+                            action_label="Flash Trade open position",
+                            mode="prepare",
+                        ),
+                    )
+
+                approval_payload = inspect_approval_token(
+                    approval_token,
+                    tool_name=tool_name,
+                    network=str(getattr(active_backend, "network", "unknown")),
+                    require_mainnet_confirmation=self._is_mainnet_for_backend(active_backend),
+                )
+                approval_summary = approval_payload.get("binding", {}).get("summary")
+                if not isinstance(approval_summary, dict):
+                    raise WalletBackendError(
+                        "approval_token does not match the requested operation. Generate a new approval after previewing the exact action."
+                    )
+                expected_summary = {
+                    "operation": "Flash Trade open position",
+                    "pool_name": pool_name.strip(),
+                    "market_symbol": market_symbol.strip(),
+                    "collateral_symbol": collateral_symbol.strip(),
+                    "collateral_amount_raw": collateral_amount_raw.strip(),
+                    "leverage": leverage.strip(),
+                    "side": side,
+                }
+                for key, expected_value in expected_summary.items():
+                    if approval_summary.get(key) != expected_value:
+                        raise WalletBackendError(
+                            "approval_token does not match the requested operation. Generate a new approval after previewing the exact action."
+                        )
+
+                approval_summary_copy = dict(approval_summary)
+                approved_preview = args.get("_approved_preview")
+                execute_preview = None
+                if isinstance(approval_summary_copy.get("_preview_digest"), str):
+                    if not isinstance(approved_preview, dict):
+                        raise WalletBackendError(
+                            "Approved Flash Trade preview payload is required for execute mode. Generate a new preview and approval before execute."
+                        )
+                    if preview_payload_digest(approved_preview) != approval_summary_copy["_preview_digest"]:
+                        raise WalletBackendError(
+                            "approved preview payload does not match the approval token. Generate a new preview and approval before execute."
+                        )
+                    preview_summary = self._build_confirmation_summary(
+                        action_label="Flash Trade open position",
+                        payload=approved_preview,
+                    )
+                    summary_without_digest = {
+                        key: value
+                        for key, value in approval_summary_copy.items()
+                        if key != "_preview_digest"
+                    }
+                    if preview_summary != summary_without_digest:
+                        raise WalletBackendError(
+                            "approved preview payload does not match the approval token. Generate a new preview and approval before execute."
+                        )
+                    execute_preview = dict(approved_preview)
+                else:
+                    execute_preview = await active_backend.preview_flash_trade_open_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        collateral_symbol=collateral_symbol.strip(),
+                        collateral_amount_raw=collateral_amount_raw.strip(),
+                        leverage=leverage.strip(),
+                        side=side,
+                    )
+
+                self._require_execute_approval(
+                    approval_token=approval_token,
+                    tool_name=tool_name,
+                    summary=approval_summary_copy,
+                    action_label="Flash Trade open position",
+                    backend=active_backend,
+                )
+                result = await active_backend.execute_flash_trade_open_position(
+                    pool_name=pool_name.strip(),
+                    market_symbol=market_symbol.strip(),
+                    collateral_symbol=collateral_symbol.strip(),
+                    collateral_amount_raw=collateral_amount_raw.strip(),
+                    leverage=leverage.strip(),
+                    side=side,
+                    approved_preview=execute_preview,
+                )
+                return AgentToolResult(
+                    tool=tool_name,
+                    ok=True,
+                    data=self._annotate_sensitive_payload(
+                        result,
+                        action_label="Flash Trade open position",
+                        mode="execute",
+                    ),
+                )
+
+            if tool_name == "flash_trade_close_position":
+                pool_name = args.get("pool_name")
+                market_symbol = args.get("market_symbol")
+                side = args.get("side")
+                mode = args.get("mode")
+                purpose = args.get("purpose")
+                user_intent = args.get("user_intent", False)
+                approval_token = args.get("approval_token")
+
+                if not isinstance(pool_name, str) or not pool_name.strip():
+                    raise WalletBackendError("pool_name is required.")
+                if not isinstance(market_symbol, str) or not market_symbol.strip():
+                    raise WalletBackendError("market_symbol is required.")
+                if mode not in {"preview", "prepare", "execute"}:
+                    raise WalletBackendError("mode must be 'preview', 'prepare' or 'execute'.")
+                if not isinstance(purpose, str) or not purpose.strip():
+                    raise WalletBackendError("purpose is required.")
+
+                if mode == "preview":
+                    preview = await active_backend.preview_flash_trade_close_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        side=side,
+                    )
+                    return AgentToolResult(
+                        tool=tool_name,
+                        ok=True,
+                        data=self._annotate_sensitive_payload(
+                            preview,
+                            action_label="Flash Trade close position",
+                            mode="preview",
+                        ),
+                    )
+
+                if mode == "prepare":
+                    self._require_prepare_intent(user_intent)
+                    prepared = await active_backend.prepare_flash_trade_close_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        side=side,
+                    )
+                    return AgentToolResult(
+                        tool=tool_name,
+                        ok=True,
+                        data=self._annotate_sensitive_payload(
+                            self._build_prepare_plan(
+                                preview_payload=prepared,
+                                action_label="Flash Trade close position",
+                            ),
+                            action_label="Flash Trade close position",
+                            mode="prepare",
+                        ),
+                    )
+
+                approval_payload = inspect_approval_token(
+                    approval_token,
+                    tool_name=tool_name,
+                    network=str(getattr(active_backend, "network", "unknown")),
+                    require_mainnet_confirmation=self._is_mainnet_for_backend(active_backend),
+                )
+                approval_summary = approval_payload.get("binding", {}).get("summary")
+                if not isinstance(approval_summary, dict):
+                    raise WalletBackendError(
+                        "approval_token does not match the requested operation. Generate a new approval after previewing the exact action."
+                    )
+                expected_summary = {
+                    "operation": "Flash Trade close position",
+                    "pool_name": pool_name.strip(),
+                    "market_symbol": market_symbol.strip(),
+                    "side": side,
+                }
+                for key, expected_value in expected_summary.items():
+                    if approval_summary.get(key) != expected_value:
+                        raise WalletBackendError(
+                            "approval_token does not match the requested operation. Generate a new approval after previewing the exact action."
+                        )
+
+                approval_summary_copy = dict(approval_summary)
+                approved_preview = args.get("_approved_preview")
+                execute_preview = None
+                if isinstance(approval_summary_copy.get("_preview_digest"), str):
+                    if not isinstance(approved_preview, dict):
+                        raise WalletBackendError(
+                            "Approved Flash Trade preview payload is required for execute mode. Generate a new preview and approval before execute."
+                        )
+                    if preview_payload_digest(approved_preview) != approval_summary_copy["_preview_digest"]:
+                        raise WalletBackendError(
+                            "approved preview payload does not match the approval token. Generate a new preview and approval before execute."
+                        )
+                    preview_summary = self._build_confirmation_summary(
+                        action_label="Flash Trade close position",
+                        payload=approved_preview,
+                    )
+                    summary_without_digest = {
+                        key: value
+                        for key, value in approval_summary_copy.items()
+                        if key != "_preview_digest"
+                    }
+                    if preview_summary != summary_without_digest:
+                        raise WalletBackendError(
+                            "approved preview payload does not match the approval token. Generate a new preview and approval before execute."
+                        )
+                    execute_preview = dict(approved_preview)
+                else:
+                    execute_preview = await active_backend.preview_flash_trade_close_position(
+                        pool_name=pool_name.strip(),
+                        market_symbol=market_symbol.strip(),
+                        side=side,
+                    )
+
+                self._require_execute_approval(
+                    approval_token=approval_token,
+                    tool_name=tool_name,
+                    summary=approval_summary_copy,
+                    action_label="Flash Trade close position",
+                    backend=active_backend,
+                )
+                result = await active_backend.execute_flash_trade_close_position(
+                    pool_name=pool_name.strip(),
+                    market_symbol=market_symbol.strip(),
+                    side=side,
+                    approved_preview=execute_preview,
+                )
+                return AgentToolResult(
+                    tool=tool_name,
+                    ok=True,
+                    data=self._annotate_sensitive_payload(
+                        result,
+                        action_label="Flash Trade close position",
+                        mode="execute",
+                    ),
+                )
+
             if tool_name == "get_kamino_lend_markets":
                 data = await self.backend.get_kamino_lend_markets()
                 return AgentToolResult(tool=tool_name, ok=True, data=data)