# { "Depends": "py-genlayer:test" } import json from dataclasses import dataclass from genlayer import * ERROR_EXPECTED = "[EXPECTED]" ERROR_LLM = "[LLM_ERROR]" @allow_storage @dataclass class Exploration: exploration_id: u256 submitter: Address chain: str address: str question: str explanation: str functions_found_json: str complexity_rating: str def _explore_prompt(chain: str, address: str, question: str) -> str: return f"""You are ContractWhisperer, an AI smart contract explorer that explains contracts in plain English. Chain: {chain} Contract Address: {address} User Question: \"\"\"{question}\"\"\" Based on your knowledge of common smart contract patterns and the specified chain: 1. Identify what type of contract this likely is (ERC-20, ERC-721, DEX, lending, etc.) 2. Explain the contract's purpose and functionality in plain English 3. Answer the user's specific question 4. List the key functions and what they do 5. Identify any notable patterns or risks Return ONLY valid JSON: {{ "explanation": "", "contract_type": "", "purpose": "", "functions": [ {{"name": "", "description": "", "access": "public|owner|internal"}} ], "complexity_rating": "SIMPLE|MODERATE|COMPLEX|ADVANCED", "risks": ["", ""], "interesting_patterns": ["", ""] }} """ class ContractWhisperer(gl.Contract): owner: Address exploration_count: u256 explorations: TreeMap[u256, Exploration] exploration_ids: DynArray[u256] def __init__(self): self.owner = gl.message.sender_address self.exploration_count = u256(0) def _analyze_contract(self, chain: str, address: str, question: str) -> dict: prompt = _explore_prompt(chain, address, question) def leader_fn(): result = gl.nondet.exec_prompt(prompt, response_format="json") if not isinstance(result, dict): raise gl.vm.UserError(f"{ERROR_LLM} Invalid response") return result def validator_fn(leaders_res): if not isinstance(leaders_res, gl.vm.Return): return False v = leader_fn() return leaders_res.calldata.get("contract_type", "").lower() == v.get("contract_type", "").lower() return gl.vm.run_nondet_unsafe(leader_fn, validator_fn) @gl.public.write def explore_contract(self, chain: str, address: str, question: str) -> u256: if len(address.strip()) < 10: raise gl.vm.UserError(f"{ERROR_EXPECTED} Invalid contract address") if len(question.strip()) < 5: raise gl.vm.UserError(f"{ERROR_EXPECTED} Question too short") analysis = self._analyze_contract(chain.strip(), address.strip(), question.strip()) exp_id = self.exploration_count functions_json = json.dumps(analysis.get("functions", []), sort_keys=True) self.explorations[exp_id] = Exploration( exploration_id=exp_id, submitter=gl.message.sender_address, chain=chain.strip(), address=address.strip(), question=question.strip(), explanation=str(analysis.get("explanation", ""))[:2000], functions_found_json=functions_json, complexity_rating=str(analysis.get("complexity_rating", "UNKNOWN")), ) self.exploration_ids.append(exp_id) self.exploration_count = exp_id + 1 return exp_id @gl.public.view def get_exploration(self, id: u256) -> dict: if id not in self.explorations: raise gl.vm.UserError(f"{ERROR_EXPECTED} Exploration not found") e = self.explorations[id] return { "exploration_id": e.exploration_id, "submitter": e.submitter.as_hex, "chain": e.chain, "address": e.address, "question": e.question, "explanation": e.explanation, "functions": json.loads(e.functions_found_json), "complexity_rating": e.complexity_rating, } @gl.public.view def get_all_explorations(self) -> list: results = [] count = int(self.exploration_count) i = count - 1 while i >= 0 and len(results) < 50: e = self.explorations[i] results.append({ "exploration_id": e.exploration_id, "submitter": e.submitter.as_hex, "chain": e.chain, "address": e.address, "question": e.question[:80] + "..." if len(e.question) > 80 else e.question, "complexity_rating": e.complexity_rating, }) i -= 1 return results