@travisennis/acai 0.0.1 → 0.0.3

package/dist/source/messages.js

@@ -1,261 +0,0 @@
-import EventEmitter from "node:events";
-import { readdir, readFile, writeFile } from "node:fs/promises";
-import { join } from "node:path";
-import { isString } from "@travisennis/stdlib/typeguards";
-import { generateText, } from "ai";
-export function createUserMessage(contentItems, prompt) {
-    const messageParts = [];
-    // Process content items (images and pre-defined texts)
-    for (const item of contentItems) {
-        if (typeof item === "string") {
-            if (item.trim().length > 0) {
-                messageParts.push({ type: "text", text: item });
-            }
-        }
-        else if (item.type === "image") {
-            messageParts.push(item);
-        }
-    }
-    // Add the main prompt text if provided
-    if (prompt && prompt.trim().length > 0) {
-        messageParts.push({ type: "text", text: prompt });
-    }
-    return {
-        role: "user",
-        content: messageParts,
-    };
-}
-export function createAssistantMessage(content) {
-    return {
-        role: "assistant",
-        content: [
-            {
-                type: "text",
-                text: content,
-            },
-        ],
-    };
-}
-export class MessageHistory extends EventEmitter {
-    history;
-    title;
-    createdAt;
-    updatedAt;
-    stateDir;
-    modelManager;
-    tokenTracker;
-    constructor({ stateDir, modelManager, tokenTracker, }) {
-        super();
-        this.history = [];
-        this.title = "";
-        this.createdAt = new Date();
-        this.updatedAt = new Date();
-        this.stateDir = stateDir;
-        this.modelManager = modelManager;
-        this.tokenTracker = tokenTracker;
-    }
-    validMessage(msg) {
-        // Filter out messages with empty content arrays
-        if (Array.isArray(msg.content) && msg.content.length === 0) {
-            return false;
-        }
-        // Filter out assistant messages with empty text fields
-        if (msg.role === "assistant" &&
-            Array.isArray(msg.content) &&
-            msg.content.length === 1 &&
-            msg.content[0]?.type === "text" &&
-            msg.content[0]?.text === "") {
-            return false;
-        }
-        return true;
-    }
-    get() {
-        return [...this.history].filter(this.validMessage);
-    }
-    clear() {
-        this.history.length = 0;
-        this.emit("clear-history");
-    }
-    appendUserMessage(msg) {
-        const now = new Date();
-        const msgObj = isString(msg) ? createUserMessage([], msg) : msg;
-        if (this.history.length === 0 &&
-            msgObj.content &&
-            msgObj.content.length > 0) {
-            const textPart = msgObj.content.at(-1);
-            if (textPart?.text && textPart.text.trim() !== "") {
-                this.generateTitle(textPart.text);
-            }
-            this.createdAt = now;
-        }
-        this.updatedAt = now;
-        this.history.push(msgObj);
-    }
-    appendAssistantMessage(msg) {
-        this.updatedAt = new Date();
-        const msgObj = isString(msg) ? createAssistantMessage(msg) : msg;
-        this.history.push(msgObj);
-    }
-    appendResponseMessages(responseMessages) {
-        this.updatedAt = new Date();
-        // Filter out messages with empty content arrays
-        const validMessages = responseMessages.filter(this.validMessage);
-        this.history.push(...validMessages);
-    }
-    isEmpty() {
-        return this.history.length === 0;
-    }
-    async save() {
-        const msgHistoryDir = this.stateDir;
-        const timestamp = new Date().toISOString().replace(/:/g, "-");
-        const fileName = `message-history-${timestamp}.json`;
-        const filePath = join(msgHistoryDir, fileName);
-        const output = {
-            title: this.title,
-            createdAt: this.createdAt,
-            updatedAt: this.updatedAt,
-            messages: this.history,
-        };
-        await writeFile(filePath, JSON.stringify(output, null, 2));
-    }
-    async generateTitle(message) {
-        // Skip title generation if message is empty
-        if (!message || message.trim() === "") {
-            return;
-        }
-        const app = "title-conversation";
-        const systemPrompt = "You are an assistant who task is to analyze messages to generate a conversation topic that can be used as a conversation title. For each message, generate a 4-7 word title that captures the topic. Return only the title with no other text.\n\nExamples:\nMessage:\nHow do I implement authentication in my Express app?\nTitle: Express Authentication Implementation\n\nMessage:\nCan you help me debug this React component that isn't rendering correctly?\nTitle:React Component Rendering Debug";
-        try {
-            const { text, usage } = await generateText({
-                model: this.modelManager.getModel(app),
-                system: systemPrompt,
-                prompt: `Request:\n${message}\nTitle:`,
-            });
-            this.tokenTracker.trackUsage(app, usage);
-            if (text && text.split(" ").length < 10) {
-                this.title = text;
-                this.emit("update-title", this.title);
-            }
-        }
-        catch (error) {
-            console.error(error);
-        }
-    }
-    getFirstUserMessage() {
-        const firstUser = this.get().find((msg) => msg.role === "user");
-        return firstUser;
-    }
-    static async load(stateDir, count = 10) {
-        try {
-            const files = await readdir(stateDir);
-            const messageHistoryFiles = files
-                .filter((file) => file.startsWith("message-history-") && file.endsWith(".json"))
-                .sort((a, b) => {
-                // Extract timestamps and compare in reverse order (newest first)
-                const timeA = a.replace("message-history-", "").replace(".json", "");
-                const timeB = b.replace("message-history-", "").replace(".json", "");
-                return timeB.localeCompare(timeA); // Newest first
-            })
-                .slice(0, count); // Use the count parameter here
-            const fileReadPromises = messageHistoryFiles.map(async (fileName) => {
-                const filePath = join(stateDir, fileName);
-                try {
-                    const content = await readFile(filePath, "utf-8");
-                    const parsed = JSON.parse(content);
-                    const result = parsed;
-                    // Basic validation - ensure messages array exists
-                    if (parsed && Array.isArray(parsed.messages)) {
-                        // Convert date strings back to Date objects
-                        result.createdAt = new Date(parsed.createdAt ?? 0);
-                        result.updatedAt = new Date(parsed.updatedAt ?? 0);
-                        return result;
-                    }
-                }
-                catch (error) {
-                    console.error(`Error reading or parsing file ${filePath}:`, error);
-                }
-                return null; // Return null for failed reads/parses
-            });
-            const results = await Promise.all(fileReadPromises);
-            // Filter out null results (failed reads/parses)
-            //sort results by result.updatedAt which is a Date
-            return results
-                .filter((result) => result !== null)
-                .sort((a, b) => b.updatedAt.getTime() - a.updatedAt.getTime());
-        }
-        catch (error) {
-            // Handle cases where the directory might not exist or other readdir errors
-            if (error.code !== "ENOENT") {
-                console.error(`Error loading message history from ${stateDir}:`, error);
-            }
-            // Return empty array if directory doesn't exist or other read errors occur
-            return [];
-        }
-    }
-    // Method to restore state from a SavedMessageHistory object
-    restore(savedHistory) {
-        this.title = savedHistory.title;
-        // Ensure dates are Date objects, though load should handle this
-        this.createdAt =
-            typeof savedHistory.createdAt === "string"
-                ? new Date(savedHistory.createdAt)
-                : savedHistory.createdAt;
-        this.updatedAt =
-            typeof savedHistory.updatedAt === "string"
-                ? new Date(savedHistory.updatedAt)
-                : savedHistory.updatedAt;
-        this.history = [...savedHistory.messages]; // Use the correct internal property name and create a copy
-    }
-}
-/**
- * Normalizes an array of messages for API consumption by:
- * 1. Filtering out progress-type messages
- * 2. Processing user and assistant messages
- * 3. Handling tool results by either:
- *    - Adding them as new messages if they're the first tool result
- *    - Adding them as new messages if the previous message wasn't a tool result
- *    - Merging them with the previous message if it was also a tool result
- *
- * This consolidation of sequential tool results into a single message
- * ensures proper formatting for API consumption while maintaining the
- * logical flow of the conversation.
- *
- * @param messages - Array of messages to normalize
- * @returns Normalized array of user and assistant messages ready for API
- */
-export function normalizeMessagesForApi(messages) {
-    const result = [];
-    for (const message of messages) {
-        switch (message.role) {
-            case "user": {
-                result.push(message);
-                continue;
-            }
-            case "tool": {
-                // If the last message is not a tool result, add it to the result
-                const lastMessage = result.at(-1);
-                if (!lastMessage ||
-                    lastMessage.role === "assistant" ||
-                    !Array.isArray(lastMessage.content) ||
-                    lastMessage.content[0]?.type !== "tool-result") {
-                    result.push(message);
-                    continue;
-                }
-                // Otherwise, merge the current message with the last message
-                result[result.indexOf(lastMessage)] = {
-                    ...lastMessage,
-                    // biome-ignore lint/suspicious/noExplicitAny: can't figure out type
-                    content: [...lastMessage.content, ...message.content], // #FIXME figure out what type this should be
-                };
-                continue;
-            }
-            case "assistant": {
-                result.push(message);
-                continue;
-            }
-            default:
-                continue;
-        }
-    }
-    return result;
-}

