synthos 0.7.2 → 0.8.0

package/src/migrations.ts

@@ -1,5 +1,7 @@
 import * as cheerio from 'cheerio';
-import { completePrompt } from 'agentm-core';
+import * as fs from 'fs/promises';
+import path from 'path';
+import { completePrompt } from './models';
 import { deduplicateInlineScripts } from './service/transformPage';
 
 /**
@@ -26,128 +28,57 @@ export async function migratePage(html: string, fromVersion: number, toVersion:
     return current;
 }
 
-/** CSS classes that belong to the shared theme and must NOT appear in page-specific <style> blocks. */
+/** CSS selectors provided by the shared theme — used by post-processing to strip leftovers. */
 const SHARED_CSS_SELECTORS = [
-    '*',
-    'body',
-    '.chat-panel',
-    '.chat-header',
-    '.chat-messages',
-    '.chat-message',
-    '.chat-message p',
-    '.chat-message strong',
-    '.chat-message pre',
-    '.chat-message code',
-    '.chat-message a',
-    '.link-group',
-    '.link-group a',
+    // Base
+    ':root', '*', 'body', 'html',
+    // Chat panel
+    '.chat-panel', '.chat-header', '.chat-messages',
+    '.chat-message', '.chat-message p', '.chat-message p strong', '.chat-message p code',
+    '.chat-message strong', '.chat-message pre', '.chat-message code', '.chat-message a',
+    '.link-group', '.link-group a', '.link-group a:hover',
     'form',
-    '.chat-input',
-    '.chat-submit',
-    '.loading-overlay',
-    '.spinner',
-    '.viewer-panel',
-    '#loadingOverlay',
-    '.chat-submit:disabled',
-    '.chat-input:disabled',
+    '.chat-input', '.chat-input:focus', '.chat-input::placeholder', '.chat-input:disabled',
+    '.chat-submit', '.chat-submit:hover', '.chat-submit:active', '.chat-submit:disabled',
+    '.chat-input-wrapper', '.chat-input-wrapper .chat-input',
+    // Viewer panel
+    '.viewer-panel', '.viewer-panel::before', '.viewer-panel.full-viewer',
+    // Loading
+    '.loading-overlay', '.spinner', '#loadingOverlay',
+    // Chat toggle
+    '.chat-toggle', '.chat-toggle:hover', '.chat-toggle-dots', '.chat-toggle-dot',
+    '.chat-toggle:hover .chat-toggle-dot',
+    'body.chat-collapsed .chat-panel', 'body.chat-collapsed .chat-toggle',
+    // Modal system
+    '.modal-overlay', '.modal-overlay.show', '.modal-content', '.modal-header',
+    '.modal-body', '.modal-footer', '.modal-footer-right',
+    '.modal-btn', '.modal-btn-primary', '.modal-btn-primary:hover',
+    '.modal-btn-secondary', '.modal-btn-secondary:hover',
+    '.modal-btn-danger', '.modal-btn-danger:hover',
+    // Form elements
+    '.form-group', '.form-group:last-child', '.form-label',
+    '.form-input', '.form-input:focus', '.form-input:read-only', '.form-input::placeholder',
+    '.checkbox-label', '.checkbox-label input[type="checkbox"]', '.checkbox-label span',
+    // Brainstorm
+    '.brainstorm-icon-btn', '.brainstorm-icon-btn:hover',
+    '.brainstorm-modal .modal-content', '.brainstorm-modal .modal-header',
+    '.brainstorm-close-btn', '.brainstorm-close-btn:hover',
+    '.brainstorm-messages', '.brainstorm-message', '.brainstorm-user', '.brainstorm-assistant',
+    '.brainstorm-input-row', '.brainstorm-input', '.brainstorm-input:focus', '.brainstorm-input::placeholder',
+    '.brainstorm-send-btn', '.brainstorm-send-btn:hover', '.brainstorm-send-btn:disabled',
+    '.brainstorm-assistant p', '.brainstorm-assistant pre', '.brainstorm-assistant code',
+    '.brainstorm-build-row', '.brainstorm-build-btn', '.brainstorm-build-btn:hover',
+    '.brainstorm-suggestions', '.brainstorm-suggestion-chip',
+    '.brainstorm-suggestion-chip:hover', '.brainstorm-suggestion-chip:disabled',
+    '.brainstorm-thinking',
 ];
 
