/** * @license * Copyright 2025 Steven Roussey * SPDX-License-Identifier: Apache-2.0 */ import type { CachePolicy, IRunConfig, TaskConfig } from "@workglow/task-graph"; import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph"; import { DataPortSchema } from "@workglow/util/schema"; import type { Capability } from "../capability/Capabilities"; export type HierarchyJoinTaskInput = { chunks?: string[] | undefined; scores?: number[] | undefined; chunk_ids?: string[] | undefined; includeParentSummaries?: boolean | undefined; includeEntities?: boolean | undefined; metadata: { [x: string]: unknown; leafNodeId?: string | undefined; summary?: string | undefined; entities?: { type: string; text: string; score: number; }[] | undefined; parentSummaries?: string[] | undefined; sectionTitles?: string[] | undefined; doc_title?: string | undefined; text: string; doc_id: string; chunkId: string; nodePath: string[]; depth: number; }[]; knowledgeBase: unknown; }; export type HierarchyJoinTaskOutput = { chunks?: string[] | undefined; scores?: number[] | undefined; chunk_ids?: string[] | undefined; metadata: { [x: string]: unknown; leafNodeId?: string | undefined; summary?: string | undefined; entities?: { type: string; text: string; score: number; }[] | undefined; parentSummaries?: string[] | undefined; sectionTitles?: string[] | undefined; doc_title?: string | undefined; text: string; doc_id: string; chunkId: string; nodePath: string[]; depth: number; }[]; count: number; }; export type HierarchyJoinTaskConfig = TaskConfig; /** * Enrich retrieval metadata with document-hierarchy context (parent summaries, * section titles, ancestor entities). Consumes only the `metadata` port of an * upstream retrieval task; other retrieval ports (chunks, chunk_ids, scores) * flow around this task via the workflow DAG. */ export declare class HierarchyJoinTask extends Task { static type: string; /** Pure-compute join task — no provider capability required. */ static readonly requires: readonly Capability[]; static category: string; static title: string; static description: string; static cachePolicy: CachePolicy; static inputSchema(): DataPortSchema; static outputSchema(): DataPortSchema; execute(input: HierarchyJoinTaskInput, context: IExecuteContext): Promise; } export declare const hierarchyJoin: (input: HierarchyJoinTaskInput, config?: HierarchyJoinTaskConfig, runConfig?: Partial) => Promise; declare module "@workglow/task-graph" { interface Workflow { hierarchyJoin: CreateWorkflow; } } //# sourceMappingURL=HierarchyJoinTask.d.ts.map