Upload SpecStream tool

README.md

@@ -1,10 +1,12 @@
 ---
-title: Specstream Get Rows
-emoji: 🏢
-colorFrom: blue
-colorTo: yellow
+title: SpecStream get rows
 sdk: static
 pinned: false
+tags:
+- smolagents
+- tool
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+SpecStream reads the live web and hands back every match with its source, so an agent can ask for data no one sells in plain words. Tool one shows the reading and the maximum price, and runs only after the person approves that maximum. Tool two reads the rows with their source URLs. Set SPECSTREAM_API_KEY. Start at https://specstream.ai/?utm_source=smolagents
+
+Load it with `Tool.from_hub("SpecStreamAI/specstream-get-rows", trust_remote_code=True)`.

index.html

@@ -1,19 +1 @@
-<!doctype html>
-<html>
-	<head>
-		<meta charset="utf-8" />
-		<meta name="viewport" content="width=device-width" />
-		<title>My static Space</title>
-		<link rel="stylesheet" href="style.css" />
-	</head>
-	<body>
-		<div class="card">
-			<h1>Welcome to your static Space!</h1>
-			<p>You can modify this app directly by editing <i>index.html</i> in the Files and versions tab.</p>
-			<p>
-				Also don't forget to check the
-				<a href="https://huggingface.co/docs/hub/spaces" target="_blank">Spaces documentation</a>.
-			</p>
-		</div>
-	</body>
-</html>
+<!doctype html><html><head><meta charset="utf-8"><title>SpecStream get rows</title></head><body style="font-family:system-ui;max-width:40rem;margin:3rem auto;padding:0 1rem"><h1>SpecStream get rows</h1><p>SpecStream reads the live web and hands back every match with its source, so an agent can ask for data no one sells in plain words. Tool one shows the reading and the maximum price, and runs only after the person approves that maximum. Tool two reads the rows with their source URLs. Set SPECSTREAM_API_KEY. Start at https://specstream.ai/?utm_source=smolagents</p><p><code>Tool.from_hub("SpecStreamAI/specstream-get-rows", trust_remote_code=True)</code></p></body></html>

requirements.txt

@@ -0,0 +1 @@
+smolagents

tool.py

@@ -0,0 +1,66 @@
+from typing import Any, Optional
+from smolagents.tools import Tool
+import json
+import urllib
+import os
+
+class SpecStreamGetRows(Tool):
+    name = "specstream_get_rows"
+    description = "Reads the results of a SpecStream run by find_id, each value with the URL of the page it came from. Also says whether the run is still going. Reading costs nothing. Needs SPECSTREAM_API_KEY."
+    inputs = {'find_id': {'type': 'string', 'description': 'The find_id returned by specstream_find_web_data.'}, 'limit': {'type': 'integer', 'description': 'How many rows to read, 1 to 100. Default 25.', 'nullable': True}, 'cursor': {'type': 'string', 'description': 'The next_cursor from a previous call, to read the next page.', 'nullable': True}}
+    output_type = "string"
+
+    def forward(self, find_id: str, limit: int | None = None, cursor: str | None = None) -> str:
+        import json
+        import os
+        from urllib.error import HTTPError, URLError
+        from urllib.parse import quote as urlquote, urlencode
+        from urllib.request import Request, urlopen
+
+        key = os.environ.get("SPECSTREAM_API_KEY", "").strip()
+        if not key:
+            return "Set SPECSTREAM_API_KEY to a SpecStream API key (https://specstream.ai/workspace?utm_source=smolagents)."
+        base = os.environ.get("SPECSTREAM_API_BASE", "https://api.specstream.ai").rstrip("/")
+        size = 25 if limit is None else max(1, min(int(limit), 100))
+        query = {"limit": str(size)}
+        if cursor:
+            query["cursor"] = cursor
+        path = "/v1/finds/%s/rows?%s" % (urlquote(find_id, safe=""), urlencode(query))
+        req = Request(
+            base + path,
+            headers={
+                "Authorization": "Bearer " + key,
+                "Accept": "application/json",
+                "User-Agent": "specstream-smolagents/0.1",
+            },
+        )
+        try:
+            with urlopen(req, timeout=120) as resp:
+                payload = json.loads(resp.read().decode("utf-8"))
+        except HTTPError as exc:
+            return "SpecStream answered HTTP %d for that find_id." % exc.code
+        except (URLError, OSError):
+            return "SpecStream could not be reached."
+        page = payload.get("data") if isinstance(payload, dict) else None
+        if not isinstance(page, dict):
+            return "SpecStream returned no rows page."
+        rows = page.get("rows") or []
+        lines = ["status: %s, accepted rows: %s" % (page.get("status"), page.get("accepted_count"))]
+        for number, row in enumerate(rows, 1):
+            parts = []
+            for cell in row.get("cells") or []:
+                shown = cell.get("display")
+                if shown in (None, ""):
+                    continue
+                source = (cell.get("provenance") or {}).get("source_url")
+                parts.append("%s: %s%s" % (cell.get("label"), shown, " (source: %s)" % source if source else ""))
+            lines.append("%d. %s" % (number, "; ".join(parts)))
+        if not rows:
+            lines.append("No rows yet. If the status is still running, call again in a minute.")
+        cursor_next = page.get("next_cursor")
+        if cursor_next:
+            lines.append("next_cursor: %s" % cursor_next)
+        return "\n".join(lines)
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False