Contract Source Code (Solidity Standard Json-Input format)

IDE

• Blockscan IDE
• 🤖 Code ReaderBeta

More Options

• Similar
• Submit Audit
• Compare

// SampleContract.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.9;

import "./interfaces/IAIOracle.sol";
import "./AIOracleCallbackReceiver.sol";

// this contract is for ai.ora.io website
contract SimplePrompt is AIOracleCallbackReceiver {

    event promptsUpdated(
        uint256 requestId,
        string output,
        bytes callbackData
    );

    event promptRequest(
        uint256 requestId,
        address sender, 
        uint256 modelId,
        string prompt
    );

    struct AIOracleRequest {
        address sender;
        uint256 modelId;
        bytes input;
        bytes output;
    }

    address immutable owner;

    modifier onlyOwner() {
        require(msg.sender == owner, "Only owner");
        _;
    }

    // requestId => AIOracleRequest
    mapping(uint256 => AIOracleRequest) public requests;

    // modelId => callback gasLimit
    mapping(uint256 => uint64) public callbackGasLimit;

    /// @notice Initialize the contract, binding it to a specified AIOracle.
    constructor(IAIOracle _aiOracle) AIOracleCallbackReceiver(_aiOracle) {
        owner = msg.sender;
        callbackGasLimit[50] = 200_000; // SD 500k
        callbackGasLimit[11] = 1_500_000; // llama
        callbackGasLimit[9] = 1_500_000; // grok
    }

    function setCallbackGasLimit(uint256 modelId, uint64 gasLimit) external onlyOwner {
        callbackGasLimit[modelId] = gasLimit;
    }

    // the callback function, only the AI Oracle can call this function
    function aiOracleCallback(uint256 requestId, bytes calldata output, bytes calldata callbackData) external override onlyAIOracleCallback() {
        emit promptsUpdated(requestId, string(output), callbackData);
    }

    function estimateFee(uint256 modelId) public view returns (uint256) {
        return aiOracle.estimateFee(modelId, callbackGasLimit[modelId]);
    }

    function calculateAIResult(uint256 modelId, string calldata prompt) payable external {
        bytes memory input = bytes(prompt);
        // we do not need to set the callbackData in this example
        uint256 requestId = aiOracle.requestCallback{value: msg.value}(
            modelId, input, address(this), callbackGasLimit[modelId], ""
        );
        emit promptRequest(requestId, msg.sender, modelId, prompt);
    }
}

// SampleContract.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.9;

import "./interfaces/IAIOracle.sol";

/// @notice A base contract for writing a AIOracle app
abstract contract AIOracleCallbackReceiver {

    // Address of the AIOracle contract
    IAIOracle public immutable aiOracle;

    // Invalid callback source error
    error UnauthorizedCallbackSource(IAIOracle expected, IAIOracle found);

    /// @notice Initialize the contract, binding it to a specified AIOracle contract
    constructor(IAIOracle _aiOracle) {
        aiOracle = _aiOracle;
    }

    /// @notice Verify this is a callback by the aiOracle contract 
    modifier onlyAIOracleCallback() {
        IAIOracle foundRelayAddress = IAIOracle(msg.sender);
        if (foundRelayAddress != aiOracle) {
            revert UnauthorizedCallbackSource(aiOracle, foundRelayAddress);
        }
        _;
    }

    /**
     * @dev the callback function in OAO, should add the modifier onlyAIOracleCallback!
     * @param requestId Id for the request in OAO (unique per request)
     * @param output AI model's output
     * @param callbackData user-defined data (The same as when the user call aiOracle.requestCallback)
     */
    function aiOracleCallback(uint256 requestId, bytes calldata output, bytes calldata callbackData) external virtual;

    function isFinalized(uint256 requestId) external view returns (bool) {
        return aiOracle.isFinalized(requestId);
    }
}

// SampleContract.sol
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.9;

interface IAIOracle {
    /// @notice Event emitted upon receiving a callback request through requestCallback.
    event AICallbackRequest(
        address indexed account,
        uint256 indexed requestId,
        uint256 modelId,
        bytes input,
        address callbackContract,
        uint64 gasLimit,
        bytes callbackData
    );