package/dist/source/middleware/audit-message.d.ts

@@ -1,5 +0,0 @@
-import type { LanguageModelV2Middleware } from "@ai-sdk/provider";
-export declare const auditMessage: ({ filePath, app, }: {
-    filePath: string;
-    app: string;
-}) => LanguageModelV2Middleware;

package/dist/source/middleware/audit-message.js

@@ -1,95 +0,0 @@
-import { mkdir, writeFile } from "node:fs/promises";
-import { dirname, join } from "node:path";
-/**
- * Writes an audit record to the specified file, overwriting any existing content.
- * Ensures the directory exists before writing.
- *
- * @param filePath - The path to the file where the audit record will be saved.
- * @param content - The audit record object to write.
- */
-const writeAuditRecord = async (app, filePath, content) => {
-    try {
-        const now = new Date();
-        const path = join(filePath, `${now.toISOString()}-${app}-message.json`);
-        // Ensure directory exists
-        await mkdir(dirname(path), { recursive: true });
-        await writeFile(path, `${JSON.stringify(content, null, 2)}`);
-    }
-    catch (error) {
-        console.error("Error writing audit file:", error);
-        throw error;
-    }
-};
-export const auditMessage = ({ filePath = "messages", app = "default", }) => {
-    const middleware = {
-        wrapGenerate: async ({ doGenerate, params, model }) => {
-            const result = await doGenerate();
-            const msg = {
-                model: model.modelId,
-                app,
-                messages: [...params.prompt].concat({
-                    role: "assistant",
-                    content: [
-                        {
-                            type: "text",
-                            // biome-ignore lint/suspicious/noExplicitAny: work-around on type issue
-                            text: result.text,
-                        },
-                    ],
-                }),
-                usage: result.usage,
-                timestamp: Date.now(),
-            };
-            await writeAuditRecord(app, filePath, msg);
-            return result;
-        },
-        wrapStream: async ({ doStream, params, model }) => {
-            const { stream, ...rest } = await doStream();
-            let generatedText = "";
-            let usage = {
-                inputTokens: 0,
-                outputTokens: 0,
-                totalTokens: 0,
-            };
-            const transformStream = new TransformStream({
-                transform(chunk, controller) {
-                    if (chunk.type === "text-delta") {
-                        generatedText += chunk.delta;
-                    }
-                    if (chunk.type === "finish") {
-                        usage = chunk.usage;
-                    }
-                    controller.enqueue(chunk);
-                },
-                async flush() {
-                    const msg = {
-                        model: model.modelId,
-                        app,
-                        messages: generatedText
-                            ? [
-                                ...params.prompt,
-                                {
-                                    role: "assistant",
-                                    content: [
-                                        {
-                                            type: "text",
-                                            text: generatedText,
-                                        },
-                                    ],
-                                },
-                            ]
-                            : [...params.prompt],
-                        usage,
-                        timestamp: Date.now(),
-                    };
-                    await writeAuditRecord(app, filePath, msg);
-                },
-            });
-            return {
-                stream: stream.pipeThrough(transformStream),
-                ...rest,
-            };
-        },
-    };
-    return middleware;
-};

