notioncode 0.1.1 → 0.1.3

package/agent-runtime-server/server/providers/utils.js

@@ -0,0 +1,29 @@
+/**
+ * Shared provider utilities.
+ *
+ * @module providers/utils
+ */
+
+/**
+ * Prefixes that indicate internal/system content which should be hidden from the UI.
+ * @type {readonly string[]}
+ */
+export const INTERNAL_CONTENT_PREFIXES = Object.freeze([
+  '<command-name>',
+  '<command-message>',
+  '<command-args>',
+  '<local-command-stdout>',
+  '<system-reminder>',
+  'Caveat:',
+  'This session is being continued from a previous',
+  '[Request interrupted',
+]);
+
+/**
+ * Check if user text content is internal/system that should be skipped.
+ * @param {string} content
+ * @returns {boolean}
+ */
+export function isInternalContent(content) {
+  return INTERNAL_CONTENT_PREFIXES.some(prefix => content.startsWith(prefix));
+}

package/agent-runtime-server/server/routes/agent-sessions.js

@@ -0,0 +1,238 @@
+import { Router } from 'express';
+
+import {
+  ProviderSelectionError,
+  abortLocalAgentSession,
+  getLocalAgentSession,
+  getLocalAgentSessionEvents,
+  listAvailableProviders,
+  sendLocalAgentSessionMessage,
+  startLocalAgentSession,
+  subscribeToLocalAgentSession,
+} from '../agent-session-service.js';
+
+const router = Router();
+
+function readString(value) {
+  return typeof value === 'string' && value.trim() ? value.trim() : '';
+}
+
+function formatRichList(items) {
+  if (!Array.isArray(items) || items.length === 0) {
+    return '';
+  }
+
+  const lines = items
+    .map((item) => {
+      if (typeof item === 'string') {
+        return item.trim();
+      }
+      if (item && typeof item === 'object') {
+        return readString(item.text || item.title || item.name || item.label || item.content);
+      }
+      return '';
+    })
+    .filter(Boolean);
+
+  return lines.length > 0 ? lines.map((line) => `- ${line}`).join('\n') : '';
+}
+
+function buildPromptFromRequest(body = {}) {
+  const directPrompt = readString(body.prompt);
+  const metadata = body.metadata && typeof body.metadata === 'object' ? body.metadata : {};
+  const notion = metadata.notion && typeof metadata.notion === 'object' ? metadata.notion : {};
+  const source = { ...metadata, ...notion, ...body };
+
+  const sections = [];
+  const title = readString(source.title || source.taskTitle || source.pageTitle);
+  const description = readString(source.description || source.taskDescription || source.pageDescription);
+  const pageContent = readString(source.pageContent || source.content || source.body || source.markdown || source.richText);
+  const findings = formatRichList(source.findings);
+  const subtasks = formatRichList(source.subtasks);
+  const checklist = formatRichList(source.checklist || source.checklistMap);
+
+  if (directPrompt) {
+    sections.push(directPrompt);
+  } else if (title) {
+    sections.push(`Task: ${title}`);
+  }
+
+  if (description) {
+    sections.push(`Description:\n${description}`);
+  }
+
+  if (pageContent) {
+    sections.push(`Notion page content:\n${pageContent}`);
+  }
+
+  if (subtasks) {
+    sections.push(`Subtasks:\n${subtasks}`);
+  }
+
+  if (findings) {
+    sections.push(`Findings:\n${findings}`);
+  }
+
+  if (checklist) {
+    sections.push(`Checklist:\n${checklist}`);
+  }
+
+  return sections.filter(Boolean).join('\n\n').trim();
+}
+
+function sendSse(res, payload) {
+  res.write(`data: ${JSON.stringify(payload)}\n\n`);
+}
+
+function parseAfterSequence(value) {
+  const parsed = Number.parseInt(typeof value === 'string' ? value : '0', 10);
+  return Number.isFinite(parsed) && parsed >= 0 ? parsed : 0;
+}
+
+router.get('/providers', async (req, res) => {
+  try {
+    const providers = await listAvailableProviders();
+    res.json(providers);
+  } catch (error) {
+    res.status(500).json({
+      error: error instanceof Error ? error.message : String(error),
+    });
+  }
+});
+
+router.post('/sessions', async (req, res) => {
+  const prompt = buildPromptFromRequest(req.body);
+  const cwd = typeof req.body?.cwd === 'string' ? req.body.cwd.trim() : '';
+
+  if (!prompt) {
+    return res.status(400).json({ error: 'prompt is required' });
+  }
+
+  if (!cwd) {
+    return res.status(400).json({ error: 'cwd is required' });
+  }
+
+  try {
+    const session = await startLocalAgentSession(
+      {
+        prompt,
+        cwd,
+        workerRole: req.body?.workerRole,
+        preferredProvider: req.body?.preferredProvider,
+        workflowDefaultProvider: req.body?.workflowDefaultProvider,
+        allowFallback: req.body?.allowFallback,
+        metadata: req.body?.metadata,
+        model: req.body?.model,
+      },
+      {
+        userId: req.user?.id ?? req.user?.userId ?? null,
+      }
+    );
+
+    return res.status(201).json(session);
+  } catch (error) {
+    if (error instanceof ProviderSelectionError) {
+      return res.status(error.code === 'session_not_found' ? 404 : 409).json({
+        error: error.message,
+        code: error.code,
+        ...error.details,
+      });
+    }
+
+    return res.status(500).json({
+      error: error instanceof Error ? error.message : String(error),
+    });
+  }
+});
+
+router.get('/sessions/:id/events', (req, res) => {
+  let unsubscribe = null;
+
+  try {
+    const afterSequence = parseAfterSequence(req.query.after);
+    const sessionId = req.params.id;
+    getLocalAgentSession(sessionId);
+
+    res.setHeader('Content-Type', 'text/event-stream');
+    res.setHeader('Cache-Control', 'no-cache');
+    res.setHeader('Connection', 'keep-alive');
+    res.setHeader('X-Accel-Buffering', 'no');
+
+    for (const event of getLocalAgentSessionEvents(sessionId, afterSequence)) {
+      sendSse(res, event);
+    }
+
+    unsubscribe = subscribeToLocalAgentSession(sessionId, (event) => {
+      sendSse(res, event);
+    });
+
+    const heartbeat = setInterval(() => {
+      sendSse(res, { type: 'status', heartbeat: true, timestamp: new Date().toISOString() });
+    }, 15000);
+
+    req.on('close', () => {
+      clearInterval(heartbeat);
+      unsubscribe?.();
+      if (!res.writableEnded) {
+        res.end();
+      }
+    });
+
+    return undefined;
+  } catch (error) {
+    unsubscribe?.();
+    return res.status(error instanceof ProviderSelectionError && error.code === 'session_not_found' ? 404 : 500).json({
+      error: error instanceof Error ? error.message : String(error),
+    });
+  }
+});
+
+router.post('/sessions/:id/messages', async (req, res) => {
+  const prompt = buildPromptFromRequest(req.body);
+
+  if (!prompt) {
+    return res.status(400).json({ error: 'prompt is required' });
+  }
+
+  try {
+    const session = await sendLocalAgentSessionMessage(req.params.id, {
+      prompt,
+      model: req.body?.model,
+    });
+
+    return res.json(session);
+  } catch (error) {
+    if (error instanceof ProviderSelectionError) {
+      return res.status(error.code === 'session_not_found' ? 404 : 409).json({
+        error: error.message,
+        code: error.code,
+        ...error.details,
+      });
+    }
+
+    return res.status(500).json({
+      error: error instanceof Error ? error.message : String(error),
+    });
+  }
+});
+
+router.post('/sessions/:id/abort', async (req, res) => {
+  try {
+    const result = await abortLocalAgentSession(req.params.id);
+    return res.json(result);
+  } catch (error) {
+    if (error instanceof ProviderSelectionError) {
+      return res.status(error.code === 'session_not_found' ? 404 : 409).json({
+        error: error.message,
+        code: error.code,
+        ...error.details,
+      });
+    }
+
+    return res.status(500).json({
+      error: error instanceof Error ? error.message : String(error),
+    });
+  }
+});
+
+export default router;