    /// @notice Event emitted when the result is uploaded or update.
    event AICallbackResult(
        address indexed account,
        uint256 indexed requestId,
        address invoker,
        bytes output
    );

    /**
     * initiate a request in OAO
     * @param modelId ID for AI model
     * @param input input for AI model
     * @param callbackContract address of callback contract
     * @param gasLimit gas limitation of calling the callback function
     * @param callbackData optional, user-defined data, will send back to the callback function
     * @return requestID
     */
    function requestCallback(
        uint256 modelId,
        bytes memory input,
        address callbackContract,
        uint64 gasLimit,
        bytes memory callbackData
    ) external payable returns (uint256);

    function estimateFee(uint256 modelId, uint256 gasLimit) external view returns (uint256);

    function isFinalized(uint256 requestId) external view returns (bool);
}

{
  "optimizer": {
    "enabled": true,
    "runs": 1000
  },
  "evmVersion": "paris",
  "outputSelection": {
    "*": {
      "*": [
        "evm.bytecode",
        "evm.deployedBytecode",
        "devdoc",
        "userdoc",
        "metadata",
        "abi"
      ]
    }
  },
  "libraries": {}
}

• No Contract Security Audit Submitted- Submit Audit Here

[{"inputs":[{"internalType":"contract IAIOracle","name":"_aiOracle","type":"address"}],"stateMutability":"nonpayable","type":"constructor"},{"inputs":[{"internalType":"contract IAIOracle","name":"expected","type":"address"},{"internalType":"contract IAIOracle","name":"found","type":"address"}],"name":"UnauthorizedCallbackSource","type":"error"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"requestId","type":"uint256"},{"indexed":false,"internalType":"address","name":"sender","type":"address"},{"indexed":false,"internalType":"uint256","name":"modelId","type":"uint256"},{"indexed":false,"internalType":"string","name":"prompt","type":"string"}],"name":"promptRequest","type":"event"},{"anonymous":false,"inputs":[{"indexed":false,"internalType":"uint256","name":"requestId","type":"uint256"},{"indexed":false,"internalType":"string","name":"output","type":"string"},{"indexed":false,"internalType":"bytes","name":"callbackData","type":"bytes"}],"name":"promptsUpdated","type":"event"},{"inputs":[],"name":"aiOracle","outputs":[{"internalType":"contract IAIOracle","name":"","type":"address"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"requestId","type":"uint256"},{"internalType":"bytes","name":"output","type":"bytes"},{"internalType":"bytes","name":"callbackData","type":"bytes"}],"name":"aiOracleCallback","outputs":[],"stateMutability":"nonpayable","type":"function"},{"inputs":[{"internalType":"uint256","name":"modelId","type":"uint256"},{"internalType":"string","name":"prompt","type":"string"}],"name":"calculateAIResult","outputs":[],"stateMutability":"payable","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"}],"name":"callbackGasLimit","outputs":[{"internalType":"uint64","name":"","type":"uint64"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"modelId","type":"uint256"}],"name":"estimateFee","outputs":[{"internalType":"uint256","name":"","type":"uint256"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"requestId","type":"uint256"}],"name":"isFinalized","outputs":[{"internalType":"bool","name":"","type":"bool"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"","type":"uint256"}],"name":"requests","outputs":[{"internalType":"address","name":"sender","type":"address"},{"internalType":"uint256","name":"modelId","type":"uint256"},{"internalType":"bytes","name":"input","type":"bytes"},{"internalType":"bytes","name":"output","type":"bytes"}],"stateMutability":"view","type":"function"},{"inputs":[{"internalType":"uint256","name":"modelId","type":"uint256"},{"internalType":"uint64","name":"gasLimit","type":"uint64"}],"name":"setCallbackGasLimit","outputs":[],"stateMutability":"nonpayable","type":"function"}]