-const V1_TO_V2_SYSTEM_PROMPT = `You are a code migration tool. You convert SynthOS v1 pages to v2 format.
-
-## Rules — What to REMOVE
-
-1. **Shared CSS rules** — Remove CSS rules that match these selectors **exactly** (they are now in theme.css):
-   \`*\`, \`body\`, \`.chat-panel\`, \`.chat-header\`, \`.chat-messages\`, \`.chat-message\` (and descendants like \`.chat-message p\`, \`.chat-message strong\`, \`.chat-message pre\`, \`.chat-message code\`, \`.chat-message a\`), \`.link-group\`, \`.link-group a\`, \`form\`, \`.chat-input\`, \`.chat-submit\`, \`.loading-overlay\`, \`.spinner\`, \`.viewer-panel\`, \`#loadingOverlay\`, \`.chat-submit:disabled\`, \`.chat-input:disabled\`
-   Also remove any \`@keyframes spin\` and scrollbar pseudo-element rules (\`::-webkit-scrollbar\`, \`::-webkit-scrollbar-track\`, \`::-webkit-scrollbar-thumb\`).
-   **Important:** Only remove rules matching these selectors exactly. Do NOT remove pseudo-element variants (\`.viewer-panel::before\`, \`.viewer-panel::after\`), pseudo-class variants (\`.chat-input:focus\`, \`.chat-submit:hover\`), or any other compound selectors that extend the shared selectors — those are page-specific styles and must be preserved. Also preserve any \`@keyframes\` referenced by page-specific rules (e.g. if a page has \`.viewer-panel::before\` using a custom animation, keep that \`@keyframes\`).
-
-2. **Shared inline JS** — Remove these specific code blocks from \`<script>\` tags:
-   - \`document.getElementById('chatInput').focus()\` line
-   - \`chatForm\` submit event listener (the one with setTimeout and loading overlay)
-   - \`saveLink\` click handler
-   - \`resetLink\` click handler
-   - \`window.onload\` that ONLY scrolls chatMessages (keep other onload logic!)
-   - Chat panel toggle IIFE (references \`synthos-chat-collapsed\`)
-   - Focus management IIFE (references \`stopImmediatePropagation\`)
-   - \`// Basic chat functionality\` comment
-
-3. **Empty \`<script>\` tags** — If a script block becomes empty after stripping, remove it entirely.
-
-## Rules — What to ADD
-
-1. In \`<head>\`, add these two lines right after the \`<title>\` tag (if not already present):
-   \`\`\`
-   <script src="/api/theme-info.js"></script>
-   <link rel="stylesheet" href="/api/theme.css">
-   \`\`\`
-
-## Rules — What to TRANSFORM
-
-### Data/Table API migration
-The data API changed from global tables to **page-scoped** tables:
-- Old: \`/api/data/:table\` → New: \`/api/data/:page/:table\`
-- Old: \`/api/data/:table/:id\` → New: \`/api/data/:page/:table/:id\`
-- Tables are now stored as sub-folders of each page's folder.
-- **Raw fetch() calls** must be updated: e.g. \`fetch('/api/data/notes')\` → \`fetch('/api/data/' + pageName + '/notes')\`
-- **synthos.data.\* helpers** handle this automatically — they read the current page name from \`window.pageInfo.name\`. Prefer converting raw fetch data calls to use the helpers instead:
-  - \`fetch('/api/data/notes')\` → \`synthos.data.list('notes')\`
-  - \`fetch('/api/data/notes/' + id)\` → \`synthos.data.get('notes', id)\`
-  - \`fetch('/api/data/notes', { method: 'POST', ... })\` → \`synthos.data.save('notes', row)\`
-  - \`fetch('/api/data/notes/' + id, { method: 'DELETE' })\` → \`synthos.data.remove('notes', id)\`
-
-### Color variables
-Replace hardcoded Nebula Dusk colors with CSS variables in **page-specific** CSS only:
-| Hardcoded | CSS Variable |
-|-----------|-------------|
-| \`#667eea\` | \`var(--accent-primary)\` |
-| \`#764ba2\` | \`var(--accent-secondary)\` |
-| \`#f093fb\` | \`var(--accent-tertiary)\` |
-| \`#b794f6\` | \`var(--text-secondary)\` |
-| \`#e0e0e0\` | \`var(--text-primary)\` |
-| \`rgba(138, 43, 226, 0.3)\` or similar purple rgba | \`var(--border-color)\` or \`var(--accent-glow)\` (use border-color for borders, accent-glow for shadows) |
-| \`#1a1a2e\` | \`var(--bg-primary)\` |
-| \`#16213e\` | \`var(--bg-secondary)\` |
-| \`#0f0f23\` | \`var(--bg-tertiary)\` |
-
-For gradients using these colors (e.g. \`linear-gradient(135deg, #667eea, #764ba2)\`), replace with \`linear-gradient(135deg, var(--accent-primary), var(--accent-secondary))\`.
-
-## Rules — What to PRESERVE (do NOT modify)
-
-- ALL viewer-panel HTML content (games, presentations, tools, etc.)
-- ALL page-specific JavaScript (game logic, presentation logic, keyboard handlers, etc.)
-- ALL page-specific CSS (game styles, presentation styles, layout rules for non-shared classes)
-- Chat message history in \`.chat-messages\`
-- \`<div id="thoughts">\` content
-- External CDN script tags (\`<script src="...">\`)
-- The two-panel layout structure (chat-panel + viewer-panel)
-- The \`<div id="loadingOverlay" class="loading-overlay"><div class="spinner"></div></div>\` element
-- The chat form, link-group, chat-header elements (structure only, their CSS is handled by theme)
-
-## V2 page structure example
-
-\`\`\`html
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>SynthOS - Page Title</title>
-    <script src="/api/theme-info.js"></script>
-    <link rel="stylesheet" href="/api/theme.css">
-    <style>
-        /* Only page-specific styles here — no shared chat/layout CSS */
-        .my-custom-element {
-            color: var(--text-primary);
-            background: var(--bg-secondary);
-        }
-    </style>
-    <!-- external CDN scripts if needed -->
-</head>
-<body>
+/** Default chat panel HTML — restored by post-processing if the LLM removes it. */
+const DEFAULT_CHAT_PANEL = `
     <div class="chat-panel">
         <div class="chat-header">SynthOS</div>
         <div class="chat-messages" id="chatMessages">
