@zihanw/pi-forge 0.2.0 → 0.3.1

package/src/types.ts

@@ -19,6 +19,42 @@ export type PromptStackSlot =
 	| "pi-docs"
 	| "variables";
 
+export type PromptStackSlotFormat = "xml" | "json" | "plain";
+
+export type PromptRegexStage = "history" | "compiled";
+
+export type PromptRegexEffect = "outgoing" | "display" | "both" | "finalize";
+
+export type PromptRegexTarget = "system" | "messages";
+
+export interface PromptRegexRule {
+	id: string;
+	name?: string;
+	enabled?: boolean;
+	stage: PromptRegexStage;
+	effect?: PromptRegexEffect;
+	pattern: string;
+	flags?: string;
+	replace?: string;
+	trimStrings?: string[];
+	roles?: string[];
+	targets?: PromptRegexTarget[];
+	maxMessages?: number;
+	maxChars?: number;
+	minDepth?: number;
+	maxDepth?: number;
+	source?: Record<string, unknown>;
+}
+
+export interface PromptRegexConfig {
+	schemaVersion?: 1;
+	rules?: PromptRegexRule[];
+}
+
+export type PromptResourcePolicy =
+	| { allow?: string[]; deny?: never }
+	| { allow?: never; deny?: string[] };
+
 export interface PromptStackDefaults {
 	syntheticMessagesVisible?: boolean;
 	unresolvedMacroPolicy?: "warn" | "keep" | "error";
@@ -50,33 +86,42 @@ export interface VariablesSlotOptions {
 	includeSession?: boolean;
 	/** Include turn variables. Default: true. */
 	includeTurn?: boolean;
-	/** Include only these scopes. Overrides includeStatic/includeSession/includeTurn when present. */
-	includeScopes?: PromptStateScope[];
-	/** Include variable names matching these exact names or wildcard prefixes, e.g. agent.*. */
-	includeNamespaces?: string[];
-	/** Exclude variable names matching these exact names or wildcard prefixes. */
-	excludeNamespaces?: string[];
-	/** Include type and description metadata from stack.state.definitions. Default: false. */
-	includeMetadata?: boolean;
 	/** Render format. Default: xml. */
-	format?: "xml" | "json";
-	/** Truncate each rendered value after this many characters. */
-	maxValueChars?: number;
+	format?: Exclude<PromptStackSlotFormat, "json">;
 }
 
 export interface PromptStackSlotOptions {
 	/** For chat-history: include the latest user message in the expanded history. Default: true. */
 	includeLastUserMessage?: boolean;
+	/** For chat-history: remove assistant thinking content blocks from inserted history. Default: false. */
+	stripAssistantThinking?: boolean;
+	/** For chat-history: keep only messages with these roles. Default: all roles. */
+	roles?: string[];
+	/** For chat-history: include Pi branch/compaction summaries. Default: true. */
+	includeSummaries?: boolean;
+	/** For chat-history: keep or drop tool-call/tool-result history. Default: keep. */
+	toolMode?: "keep" | "drop";
+	/** For chat-history: keep only the most recent N messages after filtering. */
+	maxMessages?: number;
+	/** For chat-history: keep only the most recent messages within an approximate character budget. */
+	maxChars?: number;
 	/** For variables: control which variable scopes are included. */
 	includeStatic?: boolean;
 	includeSession?: boolean;
 	includeTurn?: boolean;
-	includeScopes?: PromptStateScope[];
-	includeNamespaces?: string[];
-	excludeNamespaces?: string[];
-	includeMetadata?: boolean;
-	format?: "xml" | "json";
-	maxValueChars?: number;
+	format?: PromptStackSlotFormat;
+	/** For tools: only render tools that provide prompt snippets, matching Pi's default prompt builder. */
+	onlyWithSnippets?: boolean;
+	/** For date/date-cwd: include the exact current time. Default: false. */
+	includeTime?: boolean;
+	/** For tool-guidelines: override the plain-format section heading. */
+	heading?: string;
+	/** For tool-guidelines: include Pi's default concise/file-path guideline bullets. */
+	includePiDefaultGuidelines?: boolean;
+	/** For tool-guidelines: match Pi's default wording where it differs from pi-forge's generic wording. */
+	piStyle?: boolean;
+	/** For skills: omit the skills section unless the read tool is active, matching Pi's default prompt builder. */
+	requireReadTool?: boolean;
 	[key: string]: unknown;
 }
 
@@ -88,25 +133,9 @@ export interface PromptStackSlotItem extends PromptStackBaseItem {
 
 export type PromptStackItem = PromptStackBlockItem | PromptStackSlotItem;
 
-export type PromptStatePrimitive = string | number | boolean | null;
-
-export type PromptStateValue = PromptStatePrimitive | PromptStateValue[] | { [key: string]: PromptStateValue };
-
-export type PromptStateScope = "static" | "session" | "turn";
-
-export interface PromptStateDefinition {
-	type?: string;
-	scope?: PromptStateScope;
-	description?: string;
-	agentWritable?: boolean;
-	userWritable?: boolean;
-	default?: PromptStateValue;
-}
+export type PromptVariablePrimitive = string | number | boolean | null;
 
-export interface PromptStateConfig {
-	schemaVersion?: 1;
-	definitions?: Record<string, PromptStateDefinition>;
-}
+export type PromptVariableValue = PromptVariablePrimitive | PromptVariableValue[] | { [key: string]: PromptVariableValue };
 
 export interface PromptStack {
 	schemaVersion: 1;
@@ -118,8 +147,10 @@ export interface PromptStack {
 	mode?: PromptStackMode;
 	defaults?: PromptStackDefaults;
 	context?: PromptStackContextOptions;
+	tools?: PromptResourcePolicy;
+	skills?: PromptResourcePolicy;
 	variables?: Record<string, string>;
-	state?: PromptStateConfig;
+	regex?: PromptRegexConfig;
 	items: PromptStackItem[];
 	import?: Record<string, unknown>;
 }
@@ -139,8 +170,8 @@ export interface PromptStackDiagnostic {
 }
 
 export interface PromptVariableStore {
-	turn: Record<string, PromptStateValue>;
-	session: Record<string, PromptStateValue>;
+	turn: Record<string, PromptVariableValue>;
+	session: Record<string, PromptVariableValue>;
 	sessionDirty?: boolean;
 }
 
@@ -157,22 +188,22 @@ export interface CompileSystemPromptResult {
 	diagnostics: PromptStackDiagnostic[];
 }
 
+export type CompileMessageSourceKind = "stack-item" | "chat-history" | "implicit-history";
+
+export interface CompileMessageSource {
+	kind: CompileMessageSourceKind;
+	itemId?: string;
+	itemName?: string;
+	slot?: string;
+	historyIndex?: number;
+	historyCount?: number;
+	role?: string;
+}
+
 export interface CompileMessagesResult {
 	messages: AgentMessage[];
+	messageSources: CompileMessageSource[];
 	diagnostics: PromptStackDiagnostic[];
 }
 
-export const SUPPORTED_SLOTS = new Set<PromptStackSlot>([
-	"chat-history",
-	"tools",
-	"tool-guidelines",
-	"skills",
-	"project-context",
-	"append-system-prompt",
-	"date",
-	"cwd",
-	"date-cwd",
-	"active-model",
-	"pi-docs",
-	"variables",
-]);
+export { SUPPORTED_SLOTS } from "./slot-renderers.ts";

package/src/web-editor/index.ts

@@ -0,0 +1,2 @@
+export { DEFAULT_WEB_EDITOR_PORT, startWebEditorServer } from "./server.ts";
+export type * from "./types.ts";