@tomsun28/pizza 0.1.0 → 0.1.1

package/dist/core/messages.js

@@ -0,0 +1,134 @@
+/**
+ * Custom message types and transformers for the coding agent.
+ *
+ * Extends the base AgentMessage type with coding-agent specific message types,
+ * and provides a transformer to convert them to LLM-compatible messages.
+ */
+// Custom message types are now defined directly in src/core/agent/types.ts.
+// No declaration merging needed since pizza owns the Agent class.
+export const COMPACTION_SUMMARY_PREFIX = `The conversation history before this point was compacted into the following summary:
+
+<summary>
+`;
+export const COMPACTION_SUMMARY_SUFFIX = `
+</summary>`;
+export const BRANCH_SUMMARY_PREFIX = `The following is a summary of a branch that this conversation came back from:
+
+<summary>
+`;
+export const BRANCH_SUMMARY_SUFFIX = `</summary>`;
+/**
+ * Convert a BashExecutionMessage to user message text for LLM context.
+ */
+export function bashExecutionToText(msg) {
+    let text = `Ran \`${msg.command}\`\n`;
+    if (msg.output) {
+        text += `\`\`\`\n${msg.output}\n\`\`\``;
+    }
+    else {
+        text += "(no output)";
+    }
+    if (msg.cancelled) {
+        text += "\n\n(command cancelled)";
+    }
+    else if (msg.exitCode !== null && msg.exitCode !== undefined && msg.exitCode !== 0) {
+        text += `\n\nCommand exited with code ${msg.exitCode}`;
+    }
+    if (msg.truncated && msg.fullOutputPath) {
+        text += `\n\n[Output truncated. Full output: ${msg.fullOutputPath}]`;
+    }
+    return text;
+}
+export function createBranchSummaryMessage(summary, fromId, timestamp) {
+    return {
+        role: "branchSummary",
+        summary,
+        fromId,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+export function createCompactionSummaryMessage(summary, tokensBefore, timestamp) {
+    return {
+        role: "compactionSummary",
+        summary: summary,
+        tokensBefore,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+/** Convert CustomMessageEntry to AgentMessage format */
+export function createCustomMessage(customType, content, display, details, timestamp) {
+    return {
+        role: "custom",
+        customType,
+        content,
+        display,
+        details,
+        timestamp: new Date(timestamp).getTime(),
+    };
+}
+/**
+ * Transform AgentMessages (including custom types) to LLM-compatible Messages.
+ *
+ * This is used by:
+ * - Agent's transormToLlm option (for prompt calls and queued messages)
+ * - Compaction's generateSummary (for summarization)
+ * - Custom extensions and tools
+ */
+export function convertToLlm(messages) {
+    return messages
+        .map((m) => {
+        switch (m.role) {
+            case "bashExecution":
+                // Skip messages excluded from context (!! prefix)
+                if (m.excludeFromContext) {
+                    return undefined;
+                }
+                return {
+                    role: "user",
+                    content: [{ type: "text", text: bashExecutionToText(m) }],
+                    timestamp: m.timestamp,
+                };
+            case "custom": {
+                const content = typeof m.content === "string" ? [{ type: "text", text: m.content }] : m.content;
+                return {
+                    role: "user",
+                    content,
+                    timestamp: m.timestamp,
+                };
+            }
+            case "branchSummary":
+                return {
+                    role: "user",
+                    content: [{ type: "text", text: BRANCH_SUMMARY_PREFIX + m.summary + BRANCH_SUMMARY_SUFFIX }],
+                    timestamp: m.timestamp,
+                };
+            case "compactionSummary":
+                return {
+                    role: "user",
+                    content: [
+                        { type: "text", text: COMPACTION_SUMMARY_PREFIX + m.summary + COMPACTION_SUMMARY_SUFFIX },
+                    ],
+                    timestamp: m.timestamp,
+                };
+            case "user":
+            case "assistant":
+                return m;
+            case "toolResult": {
+                const toolResult = m;
+                return {
+                    role: "toolResult",
+                    toolCallId: toolResult.toolCallId,
+                    toolName: toolResult.toolName,
+                    content: toolResult.content,
+                    isError: toolResult.isError,
+                    timestamp: toolResult.timestamp,
+                };
+            }
+            default:
+                // biome-ignore lint/correctness/noSwitchDeclarations: fine
+                const _exhaustiveCheck = m;
+                return undefined;
+        }
+    })
+        .filter((m) => m !== undefined);
+}

package/dist/core/model-registry.d.ts

@@ -0,0 +1,131 @@
+/**
+ * Model registry - manages built-in and custom models, provides API key resolution.
+ */
+import { type Api, type AssistantMessageEventStream, type Context, type Model, type OAuthProviderInterface, type SimpleStreamOptions } from "@earendil-works/pi-ai/compat";
+import type { AuthStorage } from "./auth-storage.js";
+import { clearConfigValueCache } from "./resolve-config-value.js";
+export type ResolvedRequestAuth = {
+    ok: true;
+    apiKey?: string;
+    headers?: Record<string, string>;
+} | {
+    ok: false;
+    error: string;
+};
+/** Clear the config value command cache. Exported for testing. */
+export declare const clearApiKeyCache: typeof clearConfigValueCache;
+/**
+ * Model registry - loads and manages models, resolves API keys via AuthStorage.
+ */
+export declare class ModelRegistry {
+    readonly authStorage: AuthStorage;
+    private modelsJsonPath;
+    private models;
+    private providerRequestConfigs;
+    private modelRequestHeaders;
+    private registeredProviders;
+    private loadError;
+    private constructor();
+    static create(authStorage: AuthStorage, modelsJsonPath?: string): ModelRegistry;
+    static inMemory(authStorage: AuthStorage): ModelRegistry;
+    /**
+     * Reload models from disk (built-in + custom from models.json).
+     */
+    refresh(): void;
+    /**
+     * Get any error from loading models.json (undefined if no error).
+     */
+    getError(): string | undefined;
+    private loadModels;
+    /** Load built-in models and apply provider/model overrides */
+    private loadBuiltInModels;
+    /** Merge custom models into built-in list by provider+id (custom wins on conflicts). */
+    private mergeCustomModels;
+    private loadCustomModels;
+    private validateConfig;
+    private parseModels;
+    /**
+     * Get all models (built-in + custom).
+     * If models.json had errors, returns only built-in models.
+     */
+    getAll(): Model<Api>[];
+    /**
+     * Get only models that have auth configured.
+     * This is a fast check that doesn't refresh OAuth tokens.
+     */
+    getAvailable(): Model<Api>[];
+    /**
+     * Find a model by provider and ID.
+     */
+    find(provider: string, modelId: string): Model<Api> | undefined;
+    /**
+     * Get API key for a model.
+     */
+    hasConfiguredAuth(model: Model<Api>): boolean;
+    private getModelRequestKey;
+    private storeProviderRequestConfig;
+    private storeModelHeaders;
+    /**
+     * Get API key and request headers for a model.
+     */
+    getApiKeyAndHeaders(model: Model<Api>): Promise<ResolvedRequestAuth>;
+    /**
+     * Get API key for a provider.
+     */
+    getApiKeyForProvider(provider: string): Promise<string | undefined>;
+    /**
+     * Check if a model is using OAuth credentials (subscription).
+     */
+    isUsingOAuth(model: Model<Api>): boolean;
+    /**
+     * Register a provider dynamically (from extensions).
+     *
+     * If provider has models: replaces all existing models for this provider.
+     * If provider has only baseUrl/headers: overrides existing models' URLs.
+     * If provider has oauth: registers OAuth provider for /login support.
+     */
+    registerProvider(providerName: string, config: ProviderConfigInput): void;
+    /**
+     * Unregister a previously registered provider.
+     *
+     * Removes the provider from the registry and reloads models from disk so that
+     * built-in models overridden by this provider are restored to their original state.
+     * Also resets dynamic OAuth and API stream registrations before reapplying
+     * remaining dynamic providers.
+     * Has no effect if the provider was never registered.
+     */
+    unregisterProvider(providerName: string): void;
+    private validateProviderConfig;
+    private applyProviderConfig;
+}
+/**
+ * Input type for registerProvider API.
+ */
+export interface ProviderConfigInput {
+    baseUrl?: string;
+    apiKey?: string;
+    api?: Api;
+    streamSimple?: (model: Model<Api>, context: Context, options?: SimpleStreamOptions) => AssistantMessageEventStream;
+    headers?: Record<string, string>;
+    authHeader?: boolean;
+    /** OAuth provider for /login support */
+    oauth?: Omit<OAuthProviderInterface, "id">;
+    models?: Array<{
+        id: string;
+        name: string;
+        api?: Api;
+        baseUrl?: string;
+        reasoning: boolean;
+        input: ("text" | "image")[];
+        cost: {
+            input: number;
+            output: number;
+            cacheRead: number;
+            cacheWrite: number;
+        };
+        contextWindow: number;
+        maxTokens: number;
+        headers?: Record<string, string>;
+        compat?: Model<Api>["compat"];
+    }>;
+}