-            <!-- chat messages preserved -->
+            <div class="chat-message"><p>Welcome! How can I help you?</p></div>
         </div>
         <div class="link-group">
             <a href="#" id="saveLink">Save</a>
@@ -158,32 +89,26 @@ For gradients using these colors (e.g. \`linear-gradient(135deg, #667eea, #764ba
             <input type="text" class="chat-input" id="chatInput" name="message" placeholder="Type a message...">
             <button type="submit" class="chat-submit">Send</button>
         </form>
-    </div>
-    <div class="viewer-panel" id="viewerPanel">
-        <!-- page content preserved -->
-        <div id="loadingOverlay" class="loading-overlay"><div class="spinner"></div></div>
-    </div>
-    <div id="thoughts" style="display: none;">...</div>
-    <script>
-        // Only page-specific JS here — no shared chat handlers
-    </script>
-</body>
-</html>
-\`\`\`
+    </div>`;
 
-## Output format
-
-Return ONLY the complete migrated HTML. No markdown fences, no explanation, no commentary. Just the raw HTML starting with \`<!DOCTYPE html>\`.`;
+/**
+ * Load migration rules from the migration-rules/ folder.
+ */
+async function loadMigrationRules(filename: string): Promise<string> {
+    const rulesPath = path.join(__dirname, '..', 'migration-rules', filename);
+    return await fs.readFile(rulesPath, 'utf8');
+}
 
 /**
  * v1 -> v2: LLM-based migration that strips shared code and adds theme support.
  * Post-processes with cheerio to verify critical elements are present.
  */
 async function migrateV1toV2(html: string, completePrompt: completePrompt): Promise<string> {
-    const system = { role: 'system' as const, content: V1_TO_V2_SYSTEM_PROMPT };
+    const rules = await loadMigrationRules('v1-to-v2.md');
+    const system = { role: 'system' as const, content: rules };
     const prompt = { role: 'user' as const, content: `Convert this v1 page to v2 format:\n\n${html}` };
 
-    const result = await completePrompt({ prompt, system, maxTokens: 16000 });
+    const result = await completePrompt({ prompt, system });
     if (!result.completed || !result.value) {
         throw new Error('LLM migration failed: ' + (result.error?.message ?? 'no response'));
     }
@@ -195,16 +120,72 @@ async function migrateV1toV2(html: string, completePrompt: completePrompt): Prom
     }
 
     // Post-process with cheerio to verify and fix critical elements
-    migrated = postProcessV2(migrated);
+    migrated = postProcessV2(migrated, html);
 
     return migrated;
 }
 
 /**
  * Cheerio-based post-processing to verify the LLM output meets v2 requirements.
+ * Uses the original HTML as a fallback source for critical elements.
  */
