@zihanw/pi-forge 0.2.0 → 0.3.0

package/src/storage.ts

@@ -0,0 +1,28 @@
+import { isAbsolute, join, relative, resolve } from "node:path";
+
+export function promptStacksDir(cwd: string): string {
+	return join(cwd, ".pi", "forge", "prompt-stacks");
+}
+
+export function legacyPromptStacksDir(cwd: string): string {
+	return join(cwd, ".pi", "prompt-stacks");
+}
+
+export function promptStackReadDirs(cwd: string): string[] {
+	return [promptStacksDir(cwd), legacyPromptStacksDir(cwd)];
+}
+
+export function promptStackPath(cwd: string, id: string): string {
+	return join(promptStacksDir(cwd), `${id}.json`);
+}
+
+export function isInsidePromptStackStorage(cwd: string, filePath: string): boolean {
+	return promptStackReadDirs(cwd).some((dir) => isInsideDir(dir, filePath));
+}
+
+function isInsideDir(rootDir: string, filePath: string): boolean {
+	const root = resolve(rootDir);
+	const target = resolve(filePath);
+	const rel = relative(root, target);
+	return !!rel && !rel.startsWith("..") && !isAbsolute(rel);
+}

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,40 @@ 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 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 +131,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 +145,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 +168,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,8 +186,21 @@ 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[];
 }
 

package/src/web-editor/index.ts

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