ugly-app 0.1.0

package/dist/server/WorkerQueue.js

@@ -0,0 +1,147 @@
+import { AckPolicy, DeliverPolicy } from 'nats';
+import { ensureStream, getJetStream, getNatsConnection } from './Nats.js';
+export function createWorkerQueue(config = {}) {
+    const { streamName = 'JOBS', clockServerOnly = true, concurrency = 10, maxRetries = 3, } = config;
+    const handlers = new Map();
+    let subscription = null;
+    const isClockServer = process.env['IS_CLOCK_SERVER'] === 'true';
+    async function enqueue(type, payload, options) {
+        const js = await getJetStream();
+        const job = {
+            id: crypto.randomUUID(),
+            type,
+            payload,
+            createdAt: new Date(),
+            attempts: 0,
+            scheduledFor: options?.delay
+                ? new Date(Date.now() + options.delay)
+                : undefined,
+            working: () => { },
+        };
+        const subject = `${streamName}.${type}`;
+        await js.publish(subject, new TextEncoder().encode(JSON.stringify(job)));
+        console.log('[WorkerQueue] Job enqueued', {
+            id: job.id,
+            type,
+            scheduledFor: job.scheduledFor,
+        });
+    }
+    function registerHandler(type, handler) {
+        handlers.set(type, handler);
+    }
+    async function start() {
+        if (clockServerOnly && !isClockServer) {
+            console.log('[WorkerQueue] Not clock server (IS_CLOCK_SERVER!=true), skipping job consumer start');
+            return;
+        }
+        if (handlers.size === 0) {
+            console.log('[WorkerQueue] No handlers registered, skipping consumer start');
+            return;
+        }
+        // Ensure NATS stream exists for all job subjects
+        await ensureStream({
+            name: streamName,
+            subjects: [`${streamName}.*`],
+        });
+        const js = await getJetStream();
+        const conn = getNatsConnection();
+        const jsm = await conn.jetstreamManager();
+        // Create or get durable consumer
+        const consumerName = `worker-${process.env['HOSTNAME'] ?? 'local'}`;
+        try {
+            await jsm.consumers.info(streamName, consumerName);
+        }
+        catch {
+            await jsm.consumers.add(streamName, {
+                durable_name: consumerName,
+                deliver_policy: DeliverPolicy.New,
+                ack_policy: AckPolicy.Explicit,
+                max_ack_pending: concurrency,
+            });
+        }
+        const consumer = await js.consumers.get(streamName, consumerName);
+        const messages = await consumer.consume();
+        subscription = messages;
+        // Process messages
+        (async () => {
+            for await (const msg of messages) {
+                try {
+                    const job = JSON.parse(new TextDecoder().decode(msg.data));
+                    const handler = handlers.get(job.type);
+                    if (!handler) {
+                        console.warn('[WorkerQueue] No handler for job type, discarding', {
+                            type: job.type,
+                            id: job.id,
+                        });
+                        msg.ack();
+                        continue;
+                    }
+                    // Check if job should be delayed
+                    if (job.scheduledFor && new Date(job.scheduledFor) > new Date()) {
+                        const delayMs = new Date(job.scheduledFor).getTime() - Date.now();
+                        console.log('[WorkerQueue] Job not ready yet, re-queuing', {
+                            id: job.id,
+                            type: job.type,
+                            delayMs,
+                        });
+                        msg.nak(Math.min(delayMs, 30_000)); // NAK with delay, max 30s (NATS limit)
+                        continue;
+                    }
+                    try {
+                        const jobWithWorking = {
+                            ...job,
+                            working: () => {
+                                msg.working();
+                            },
+                        };
+                        await handler(jobWithWorking);
+                        msg.ack();
+                        console.log('[WorkerQueue] Job completed', {
+                            id: job.id,
+                            type: job.type,
+                        });
+                    }
+                    catch (err) {
+                        // Use NATS delivery count rather than job.attempts from the payload:
+                        // NATS re-delivers the *original* message bytes, so job.attempts in
+                        // the payload is always 0 on re-delivery and cannot track retries.
+                        const deliveryCount = msg.info.deliveryCount;
+                        if (deliveryCount >= maxRetries) {
+                            console.error('[WorkerQueue] Job failed after max retries, discarding', { id: job.id, type: job.type, deliveryCount, err });
+                            msg.ack(); // Discard
+                        }
+                        else {
+                            console.error('[WorkerQueue] Job failed, will retry', {
+                                id: job.id,
+                                type: job.type,
+                                deliveryCount,
+                                err,
+                            });
+                            msg.nak(5000); // Retry after 5 seconds
+                        }
+                    }
+                }
+                catch (err) {
+                    console.error('[WorkerQueue] Failed to parse job message', { err });
+                    msg.ack(); // Discard malformed messages
+                }
+            }
+        })().catch((err) => {
+            console.error('[WorkerQueue] Consumer loop error', { err });
+        });
+        console.log('[WorkerQueue] Started', {
+            streamName,
+            consumerName,
+            handlers: [...handlers.keys()],
+        });
+    }
+    async function stop() {
+        if (subscription) {
+            subscription.stop();
+            subscription = null;
+        }
+        console.log('[WorkerQueue] Stopped');
+    }
+    return { start, stop, enqueue, registerHandler };
+}
+//# sourceMappingURL=WorkerQueue.js.map

