bloby-bot 0.46.2 → 0.47.0

package/worker/index.ts

@@ -16,6 +16,9 @@ import {
 import { startClaudeOAuth, exchangeClaudeCode, getClaudeAuthStatus, readClaudeAccessToken } from './claude-auth.js';
 import { checkAvailability, registerHandle, claimReservedHandle, releaseHandle, updateTunnelUrl, startHeartbeat, stopHeartbeat } from '../shared/relay.js';
 import { ensureFileDirs } from '../supervisor/file-saver.js';
+import { readPiAuth, writePiAuth, clearPiAuth, getPiAuthStatus } from '../supervisor/harnesses/pi/auth-storage.js';
+import { runPiTestCompletion } from '../supervisor/harnesses/pi/test-completion.js';
+import { PI_SUB_PROVIDERS, getPiSubProvider } from '../supervisor/harnesses/pi/sub-providers.js';
 
 // ── Password hashing (scrypt) ──
 
@@ -133,12 +136,12 @@ app.post('/api/conversations/:id/messages', (req, res) => {
 });
 app.get('/api/conversations/:id/messages/recent', (req, res) => {
   const limit = parseInt(req.query.limit as string) || 20;
-  const msgs = getRecentMessages(req.params.id, Math.min(limit, 100));
+  const msgs = getRecentMessages(req.params.id, Math.min(limit, 1000));
   res.json(msgs);
 });
 app.get('/api/conversations/:id/messages', (req, res) => {
   const before = req.query.before as string;
-  const limit = Math.min(parseInt(req.query.limit as string) || 20, 100);
+  const limit = Math.min(parseInt(req.query.limit as string) || 20, 1000);
   if (before) {
     res.json(getMessagesBefore(req.params.id, before, limit));
   } else {
@@ -228,6 +231,89 @@ app.get('/api/auth/claude/status', async (_req, res) => {
   res.json(await getClaudeAuthStatus());
 });
 
+// ── Pi (Bloby third harness) auth routes ──
+
+app.get('/api/auth/pi/providers', (_req, res) => {
+  res.json({
+    providers: PI_SUB_PROVIDERS.map((p) => ({
+      id: p.id,
+      name: p.name,
+      subtitle: p.subtitle,
+      flavor: p.flavor,
+      baseUrl: p.baseUrl,
+      needsBaseUrl: !!p.needsBaseUrl,
+      needsApiKey: p.needsApiKey !== false,
+      apiKeyUrl: p.apiKeyUrl,
+      models: p.models,
+      defaultModel: p.defaultModel,
+    })),
+  });
+});
+
+app.get('/api/auth/pi/status', (_req, res) => {
+  res.json(getPiAuthStatus());
+});
+
+app.post('/api/auth/pi/test', async (req, res) => {
+  const { subProvider, apiKey, baseUrl, modelId } = req.body || {};
+  if (!subProvider || typeof subProvider !== 'string') {
+    res.json({ ok: false, error: 'Missing subProvider' });
+    return;
+  }
+  const provider = getPiSubProvider(subProvider);
+  if (!provider) {
+    res.json({ ok: false, error: `Unknown sub-provider: ${subProvider}` });
+    return;
+  }
+  const prompt = 'Reply with the single word OK so we can confirm this LLM endpoint is reachable.';
+  const result = await runPiTestCompletion({ subProvider, apiKey, baseUrl, modelId, prompt });
+  res.json(result);
+});
+
+app.post('/api/auth/pi/save', (req, res) => {
+  const { subProvider, apiKey, baseUrl, modelId } = req.body || {};
+  if (!subProvider || typeof subProvider !== 'string') {
+    res.json({ ok: false, error: 'Missing subProvider' });
+    return;
+  }
+  const provider = getPiSubProvider(subProvider);
+  if (!provider) {
+    res.json({ ok: false, error: `Unknown sub-provider: ${subProvider}` });
+    return;
+  }
+  const saved = writePiAuth({
+    subProvider,
+    apiKey: typeof apiKey === 'string' ? apiKey : undefined,
+    baseUrl: typeof baseUrl === 'string' && baseUrl.trim() ? baseUrl.trim() : provider.baseUrl,
+    modelId: typeof modelId === 'string' && modelId.trim() ? modelId.trim() : provider.defaultModel,
+  });
+  res.json({ ok: true, status: { configured: true, subProvider: saved.subProvider, modelId: saved.modelId, baseUrl: saved.baseUrl } });
+});
+
+app.delete('/api/auth/pi', (_req, res) => {
+  clearPiAuth();
+  res.json({ ok: true });
+});
+
+app.post('/api/auth/pi/completion', async (req, res) => {
+  const auth = readPiAuth();
+  if (!auth) {
+    res.json({ ok: false, error: 'Bloby provider is not configured yet' });
+    return;
+  }
+  const prompt = (req.body?.prompt && typeof req.body.prompt === 'string')
+    ? req.body.prompt
+    : 'In one short sentence, introduce yourself as the model behind this endpoint and confirm the connection is working.';
+  const result = await runPiTestCompletion({
+    subProvider: auth.subProvider,
+    apiKey: auth.apiKey,
+    baseUrl: auth.baseUrl,
+    modelId: auth.modelId,
+    prompt,
+  });
+  res.json(result);
+});
+
 // ── Handle registration ──
 
 app.get('/api/handle/check/:username', async (req, res) => {