-export function postProcessV2(html: string): string {
+export function postProcessV2(html: string, originalHtml?: string): string {
     const $ = cheerio.load(html, { decodeEntities: false });
+    const $original = originalHtml ? cheerio.load(originalHtml, { decodeEntities: false }) : null;
+
+    // --- Critical structural checks ---
+
+    // Ensure chat-panel exists with chatForm
+    if ($('#chatForm').length === 0) {
+        if ($('.chat-panel').length > 0) {
+            // Chat panel exists but form is missing — restore form from original or default
+            const originalForm = $original?.('#chatForm').parent('.chat-panel');
+            if (originalForm && originalForm.length > 0) {
+                $('.chat-panel').replaceWith(originalForm.html()!);
+            } else {
+                // Append default form
+                $('.chat-panel').append(`
+        <form action="/" method="POST" id="chatForm">
+            <input type="text" class="chat-input" id="chatInput" name="message" placeholder="Type a message...">
+            <button type="submit" class="chat-submit">Send</button>
+        </form>`);
+            }
+        } else {
+            // Entire chat panel is missing — restore from original or use default
+            const originalPanel = $original?.('.chat-panel');
+            if (originalPanel && originalPanel.length > 0) {
+                $('body').prepend($.html(originalPanel));
+            } else {
+                $('body').prepend(DEFAULT_CHAT_PANEL);
+            }
+        }
+    }
+
+    // Ensure thoughts div exists
+    if ($('#thoughts').length === 0) {
+        const originalThoughts = $original?.('#thoughts');
+        if (originalThoughts && originalThoughts.length > 0) {
+            $('body').append($.html(originalThoughts));
+        } else {
+            $('body').append('<div id="thoughts" style="display: none;"></div>');
+        }
+    }
+
+    // Ensure loadingOverlay exists inside viewer-panel
+    const overlay = $('#loadingOverlay');
+    const viewerPanel = $('.viewer-panel');
+    if (overlay.length === 0 && viewerPanel.length > 0) {
+        viewerPanel.append('<div id="loadingOverlay" class="loading-overlay"><div class="spinner"></div></div>');
+    } else if (overlay.length > 0 && viewerPanel.length > 0) {
+        // Move inside viewer-panel if it's outside
+        if (overlay.closest('.viewer-panel').length === 0) {
+            const overlayHtml = $.html(overlay);
+            overlay.remove();
+            viewerPanel.append(overlayHtml);
+        }
+    }
+
+    // --- Theme refs ---
 
     // Ensure theme-info.js is in <head>
     if ($('script[src="/api/theme-info.js"]').length === 0) {
@@ -226,7 +207,8 @@ export function postProcessV2(html: string): string {
         }
     }
 
-    // Remove leftover shared CSS selectors from <style> blocks
+    // --- Strip leftover shared CSS ---
+
     $('style').each(function (_, el) {
         let css = $(el).html() ?? '';
         for (const selector of SHARED_CSS_SELECTORS) {
@@ -236,10 +218,11 @@ export function postProcessV2(html: string): string {
             const pattern = new RegExp(`(?:^|\\n)\\s*${escaped}\\s*\\{[^}]*\\}`, 'g');
             css = css.replace(pattern, '');
         }
-        // Remove @keyframes spin
+        // Remove @keyframes spin and nebula-pulse
         css = css.replace(/@keyframes\s+spin\s*\{[^}]*(?:\{[^}]*\}[^}]*)*\}/g, '');
+        css = css.replace(/@keyframes\s+nebula-pulse\s*\{[^}]*(?:\{[^}]*\}[^}]*)*\}/g, '');
         // Remove scrollbar pseudo-element rules