package/dist/server/WorkerQueue.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerQueue.js","sourceRoot":"","sources":["../../src/server/WorkerQueue.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,MAAM,CAAC;AAEhD,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,WAAW,CAAC;AA4B1E,MAAM,UAAU,iBAAiB,CAC/B,SAA4B,EAAE;IAS9B,MAAM,EACJ,UAAU,GAAG,MAAM,EACnB,eAAe,GAAG,IAAI,EACtB,WAAW,GAAG,EAAE,EAChB,UAAU,GAAG,CAAC,GACf,GAAG,MAAM,CAAC;IAEX,MAAM,QAAQ,GAAG,IAAI,GAAG,EAA+B,CAAC;IACxD,IAAI,YAAY,GAA4B,IAAI,CAAC;IAEjD,MAAM,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,KAAK,MAAM,CAAC;IAEhE,KAAK,UAAU,OAAO,CACpB,IAAY,EACZ,OAAU,EACV,OAA4B;QAE5B,MAAM,EAAE,GAAG,MAAM,YAAY,EAAE,CAAC;QAChC,MAAM,GAAG,GAAW;YAClB,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE;YACvB,IAAI;YACJ,OAAO;YACP,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,QAAQ,EAAE,CAAC;YACX,YAAY,EAAE,OAAO,EAAE,KAAK;gBAC1B,CAAC,CAAC,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,OAAO,CAAC,KAAK,CAAC;gBACtC,CAAC,CAAC,SAAS;YACb,OAAO,EAAE,GAAG,EAAE,GAAE,CAAC;SAClB,CAAC;QACF,MAAM,OAAO,GAAG,GAAG,UAAU,IAAI,IAAI,EAAE,CAAC;QACxC,MAAM,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;QACzE,OAAO,CAAC,GAAG,CAAC,4BAA4B,EAAE;YACxC,EAAE,EAAE,GAAG,CAAC,EAAE;YACV,IAAI;YACJ,YAAY,EAAE,GAAG,CAAC,YAAY;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,SAAS,eAAe,CAAI,IAAY,EAAE,OAAsB;QAC9D,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAA8B,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,UAAU,KAAK;QAClB,IAAI,eAAe,IAAI,CAAC,aAAa,EAAE,CAAC;YACtC,OAAO,CAAC,GAAG,CACT,qFAAqF,CACtF,CAAC;YACF,OAAO;QACT,CAAC;QAED,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;YACxB,OAAO,CAAC,GAAG,CACT,+DAA+D,CAChE,CAAC;YACF,OAAO;QACT,CAAC;QAED,iDAAiD;QACjD,MAAM,YAAY,CAAC;YACjB,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE,CAAC,GAAG,UAAU,IAAI,CAAC;SAC9B,CAAC,CAAC;QAEH,MAAM,EAAE,GAAG,MAAM,YAAY,EAAE,CAAC;QAChC,MAAM,IAAI,GAAG,iBAAiB,EAAE,CAAC;QACjC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,gBAAgB,EAAE,CAAC;QAE1C,iCAAiC;QACjC,MAAM,YAAY,GAAG,UAAU,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC;QAEpE,IAAI,CAAC;YACH,MAAM,GAAG,CAAC,SAAS,CAAC,IAAI,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QACrD,CAAC;QAAC,MAAM,CAAC;YACP,MAAM,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE;gBAClC,YAAY,EAAE,YAAY;gBAC1B,cAAc,EAAE,aAAa,CAAC,GAAG;gBACjC,UAAU,EAAE,SAAS,CAAC,QAAQ;gBAC9B,eAAe,EAAE,WAAW;aAC7B,CAAC,CAAC;QACL,CAAC;QAED,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QAClE,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,OAAO,EAAE,CAAC;QAE1C,YAAY,GAAG,QAAQ,CAAC;QAExB,mBAAmB;QACnB,CAAC,KAAK,IAAI,EAAE;YACV,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;gBACjC,IAAI,CAAC;oBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAQ,CAAC;oBAClE,MAAM,OAAO,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBAEvC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,OAAO,CAAC,IAAI,CAAC,mDAAmD,EAAE;4BAChE,IAAI,EAAE,GAAG,CAAC,IAAI;4BACd,EAAE,EAAE,GAAG,CAAC,EAAE;yBACX,CAAC,CAAC;wBACH,GAAG,CAAC,GAAG,EAAE,CAAC;wBACV,SAAS;oBACX,CAAC;oBAED,iCAAiC;oBACjC,IAAI,GAAG,CAAC,YAAY,IAAI,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC;wBAChE,MAAM,OAAO,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,OAAO,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;wBAClE,OAAO,CAAC,GAAG,CAAC,6CAA6C,EAAE;4BACzD,EAAE,EAAE,GAAG,CAAC,EAAE;4BACV,IAAI,EAAE,GAAG,CAAC,IAAI;4BACd,OAAO;yBACR,CAAC,CAAC;wBACH,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,uCAAuC;wBAC3E,SAAS;oBACX,CAAC;oBAED,IAAI,CAAC;wBACH,MAAM,cAAc,GAAQ;4BAC1B,GAAG,GAAG;4BACN,OAAO,EAAE,GAAG,EAAE;gCACZ,GAAG,CAAC,OAAO,EAAE,CAAC;4BAChB,CAAC;yBACF,CAAC;wBACF,MAAM,OAAO,CAAC,cAAc,CAAC,CAAC;wBAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;wBACV,OAAO,CAAC,GAAG,CAAC,6BAA6B,EAAE;4BACzC,EAAE,EAAE,GAAG,CAAC,EAAE;4BACV,IAAI,EAAE,GAAG,CAAC,IAAI;yBACf,CAAC,CAAC;oBACL,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,qEAAqE;wBACrE,oEAAoE;wBACpE,mEAAmE;wBACnE,MAAM,aAAa,GAAG,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC;wBAC7C,IAAI,aAAa,IAAI,UAAU,EAAE,CAAC;4BAChC,OAAO,CAAC,KAAK,CACX,wDAAwD,EACxD,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,CACnD,CAAC;4BACF,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,UAAU;wBACvB,CAAC;6BAAM,CAAC;4BACN,OAAO,CAAC,KAAK,CAAC,sCAAsC,EAAE;gCACpD,EAAE,EAAE,GAAG,CAAC,EAAE;gCACV,IAAI,EAAE,GAAG,CAAC,IAAI;gCACd,aAAa;gCACb,GAAG;6BACJ,CAAC,CAAC;4BACH,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,wBAAwB;wBACzC,CAAC;oBACH,CAAC;gBACH,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,OAAO,CAAC,KAAK,CAAC,2CAA2C,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;oBACpE,GAAG,CAAC,GAAG,EAAE,CAAC,CAAC,6BAA6B;gBAC1C,CAAC;YACH,CAAC;QACH,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;YAC1B,OAAO,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,OAAO,CAAC,GAAG,CAAC,uBAAuB,EAAE;YACnC,UAAU;YACV,YAAY;YACZ,QAAQ,EAAE,CAAC,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;SAC/B,CAAC,CAAC;IACL,CAAC;IAED,KAAK,UAAU,IAAI;QACjB,IAAI,YAAY,EAAE,CAAC;YACjB,YAAY,CAAC,IAAI,EAAE,CAAC;YACpB,YAAY,GAAG,IAAI,CAAC;QACtB,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;IACvC,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC;AACnD,CAAC"}

