support llm precompile

app/app.go

@@ -158,6 +158,7 @@ import (
 
 	memiavlstore "github.com/crypto-org-chain/cronos/store"
 	"github.com/crypto-org-chain/cronos/v2/client/docs"
+	"github.com/crypto-org-chain/cronos/v2/llm"
 	"github.com/crypto-org-chain/cronos/v2/x/cronos"
 	cronosclient "github.com/crypto-org-chain/cronos/v2/x/cronos/client"
 	cronoskeeper "github.com/crypto-org-chain/cronos/v2/x/cronos/keeper"
@@ -685,6 +686,9 @@ func New(
 			func(ctx sdk.Context, rules ethparams.Rules) vm.PrecompiledContract {
 				return cronosprecompiles.NewIcaContract(ctx, app.ICAControllerKeeper, &app.CronosKeeper, appCodec, gasConfig)
 			},
+			func(ctx sdk.Context, rules ethparams.Rules) vm.PrecompiledContract {
+				return cronosprecompiles.NewLLamaContract(gasConfig, llm.NewBuiltinModel())
+			},
 		},
 	)
 

integration_tests/contracts/contracts/TestLLama.sol

@@ -0,0 +1,13 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.4;
+
+import {ILLamaModule} from "./src/LLama.sol";
+
+contract TestLLama {
+    address constant llamaContract = 0x0000000000000000000000000000000000000067;
+    ILLamaModule llama = ILLamaModule(llamaContract);
+
+    function inference(string calldata prompt, int64 seed, int32 steps) public returns (string memory) {
+        return llama.inference(prompt, seed, steps);
+    }
+}

integration_tests/test_llama.py

@@ -0,0 +1,14 @@
+from .utils import ADDRS, CONTRACTS, deploy_contract, send_transaction
+
+
+def test_call(cronos):
+    w3 = cronos.w3
+    addr = ADDRS["validator"]
+    contract = deploy_contract(w3, CONTRACTS["TestLLama"])
+    prompt = ""
+    seed = 2
+    steps = 256
+    data = {"from": addr, "gasPrice": w3.eth.gas_price, "gas": 600000}
+    tx = contract.functions.inference(prompt, seed, steps).build_transaction(data)
+    receipt = send_transaction(w3, tx)
+    assert receipt.status == 1

integration_tests/utils.py

@@ -62,6 +62,7 @@
     "TestICA": "TestICA.sol",
     "Random": "Random.sol",
     "TestRelayer": "TestRelayer.sol",
+    "TestLLama": "TestLLama.sol",
 }