Flap AI Oracle

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.13;

import {FlapAIConsumerBase, IFlapAIProvider}
    from "src/plugins/AIProvider/IFlapAIProvider.sol";

contract MarketMakingVault is FlapAIConsumerBase {
    address public immutable TOKEN = 0x00000000000000000000000000000000000077777;

    uint256 private _lastRequestId;
    uint256 private _lastReasoningBlock;

    // Rate-limit: at most one AI call per ~1 200 blocks
    uint256 constant REASON_COOLDOWN = 1200;

    function lastRequestId() public view override returns (uint256) {
        return _lastRequestId;
    }

    function _shouldReason() internal view returns (bool) {
        if (_lastRequestId != 0) return false;       // request pending
        if (block.number < _lastReasoningBlock + REASON_COOLDOWN) return false;
        return true;
    }

    // Triggered on every incoming BNB; ask AI only when cooldown passed
    receive() external payable {
        if (!_shouldReason()) return;

        IFlapAIProvider p = IFlapAIProvider(_getFlapAIProvider());
        uint256 fee = p.getModel(0).price; // gemini-flash

        // Prompt built entirely on-chain — AI will call ave_token_info
        // to fetch live market data before deciding
        string memory prompt = string.concat(
            "I am managing a vault for token ",
            _toHexString(TOKEN),
            ", my main goal is to use my fund to do market making for this token. "
            "Check the market data (use ave_token_info) of this token and then "
            "decide what to do: "
            "(0) buy tokens to support the floor "
            "(1) sell my holdings to gain more funds for future market making "
            "(2) generate volumes only (i.e: buy and then sell immediately). "
            "You must choose exactly one option. "
            "Reply with a single integer: 0 to buy, 1 to sell, 2 to generate volumes."
        );

        _lastRequestId      = p.reason{value: fee}(0, prompt, 3);
        _lastReasoningBlock = block.number;
    }

    // AI decision arrives here after ave_token_info tool call + reasoning
    function _fulfillReasoning(uint256 id, uint8 choice)
        internal override
    {
        require(id == _lastRequestId);
        _lastRequestId = 0;
        if      (choice == 0) _buyTokens();          // support the floor
        else if (choice == 1) _sellHoldings();       // accumulate BNB
        else if (choice == 2) _generateVolume();     // buy then sell
    }

    function _onFlapAIRequestRefunded(uint256 id) internal override {
        require(id == _lastRequestId);
        _lastRequestId = 0; // cooldown still applies; will retry later
    }
}