package/dist/server/ai/ImageGenClient.d.ts

@@ -0,0 +1,6 @@
+import type { ImageGenOptions } from './types.js';
+export interface ImageGenClient {
+    generate(prompt: string, options?: ImageGenOptions): Promise<string>;
+}
+export declare function createImageGenClient(userId: string, defaults?: ImageGenOptions): ImageGenClient;
+//# sourceMappingURL=ImageGenClient.d.ts.map

package/dist/server/ai/ImageGenClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ImageGenClient.d.ts","sourceRoot":"","sources":["../../../src/server/ai/ImageGenClient.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAGlD,MAAM,WAAW,cAAc;IAC7B,QAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CACtE;AAED,wBAAgB,oBAAoB,CAClC,MAAM,EAAE,MAAM,EACd,QAAQ,CAAC,EAAE,eAAe,GACzB,cAAc,CAyBhB"}

package/dist/server/ai/ImageGenClient.js

@@ -0,0 +1,29 @@
+// src/server/ai/ImageGenClient.ts
+import { getBillingGateway } from '../billing/BillingGateway.js';
+import { selectImageGenProvider } from './ProviderSelector.js';
+import { TextGenNoProviderError } from './types.js';
+export function createImageGenClient(userId, defaults) {
+    return {
+        async generate(prompt, options) {
+            const opts = {
+                provider: 'auto',
+                ...defaults,
+                ...options,
+            };
+            const provider = selectImageGenProvider({
+                provider: opts.provider ?? 'auto',
+            });
+            if (!provider)
+                throw new TextGenNoProviderError('No image gen provider available — set TOGETHER_API_KEY, FAL_API_KEY, or GOOGLE_API_KEY.');
+            await getBillingGateway().charge({
+                userId,
+                provider: provider.name,
+                model: provider.name,
+                type: 'imageGen',
+                costUsd: 0.003,
+            });
+            return provider.generate(prompt, opts);
+        },
+    };
+}
+//# sourceMappingURL=ImageGenClient.js.map

package/dist/server/ai/ImageGenClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ImageGenClient.js","sourceRoot":"","sources":["../../../src/server/ai/ImageGenClient.ts"],"names":[],"mappings":"AAAA,kCAAkC;AAClC,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAMpD,MAAM,UAAU,oBAAoB,CAClC,MAAc,EACd,QAA0B;IAE1B,OAAO;QACL,KAAK,CAAC,QAAQ,CAAC,MAAM,EAAE,OAAO;YAC5B,MAAM,IAAI,GAAoB;gBAC5B,QAAQ,EAAE,MAAM;gBAChB,GAAG,QAAQ;gBACX,GAAG,OAAO;aACX,CAAC;YACF,MAAM,QAAQ,GAAG,sBAAsB,CAAC;gBACtC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,MAAM;aAClC,CAAC,CAAC;YACH,IAAI,CAAC,QAAQ;gBACX,MAAM,IAAI,sBAAsB,CAC9B,yFAAyF,CAC1F,CAAC;YACJ,MAAM,iBAAiB,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,KAAK,EAAE,QAAQ,CAAC,IAAI;gBACpB,IAAI,EAAE,UAAU;gBAChB,OAAO,EAAE,KAAK;aACf,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QACzC,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/server/ai/ProviderBalance.d.ts

@@ -0,0 +1,15 @@
+export interface ProviderBalance {
+    provider: string;
+    externalBalance: number | null;
+    usageToday: number;
+    usageThisMonth: number;
+    requestsToday: number;
+    requestsThisMonth: number;
+    rateLimitStatus: {
+        remainingPoints: number;
+        totalPoints: number;
+    } | null;
+    healthy: boolean;
+}
+export declare function getProviderBalances(): Promise<ProviderBalance[]>;
+//# sourceMappingURL=ProviderBalance.d.ts.map

package/dist/server/ai/ProviderBalance.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderBalance.d.ts","sourceRoot":"","sources":["../../../src/server/ai/ProviderBalance.ts"],"names":[],"mappings":"AAIA,MAAM,WAAW,eAAe;IAC9B,QAAQ,EAAE,MAAM,CAAC;IACjB,eAAe,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,cAAc,EAAE,MAAM,CAAC;IACvB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,eAAe,EAAE;QAAE,eAAe,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IACzE,OAAO,EAAE,OAAO,CAAC;CAClB;AAgHD,wBAAsB,mBAAmB,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC,CAsBtE"}

package/dist/server/ai/ProviderBalance.js