60c060405234801561001057600080fd5b50604051610c96380380610c9683398101604081905261002f916100dd565b6001600160a01b03166080523360a05260016020527ff7459e621ec1fc60fd62c436d4281de72cb9f48787dc9c8288a3050292b0900480546001600160401b031990811662030d40179091557ffc80cd5fe514767bc6e66ec558e68a5429ea70b50fa6caa3b53fc9278e918632805482166216e36090811790915560096000527f74a5fbcb419ab7dbacbb2c92a4e163730f0da5c72b911deecf4f05a6b327d0a4805490921617905561010d565b6000602082840312156100ef57600080fd5b81516001600160a01b038116811461010657600080fd5b9392505050565b60805160a051610b41610155600039600061037a01526000818161011501528181610252015281816102fc01528181610584015281816105da01526106cc0152610b416000f3fe60806040526004361061007b5760003560e01c8063813d1e151161004e578063813d1e151461017f57806381d12c58146101a1578063b0347814146101d1578063fac5d7e1146101f157600080fd5b8063127e8e4d1461008057806320f38718146100b357806331b221cd1461010357806333727c4d1461014f575b600080fd5b34801561008c57600080fd5b506100a061009b3660046107a2565b610204565b6040519081526020015b60405180910390f35b3480156100bf57600080fd5b506100ea6100ce3660046107a2565b60016020526000908152604090205467ffffffffffffffff1681565b60405167ffffffffffffffff90911681526020016100aa565b34801561010f57600080fd5b506101377f000000000000000000000000000000000000000000000000000000000000000081565b6040516001600160a01b0390911681526020016100aa565b34801561015b57600080fd5b5061016f61016a3660046107a2565b6102ca565b60405190151581526020016100aa565b34801561018b57600080fd5b5061019f61019a3660046107bb565b61036f565b005b3480156101ad57600080fd5b506101c16101bc3660046107a2565b610436565b6040516100aa949392919061083e565b3480156101dd57600080fd5b5061019f6101ec3660046108cc565b610579565b61019f6101ff366004610946565b610656565b6000818152600160205260408082205490517ee1d8d00000000000000000000000000000000000000000000000000000000081526004810184905267ffffffffffffffff90911660248201527f00000000000000000000000000000000000000000000000000000000000000006001600160a01b03169062e1d8d090604401602060405180830381865afa1580156102a0573d6000803e3d6000fd5b505050506040513d601f19601f820116820180604052508101906102c49190610992565b92915050565b6040517f33727c4d000000000000000000000000000000000000000000000000000000008152600481018290526000907f00000000000000000000000000000000000000000000000000000000000000006001600160a01b0316906333727c4d90602401602060405180830381865afa15801561034b573d6000803e3d6000fd5b505050506040513d601f19601f820116820180604052508101906102c491906109ab565b336001600160a01b037f00000000000000000000000000000000000000000000000000000000000000001614610406576040517f08c379a000000000000000000000000000000000000000000000000000000000815260206004820152600a60248201527f4f6e6c79206f776e65720000000000000000000000000000000000000000000060448201526064015b60405180910390fd5b600091825260016020526040909120805467ffffffffffffffff191667ffffffffffffffff909216919091179055565b6000602081905290815260409020805460018201546002830180546001600160a01b03909316939192610468906109d4565b80601f0160208091040260200160405190810160405280929190818152602001828054610494906109d4565b80156104e15780601f106104b6576101008083540402835291602001916104e1565b820191906000526020600020905b8154815290600101906020018083116104c457829003601f168201915b5050505050908060030180546104f6906109d4565b80601f0160208091040260200160405190810160405280929190818152602001828054610522906109d4565b801561056f5780601f106105445761010080835404028352916020019161056f565b820191906000526020600020905b81548152906001019060200180831161055257829003601f168201915b5050505050905084565b336001600160a01b037f000000000000000000000000000000000000000000000000000000000000000016811461060f576040517f865c066e0000000000000000000000000000000000000000000000000000000081526001600160a01b037f000000000000000000000000000000000000000000000000000000000000000081166004830152821660248201526044016103fd565b7f06a0e7c58dc0deb40e0fe5d0b6f113abd3b70be188511800ff197ef9b3d9bdf98686868686604051610646959493929190610a50565b60405180910390a1505050505050565b600082828080601f0160208091040260200160405190810160405280939291908181526020018383808284376000920182905250888152600160205260408082205490517f5b95380600000000000000000000000000000000000000000000000000000000815295965090946001600160a01b037f0000000000000000000000000000000000000000000000000000000000000000169450635b9538069350349250610717918a918891309167ffffffffffffffff90911690600401610a89565b60206040518083038185885af1158015610735573d6000803e3d6000fd5b50505050506040513d601f19601f8201168201806040525081019061075a9190610992565b90507fa0faead83d70148ae18b694377f9bef079251342ab90e14af0f9ef68b891269f8133878787604051610793959493929190610adc565b60405180910390a15050505050565b6000602082840312156107b457600080fd5b5035919050565b600080604083850312156107ce57600080fd5b82359150602083013567ffffffffffffffff811681146107ed57600080fd5b809150509250929050565b6000815180845260005b8181101561081e57602081850181015186830182015201610802565b506000602082860101526020601f19601f83011685010191505092915050565b6001600160a01b038516815283602082015260806040820152600061086660808301856107f8565b828103606084015261087881856107f8565b979650505050505050565b60008083601f84011261089557600080fd5b50813567ffffffffffffffff8111156108ad57600080fd5b6020830191508360208285010111156108c557600080fd5b9250929050565b6000806000806000606086880312156108e457600080fd5b85359450602086013567ffffffffffffffff8082111561090357600080fd5b61090f89838a01610883565b9096509450604088013591508082111561092857600080fd5b5061093588828901610883565b969995985093965092949392505050565b60008060006040848603121561095b57600080fd5b83359250602084013567ffffffffffffffff81111561097957600080fd5b61098586828701610883565b9497909650939450505050565b6000602082840312156109a457600080fd5b5051919050565b6000602082840312156109bd57600080fd5b815180151581146109cd57600080fd5b9392505050565b600181811c908216806109e857607f821691505b602082108103610a21577f4e487b7100000000000000000000000000000000000000000000000000000000600052602260045260246000fd5b50919050565b81835281816020850137506000828201602090810191909152601f909101601f19169091010190565b858152606060208201526000610a6a606083018688610a27565b8281036040840152610a7d818587610a27565b98975050505050505050565b84815260a060208201526000610aa260a08301866107f8565b6001600160a01b039490941660408301525067ffffffffffffffff9190911660608201528082036080909101526000815260200192915050565b8581526001600160a01b0385166020820152836040820152608060608201526000610878608083018486610a2756fea2646970667358221220faa52b47697932f166309d65ea86a360dcfaa4ed66cfb68938dd27e7e1dc0c6a64736f6c634300081800330000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a0