package/dist/source/middleware/index.d.ts

@@ -1,2 +0,0 @@
-export { auditMessage } from "./audit-message.ts";
-export { createRateLimitMiddleware } from "./rate-limit.ts";

package/dist/source/middleware/index.js

@@ -1,2 +0,0 @@
-export { auditMessage } from "./audit-message.js";
-export { createRateLimitMiddleware } from "./rate-limit.js";

package/dist/source/middleware/rate-limit.d.ts

@@ -1,4 +0,0 @@
-import type { LanguageModelV2Middleware } from "@ai-sdk/provider";
-export declare const createRateLimitMiddleware: ({ requestsPerMinute, }: {
-    requestsPerMinute: number;
-}) => LanguageModelV2Middleware;

package/dist/source/middleware/rate-limit.js

@@ -1,17 +0,0 @@
-import pThrottle from "p-throttle";
-export const createRateLimitMiddleware = ({ requestsPerMinute, }) => {
-    const throttle = pThrottle({
-        limit: requestsPerMinute,
-        interval: 60 * 1000, // 1 minute
-    });
-    return {
-        wrapGenerate: ({ doGenerate }) => {
-            const throttledGenerate = throttle(doGenerate);
-            return Promise.resolve(throttledGenerate());
-        },
-        wrapStream: ({ doStream }) => {
-            const throttledStream = throttle(doStream);
-            return throttledStream();
-        },
-    };
-};

package/dist/source/models/ai-config.d.ts

@@ -1,12 +0,0 @@
-import type { SharedV2ProviderMetadata } from "@ai-sdk/provider";
-import type { ModelMetadata } from "./providers.ts";
-export declare class AiConfig {
-    private modelMetadata;
-    private prompt;
-    constructor({ modelMetadata, prompt, }: {
-        modelMetadata: ModelMetadata;
-        prompt: string;
-    });
-    getMaxTokens(): number;
-    getProviderOptions(): SharedV2ProviderMetadata;
-}