-        css = css.replace(/(?:^|\n)\s*(?:\*|body|)::-webkit-scrollbar(?:-(?:track|thumb))?\s*\{[^}]*\}/g, '');
+        css = css.replace(/(?:^|\n)\s*(?:\*|body|)::-webkit-scrollbar(?:-(?:track|thumb|corner))?(?::hover)?\s*\{[^}]*\}/g, '');
         $(el).html(css);
     });
 

package/src/models/anthropic.ts

@@ -0,0 +1,89 @@
+import Anthropic from '@anthropic-ai/sdk';
+import { AgentCompletion, completePrompt, PromptCompletionArgs, RequestError } from './types';
+
+export interface AnthropicArgs {
+    apiKey: string;
+    model: string;
+    baseURL?: string;
+    temperature?: number;
+    maxRetries?: number;
+}
+
+/**
+ * Build the messages array and system content for an Anthropic API request.
+ * Pure function — no SDK dependency.
+ */
+export function buildAnthropicRequest(args: PromptCompletionArgs, defaultTemp: number): {
+    messages: { role: string; content: string }[];
+    system: string | undefined;
+    temperature: number;
+} {
+    const reqTemp = args.temperature ?? defaultTemp;
+
+    const messages: { role: string; content: string }[] = [];
+    if (args.history) {
+        for (const msg of args.history) {
+            messages.push({ role: msg.role, content: msg.content });
+        }
+    }
+
+    const useJsonPrefill = args.jsonMode || args.jsonSchema;
+    if (useJsonPrefill) {
+        messages.push({ role: 'user', content: args.prompt.content });
+        messages.push({ role: 'assistant', content: '{' });
+    } else {
+        messages.push({ role: 'user', content: args.prompt.content });
+    }
+
+    let system = args.system?.content;
+    if (args.jsonSchema) {
+        const schemaInstruction = `\n\nYou must return valid JSON conforming to this schema:\n${JSON.stringify(args.jsonSchema)}`;
+        system = system ? system + schemaInstruction : schemaInstruction;
+    }
+
+    return { messages, system, temperature: reqTemp };
+}
+
+export function anthropic(args: AnthropicArgs): completePrompt {
+    const { apiKey, model, baseURL, temperature = 0.0, maxRetries } = args;
+
+    const client = new Anthropic({ apiKey, baseURL, maxRetries });
+
+    return async (completionArgs: PromptCompletionArgs): Promise<AgentCompletion<string>> => {
+        const { messages, system: systemContent, temperature: reqTemp } = buildAnthropicRequest(completionArgs, temperature);
+
+        const useJsonPrefill = completionArgs.jsonMode || completionArgs.jsonSchema;
+
+        try {
+            const stream = await client.messages.create({
+                model,
+                max_tokens: 32768,
+                temperature: reqTemp,
+                system: systemContent,
+                messages: messages as Anthropic.MessageParam[],
+                stream: true,
+            });
+
+            let text = '';
+            for await (const event of stream) {
+                if (event.type === 'content_block_delta' && event.delta.type === 'text_delta') {
+                    text += event.delta.text;
+                }
+            }
+
+            if (useJsonPrefill) {
+                text = '{' + text;
+            }
+
+            return { completed: true, value: text };
+        } catch (err: unknown) {
+            let error: Error;
+            if (err instanceof Anthropic.APIError && (err as any).status !== undefined) {
+                error = new RequestError(err.message, (err as any).status, err.name);
+            } else {
+                error = err as Error;
+            }
+            return { completed: false, error };
+        }
+    };
+}

package/src/models/chainOfThought.ts