0x60806040526004361061007b5760003560e01c8063813d1e151161004e578063813d1e151461017f57806381d12c58146101a1578063b0347814146101d1578063fac5d7e1146101f157600080fd5b8063127e8e4d1461008057806320f38718146100b357806331b221cd1461010357806333727c4d1461014f575b600080fd5b34801561008c57600080fd5b506100a061009b3660046107a2565b610204565b6040519081526020015b60405180910390f35b3480156100bf57600080fd5b506100ea6100ce3660046107a2565b60016020526000908152604090205467ffffffffffffffff1681565b60405167ffffffffffffffff90911681526020016100aa565b34801561010f57600080fd5b506101377f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a081565b6040516001600160a01b0390911681526020016100aa565b34801561015b57600080fd5b5061016f61016a3660046107a2565b6102ca565b60405190151581526020016100aa565b34801561018b57600080fd5b5061019f61019a3660046107bb565b61036f565b005b3480156101ad57600080fd5b506101c16101bc3660046107a2565b610436565b6040516100aa949392919061083e565b3480156101dd57600080fd5b5061019f6101ec3660046108cc565b610579565b61019f6101ff366004610946565b610656565b6000818152600160205260408082205490517ee1d8d00000000000000000000000000000000000000000000000000000000081526004810184905267ffffffffffffffff90911660248201527f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a06001600160a01b03169062e1d8d090604401602060405180830381865afa1580156102a0573d6000803e3d6000fd5b505050506040513d601f19601f820116820180604052508101906102c49190610992565b92915050565b6040517f33727c4d000000000000000000000000000000000000000000000000000000008152600481018290526000907f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a06001600160a01b0316906333727c4d90602401602060405180830381865afa15801561034b573d6000803e3d6000fd5b505050506040513d601f19601f820116820180604052508101906102c491906109ab565b336001600160a01b037f000000000000000000000000f5aeb5a4b35be7af7dbfdb765f99bcf479c917bd1614610406576040517f08c379a000000000000000000000000000000000000000000000000000000000815260206004820152600a60248201527f4f6e6c79206f776e65720000000000000000000000000000000000000000000060448201526064015b60405180910390fd5b600091825260016020526040909120805467ffffffffffffffff191667ffffffffffffffff909216919091179055565b6000602081905290815260409020805460018201546002830180546001600160a01b03909316939192610468906109d4565b80601f0160208091040260200160405190810160405280929190818152602001828054610494906109d4565b80156104e15780601f106104b6576101008083540402835291602001916104e1565b820191906000526020600020905b8154815290600101906020018083116104c457829003601f168201915b5050505050908060030180546104f6906109d4565b80601f0160208091040260200160405190810160405280929190818152602001828054610522906109d4565b801561056f5780601f106105445761010080835404028352916020019161056f565b820191906000526020600020905b81548152906001019060200180831161055257829003601f168201915b5050505050905084565b336001600160a01b037f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a016811461060f576040517f865c066e0000000000000000000000000000000000000000000000000000000081526001600160a01b037f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a081166004830152821660248201526044016103fd565b7f06a0e7c58dc0deb40e0fe5d0b6f113abd3b70be188511800ff197ef9b3d9bdf98686868686604051610646959493929190610a50565b60405180910390a1505050505050565b600082828080601f0160208091040260200160405190810160405280939291908181526020018383808284376000920182905250888152600160205260408082205490517f5b95380600000000000000000000000000000000000000000000000000000000815295965090946001600160a01b037f0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a0169450635b9538069350349250610717918a918891309167ffffffffffffffff90911690600401610a89565b60206040518083038185885af1158015610735573d6000803e3d6000fd5b50505050506040513d601f19601f8201168201806040525081019061075a9190610992565b90507fa0faead83d70148ae18b694377f9bef079251342ab90e14af0f9ef68b891269f8133878787604051610793959493929190610adc565b60405180910390a15050505050565b6000602082840312156107b457600080fd5b5035919050565b600080604083850312156107ce57600080fd5b82359150602083013567ffffffffffffffff811681146107ed57600080fd5b809150509250929050565b6000815180845260005b8181101561081e57602081850181015186830182015201610802565b506000602082860101526020601f19601f83011685010191505092915050565b6001600160a01b038516815283602082015260806040820152600061086660808301856107f8565b828103606084015261087881856107f8565b979650505050505050565b60008083601f84011261089557600080fd5b50813567ffffffffffffffff8111156108ad57600080fd5b6020830191508360208285010111156108c557600080fd5b9250929050565b6000806000806000606086880312156108e457600080fd5b85359450602086013567ffffffffffffffff8082111561090357600080fd5b61090f89838a01610883565b9096509450604088013591508082111561092857600080fd5b5061093588828901610883565b969995985093965092949392505050565b60008060006040848603121561095b57600080fd5b83359250602084013567ffffffffffffffff81111561097957600080fd5b61098586828701610883565b9497909650939450505050565b6000602082840312156109a457600080fd5b5051919050565b6000602082840312156109bd57600080fd5b815180151581146109cd57600080fd5b9392505050565b600181811c908216806109e857607f821691505b602082108103610a21577f4e487b7100000000000000000000000000000000000000000000000000000000600052602260045260246000fd5b50919050565b81835281816020850137506000828201602090810191909152601f909101601f19169091010190565b858152606060208201526000610a6a606083018688610a27565b8281036040840152610a7d818587610a27565b98975050505050505050565b84815260a060208201526000610aa260a08301866107f8565b6001600160a01b039490941660408301525067ffffffffffffffff9190911660608201528082036080909101526000815260200192915050565b8581526001600160a01b0385166020820152836040820152608060608201526000610878608083018486610a2756fea2646970667358221220faa52b47697932f166309d65ea86a360dcfaa4ed66cfb68938dd27e7e1dc0c6a64736f6c63430008180033

0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a0

-----Decoded View---------------
Arg [0] : _aiOracle (address): 0x0A0f4321214BB6C7811dD8a71cF587bdaF03f0A0

-----Encoded View---------------
1 Constructor Arguments found :
Arg [0] : 0000000000000000000000000a0f4321214bb6c7811dd8a71cf587bdaf03f0a0