package/dist/source/models/ai-config.js

@@ -1,87 +0,0 @@
-const THINKING_TIERS = [
-    {
-        pattern: /\b(ultrathink|think super hard|think really hard|think intensely)\b/i,
-        budget: 31999,
-        effort: "high",
-    },
-    {
-        pattern: /\b(megathink|think (very )?hard|think (a lot|more|about it))\b/i,
-        budget: 10000,
-        effort: "medium",
-    },
-    {
-        pattern: /\bthink\b/i, // Catch-all for standalone "think"
-        budget: 4000,
-        effort: "low",
-    },
-];
-function calculateThinkingLevel(userInput) {
-    let tokenBudget = 0; // Default
-    let effort = "none";
-    for (const tier of THINKING_TIERS) {
-        if (tier.pattern.test(userInput)) {
-            tokenBudget = tier.budget;
-            effort = tier.effort;
-            break; // Use highest priority match
-        }
-    }
-    return { tokenBudget, effort };
-}
-export class AiConfig {
-    modelMetadata;
-    prompt;
-    constructor({ modelMetadata, prompt, }) {
-        this.modelMetadata = modelMetadata;
-        this.prompt = prompt;
-    }
-    getMaxTokens() {
-        const modelConfig = this.modelMetadata;
-        const thinkingLevel = calculateThinkingLevel(this.prompt);
-        const maxTokens = modelConfig.provider === "anthropic" && modelConfig.supportsReasoning
-            ? modelConfig.maxOutputTokens - thinkingLevel.tokenBudget
-            : modelConfig.maxOutputTokens;
-        return maxTokens;
-    }
-    getProviderOptions() {
-        const modelConfig = this.modelMetadata;
-        const thinkingLevel = calculateThinkingLevel(this.prompt);
-        if (modelConfig.supportsReasoning && thinkingLevel.effort !== "none") {
-            switch (modelConfig.provider) {
-                case "anthropic":
-                    return {
-                        anthropic: {
-                            thinking: {
-                                type: "enabled",
-                                budgetTokens: thinkingLevel.tokenBudget,
-                            },
-                        },
-                    };
-                case "openai":
-                    return { openai: { reasoningEffort: thinkingLevel.effort } };
-                case "google": {
-                    return {
-                        google: {
-                            thinkingConfig: {
-                                thinkingBudget: thinkingLevel.tokenBudget,
-                            },
-                        },
-                    };
-                }
-                case "openrouter": {
-                    return {
-                        openrouter: {
-                            reasoning: {
-                                enabled: true,
-                                effort: thinkingLevel.effort,
-                            },
-                        },
-                    };
-                }
-                default:
-                    return {};
-            }
-        }
-        // If supportsReasoning is false, or no provider case matched
-        return {};
-    }
-}

package/dist/source/models/anthropic-provider.d.ts

@@ -1,25 +0,0 @@
-import type { ModelMetadata } from "./providers.ts";
-declare const anthropicModels: {
-    readonly opus: import("@ai-sdk/provider").LanguageModelV2;
-    readonly sonnet: import("@ai-sdk/provider").LanguageModelV2;
-    readonly sonnet37: import("@ai-sdk/provider").LanguageModelV2;
-    readonly "sonnet37-token-efficient-tools": import("@ai-sdk/provider").LanguageModelV2;
-    readonly "sonnet37-128k": import("@ai-sdk/provider").LanguageModelV2;
-    readonly sonnet35: import("@ai-sdk/provider").LanguageModelV2;
-    readonly haiku: import("@ai-sdk/provider").LanguageModelV2;
-};
-type ModelName = `anthropic:${keyof typeof anthropicModels}`;
-export declare const anthropicModelNames: ModelName[];
-export declare const anthropicProvider: {
-    anthropic: import("@ai-sdk/provider").ProviderV2 & {
-        languageModel(modelId: "opus" | "sonnet" | "sonnet37" | "sonnet37-token-efficient-tools" | "sonnet37-128k" | "sonnet35" | "haiku"): import("@ai-sdk/provider").LanguageModelV2;
-        textEmbeddingModel(modelId: string): import("@ai-sdk/provider").EmbeddingModelV2<string>;
-        imageModel(modelId: string): import("@ai-sdk/provider").ImageModelV2;
-        transcriptionModel(modelId: string): import("@ai-sdk/provider").TranscriptionModelV2;
-        speechModel(modelId: string): import("@ai-sdk/provider").SpeechModelV2;
-    };
-};
-export declare const anthropicModelRegistry: {
-    [K in ModelName]: ModelMetadata<ModelName>;
-};
-export {};