feat: cross chain transfer service

Author: andrew-virtuals

examples/acp_base/cross_chain_transfer_service/seller.py

@@ -0,0 +1,125 @@
+import logging
+import threading
+
+import sys
+sys.path.append("../../../")
+
+from typing import Optional
+from dotenv import load_dotenv
+
+from virtuals_acp.client import VirtualsACP
+from virtuals_acp.configs.configs import BASE_SEPOLIA_ACP_X402_CONFIG_V2
+from virtuals_acp.contract_clients.contract_client_v2 import ACPContractClientV2
+from virtuals_acp.env import EnvSettings
+from virtuals_acp.fare import Fare, FareAmount
+from virtuals_acp.job import ACPJob
+from virtuals_acp.memo import ACPMemo
+from virtuals_acp.models import ACPJobPhase, ChainConfig, MemoType
+
+# Configure logging
+logging.basicConfig(
+    level=logging.INFO,
+    format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
+)
+logger = logging.getLogger("SellerAgent")
+
+load_dotenv(override=True)
+
+REJECT_JOB_IN_REQUEST_PHASE = False
+REJECT_JOB_IN_OTHER_PHASE = False
+SOURCE_TOKEN_ADDRESS = ""
+TARGET_TOKEN_ADDRESS = ""
+TARGET_CHAIN_ID = 97
+
+config = BASE_SEPOLIA_ACP_X402_CONFIG_V2
+config.chains = [
+    ChainConfig(
+        chain_id=TARGET_CHAIN_ID,
+        rpc_url="https://bsc-testnet-dataseed.bnbchain.org"
+    )
+]
+
+def seller():
+    env = EnvSettings()
+
+    def on_new_task(job: ACPJob, memo_to_sign: Optional[ACPMemo] = None):
+        logger.info(f"[on_new_task] Received job {job.id} (phase: {job.phase})")
+
+        if (
+            job.phase == ACPJobPhase.REQUEST
+            and memo_to_sign is not None
+            and memo_to_sign.next_phase == ACPJobPhase.NEGOTIATION
+        ):
+            logger.info(f"Responding to job {job.id} with requirement: {job.requirement}")
+            if REJECT_JOB_IN_REQUEST_PHASE:
+                job.reject("Job requirement does not meet agent capability")
+            else:
+                job.accept("Job requirement matches agent capability")
+
+                swappedToken = FareAmount(
+                    1,
+                    Fare.from_contract_address(
+                        TARGET_TOKEN_ADDRESS,
+                        config,
+                        TARGET_CHAIN_ID
+                    )
+                )
+
+                job.create_payable_requirement(
+                    "Requesting token from client on destination chain",
+                    MemoType.PAYABLE_REQUEST,
+                    swappedToken,
+                    job.client_address,
+                )
+
+            logger.info(f"Job {job.id} responded with {'rejected' if REJECT_JOB_IN_REQUEST_PHASE else 'accepted'}")
+
+        elif (
+            job.phase == ACPJobPhase.TRANSACTION
+            and memo_to_sign is not None
+            and memo_to_sign.next_phase == ACPJobPhase.EVALUATION
+        ):
+            # to cater cases where agent decide to reject job after payment has been made
+            if REJECT_JOB_IN_OTHER_PHASE:  # conditional check for job rejection logic
+                reason = "Job requirement does not meet agent capability"
+                logger.info(f"Rejecting job {job.id} with reason: {reason}")
+                job.reject(reason)
+                logger.info(f"Job {job.id} rejected")
+                return
+
+            deliverable = FareAmount(
+                1,
+                Fare.from_contract_address(
+                    TARGET_TOKEN_ADDRESS,
+                    config,
+                    TARGET_CHAIN_ID
+                )
+            )
+            logger.info(f"Delivering job {job.id} with deliverable {deliverable}")
+            # job.deliver(deliverable)
+            job.deliver_payable("Deliver swapped token on destination chain", deliverable)
+            logger.info(f"Job {job.id} delivered")
+            return
+
+        elif job.phase == ACPJobPhase.COMPLETED:
+            logger.info(f"Job {job.id} completed")
+
+        elif job.phase == ACPJobPhase.REJECTED:
+            logger.info(f"Job {job.id} rejected")
+
+    VirtualsACP(
+        acp_contract_clients=ACPContractClientV2(
+            wallet_private_key=env.WHITELISTED_WALLET_PRIVATE_KEY,
+            agent_wallet_address=env.SELLER_AGENT_WALLET_ADDRESS,
+            entity_id=env.SELLER_ENTITY_ID,
+            config=config
+        ),
+        on_new_task=on_new_task
+    )
+
+    logger.info("Seller agent is running, waiting for new tasks...")
+    threading.Event().wait()
+
+
+if __name__ == "__main__":
+    seller()