# { "Depends": "py-genlayer:test" } import json from dataclasses import dataclass from genlayer import * ERROR_EXPECTED = "[EXPECTED]" ERROR_LLM = "[LLM_ERROR]" @allow_storage @dataclass class Intent: intent_id: u256 submitter: Address intent_text: str parsed_json: str status: str token_in: str token_out: str amount: str slippage: str chain: str def _parse_intent_prompt(intent_text: str) -> str: return f"""You are IntentSwap, an AI-powered natural language DEX. Parse this trading intent into structured parameters. User Intent: \"\"\"{intent_text}\"\"\" Parse the natural language into structured trade parameters. Handle variations like: - "swap 1 ETH for USDC" -> token_in: ETH, token_out: USDC, amount: 1 - "buy $500 worth of SOL" -> token_in: USD, token_out: SOL, amount: 500 - "sell half my LINK for ETH on Arbitrum" -> token_in: LINK, token_out: ETH, amount: 50%, chain: arbitrum Return ONLY valid JSON: {{ "token_in": "", "token_out": "", "amount": "", "slippage": "", "chain": "", "order_type": "market|limit", "limit_price": "", "urgency": "low|medium|high", "confidence": <0-100 how confident the parse is>, "warnings": [""], "human_readable": "" }} """ class IntentSwap(gl.Contract): owner: Address intent_count: u256 intents: TreeMap[u256, Intent] intent_ids: DynArray[u256] def __init__(self): self.owner = gl.message.sender_address self.intent_count = u256(0) def _parse_intent(self, intent_text: str) -> dict: prompt = _parse_intent_prompt(intent_text) 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 parse result") return result def validator_fn(leaders_res): if not isinstance(leaders_res, gl.vm.Return): return False v = leader_fn() leader = leaders_res.calldata if leader.get("token_in", "").upper() != v.get("token_in", "").upper(): return False if leader.get("token_out", "").upper() != v.get("token_out", "").upper(): return False return True return gl.vm.run_nondet_unsafe(leader_fn, validator_fn) @gl.public.write def submit_intent(self, intent_text: str) -> u256: cleaned = intent_text.strip() if len(cleaned) < 5: raise gl.vm.UserError(f"{ERROR_EXPECTED} Intent too short") if len(cleaned) > 1000: raise gl.vm.UserError(f"{ERROR_EXPECTED} Intent too long") parsed = self._parse_intent(cleaned) intent_id = self.intent_count self.intents[intent_id] = Intent( intent_id=intent_id, submitter=gl.message.sender_address, intent_text=cleaned, parsed_json=json.dumps(parsed, sort_keys=True), status="PARSED", token_in=str(parsed.get("token_in", "")).upper(), token_out=str(parsed.get("token_out", "")).upper(), amount=str(parsed.get("amount", "0")), slippage=str(parsed.get("slippage", "0.5%")), chain=str(parsed.get("chain", "ethereum")).lower(), ) self.intent_ids.append(intent_id) self.intent_count = intent_id + 1 return intent_id @gl.public.view def get_intent(self, id: u256) -> dict: if id not in self.intents: raise gl.vm.UserError(f"{ERROR_EXPECTED} Intent not found") i = self.intents[id] return { "intent_id": i.intent_id, "submitter": i.submitter.as_hex, "intent_text": i.intent_text, "parsed": json.loads(i.parsed_json), "status": i.status, "token_in": i.token_in, "token_out": i.token_out, "amount": i.amount, "slippage": i.slippage, "chain": i.chain, } @gl.public.view def get_all_intents(self) -> list: results = [] count = int(self.intent_count) i = count - 1 while i >= 0 and len(results) < 50: intent = self.intents[i] results.append({ "intent_id": intent.intent_id, "submitter": intent.submitter.as_hex, "intent_text": intent.intent_text[:80] + "..." if len(intent.intent_text) > 80 else intent.intent_text, "status": intent.status, "token_in": intent.token_in, "token_out": intent.token_out, "chain": intent.chain, }) i -= 1 return results