编译器
0.8.24+commit.e11b9ed9
文件 1 的 3:AIOracleCallbackReceiver.sol
pragma solidity ^0.8.9;
import "./interfaces/IAIOracle.sol";
abstract contract AIOracleCallbackReceiver {
IAIOracle public immutable aiOracle;
error UnauthorizedCallbackSource(IAIOracle expected, IAIOracle found);
constructor(IAIOracle _aiOracle) {
aiOracle = _aiOracle;
}
modifier onlyAIOracleCallback() {
IAIOracle foundRelayAddress = IAIOracle(msg.sender);
if (foundRelayAddress != aiOracle) {
revert UnauthorizedCallbackSource(aiOracle, foundRelayAddress);
}
_;
}
function aiOracleCallback(uint256 requestId, bytes calldata output, bytes calldata callbackData) external virtual;
function isFinalized(uint256 requestId) external view returns (bool) {
return aiOracle.isFinalized(requestId);
}
}
文件 2 的 3:IAIOracle.sol
pragma solidity ^0.8.9;
interface IAIOracle {
event AICallbackRequest(
address indexed account,
uint256 indexed requestId,
uint256 modelId,
bytes input,
address callbackContract,
uint64 gasLimit,
bytes callbackData
);
event AICallbackResult(
address indexed account,
uint256 indexed requestId,
address invoker,
bytes output
);
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);
}
文件 3 的 3:SimplePrompt.sol
pragma solidity ^0.8.9;
import "./interfaces/IAIOracle.sol";
import "./AIOracleCallbackReceiver.sol";
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");
_;
}
mapping(uint256 => AIOracleRequest) public requests;
mapping(uint256 => uint64) public callbackGasLimit;
constructor(IAIOracle _aiOracle) AIOracleCallbackReceiver(_aiOracle) {
owner = msg.sender;
callbackGasLimit[50] = 200_000;
callbackGasLimit[11] = 1_500_000;
callbackGasLimit[9] = 1_500_000;
}
function setCallbackGasLimit(uint256 modelId, uint64 gasLimit) external onlyOwner {
callbackGasLimit[modelId] = gasLimit;
}
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);
uint256 requestId = aiOracle.requestCallback{value: msg.value}(
modelId, input, address(this), callbackGasLimit[modelId], ""
);
emit promptRequest(requestId, msg.sender, modelId, prompt);
}
}
{
"compilationTarget": {
"contracts/SimplePrompt.sol": "SimplePrompt"
},
"evmVersion": "paris",
"libraries": {},
"metadata": {
"bytecodeHash": "ipfs"
},
"optimizer": {
"enabled": true,
"runs": 1000
},
"remappings": []
}
[{"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"}]