@@ -0,0 +1,110 @@
+// src/server/ai/ProviderBalance.ts
+import { getDbName, getMongoClient } from '../DB.js';
+import { getAllTextGenProviders } from './registry.js';
+// Provider API key environment variable names
+const providerApiKeyEnvs = {
+    together: 'TOGETHER_API_KEY',
+    kie: 'KIE_API_KEY',
+};
+// ---- Private balance fetchers ----
+async function fetchTogetherBalance() {
+    const apiKey = process.env[providerApiKeyEnvs['together'] ?? ''];
+    if (!apiKey)
+        return null;
+    try {
+        const res = await fetch('https://api.together.xyz/v1/credits', {
+            method: 'GET',
+            headers: { Authorization: `Bearer ${apiKey}` },
+        });
+        if (!res.ok)
+            return null;
+        const data = (await res.json());
+        if (data.available_credit == null)
+            return null;
+        // Together returns credits in dollars; convert to nano-dollars
+        return data.available_credit * 1_000_000_000;
+    }
+    catch {
+        return null;
+    }
+}
+async function fetchKieBalance() {
+    const apiKey = process.env[providerApiKeyEnvs['kie'] ?? ''];
+    if (!apiKey)
+        return null;
+    try {
+        const res = await fetch('https://api.kie.ai/api/v1/chat/credit', {
+            method: 'GET',
+            headers: { Authorization: `Bearer ${apiKey}` },
+        });
+        if (!res.ok)
+            return null;
+        const data = (await res.json());
+        if (data.credit == null)
+            return null;
+        // Kie.ai returns credit in dollars; convert to nano-dollars
+        return data.credit * 1_000_000_000;
+    }
+    catch {
+        return null;
+    }
+}
+const balanceFetchers = {
+    together: fetchTogetherBalance,
+    kie: fetchKieBalance,
+};
+// ---- Usage stats from MongoDB ----
+async function getProviderUsageStats(provider) {
+    const now = new Date();
+    const startOfDay = new Date(now.getFullYear(), now.getMonth(), now.getDate());
+    const startOfMonth = new Date(now.getFullYear(), now.getMonth(), 1);
+    try {
+        const mongo = await getMongoClient();
+        const db = mongo.db(getDbName());
+        const collection = db.collection('billingCharge');
+        const [todayCharges, monthCharges] = await Promise.all([
+            collection
+                .find({ provider, created: { $gte: startOfDay.getTime() } })
+                .toArray(),
+            collection
+                .find({ provider, created: { $gte: startOfMonth.getTime() } })
+                .toArray(),
+        ]);
+        const usageToday = todayCharges.reduce((sum, c) => sum + (c.priceNanoDollars ?? 0), 0);
+        const usageThisMonth = monthCharges.reduce((sum, c) => sum + (c.priceNanoDollars ?? 0), 0);
+        return {
+            usageToday,
+            usageThisMonth,
+            requestsToday: todayCharges.length,
+            requestsThisMonth: monthCharges.length,
+        };
+    }
+    catch {
+        return {
+            usageToday: 0,
+            usageThisMonth: 0,
+            requestsToday: 0,
+            requestsThisMonth: 0,
+        };
+    }
+}
+// ---- Public API ----
+export async function getProviderBalances() {
+    const providers = getAllTextGenProviders();
+    return Promise.all(providers.map(async (p) => {
+        const fetcher = balanceFetchers[p.name];
+        const externalBalance = fetcher ? await fetcher() : null;
+        const usage = await getProviderUsageStats(p.name);
+        return {
+            provider: p.name,
+            externalBalance,
+            usageToday: usage.usageToday,
+            usageThisMonth: usage.usageThisMonth,
+            requestsToday: usage.requestsToday,
+            requestsThisMonth: usage.requestsThisMonth,
+            rateLimitStatus: null,
+            healthy: externalBalance === null || externalBalance > 0,
+        };
+    }));
+}
+//# sourceMappingURL=ProviderBalance.js.map

package/dist/server/ai/ProviderBalance.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderBalance.js","sourceRoot":"","sources":["../../../src/server/ai/ProviderBalance.ts"],"names":[],"mappings":"AAAA,mCAAmC;AACnC,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AACrD,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAmBvD,8CAA8C;AAC9C,MAAM,kBAAkB,GAA2B;IACjD,QAAQ,EAAE,kBAAkB;IAC5B,GAAG,EAAE,aAAa;CACnB,CAAC;AAEF,qCAAqC;AAErC,KAAK,UAAU,oBAAoB;IACjC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;IACjE,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAEzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,qCAAqC,EAAE;YAC7D,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;SAC/C,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QACzB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAkC,CAAC;QACjE,IAAI,IAAI,CAAC,gBAAgB,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QAC/C,+DAA+D;QAC/D,OAAO,IAAI,CAAC,gBAAgB,GAAG,aAAa,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,KAAK,UAAU,eAAe;IAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAC5D,IAAI,CAAC,MAAM;QAAE,OAAO,IAAI,CAAC;IAEzB,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,uCAAuC,EAAE;YAC/D,MAAM,EAAE,KAAK;YACb,OAAO,EAAE,EAAE,aAAa,EAAE,UAAU,MAAM,EAAE,EAAE;SAC/C,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QACzB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAwB,CAAC;QACvD,IAAI,IAAI,CAAC,MAAM,IAAI,IAAI;YAAE,OAAO,IAAI,CAAC;QACrC,4DAA4D;QAC5D,OAAO,IAAI,CAAC,MAAM,GAAG,aAAa,CAAC;IACrC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,MAAM,eAAe,GAAiD;IACpE,QAAQ,EAAE,oBAAoB;IAC9B,GAAG,EAAE,eAAe;CACrB,CAAC;AAEF,qCAAqC;AAErC,KAAK,UAAU,qBAAqB,CAAC,QAAgB;IAMnD,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;IACvB,MAAM,UAAU,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;IAC9E,MAAM,YAAY,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,CAAC;IAEpE,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,MAAM,cAAc,EAAE,CAAC;QACrC,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;QACjC,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAmB,eAAe,CAAC,CAAC;QAEpE,MAAM,CAAC,YAAY,EAAE,YAAY,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;YACrD,UAAU;iBACP,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;iBAC3D,OAAO,EAAE;YACZ,UAAU;iBACP,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE,EAAE,EAAE,CAAC;iBAC7D,OAAO,EAAE;SACb,CAAC,CAAC;QAEH,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CACpC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,gBAAgB,IAAI,CAAC,CAAC,EAC3C,CAAC,CACF,CAAC;QACF,MAAM,cAAc,GAAG,YAAY,CAAC,MAAM,CACxC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,gBAAgB,IAAI,CAAC,CAAC,EAC3C,CAAC,CACF,CAAC;QAEF,OAAO;YACL,UAAU;YACV,cAAc;YACd,aAAa,EAAE,YAAY,CAAC,MAAM;YAClC,iBAAiB,EAAE,YAAY,CAAC,MAAM;SACvC,CAAC;IACJ,CAAC;IAAC,MAAM,CAAC;QACP,OAAO;YACL,UAAU,EAAE,CAAC;YACb,cAAc,EAAE,CAAC;YACjB,aAAa,EAAE,CAAC;YAChB,iBAAiB,EAAE,CAAC;SACrB,CAAC;IACJ,CAAC;AACH,CAAC;AAED,uBAAuB;AAEvB,MAAM,CAAC,KAAK,UAAU,mBAAmB;IACvC,MAAM,SAAS,GAAG,sBAAsB,EAAE,CAAC;IAE3C,OAAO,OAAO,CAAC,GAAG,CAChB,SAAS,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE;QACxB,MAAM,OAAO,GAAG,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QACxC,MAAM,eAAe,GAAG,OAAO,CAAC,CAAC,CAAC,MAAM,OAAO,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QAEzD,MAAM,KAAK,GAAG,MAAM,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAElD,OAAO;YACL,QAAQ,EAAE,CAAC,CAAC,IAAI;YAChB,eAAe;YACf,UAAU,EAAE,KAAK,CAAC,UAAU;YAC5B,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,aAAa,EAAE,KAAK,CAAC,aAAa;YAClC,iBAAiB,EAAE,KAAK,CAAC,iBAAiB;YAC1C,eAAe,EAAE,IAAI;YACrB,OAAO,EAAE,eAAe,KAAK,IAAI,IAAI,eAAe,GAAG,CAAC;SACzD,CAAC;IACJ,CAAC,CAAC,CACH,CAAC;AACJ,CAAC"}