@@ -0,0 +1,56 @@
+import { AgentCompletion, AgentArgs, SystemMessage, UserMessage } from './types';
+
+export interface ExplainedAnswer {
+    explanation: string;
+    answer: string;
+}
+
+export interface ChainOfThoughtArgs extends AgentArgs {
+    question: string;
+    temperature?: number;
+    instructions?: string;
+}
+
+export async function chainOfThought(args: ChainOfThoughtArgs): Promise<AgentCompletion<ExplainedAnswer>> {
+    const { completePrompt, question, temperature, instructions } = args;
+
+    const systemContent = instructions
+        ? `${instructions}\n\nYou must return your response as a JSON object with "explanation" and "answer" fields.`
+        : 'You must return your response as a JSON object with "explanation" and "answer" fields.';
+
+    const system: SystemMessage = { role: 'system', content: systemContent };
+    const prompt: UserMessage = { role: 'user', content: question };
+
+    const result = await completePrompt({ prompt, system, temperature, jsonMode: true });
+
+    if (!result.completed || !result.value) {
+        return { completed: false, error: result.error };
+    }
+
+    try {
+        let parsed: any;
+        if (typeof result.value === 'object') {
+            parsed = result.value;
+        } else {
+            let text = result.value as string;
+            // Strip markdown code fences if present
+            text = text.replace(/^```(?:json)?\s*/i, '').replace(/\s*```\s*$/, '');
+            // Extract the first JSON object from the text
+            const start = text.indexOf('{');
+            const end = text.lastIndexOf('}');
+            if (start !== -1 && end > start) {
+                text = text.substring(start, end + 1);
+            }
+            parsed = JSON.parse(text);
+        }
+        return {
+            completed: true,
+            value: {
+                explanation: parsed.explanation ?? '',
+                answer: parsed.answer ?? '',
+            },
+        };
+    } catch {
+        return { completed: false, error: new Error('Failed to parse chain-of-thought response') };
+    }
+}

package/src/models/fireworksai.ts

