Merge pull request #194 from veithly/fix/x420-demo

Enhance X402ReactAgent with default URL handling and prompt management

Author: moonpyt

examples/x402_agent_demo.py

@@ -25,6 +25,8 @@
 from spoon_ai.chat import ChatBot  # noqa: E402
 from spoon_ai.schema import Message, Role  # noqa: E402
 from spoon_ai.payments import X402PaymentReceipt, X402PaymentService  # noqa: E402
+from pydantic import Field  # noqa: E402
+from spoon_ai.prompts.spoon_react import NEXT_STEP_PROMPT_TEMPLATE  # noqa: E402
 from spoon_ai.tools.base import BaseTool, ToolResult  # noqa: E402
 from spoon_ai.tools.tool_manager import ToolManager  # noqa: E402
 from spoon_ai.tools.x402_payment import X402PaywalledRequestTool  # noqa: E402
@@ -42,7 +44,11 @@ class HttpProbeTool(BaseTool):
     parameters: Dict[str, Any] = {
         "type": "object",
         "properties": {
-            "url": {"type": "string", "description": "Resource URL to fetch"},
+            "url": {
+                "type": "string",
+                "description": "Resource URL to fetch",
+                "default": PAYWALLED_URL,
+            },
             "timeout": {
                 "type": "number",
                 "description": "Timeout in seconds for the request",
@@ -60,12 +66,14 @@ class HttpProbeTool(BaseTool):
 
     async def execute(
         self,
-        url: str,
+        url: str = PAYWALLED_URL,
         timeout: float = 20.0,
         headers: Optional[Dict[str, str]] = None,
     ) -> ToolResult:
+        # Fall back to the demo URL so the agent never fails on a missing argument.
+        target = url or PAYWALLED_URL
         async with httpx.AsyncClient(timeout=timeout) as client:
-            response = await client.get(url, headers=headers)
+            response = await client.get(target, headers=headers)
         try:
             body = response.json()
         except json.JSONDecodeError:
@@ -125,6 +133,25 @@ def decode_receipt(header_value: str) -> Dict[str, Any]:
     return payload.model_dump()
 
 
+def extract_music_url(content: str) -> Optional[str]:
+    """Extract SoundCloud (or other) music link from HTML/text."""
+    if not content:
+        return None
+    cleaned = html.unescape(content)
+    # Look for SoundCloud embed first
+    match = re.search(r"https?://w\.soundcloud\.com/player/\?[^\s\"'<>]+", cleaned)
+    if match:
+        return match.group(0)
+    match = re.search(r"https?://soundcloud\.com/[^\s\"'<>]+", cleaned)
+    if match:
+        return match.group(0)
+    # Fallback: first http(s) link
+    match = re.search(r"https?://[^\s\"'<>]+", cleaned)
+    if match:
+        return match.group(0)
+    return None
+
+
 def parse_tool_output(raw: str) -> Any:
     segment: Optional[str] = None
     if "Output:" in raw:
@@ -198,6 +225,8 @@ def print_conversation(messages: list[Message]) -> None:
 class X402ReactAgent(SpoonReactAI):
     name: str = "x402_react_agent"
     description: str = "ReAct agent that pays x402 invoices to reach protected resources"
+    target_url: str = PAYWALLED_URL
+    service: X402PaymentService = Field(default_factory=X402PaymentService, exclude=True)
 
     template_system_prompt: str = (
         "You are an autonomous ReAct agent with tool access."
@@ -213,23 +242,31 @@ class X402ReactAgent(SpoonReactAI):
     )
 
     def __init__(self, service: X402PaymentService, url: str, **kwargs: Any) -> None:
-        super().__init__(**kwargs)
-        self.service = service
-        self.target_url = url
+        super().__init__(service=service, target_url=url, **kwargs)
         self.http_probe_tool = HttpProbeTool()
         self.payment_tool: Optional[X402PaywalledRequestTool] = None
-        self.system_prompt = self.template_system_prompt.format(
-            target_url=self.target_url,
-            amount=str(PAYMENT_USDC),
-        )
         self.max_steps = 6
         self.x402_enabled = False  # prevent base class from auto-attaching duplicate tools
         self.available_tools = ToolManager([])
+        self._refresh_prompts()
 
     async def initialize(self) -> None:
         ensure_wallet_configuration(self.service)
         self.payment_tool = X402PaywalledRequestTool(service=self.service)
         self.available_tools = ToolManager([self.http_probe_tool, self.payment_tool])
+        self._refresh_prompts()
+
+    def _refresh_prompts(self) -> None:
+        """Keep the customised x402 playbook while still listing current tools."""
+        tool_list = self._build_tool_list()
+        self.system_prompt = (
+            self.template_system_prompt.format(
+                target_url=self.target_url,
+                amount=str(PAYMENT_USDC),
+            )
+            + f"\n\nAvailable tools:\n{tool_list}"
+        )
+        self.next_step_prompt = NEXT_STEP_PROMPT_TEMPLATE.format(tool_list=tool_list)
 
 
 async def main() -> None:
@@ -276,13 +313,21 @@ async def main() -> None:
     http_probe_result = extract_tool_payload(messages, "http_probe")
     payment_result = extract_tool_payload(messages, "x402_paywalled_request")
     assistant_summary = extract_last_assistant(messages)
+    music_url: Optional[str] = None
 
     if not assistant_summary and payment_result:
         body = payment_result.get("body")
         if isinstance(body, dict):
             assistant_summary = summarise_text(json.dumps(body, ensure_ascii=False))
         elif isinstance(body, str):
             assistant_summary = summarise_text(body)
+            music_url = extract_music_url(body)
+
+    # Attempt music URL extraction even if assistant summary already exists.
+    if music_url is None and payment_result:
+        body = payment_result.get("body")
+        if isinstance(body, str):
+            music_url = extract_music_url(body)
 
     if http_probe_result:
         preview_body = http_probe_result.get("body")
@@ -310,6 +355,9 @@ async def main() -> None:
     if assistant_summary:
         rprint("\n[bold green]Agent Final Summary[/]")
         rprint(assistant_summary)
+    if music_url:
+        rprint("\n[bold green]Music URL[/]")
+        rprint(music_url)
 
     if payment_header:
         rprint("\n[bold blue]Signed X-PAYMENT Header[/]")