package/dist/server/ai/ProviderSelector.d.ts

@@ -0,0 +1,13 @@
+import type { ImageGenOptions, ImageGenProvider, TextGenOptions, TextGenProvider } from './types.js';
+interface SelectTextGenOpts {
+    provider?: TextGenOptions['provider'];
+    needsVision?: boolean;
+    needsTools?: boolean;
+    needsJson?: boolean;
+}
+export declare function selectTextGenProvider(opts: SelectTextGenOpts): TextGenProvider | undefined;
+export declare function selectImageGenProvider(opts: {
+    provider?: ImageGenOptions['provider'];
+}): ImageGenProvider | undefined;
+export {};
+//# sourceMappingURL=ProviderSelector.d.ts.map

package/dist/server/ai/ProviderSelector.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderSelector.d.ts","sourceRoot":"","sources":["../../../src/server/ai/ProviderSelector.ts"],"names":[],"mappings":"AAOA,OAAO,KAAK,EACV,eAAe,EACf,gBAAgB,EAChB,cAAc,EACd,eAAe,EAChB,MAAM,YAAY,CAAC;AAMpB,UAAU,iBAAiB;IACzB,QAAQ,CAAC,EAAE,cAAc,CAAC,UAAU,CAAC,CAAC;IACtC,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,CAAC,EAAE,OAAO,CAAC;IACrB,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED,wBAAgB,qBAAqB,CACnC,IAAI,EAAE,iBAAiB,GACtB,eAAe,GAAG,SAAS,CAa7B;AAED,wBAAgB,sBAAsB,CAAC,IAAI,EAAE;IAC3C,QAAQ,CAAC,EAAE,eAAe,CAAC,UAAU,CAAC,CAAC;CACxC,GAAG,gBAAgB,GAAG,SAAS,CAM/B"}

package/dist/server/ai/ProviderSelector.js

@@ -0,0 +1,25 @@
+// src/server/ai/ProviderSelector.ts
+import { getAllImageGenProviders, getAllTextGenProviders, getImageGenProvider, getTextGenProvider, } from './registry.js';
+function hasKey(p) {
+    return !!process.env[p.apiKeyEnv];
+}
+export function selectTextGenProvider(opts) {
+    if (opts.provider && opts.provider !== 'auto') {
+        const p = getTextGenProvider(opts.provider);
+        return p && hasKey(p) ? p : undefined;
+    }
+    return getAllTextGenProviders()
+        .filter((p) => hasKey(p))
+        .filter((p) => !opts.needsVision || p.capabilities.vision)
+        .filter((p) => !opts.needsTools || p.capabilities.tools)
+        .filter((p) => !opts.needsJson || p.capabilities.json)
+        .sort((a, b) => a.costPerMillionInputTokens - b.costPerMillionInputTokens)[0];
+}
+export function selectImageGenProvider(opts) {
+    if (opts.provider && opts.provider !== 'auto') {
+        const p = getImageGenProvider(opts.provider);
+        return p && hasKey(p) ? p : undefined;
+    }
+    return getAllImageGenProviders().find((p) => hasKey(p));
+}
+//# sourceMappingURL=ProviderSelector.js.map

package/dist/server/ai/ProviderSelector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ProviderSelector.js","sourceRoot":"","sources":["../../../src/server/ai/ProviderSelector.ts"],"names":[],"mappings":"AAAA,oCAAoC;AACpC,OAAO,EACL,uBAAuB,EACvB,sBAAsB,EACtB,mBAAmB,EACnB,kBAAkB,GACnB,MAAM,eAAe,CAAC;AAQvB,SAAS,MAAM,CAAC,CAAwB;IACtC,OAAO,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;AACpC,CAAC;AASD,MAAM,UAAU,qBAAqB,CACnC,IAAuB;IAEvB,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;QAC9C,MAAM,CAAC,GAAG,kBAAkB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC5C,OAAO,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IACD,OAAO,sBAAsB,EAAE;SAC5B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;SACxB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,WAAW,IAAI,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC;SACzD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC,YAAY,CAAC,KAAK,CAAC;SACvD,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC;SACrD,IAAI,CACH,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,yBAAyB,GAAG,CAAC,CAAC,yBAAyB,CACpE,CAAC,CAAC,CAAC,CAAC;AACT,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,IAEtC;IACC,IAAI,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,QAAQ,KAAK,MAAM,EAAE,CAAC;QAC9C,MAAM,CAAC,GAAG,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7C,OAAO,CAAC,IAAI,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACxC,CAAC;IACD,OAAO,uBAAuB,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;AAC1D,CAAC"}

package/dist/server/ai/TextGenClient.d.ts

@@ -0,0 +1,9 @@
+import { z } from 'zod';
+import type { Message, TextGenOptions, ToolCallResult, ToolSet } from './types.js';
+export interface TextGenClient {
+    generate(messages: Message[], options?: TextGenOptions): Promise<string>;
+    generateJson<T extends z.ZodTypeAny>(schema: T, messages: Message[], options?: TextGenOptions): Promise<z.infer<T>>;
+    generateWithTools(messages: Message[], tools: ToolSet, options?: TextGenOptions): Promise<ToolCallResult>;
+}
+export declare function createTextGenClient(userId: string, defaults?: TextGenOptions): TextGenClient;
+//# sourceMappingURL=TextGenClient.d.ts.map