@@ -0,0 +1,136 @@
+import OpenAI from 'openai';
+import { AgentCompletion, completePrompt, PromptCompletionArgs, RequestError } from './types';
+
+export interface FireworksAIArgs {
+    apiKey: string;
+    model: string;
+    temperature?: number;
+}
+
+/** Known short-name → full Fireworks model path mappings. */
+const FIREWORKS_MODEL_MAP: Record<string, string> = {
+    'fireworks-glm-5': 'accounts/fireworks/models/glm-5',
+    'fireworks-minimax-m2p5': 'accounts/fireworks/models/minimax-m2p5',
+    'fireworks-kimi-k2p5': 'accounts/fireworks/models/kimi-k2p5',
+};
+
+const FIREWORKS_BASE_URL = 'https://api.fireworks.ai/inference/v1';
+
+/**
+ * Extract JSON from a string that may be wrapped in markdown fences or prose.
+ * Returns the original string if no JSON object/array is found.
+ */
+export function extractJSON(text: string): string {
+    // Strip markdown code fences (```json ... ``` or ``` ... ```)
+    const fenceMatch = text.match(/```(?:json)?\s*\n?([\s\S]*?)\n?\s*```/);
+    if (fenceMatch) {
+        return fenceMatch[1].trim();
+    }
+
+    // Find the first top-level { ... } or [ ... ]
+    const start = text.search(/[\[{]/);
+    if (start === -1) return text;
+
+    const open = text[start];
+    const close = open === '{' ? '}' : ']';
+    let depth = 0;
+    let inString = false;
+    let escape = false;
+    for (let i = start; i < text.length; i++) {
+        const ch = text[i];
+        if (escape) { escape = false; continue; }
+        if (ch === '\\' && inString) { escape = true; continue; }
+        if (ch === '"') { inString = !inString; continue; }
+        if (inString) continue;
+        if (ch === open || ch === (open === '{' ? '[' : '{')) depth++;
+        else if (ch === close || ch === (close === '}' ? ']' : '}')) depth--;
+        if (depth === 0) {
+            return text.slice(start, i + 1);
+        }
+    }
+
+    // Fallback: return from the first brace to end
+    return text.slice(start);
+}
+
+/**
+ * Resolve a short Fireworks model name to the full API path.
+ * Returns the input unchanged if not a known short name.
+ */
+export function resolveFireworksModel(model: string): string {
+    return FIREWORKS_MODEL_MAP[model] || model;
+}
+
+/**
+ * Build the messages array, temperature, and JSON flag for a FireworksAI request.
+ * Pure function — no SDK dependency.
+ */
+export function buildFireworksRequest(args: PromptCompletionArgs, defaultTemp: number): {
+    messages: { role: string; content: string }[];
+    temperature: number;
+    useJson: boolean;
+} {
+    const temperature = args.temperature ?? defaultTemp;
+    const useJson = !!(args.jsonMode || args.jsonSchema);
+
+    const messages: { role: string; content: string }[] = [];
+    if (args.system) {
+        messages.push({ role: 'system', content: args.system.content });
+    }
+    if (args.history) {
+        for (const msg of args.history) {
+            messages.push({ role: msg.role, content: msg.content });
+        }
+    }
+
+    let userContent = args.prompt.content;
+    if (useJson) {
+        userContent += '\n\nRespond with valid JSON only. No markdown fences.';
+    }
+    messages.push({ role: 'user', content: userContent });
+
+    return { messages, temperature, useJson };
+}
+
+export function fireworksai(args: FireworksAIArgs): completePrompt {
+    const { apiKey, temperature = 0.0 } = args;
+    const model = resolveFireworksModel(args.model);
+
+    const client = new OpenAI({ apiKey, baseURL: FIREWORKS_BASE_URL });
+
+    return async (completionArgs: PromptCompletionArgs): Promise<AgentCompletion<string>> => {
+        const { messages, temperature: reqTemp, useJson } = buildFireworksRequest(completionArgs, temperature);
+
+        try {
+            const response = await client.chat.completions.create({
+                model,
+                messages: messages as OpenAI.ChatCompletionMessageParam[],
+                temperature: reqTemp,
+                max_tokens: 32768,
+                ...(useJson ? { response_format: { type: 'json_object' } } : {}),
+            });
+
+            const choice = response.choices[0];
+            if (!choice?.message) {
+                return { completed: false, error: new Error('No response choice returned') };
+            }
+
+            let text = choice.message.content ?? '';
+
+            // Robust JSON extraction as a safety net
+            if (useJson && text) {
+                text = extractJSON(text);
+            }
+
+            return { completed: true, value: text };
+        } catch (err: unknown) {
+            let error: Error;
+            if (err instanceof OpenAI.APIError && (err as any).status !== undefined) {
+                error = new RequestError(err.message, (err as any).status, err.name ?? 'APIError');
+            } else {
+                error = err as Error;
+            }
+            return { completed: false, error };
+        }
+    };
+}

package/src/models/index.ts

@@ -1,2 +1,8 @@
-export { ProviderName, ProviderConfig, ModelEntry, Provider } from './types';
+export { ProviderName, ProviderConfig, ModelEntry, Provider, SystemMessage, UserMessage, Message, AgentCompletion, completePrompt, PromptCompletionArgs, AgentArgs, RequestError } from './types';
 export { AnthropicProvider, OpenAIProvider, PROVIDERS, getProvider, detectProvider } from './providers';
+export { anthropic, AnthropicArgs, buildAnthropicRequest } from './anthropic';
+export { openai, OpenaiArgs, buildOpenAIRequest } from './openai';
+export { fireworksai, FireworksAIArgs, resolveFireworksModel, buildFireworksRequest } from './fireworksai';
+export { chainOfThought, ChainOfThoughtArgs, ExplainedAnswer } from './chainOfThought';
+export { logCompletePrompt } from './logCompletePrompt';
+export { variableToString } from './utils';

package/src/models/logCompletePrompt.ts

@@ -0,0 +1,25 @@
+import { AgentCompletion, completePrompt, PromptCompletionArgs } from './types';
+
+export function logCompletePrompt<TValue = string>(
+    inner: completePrompt<TValue>,
+    logDetails?: boolean
+): completePrompt<TValue> {
+    return async (args: PromptCompletionArgs): Promise<AgentCompletion<TValue>> => {
+        const result = await inner(args);
+
+        if (result.completed) {
+            const value = typeof result.value === 'string'
+                ? result.value
+                : JSON.stringify(result.value, null, 2);
+            console.log(`\x1b[32m${value}\x1b[0m`);
+        } else if (result.error) {
+            console.log(`\x1b[31m${result.error.message}\x1b[0m`);
+        }
+
+        if (logDetails) {
+            console.log('─'.repeat(80));
+        }
+
+        return result;
+    };
+}