Upload folder using huggingface_hub

server/api.js

@@ -191,9 +191,10 @@ export function apiRouter() {
 
   // --- mention long-poll ---
 
-  router.get('/mentions', requireUser, async (req, res) => {
-    const wait = Math.max(0, Math.min(Number(req.query.wait) || 0, 55))
-    const result = await collab.pollMentions(req.user.username, wait, req.agent?.handle)
+  // Instant snapshot of outstanding work (nothing is claimed) — for debugging.
+  // The `wait` param is gone: waiting happens on /mentions/stream.
+  router.get('/mentions', requireUser, (req, res) => {
+    const result = collab.snapshotMentions(req.user.username, req.agent?.handle)
     if (result.error) return res.status(400).json(result)
     res.json(result)
   })
@@ -288,7 +289,7 @@ Work loop — repeat until the user tells you to stop:
    The stream sends ":hb" heartbeat lines while idle; the final line is JSON:
    {"mentions": [...]}. Empty output or an empty list just means the wait expired —
    immediately make the same call again. This is the normal idle state; one call per
-   ~50 minutes costs almost nothing. (A short poll also exists: GET /api/mentions?wait=50)
+   ~50 minutes costs almost nothing.
    Each mention has: mention_id, doc_id, thread_id, instruction, anchored_text (the text the
    comment is attached to), thread_messages (conversation so far), requested_by.
    If "about_suggestion" is set, the comment discusses one of your earlier suggestions —

server/collab.js

@@ -302,7 +302,7 @@ async function claimAndEnrich(tasks) {
   return out
 }
 
-export function pollMentions(username, waitSeconds, onlyHandle = null, opts = {}) {
+function pollableHandles(username, onlyHandle) {
   let handles = handlesOwnedBy(username)
   if (onlyHandle) {
     // agent-key poll: exactly this handle
@@ -314,7 +314,25 @@ export function pollMentions(username, waitSeconds, onlyHandle = null, opts = {}
     const agents = store.getAgents()
     handles = handles.filter(h => !agents[h]?.keyHash)
   }
-  if (!handles.length) return Promise.resolve({ error: 'no agent handles registered for your account (handles with a key must poll with that key) — POST /api/agents first' })
+  return handles
+}
+
+const NO_HANDLES_ERROR = 'no agent handles registered for your account (handles with a key must poll with that key) — POST /api/agents first'
+
+// Instant, non-claiming view of outstanding work — for debugging. Receiving
+// (and claiming) work happens through the streaming poll.
+export function snapshotMentions(username, onlyHandle = null) {
+  const handles = pollableHandles(username, onlyHandle)
+  if (!handles.length) return { error: NO_HANDLES_ERROR }
+  const mentions = Object.values(mentionState.tasks)
+    .filter(t => handles.includes(t.handle) && (t.status === 'pending' || t.status === 'claimed'))
+    .map(t => ({ mention_id: t.id, doc_id: t.docId, thread_id: t.threadId, handle: t.handle, status: t.status, instruction: t.text, created_at: t.ts }))
+  return { snapshot: true, mentions }
+}
+
+export function pollMentions(username, waitSeconds, onlyHandle = null, opts = {}) {
+  const handles = pollableHandles(username, onlyHandle)
+  if (!handles.length) return Promise.resolve({ error: NO_HANDLES_ERROR })
   for (const h of handles) agentLastPoll.set(h, Date.now())
 
   const now = claimable(handles)

test/prod.js

@@ -82,7 +82,12 @@ async function main() {
     doc.getMap('threads').set(threadId, t)
   })
 
-  const poll = await agentApi('/api/mentions?wait=15')
+  const poll = await (async () => {
+    const resp = await fetch(`${BASE}/api/mentions/stream?wait=15`, { headers: { authorization: `Bearer ${AGENT}` } })
+    let text = ''
+    for await (const chunk of resp.body) text += Buffer.from(chunk).toString()
+    return JSON.parse(text.trim().split('\n').filter(l => !l.startsWith(':')).pop())
+  })()
   assert.equal(poll.mentions?.length, 1, JSON.stringify(poll))
   console.log('✓ mention delivered via long-poll')
 

test/smoke.js

@@ -54,6 +54,14 @@ function api(token, path_, { method = 'GET', body } = {}) {
   }).then(r => r.json())
 }
 
+async function streamPoll(key, wait = 10) {
+  const resp = await fetch(`${BASE}/api/mentions/stream?wait=${wait}`, { headers: { authorization: `Bearer ${key}` } })
+  let text = ''
+  for await (const chunk of resp.body) text += Buffer.from(chunk).toString()
+  const last = text.trim().split('\n').filter(l => !l.startsWith(':')).pop()
+  return JSON.parse(last)
+}
+
 function connect(docId, token) {
   const doc = new Y.Doc()
   const provider = new HocuspocusProvider({
@@ -153,8 +161,12 @@ async function main() {
   }, 'mention chip')
   console.log('✓ mention detected, chip set')
 
-  // 6. the agent long-polls with its key
-  const poll = await api(AGENT, '/api/mentions?wait=10')
+  // 6. snapshot shows the mention without claiming; the stream claims it
+  const snap0 = await api(AGENT, '/api/mentions')
+  assert.equal(snap0.snapshot, true, 'snapshot flagged')
+  assert.equal(snap0.mentions?.length, 1, 'snapshot shows pending: ' + JSON.stringify(snap0))
+  assert.equal(snap0.mentions[0].status, 'pending', 'snapshot does not claim')
+  const poll = await streamPoll(AGENT, 10)
   assert.equal(poll.mentions?.length, 1, 'mention delivered: ' + JSON.stringify(poll))
   const mention = poll.mentions[0]
   assert.equal(mention.handle, 'test-agent')
@@ -205,7 +217,7 @@ async function main() {
       { id: 'msg-followup', author: 'alice', authorType: 'user', text: 'hmm, can you double check the link?', ts: Date.now() },
     ])
   })
-  const followPoll = await api(AGENT, '/api/mentions?wait=10')
+  const followPoll = await streamPoll(AGENT, 10)
   assert.equal(followPoll.mentions?.length, 1, 'follow-up delivered')
   assert.equal(followPoll.mentions[0].implicit_follow_up, true, 'marked implicit')
   const dismissed = await api(AGENT, `/api/mentions/${followPoll.mentions[0].mention_id}/dismiss`, { method: 'POST' })