package/dist/server/ai/TextGenClient.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TextGenClient.d.ts","sourceRoot":"","sources":["../../../src/server/ai/TextGenClient.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAOxB,OAAO,KAAK,EACV,OAAO,EACP,cAAc,EACd,cAAc,EACd,OAAO,EACR,MAAM,YAAY,CAAC;AAGpB,MAAM,WAAW,aAAa;IAC5B,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACzE,YAAY,CAAC,CAAC,SAAS,CAAC,CAAC,UAAU,EACjC,MAAM,EAAE,CAAC,EACT,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACvB,iBAAiB,CACf,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,OAAO,EACd,OAAO,CAAC,EAAE,cAAc,GACvB,OAAO,CAAC,cAAc,CAAC,CAAC;CAC5B;AAoBD,wBAAgB,mBAAmB,CACjC,MAAM,EAAE,MAAM,EACd,QAAQ,CAAC,EAAE,cAAc,GACxB,aAAa,CAuFf"}

package/dist/server/ai/TextGenClient.js

@@ -0,0 +1,72 @@
+import { getBillingGateway } from '../billing/BillingGateway.js';
+import { selectTextGenProvider } from './ProviderSelector.js';
+import { generateJsonFallback, generateWithToolsFallback, } from './fallbacks.js';
+import { TextGenNoProviderError } from './types.js';
+function estimateCost(messages, maxTokens, inputCost, outputCost) {
+    const inputTokens = messages.reduce((n, m) => {
+        const text = typeof m.content === 'string'
+            ? m.content
+            : m.content.map((p) => p.text ?? '').join('');
+        return n + Math.ceil(text.length / 4);
+    }, 0);
+    return ((inputTokens * inputCost) / 1_000_000 + (maxTokens * outputCost) / 1_000_000);
+}
+export function createTextGenClient(userId, defaults) {
+    function resolve(options, needsVision, needsTools, needsJson) {
+        const opts = { ...defaults, ...options };
+        const provider = selectTextGenProvider({
+            provider: opts.provider ?? 'auto',
+            needsVision,
+            needsTools,
+            needsJson,
+        });
+        if (!provider)
+            throw new TextGenNoProviderError('No provider available — check that at least one API key is configured.');
+        return { provider, opts };
+    }
+    return {
+        async generate(messages, options) {
+            const hasVision = messages.some((m) => Array.isArray(m.content) &&
+                m.content.some((p) => p.type === 'image_url'));
+            const { provider, opts } = resolve(options, hasVision);
+            const cost = estimateCost(messages, opts.maxTokens ?? 1024, provider.costPerMillionInputTokens, provider.costPerMillionOutputTokens);
+            await getBillingGateway().charge({
+                userId,
+                provider: provider.name,
+                model: provider.name,
+                type: 'textGen',
+                costUsd: cost,
+            });
+            return provider.generate(messages, opts);
+        },
+        async generateJson(schema, messages, options) {
+            const { provider, opts } = resolve(options, false, false, true);
+            const cost = estimateCost(messages, opts.maxTokens ?? 1024, provider.costPerMillionInputTokens, provider.costPerMillionOutputTokens);
+            await getBillingGateway().charge({
+                userId,
+                provider: provider.name,
+                model: provider.name,
+                type: 'textGen',
+                costUsd: cost,
+            });
+            if (provider.generateJson)
+                return provider.generateJson(schema, messages, opts);
+            return generateJsonFallback(schema, messages, provider);
+        },
+        async generateWithTools(messages, tools, options) {
+            const { provider, opts } = resolve(options, false, true);
+            const cost = estimateCost(messages, opts.maxTokens ?? 1024, provider.costPerMillionInputTokens, provider.costPerMillionOutputTokens);
+            await getBillingGateway().charge({
+                userId,
+                provider: provider.name,
+                model: provider.name,
+                type: 'textGen',
+                costUsd: cost,
+            });
+            if (provider.generateWithTools)
+                return provider.generateWithTools(messages, tools, opts);
+            return generateWithToolsFallback(messages, tools, provider);
+        },
+    };
+}
+//# sourceMappingURL=TextGenClient.js.map

