snapshot of https://github.com/tldraw/tldraw.git (main, 2026-09-23; blobs > 131072B omitted)

templates/agent/worker/prompt/getSystemPromptFlags.ts

@@ -0,0 +1,81 @@
+import { getActionMeta } from '../../shared/types/AgentAction'
+import type { AgentAction } from '../../shared/types/AgentAction'
+import type { PromptPart } from '../../shared/types/PromptPart'
+import type { SystemPromptCategory } from '../../shared/types/SystemPromptCategory'
+
+export function getSystemPromptFlags(actions: AgentAction['_type'][], parts: PromptPart['type'][]) {
+	return {
+		// Communication
+		hasMessage: actions.includes('message'),
+
+		// Planning
+		hasThink: actions.includes('think'),
+		hasReview: actions.includes('review'),
+		hasSetMyView: actions.includes('setMyView'),
+		hasTodoList: actions.includes('update-todo-list') && parts.includes('todoList'),
+		hasAddDetail: actions.includes('add-detail'),
+
+		// Individual shapes
+		hasCreate: actions.includes('create'),
+		hasDelete: actions.includes('delete'),
+		hasUpdate: actions.includes('update'),
+		hasLabel: actions.includes('label'),
+		hasMove: actions.includes('move'),
+
+		// Groups of shapes
+		hasPlace: actions.includes('place'),
+		hasBringToFront: actions.includes('bringToFront'),
+		hasSendToBack: actions.includes('sendToBack'),
+		hasRotate: actions.includes('rotate'),
+		hasResize: actions.includes('resize'),
+		hasAlign: actions.includes('align'),
+		hasDistribute: actions.includes('distribute'),
+		hasStack: actions.includes('stack'),
+		hasClear: actions.includes('clear'),
+
+		// Drawing
+		hasPen: actions.includes('pen'),
+
+		// Request
+		hasMessagesPart: parts.includes('messages'),
+		hasDataPart: parts.includes('data'),
+		hasContextItemsPart: parts.includes('contextItems'),
+
+		// Viewport
+		hasScreenshotPart: parts.includes('screenshot'),
+		hasUserViewportBoundsPart: parts.includes('userViewportBounds'),
+		hasAgentViewportBoundsPart: parts.includes('agentViewportBounds'),
+
+		// Shapes
+		hasBlurryShapesPart: parts.includes('blurryShapes'),
+		hasPeripheralShapesPart: parts.includes('peripheralShapes'),
+		hasSelectedShapesPart: parts.includes('selectedShapes'),
+
+		// History
+		hasChatHistoryPart: parts.includes('chatHistory'),
+		hasUserActionHistoryPart: parts.includes('userActionHistory'),
+		hasTodoListPart: parts.includes('todoList'),
+
+		// Lints
+		hasCanvasLintsPart: parts.includes('canvasLints'),
+
+		// Metadata
+		hasTimePart: parts.includes('time'),
+
+		// Derived flags for convenience
+		canEdit: actions.some(isEditAction),
+	}
+}
+
+export type SystemPromptFlags = ReturnType<typeof getSystemPromptFlags>
+
+function isEditAction(type: AgentAction['_type']): boolean {
+	return isActionCategory(type, 'edit')
+}
+
+function isActionCategory(type: AgentAction['_type'], category: SystemPromptCategory): boolean {
+	const meta = getActionMeta(type)
+	if (!meta) return false
+	if (!meta._systemPromptCategory) return false
+	return meta._systemPromptCategory === category
+}