package/dist/server/ai/TextGenClient.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"TextGenClient.js","sourceRoot":"","sources":["../../../src/server/ai/TextGenClient.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,iBAAiB,EAAE,MAAM,8BAA8B,CAAC;AACjE,OAAO,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,EACL,oBAAoB,EACpB,yBAAyB,GAC1B,MAAM,gBAAgB,CAAC;AAOxB,OAAO,EAAE,sBAAsB,EAAE,MAAM,YAAY,CAAC;AAgBpD,SAAS,YAAY,CACnB,QAAmB,EACnB,SAAiB,EACjB,SAAiB,EACjB,UAAkB;IAElB,MAAM,WAAW,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;QAC3C,MAAM,IAAI,GACR,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ;YAC3B,CAAC,CAAC,CAAC,CAAC,OAAO;YACX,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClD,OAAO,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACxC,CAAC,EAAE,CAAC,CAAC,CAAC;IACN,OAAO,CACL,CAAC,WAAW,GAAG,SAAS,CAAC,GAAG,SAAS,GAAG,CAAC,SAAS,GAAG,UAAU,CAAC,GAAG,SAAS,CAC7E,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,MAAc,EACd,QAAyB;IAEzB,SAAS,OAAO,CACd,OAAwB,EACxB,WAAqB,EACrB,UAAoB,EACpB,SAAmB;QAEnB,MAAM,IAAI,GAAG,EAAE,GAAG,QAAQ,EAAE,GAAG,OAAO,EAAE,CAAC;QACzC,MAAM,QAAQ,GAAG,qBAAqB,CAAC;YACrC,QAAQ,EAAE,IAAI,CAAC,QAAQ,IAAI,MAAM;YACjC,WAAW;YACX,UAAU;YACV,SAAS;SACV,CAAC,CAAC;QACH,IAAI,CAAC,QAAQ;YACX,MAAM,IAAI,sBAAsB,CAC9B,wEAAwE,CACzE,CAAC;QACJ,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC;IAC5B,CAAC;IAED,OAAO;QACL,KAAK,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO;YAC9B,MAAM,SAAS,GAAG,QAAQ,CAAC,IAAI,CAC7B,CAAC,CAAC,EAAE,EAAE,CACJ,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC;gBACxB,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAChD,CAAC;YACF,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACvD,MAAM,IAAI,GAAG,YAAY,CACvB,QAAQ,EACR,IAAI,CAAC,SAAS,IAAI,IAAI,EACtB,QAAQ,CAAC,yBAAyB,EAClC,QAAQ,CAAC,0BAA0B,CACpC,CAAC;YACF,MAAM,iBAAiB,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,KAAK,EAAE,QAAQ,CAAC,IAAI;gBACpB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,OAAO,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,CAAC;QAC3C,CAAC;QAED,KAAK,CAAC,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE,OAAO;YAC1C,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,YAAY,CACvB,QAAQ,EACR,IAAI,CAAC,SAAS,IAAI,IAAI,EACtB,QAAQ,CAAC,yBAAyB,EAClC,QAAQ,CAAC,0BAA0B,CACpC,CAAC;YACF,MAAM,iBAAiB,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,KAAK,EAAE,QAAQ,CAAC,IAAI;gBACpB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,YAAY;gBACvB,OAAO,QAAQ,CAAC,YAAY,CAAC,MAAM,EAAE,QAAQ,EAAE,IAAI,CAElD,CAAC;YACJ,OAAO,oBAAoB,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1D,CAAC;QAED,KAAK,CAAC,iBAAiB,CAAC,QAAQ,EAAE,KAAK,EAAE,OAAO;YAC9C,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YACzD,MAAM,IAAI,GAAG,YAAY,CACvB,QAAQ,EACR,IAAI,CAAC,SAAS,IAAI,IAAI,EACtB,QAAQ,CAAC,yBAAyB,EAClC,QAAQ,CAAC,0BAA0B,CACpC,CAAC;YACF,MAAM,iBAAiB,EAAE,CAAC,MAAM,CAAC;gBAC/B,MAAM;gBACN,QAAQ,EAAE,QAAQ,CAAC,IAAI;gBACvB,KAAK,EAAE,QAAQ,CAAC,IAAI;gBACpB,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,IAAI;aACd,CAAC,CAAC;YACH,IAAI,QAAQ,CAAC,iBAAiB;gBAC5B,OAAO,QAAQ,CAAC,iBAAiB,CAAC,QAAQ,EAAE,KAAK,EAAE,IAAI,CAAC,CAAC;YAC3D,OAAO,yBAAyB,CAAC,QAAQ,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QAC9D,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/server/ai/fallbacks.d.ts

@@ -0,0 +1,9 @@
+import { z } from 'zod';
+import type { Message, ToolCallResult, ToolSet } from './types.js';
+interface MinimalProvider {
+    generate(messages: Message[]): Promise<string>;
+}
+export declare function generateJsonFallback<T extends z.ZodTypeAny>(schema: T, messages: Message[], provider: MinimalProvider): Promise<z.infer<T>>;
+export declare function generateWithToolsFallback(messages: Message[], tools: ToolSet, provider: MinimalProvider, maxRounds?: number): Promise<ToolCallResult>;
+export {};
+//# sourceMappingURL=fallbacks.d.ts.map

package/dist/server/ai/fallbacks.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"fallbacks.d.ts","sourceRoot":"","sources":["../../../src/server/ai/fallbacks.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AAGnE,UAAU,eAAe;IACvB,QAAQ,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;CAChD;AAED,wBAAsB,oBAAoB,CAAC,CAAC,SAAS,CAAC,CAAC,UAAU,EAC/D,MAAM,EAAE,CAAC,EACT,QAAQ,EAAE,OAAO,EAAE,EACnB,QAAQ,EAAE,eAAe,GACxB,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAsCrB;AAED,wBAAsB,yBAAyB,CAC7C,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,OAAO,EACd,QAAQ,EAAE,eAAe,EACzB,SAAS,SAAI,GACZ,OAAO,CAAC,cAAc,CAAC,CA0CzB"}

package/dist/server/ai/fallbacks.js

@@ -0,0 +1,77 @@
+// src/server/ai/fallbacks.ts
+import { z } from 'zod';
+import { TextGenSchemaError } from './types.js';
+export async function generateJsonFallback(schema, messages, provider) {
+    const schemaInstruction = `\n\nRespond with valid JSON only matching this schema:\n${JSON.stringify(z.toJSONSchema(schema), null, 2)}`;
+    const augmented = messages.map((m, i) => i === messages.length - 1 && m.role === 'user'
+        ? { ...m, content: toText(m.content) + schemaInstruction }
+        : m);
+    let lastError;
+    let lastRaw = '';
+    for (let attempt = 0; attempt < 3; attempt++) {
+        const msgs = attempt === 0
+            ? augmented
+            : [
+                ...augmented,
+                { role: 'assistant', content: lastRaw },
+                {
+                    role: 'user',
+                    content: `Invalid JSON or schema mismatch. Error: ${lastError}. Respond with valid JSON only.`,
+                },
+            ];
+        lastRaw = await provider.generate(msgs);
+        try {
+            const clean = lastRaw
+                .trim()
+                .replace(/^```json\n?/, '')
+                .replace(/\n?```$/, '');
+            return schema.parse(JSON.parse(clean));
+        }
+        catch (err) {
+            lastError = err;
+        }
+    }
+    throw new TextGenSchemaError(lastError, lastRaw);
+}
+export async function generateWithToolsFallback(messages, tools, provider, maxRounds = 5) {
+    if (messages.length === 0) {
+        throw new Error('[TextGen] generateWithToolsFallback requires at least one message');
+    }
+    const toolDesc = Object.entries(tools)
+        .map(([name, def]) => `TOOL: ${name}\nDescription: ${def.description}\nInput JSON schema: ${JSON.stringify(z.toJSONSchema(def.input))}`)
+        .join('\n\n');
+    const instruction = `\n\nAvailable tools:\n${toolDesc}\n\nTo call a tool respond with: TOOL_CALL:<name>:<json_input>\nTo give the final answer, respond normally.`;
+    const conv = [
+        ...messages.slice(0, -1),
+        {
+            role: messages.at(-1).role,
+            content: toText(messages.at(-1).content) + instruction,
+        },
+    ];
+    const allToolCalls = [];
+    for (let round = 0; round < maxRounds; round++) {
+        const response = await provider.generate(conv);
+        const match = response.match(/^TOOL_CALL:(\w+):(.+)$/m);
+        if (!match)
+            return { response, toolCalls: allToolCalls };
+        const [, name, argsJson] = match;
+        const tool = tools[name];
+        if (!tool)
+            return { response, toolCalls: allToolCalls };
+        const input = tool.input.parse(JSON.parse(argsJson));
+        const output = await tool.handler(input);
+        allToolCalls.push({ name: name, input, output });
+        conv.push({ role: 'assistant', content: response });
+        conv.push({
+            role: 'user',
+            content: `Tool result for ${name}: ${JSON.stringify(output)}`,
+        });
+    }
+    return { response: '', toolCalls: allToolCalls };
+}
+function toText(content) {
+    return typeof content === 'string'
+        ? content
+        : content.map((p) => p.text ?? '').join('');
+}
+//# sourceMappingURL=fallbacks.js.map

package/dist/server/ai/fallbacks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"fallbacks.js","sourceRoot":"","sources":["../../../src/server/ai/fallbacks.ts"],"names":[],"mappings":"AAAA,6BAA6B;AAC7B,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAMhD,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,MAAS,EACT,QAAmB,EACnB,QAAyB;IAEzB,MAAM,iBAAiB,GAAG,2DAA2D,IAAI,CAAC,SAAS,CACjG,CAAC,CAAC,YAAY,CAAC,MAAM,CAAC,EACtB,IAAI,EACJ,CAAC,CACF,EAAE,CAAC;IACJ,MAAM,SAAS,GAAc,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CACjD,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM;QAC5C,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,iBAAiB,EAAE;QAC1D,CAAC,CAAC,CAAC,CACN,CAAC;IAEF,IAAI,SAAkB,CAAC;IACvB,IAAI,OAAO,GAAG,EAAE,CAAC;IACjB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC7C,MAAM,IAAI,GACR,OAAO,KAAK,CAAC;YACX,CAAC,CAAC,SAAS;YACX,CAAC,CAAC;gBACE,GAAG,SAAS;gBACZ,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE,OAAO,EAAE;gBAChD;oBACE,IAAI,EAAE,MAAe;oBACrB,OAAO,EAAE,2CAA2C,SAAS,iCAAiC;iBAC/F;aACF,CAAC;QACR,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACxC,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,OAAO;iBAClB,IAAI,EAAE;iBACN,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC;iBAC1B,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAC1B,OAAO,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAe,CAAC;QACvD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,SAAS,GAAG,GAAG,CAAC;QAClB,CAAC;IACH,CAAC;IACD,MAAM,IAAI,kBAAkB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;AACnD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC7C,QAAmB,EACnB,KAAc,EACd,QAAyB,EACzB,SAAS,GAAG,CAAC;IAEb,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;IACJ,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC;SACnC,GAAG,CACF,CAAC,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,EAAE,CACd,SAAS,IAAI,kBACX,GAAG,CAAC,WACN,wBAAwB,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,EAAE,CACtE;SACA,IAAI,CAAC,MAAM,CAAC,CAAC;IAChB,MAAM,WAAW,GAAG,yBAAyB,QAAQ,6GAA6G,CAAC;IAEnK,MAAM,IAAI,GAAc;QACtB,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;QACxB;YACE,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,IAAI;YAC3B,OAAO,EAAE,MAAM,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,OAAO,CAAC,GAAG,WAAW;SAC7C;KACb,CAAC;IACF,MAAM,YAAY,GAAgC,EAAE,CAAC;IAErD,KAAK,IAAI,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACxD,IAAI,CAAC,KAAK;YAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;QACzD,MAAM,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,GAAG,KAAK,CAAC;QACjC,MAAM,IAAI,GAAG,KAAK,CAAC,IAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,IAAI;YAAE,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,QAAS,CAAC,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QACzC,YAAY,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAK,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,IAAI,CAAC;YACR,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,mBAAmB,IAAI,KAAK,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;SAC9D,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE,CAAC;AACnD,CAAC;AAED,SAAS,MAAM,CAAC,OAA2B;IACzC,OAAO,OAAO,OAAO,KAAK,QAAQ;QAChC,CAAC,CAAC,OAAO;QACT,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AAChD,CAAC"}

package/dist/server/ai/index.d.ts

@@ -0,0 +1,12 @@
+export { createImageGenClient } from './ImageGenClient.js';
+export type { ImageGenClient } from './ImageGenClient.js';
+export { fireworksTextProvider } from './providers/Fireworks.js';
+export { kieTextProvider } from './providers/Kie.js';
+export { kieImageProvider } from './providers/KieImage.js';
+export { wavespeedImageProvider } from './providers/Wavespeed.js';
+export { registerImageGenProvider, registerTextGenProvider, } from './registry.js';
+export { createTextGenClient } from './TextGenClient.js';
+export type { TextGenClient } from './TextGenClient.js';
+export { TextGenNoProviderError, TextGenSchemaError } from './types.js';
+export type { ImageGenOptions, ImageGenProvider, Message, TextGenOptions, TextGenProvider, ToolCallResult, ToolSet, } from './types.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/server/ai/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/server/ai/index.ts"],"names":[],"mappings":"AAiCA,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAC3D,YAAY,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,0BAA0B,CAAC;AACjE,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EACL,wBAAwB,EACxB,uBAAuB,GACxB,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,YAAY,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,EAAE,sBAAsB,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AACxE,YAAY,EACV,eAAe,EACf,gBAAgB,EAChB,OAAO,EACP,cAAc,EACd,eAAe,EACf,cAAc,EACd,OAAO,GACR,MAAM,YAAY,CAAC"}