@workglow/ai 0.1.1 → 0.2.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +8 -7
- package/dist/browser.js +1436 -397
- package/dist/browser.js.map +64 -58
- package/dist/bun.js +1436 -397
- package/dist/bun.js.map +64 -58
- package/dist/common.d.ts +1 -0
- package/dist/common.d.ts.map +1 -1
- package/dist/execution/DirectExecutionStrategy.d.ts +1 -1
- package/dist/execution/DirectExecutionStrategy.d.ts.map +1 -1
- package/dist/execution/QueuedExecutionStrategy.d.ts +2 -2
- package/dist/execution/QueuedExecutionStrategy.d.ts.map +1 -1
- package/dist/job/AiJob.d.ts +2 -1
- package/dist/job/AiJob.d.ts.map +1 -1
- package/dist/model/ModelRegistry.d.ts.map +1 -1
- package/dist/model/ModelRepository.d.ts +11 -15
- package/dist/model/ModelRepository.d.ts.map +1 -1
- package/dist/model/ModelSchema.d.ts.map +1 -1
- package/dist/node.js +1436 -397
- package/dist/node.js.map +64 -58
- package/dist/provider/AiProvider.d.ts +2 -2
- package/dist/provider/AiProvider.d.ts.map +1 -1
- package/dist/provider/AiProviderRegistry.d.ts +3 -2
- package/dist/provider/AiProviderRegistry.d.ts.map +1 -1
- package/dist/provider/QueuedAiProvider.d.ts +2 -1
- package/dist/provider/QueuedAiProvider.d.ts.map +1 -1
- package/dist/task/AgentTask.d.ts +525 -0
- package/dist/task/AgentTask.d.ts.map +1 -0
- package/dist/task/AgentTypes.d.ts +183 -0
- package/dist/task/AgentTypes.d.ts.map +1 -0
- package/dist/task/AgentUtils.d.ts +55 -0
- package/dist/task/AgentUtils.d.ts.map +1 -0
- package/dist/task/BackgroundRemovalTask.d.ts +6 -34
- package/dist/task/BackgroundRemovalTask.d.ts.map +1 -1
- package/dist/task/ChunkRetrievalTask.d.ts +6 -4
- package/dist/task/ChunkRetrievalTask.d.ts.map +1 -1
- package/dist/task/ChunkToVectorTask.d.ts +6 -26
- package/dist/task/ChunkToVectorTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorHybridSearchTask.d.ts +6 -25
- package/dist/task/ChunkVectorHybridSearchTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorSearchTask.d.ts +6 -12
- package/dist/task/ChunkVectorSearchTask.d.ts.map +1 -1
- package/dist/task/ChunkVectorUpsertTask.d.ts +6 -13
- package/dist/task/ChunkVectorUpsertTask.d.ts.map +1 -1
- package/dist/task/ContextBuilderTask.d.ts +6 -30
- package/dist/task/ContextBuilderTask.d.ts.map +1 -1
- package/dist/task/CountTokensTask.d.ts +6 -21
- package/dist/task/CountTokensTask.d.ts.map +1 -1
- package/dist/task/DocumentEnricherTask.d.ts +6 -40
- package/dist/task/DocumentEnricherTask.d.ts.map +1 -1
- package/dist/task/DownloadModelTask.d.ts +7 -21
- package/dist/task/DownloadModelTask.d.ts.map +1 -1
- package/dist/task/FaceDetectorTask.d.ts +6 -36
- package/dist/task/FaceDetectorTask.d.ts.map +1 -1
- package/dist/task/FaceLandmarkerTask.d.ts +6 -40
- package/dist/task/FaceLandmarkerTask.d.ts.map +1 -1
- package/dist/task/GestureRecognizerTask.d.ts +6 -38
- package/dist/task/GestureRecognizerTask.d.ts.map +1 -1
- package/dist/task/HandLandmarkerTask.d.ts +6 -38
- package/dist/task/HandLandmarkerTask.d.ts.map +1 -1
- package/dist/task/HierarchicalChunkerTask.d.ts +6 -26
- package/dist/task/HierarchicalChunkerTask.d.ts.map +1 -1
- package/dist/task/HierarchyJoinTask.d.ts +6 -29
- package/dist/task/HierarchyJoinTask.d.ts.map +1 -1
- package/dist/task/ImageClassificationTask.d.ts +6 -36
- package/dist/task/ImageClassificationTask.d.ts.map +1 -1
- package/dist/task/ImageEmbeddingTask.d.ts +6 -47
- package/dist/task/ImageEmbeddingTask.d.ts.map +1 -1
- package/dist/task/ImageSegmentationTask.d.ts +6 -36
- package/dist/task/ImageSegmentationTask.d.ts.map +1 -1
- package/dist/task/ImageToTextTask.d.ts +6 -35
- package/dist/task/ImageToTextTask.d.ts.map +1 -1
- package/dist/task/MessageConversion.d.ts +52 -0
- package/dist/task/MessageConversion.d.ts.map +1 -0
- package/dist/task/ModelInfoTask.d.ts +7 -22
- package/dist/task/ModelInfoTask.d.ts.map +1 -1
- package/dist/task/ModelSearchTask.d.ts +6 -7
- package/dist/task/ModelSearchTask.d.ts.map +1 -1
- package/dist/task/ObjectDetectionTask.d.ts +6 -36
- package/dist/task/ObjectDetectionTask.d.ts.map +1 -1
- package/dist/task/PoseLandmarkerTask.d.ts +6 -39
- package/dist/task/PoseLandmarkerTask.d.ts.map +1 -1
- package/dist/task/QueryExpanderTask.d.ts +6 -9
- package/dist/task/QueryExpanderTask.d.ts.map +1 -1
- package/dist/task/RerankerTask.d.ts +6 -28
- package/dist/task/RerankerTask.d.ts.map +1 -1
- package/dist/task/StructuralParserTask.d.ts +6 -10
- package/dist/task/StructuralParserTask.d.ts.map +1 -1
- package/dist/task/StructuredGenerationTask.d.ts +6 -26
- package/dist/task/StructuredGenerationTask.d.ts.map +1 -1
- package/dist/task/TextChunkerTask.d.ts +6 -9
- package/dist/task/TextChunkerTask.d.ts.map +1 -1
- package/dist/task/TextClassificationTask.d.ts +6 -23
- package/dist/task/TextClassificationTask.d.ts.map +1 -1
- package/dist/task/TextEmbeddingTask.d.ts +6 -21
- package/dist/task/TextEmbeddingTask.d.ts.map +1 -1
- package/dist/task/TextFillMaskTask.d.ts +6 -21
- package/dist/task/TextFillMaskTask.d.ts.map +1 -1
- package/dist/task/TextGenerationTask.d.ts +6 -26
- package/dist/task/TextGenerationTask.d.ts.map +1 -1
- package/dist/task/TextLanguageDetectionTask.d.ts +6 -22
- package/dist/task/TextLanguageDetectionTask.d.ts.map +1 -1
- package/dist/task/TextNamedEntityRecognitionTask.d.ts +6 -22
- package/dist/task/TextNamedEntityRecognitionTask.d.ts.map +1 -1
- package/dist/task/TextQuestionAnswerTask.d.ts +6 -22
- package/dist/task/TextQuestionAnswerTask.d.ts.map +1 -1
- package/dist/task/TextRewriterTask.d.ts +6 -22
- package/dist/task/TextRewriterTask.d.ts.map +1 -1
- package/dist/task/TextSummaryTask.d.ts +6 -21
- package/dist/task/TextSummaryTask.d.ts.map +1 -1
- package/dist/task/TextTranslationTask.d.ts +6 -23
- package/dist/task/TextTranslationTask.d.ts.map +1 -1
- package/dist/task/ToolCallingTask.d.ts +348 -0
- package/dist/task/ToolCallingTask.d.ts.map +1 -0
- package/dist/task/ToolCallingUtils.d.ts +75 -0
- package/dist/task/ToolCallingUtils.d.ts.map +1 -0
- package/dist/task/TopicSegmenterTask.d.ts +6 -10
- package/dist/task/TopicSegmenterTask.d.ts.map +1 -1
- package/dist/task/UnloadModelTask.d.ts +6 -20
- package/dist/task/UnloadModelTask.d.ts.map +1 -1
- package/dist/task/VectorQuantizeTask.d.ts +6 -8
- package/dist/task/VectorQuantizeTask.d.ts.map +1 -1
- package/dist/task/VectorSimilarityTask.d.ts +7 -10
- package/dist/task/VectorSimilarityTask.d.ts.map +1 -1
- package/dist/task/base/AiTask.d.ts +8 -5
- package/dist/task/base/AiTask.d.ts.map +1 -1
- package/dist/task/base/AiTaskSchemas.d.ts.map +1 -1
- package/dist/task/base/AiVisionTask.d.ts +3 -7
- package/dist/task/base/AiVisionTask.d.ts.map +1 -1
- package/dist/task/base/StreamingAiTask.d.ts +4 -7
- package/dist/task/base/StreamingAiTask.d.ts.map +1 -1
- package/dist/task/index.d.ts +9 -1
- package/dist/task/index.d.ts.map +1 -1
- package/dist/worker.d.ts +2 -0
- package/dist/worker.d.ts.map +1 -1
- package/dist/worker.js +247 -25
- package/dist/worker.js.map +9 -7
- package/package.json +16 -11
|
@@ -3,9 +3,10 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow
|
|
6
|
+
import { CreateWorkflow } from "@workglow/task-graph";
|
|
7
|
+
import type { IExecuteContext, TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
|
-
import { AiTask
|
|
9
|
+
import { AiTask } from "./base/AiTask";
|
|
9
10
|
declare const ModelInfoInputSchema: {
|
|
10
11
|
readonly type: "object";
|
|
11
12
|
readonly properties: {
|
|
@@ -167,10 +168,11 @@ declare const ModelInfoOutputSchema: {
|
|
|
167
168
|
};
|
|
168
169
|
export type ModelInfoTaskInput = FromSchema<typeof ModelInfoInputSchema>;
|
|
169
170
|
export type ModelInfoTaskOutput = FromSchema<typeof ModelInfoOutputSchema>;
|
|
171
|
+
export type ModelInfoTaskConfig = TaskConfig<ModelInfoTaskInput>;
|
|
170
172
|
/**
|
|
171
173
|
* Retrieve runtime metadata about a model: locality, browser support, cache status, and file sizes.
|
|
172
174
|
*/
|
|
173
|
-
export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoTaskOutput,
|
|
175
|
+
export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoTaskOutput, ModelInfoTaskConfig> {
|
|
174
176
|
static type: string;
|
|
175
177
|
static category: string;
|
|
176
178
|
static cacheable: boolean;
|
|
@@ -186,24 +188,7 @@ export declare class ModelInfoTask extends AiTask<ModelInfoTaskInput, ModelInfoT
|
|
|
186
188
|
* @param input - Input containing the model to query
|
|
187
189
|
* @returns Promise resolving to model info including locality and cache status
|
|
188
190
|
*/
|
|
189
|
-
export declare const modelInfo: (input: {
|
|
190
|
-
detail?: "cached_status" | "files" | "files_with_metadata" | undefined;
|
|
191
|
-
model: string | {
|
|
192
|
-
[x: string]: unknown;
|
|
193
|
-
description?: string | undefined;
|
|
194
|
-
metadata?: {
|
|
195
|
-
[x: string]: unknown;
|
|
196
|
-
} | undefined;
|
|
197
|
-
model_id?: string | undefined;
|
|
198
|
-
provider: string;
|
|
199
|
-
provider_config: {
|
|
200
|
-
[x: string]: unknown;
|
|
201
|
-
credential_key?: string | undefined;
|
|
202
|
-
};
|
|
203
|
-
tasks?: string[] | undefined;
|
|
204
|
-
title?: string | undefined;
|
|
205
|
-
};
|
|
206
|
-
}, config?: import("@workglow/task-graph").TaskConfig | undefined) => Promise<{
|
|
191
|
+
export declare const modelInfo: (input: ModelInfoTaskInput, config?: ModelInfoTaskConfig) => Promise<{
|
|
207
192
|
file_sizes: unknown;
|
|
208
193
|
is_cached: boolean;
|
|
209
194
|
is_loaded: boolean;
|
|
@@ -230,7 +215,7 @@ export declare const modelInfo: (input: {
|
|
|
230
215
|
}>;
|
|
231
216
|
declare module "@workglow/task-graph" {
|
|
232
217
|
interface Workflow {
|
|
233
|
-
modelInfo: CreateWorkflow<ModelInfoTaskInput, ModelInfoTaskOutput,
|
|
218
|
+
modelInfo: CreateWorkflow<ModelInfoTaskInput, ModelInfoTaskOutput, ModelInfoTaskConfig>;
|
|
234
219
|
}
|
|
235
220
|
}
|
|
236
221
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelInfoTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelInfoTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,
|
|
1
|
+
{"version":3,"file":"ModelInfoTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelInfoTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAEhE,OAAO,KAAK,EAAE,eAAe,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACxE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAGnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,oBAAoB;mBAClB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,OAAO,EAAE,qBAAqB;;;;;CAKD,CAAC;AAEpC,QAAA,MAAM,qBAAqB;mBACnB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,QAAQ;qBAAI,IAAI,EAAE,SAAS;;iBAC3B,SAAS;qBAAI,IAAI,EAAE,SAAS;;iBAC5B,gBAAgB;qBAAI,IAAI,EAAE,SAAS;;iBACnC,aAAa;qBAAI,IAAI,EAAE,SAAS;;iBAChC,SAAS;qBAAI,IAAI,EAAE,SAAS;;iBAC5B,SAAS;qBAAI,IAAI,EAAE,SAAS;;iBAC5B,UAAU;iBACV,aAAa;qBACX,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,WAAW,EACT,2GAA2G;;;;;CAchF,CAAC;AAEpC,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,oBAAoB,CAAC,CAAC;AACzE,MAAM,MAAM,mBAAmB,GAAG,UAAU,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAC3E,MAAM,MAAM,mBAAmB,GAAG,UAAU,CAAC,kBAAkB,CAAC,CAAC;AAEjE;;GAEG;AACH,qBAAa,aAAc,SAAQ,MAAM,CACvC,kBAAkB,EAClB,mBAAmB,EACnB,mBAAmB,CACpB;IACC,OAAuB,IAAI,SAAmB;IAC9C,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,SAAS,UAAS;IACzC,OAAuB,KAAK,SAAgB;IAC5C,OAAuB,WAAW,SAC6D;IAC/F,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,kBAAkB,EACzB,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,mBAAmB,CAAC,CAS9B;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,SAAS,UAAW,kBAAkB,WAAW,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;;EAEhF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,SAAS,EAAE,cAAc,CAAC,kBAAkB,EAAE,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;KACzF;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow, IExecuteContext, Task
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
import type { ModelRecord } from "../model/ModelSchema";
|
|
9
10
|
/**
|
|
@@ -38,10 +39,11 @@ export type ModelSearchTaskInput = FromSchema<typeof ModelSearchInputSchema>;
|
|
|
38
39
|
export type ModelSearchTaskOutput = {
|
|
39
40
|
results: ModelSearchResultItem[];
|
|
40
41
|
};
|
|
42
|
+
export type ModelSearchTaskConfig = TaskConfig<ModelSearchTaskInput>;
|
|
41
43
|
/**
|
|
42
44
|
* Search for models using a provider-specific run function from the AiProviderRegistry.
|
|
43
45
|
*/
|
|
44
|
-
export declare class ModelSearchTask extends Task<ModelSearchTaskInput, ModelSearchTaskOutput,
|
|
46
|
+
export declare class ModelSearchTask extends Task<ModelSearchTaskInput, ModelSearchTaskOutput, ModelSearchTaskConfig> {
|
|
45
47
|
static type: string;
|
|
46
48
|
static category: string;
|
|
47
49
|
static title: string;
|
|
@@ -56,13 +58,10 @@ export declare class ModelSearchTask extends Task<ModelSearchTaskInput, ModelSea
|
|
|
56
58
|
/**
|
|
57
59
|
* Search for models using a provider-specific search function.
|
|
58
60
|
*/
|
|
59
|
-
export declare const modelSearch: (input:
|
|
60
|
-
provider: string;
|
|
61
|
-
query?: string | undefined;
|
|
62
|
-
}, config?: TaskConfig | undefined) => Promise<ModelSearchTaskOutput>;
|
|
61
|
+
export declare const modelSearch: (input: ModelSearchTaskInput, config?: ModelSearchTaskConfig) => Promise<ModelSearchTaskOutput>;
|
|
63
62
|
declare module "@workglow/task-graph" {
|
|
64
63
|
interface Workflow {
|
|
65
|
-
modelSearch: CreateWorkflow<ModelSearchTaskInput, ModelSearchTaskOutput,
|
|
64
|
+
modelSearch: CreateWorkflow<ModelSearchTaskInput, ModelSearchTaskOutput, ModelSearchTaskConfig>;
|
|
66
65
|
}
|
|
67
66
|
}
|
|
68
67
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ModelSearchTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelSearchTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"ModelSearchTask.d.ts","sourceRoot":"","sources":["../../src/task/ModelSearchTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,sBAAsB,CAAC;AAGxD;;GAEG;AACH,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,KAAK,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,MAAM,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;IACtC,QAAQ,CAAC,GAAG,EAAE,OAAO,CAAC;CACvB;AAED,4IAA4I;AAC5I,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,QAAQ;qBACN,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,UAAU;qBACjB,WAAW,EACT,+HAA+H;;iBAEnI,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EACT,gGAAgG;;;;;CAKrE,CAAC;AAkDpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAC7E,MAAM,MAAM,qBAAqB,GAAG;IAAE,OAAO,EAAE,qBAAqB,EAAE,CAAA;CAAE,CAAC;AACzE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAErE;;GAEG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CACvC,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SAAgE;IAClG,OAAuB,SAAS,UAAS;IACzC,OAAuB,iBAAiB,UAAQ;IAEhD,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEe,WAAW,IAAI,cAAc,CAE5C;IAEc,OAAO,CACpB,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC,CAQhC;CACF;AAED;;GAEG;AACH,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB,mCAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow
|
|
6
|
+
import { CreateWorkflow } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
import { AiVisionTask } from "./base/AiVisionTask";
|
|
9
10
|
export declare const ObjectDetectionInputSchema: {
|
|
@@ -292,11 +293,12 @@ export declare const ObjectDetectionOutputSchema: {
|
|
|
292
293
|
};
|
|
293
294
|
export type ObjectDetectionTaskInput = FromSchema<typeof ObjectDetectionInputSchema>;
|
|
294
295
|
export type ObjectDetectionTaskOutput = FromSchema<typeof ObjectDetectionOutputSchema>;
|
|
296
|
+
export type ObjectDetectionTaskConfig = TaskConfig<ObjectDetectionTaskInput>;
|
|
295
297
|
/**
|
|
296
298
|
* Detects objects in images using vision models.
|
|
297
299
|
* Automatically selects between regular and zero-shot detection based on whether labels are provided.
|
|
298
300
|
*/
|
|
299
|
-
export declare class ObjectDetectionTask extends AiVisionTask<ObjectDetectionTaskInput, ObjectDetectionTaskOutput,
|
|
301
|
+
export declare class ObjectDetectionTask extends AiVisionTask<ObjectDetectionTaskInput, ObjectDetectionTaskOutput, ObjectDetectionTaskConfig> {
|
|
300
302
|
static type: string;
|
|
301
303
|
static category: string;
|
|
302
304
|
static title: string;
|
|
@@ -310,39 +312,7 @@ export declare class ObjectDetectionTask extends AiVisionTask<ObjectDetectionTas
|
|
|
310
312
|
* @param input The input parameters for object detection (image, model, and optional labels)
|
|
311
313
|
* @returns Promise resolving to the detected objects with labels, scores, and bounding boxes
|
|
312
314
|
*/
|
|
313
|
-
export declare const objectDetection: (input: {
|
|
314
|
-
image: string | {
|
|
315
|
-
channels: number;
|
|
316
|
-
data: {
|
|
317
|
-
[x: string]: unknown;
|
|
318
|
-
} | {
|
|
319
|
-
channels: number;
|
|
320
|
-
data: number[];
|
|
321
|
-
height: number;
|
|
322
|
-
rawChannels?: number | undefined;
|
|
323
|
-
width: number;
|
|
324
|
-
};
|
|
325
|
-
height: number;
|
|
326
|
-
width: number;
|
|
327
|
-
};
|
|
328
|
-
labels?: string[] | undefined;
|
|
329
|
-
model: string | {
|
|
330
|
-
[x: string]: unknown;
|
|
331
|
-
description?: string | undefined;
|
|
332
|
-
metadata?: {
|
|
333
|
-
[x: string]: unknown;
|
|
334
|
-
} | undefined;
|
|
335
|
-
model_id?: string | undefined;
|
|
336
|
-
provider: string;
|
|
337
|
-
provider_config: {
|
|
338
|
-
[x: string]: unknown;
|
|
339
|
-
credential_key?: string | undefined;
|
|
340
|
-
};
|
|
341
|
-
tasks?: string[] | undefined;
|
|
342
|
-
title?: string | undefined;
|
|
343
|
-
};
|
|
344
|
-
threshold?: number | undefined;
|
|
345
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
315
|
+
export declare const objectDetection: (input: ObjectDetectionTaskInput, config?: ObjectDetectionTaskConfig) => Promise<{
|
|
346
316
|
detections: {
|
|
347
317
|
box: {
|
|
348
318
|
height: number;
|
|
@@ -365,7 +335,7 @@ export declare const objectDetection: (input: {
|
|
|
365
335
|
}>;
|
|
366
336
|
declare module "@workglow/task-graph" {
|
|
367
337
|
interface Workflow {
|
|
368
|
-
objectDetection: CreateWorkflow<ObjectDetectionTaskInput, ObjectDetectionTaskOutput,
|
|
338
|
+
objectDetection: CreateWorkflow<ObjectDetectionTaskInput, ObjectDetectionTaskOutput, ObjectDetectionTaskConfig>;
|
|
369
339
|
}
|
|
370
340
|
}
|
|
371
341
|
//# sourceMappingURL=ObjectDetectionTask.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"ObjectDetectionTask.d.ts","sourceRoot":"","sources":["../../src/task/ObjectDetectionTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAyBnD,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;;qBAEhB,KAAK,EAAE,QAAQ;qBACf,WAAW,EACT,kFAAkF;qBACpF,YAAY,EAAE,eAAe;;iBAE/B,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,WAAW;qBAClB,WAAW,EAAE,iDAAiD;qBAC9D,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,2BAA2B;mBAChC,QAAQ;;;;+BAIA,OAAO;;mCAtDf,QAAQ;;iCAEZ,KAAK;qCACH,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,OAAO;qCACd,WAAW,EAAE,kCAAkC;;iCAEjD,KAAK;qCACH,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,kBAAkB;qCACzB,WAAW,EAAE,yCAAyC;qCACtD,OAAO,EAAE,CAAC;qCACV,OAAO,EAAE,CAAC;;iCAEZ,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iDAEM,OAAO,EAAE,OAAO,EAAE,KAAK;;;;+BAuCpB,OAAO;;6BAAW,IAAI,EAAE,OAAO;6BAAE,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4BAEzC,YAAY;kCACN,oEAAoE;;;;;CAKpD,CAAC;AAEpC,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,2BAA2B,CAAC,CAAC;AACvF,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,wBAAwB,CAAC,CAAC;AAE7E;;;GAGG;AACH,qBAAa,mBAAoB,SAAQ,YAAY,CACnD,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B;IACC,OAAuB,IAAI,SAAyB;IACpD,OAAuB,QAAQ,SAAqB;IACpD,OAAuB,KAAK,SAAsB;IAClD,OAAuB,WAAW,SACwE;IAC1G,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,eAAe,UACnB,wBAAwB,WACtB,yBAAyB;;;;;;;;;;;;;;;;;;;;EAGnC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,eAAe,EAAE,cAAc,CAC7B,wBAAwB,EACxB,yBAAyB,EACzB,yBAAyB,CAC1B,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow
|
|
6
|
+
import { CreateWorkflow } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
import { AiVisionTask } from "./base/AiVisionTask";
|
|
9
10
|
export declare const PoseLandmarkerInputSchema: {
|
|
@@ -427,11 +428,12 @@ export declare const PoseLandmarkerOutputSchema: {
|
|
|
427
428
|
};
|
|
428
429
|
export type PoseLandmarkerTaskInput = FromSchema<typeof PoseLandmarkerInputSchema>;
|
|
429
430
|
export type PoseLandmarkerTaskOutput = FromSchema<typeof PoseLandmarkerOutputSchema>;
|
|
431
|
+
export type PoseLandmarkerTaskConfig = TaskConfig<PoseLandmarkerTaskInput>;
|
|
430
432
|
/**
|
|
431
433
|
* Detects pose landmarks in images using MediaPipe Pose Landmarker.
|
|
432
434
|
* Identifies 33 body landmarks for pose estimation and optional segmentation.
|
|
433
435
|
*/
|
|
434
|
-
export declare class PoseLandmarkerTask extends AiVisionTask<PoseLandmarkerTaskInput, PoseLandmarkerTaskOutput,
|
|
436
|
+
export declare class PoseLandmarkerTask extends AiVisionTask<PoseLandmarkerTaskInput, PoseLandmarkerTaskOutput, PoseLandmarkerTaskConfig> {
|
|
435
437
|
static type: string;
|
|
436
438
|
static category: string;
|
|
437
439
|
static title: string;
|
|
@@ -445,42 +447,7 @@ export declare class PoseLandmarkerTask extends AiVisionTask<PoseLandmarkerTaskI
|
|
|
445
447
|
* @param input The input parameters for pose landmark detection (image, model, and optional configuration)
|
|
446
448
|
* @returns Promise resolving to the detected pose landmarks and optional segmentation masks
|
|
447
449
|
*/
|
|
448
|
-
export declare const poseLandmarker: (input: {
|
|
449
|
-
image: string | {
|
|
450
|
-
channels: number;
|
|
451
|
-
data: {
|
|
452
|
-
[x: string]: unknown;
|
|
453
|
-
} | {
|
|
454
|
-
channels: number;
|
|
455
|
-
data: number[];
|
|
456
|
-
height: number;
|
|
457
|
-
rawChannels?: number | undefined;
|
|
458
|
-
width: number;
|
|
459
|
-
};
|
|
460
|
-
height: number;
|
|
461
|
-
width: number;
|
|
462
|
-
};
|
|
463
|
-
minPoseDetectionConfidence?: number | undefined;
|
|
464
|
-
minPosePresenceConfidence?: number | undefined;
|
|
465
|
-
minTrackingConfidence?: number | undefined;
|
|
466
|
-
model: string | {
|
|
467
|
-
[x: string]: unknown;
|
|
468
|
-
description?: string | undefined;
|
|
469
|
-
metadata?: {
|
|
470
|
-
[x: string]: unknown;
|
|
471
|
-
} | undefined;
|
|
472
|
-
model_id?: string | undefined;
|
|
473
|
-
provider: string;
|
|
474
|
-
provider_config: {
|
|
475
|
-
[x: string]: unknown;
|
|
476
|
-
credential_key?: string | undefined;
|
|
477
|
-
};
|
|
478
|
-
tasks?: string[] | undefined;
|
|
479
|
-
title?: string | undefined;
|
|
480
|
-
};
|
|
481
|
-
numPoses?: number | undefined;
|
|
482
|
-
outputSegmentationMasks?: boolean | undefined;
|
|
483
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
450
|
+
export declare const poseLandmarker: (input: PoseLandmarkerTaskInput, config?: PoseLandmarkerTaskConfig) => Promise<{
|
|
484
451
|
poses: {
|
|
485
452
|
landmarks: {
|
|
486
453
|
presence?: number | undefined;
|
|
@@ -529,7 +496,7 @@ export declare const poseLandmarker: (input: {
|
|
|
529
496
|
}>;
|
|
530
497
|
declare module "@workglow/task-graph" {
|
|
531
498
|
interface Workflow {
|
|
532
|
-
poseLandmarker: CreateWorkflow<PoseLandmarkerTaskInput, PoseLandmarkerTaskOutput,
|
|
499
|
+
poseLandmarker: CreateWorkflow<PoseLandmarkerTaskInput, PoseLandmarkerTaskOutput, PoseLandmarkerTaskConfig>;
|
|
533
500
|
}
|
|
534
501
|
}
|
|
535
502
|
//# sourceMappingURL=PoseLandmarkerTask.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"PoseLandmarkerTask.d.ts","sourceRoot":"","sources":["../../src/task/PoseLandmarkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AA0FnD,eAAO,MAAM,yBAAyB;mBAC9B,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,QAAQ;qBACN,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;qBACX,OAAO,EAAE,CAAC;qBACV,KAAK,EAAE,iBAAiB;qBACxB,WAAW,EAAE,uCAAuC;qBACpD,YAAY,EAAE,eAAe;;iBAE/B,0BAA0B;qBACxB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,+BAA+B;qBACtC,WAAW,EAAE,6CAA6C;qBAC1D,YAAY,EAAE,eAAe;;iBAE/B,yBAAyB;qBACvB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,8BAA8B;qBACrC,WAAW,EAAE,4CAA4C;qBACzD,YAAY,EAAE,eAAe;;iBAE/B,qBAAqB;qBACnB,IAAI,EAAE,QAAQ;qBACd,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;qBACZ,KAAK,EAAE,yBAAyB;qBAChC,WAAW,EAAE,4CAA4C;qBACzD,YAAY,EAAE,eAAe;;iBAE/B,uBAAuB;qBACrB,IAAI,EAAE,SAAS;qBACf,OAAO;qBACP,KAAK,EAAE,2BAA2B;qBAClC,WAAW,EAAE,yDAAyD;qBACtE,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,0BAA0B;mBAC/B,QAAQ;;;;+BAIA,OAAO;;mCA9Ef,QAAQ;;iCAEZ,SAAS;qCACP,IAAI,EAAE,OAAO;qCACb,KAAK;+CAlEH,QAAQ;;6CAEZ,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,uCAAuC;;6CAEtD,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,uCAAuC;;6CAEtD,CAAC;iDACC,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,cAAc;iDACrB,WAAW,EAAE,sBAAsB;;6CAErC,UAAU;iDACR,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,YAAY;iDACnB,WAAW,EAAE,2DAA2D;;6CAE1E,QAAQ;iDACN,IAAI,EAAE,QAAQ;iDACd,KAAK,EAAE,UAAU;iDACjB,WAAW,EAAE,uDAAuD;;;6DAG7D,GAAG,EAAE,GAAG,EAAE,GAAG;;;qCAuCpB,KAAK,EAAE,WAAW;qCAClB,WAAW,EAAE,wCAAwC;;iCAEvD,cAAc;qCACZ,IAAI,EAAE,OAAO;qCACb,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;qCACL,KAAK,EAAE,iBAAiB;qCACxB,WAAW,EAAE,oDAAoD;;iCAEnE,gBAAgB;2CAxCZ,QAAQ;;yCAEZ,IAAI;6CACF,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,WAAW;6CAClB,WAAW,EAAE,uDAAuD;;yCAEtE,KAAK;6CACH,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,OAAO;6CACd,WAAW,EAAE,gCAAgC;;yCAE/C,MAAM;6CACJ,IAAI,EAAE,QAAQ;6CACd,KAAK,EAAE,QAAQ;6CACf,WAAW,EAAE,iCAAiC;;;yDAGvC,MAAM,EAAE,OAAO,EAAE,QAAQ;;;;iDAwBzB,WAAW,EAAE,gBAAgB;;;;+BA+D1B,OAAO;;6BAAW,IAAI,EAAE,OAAO;6BAAE,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;4BAEzC,iBAAiB;kCACX,+DAA+D;;;;;CAK/C,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,yBAAyB,CAAC,CAAC;AACnF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,0BAA0B,CAAC,CAAC;AACrF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAE3E;;;GAGG;AACH,qBAAa,kBAAmB,SAAQ,YAAY,CAClD,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB;IACC,OAAuB,IAAI,SAAwB;IACnD,OAAuB,QAAQ,SAAqB;IACpD,OAAuB,KAAK,SAAqB;IACjD,OAAuB,WAAW,SACgF;IAClH,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;;;;GAKG;AACH,eAAO,MAAM,cAAc,UAClB,uBAAuB,WACrB,wBAAwB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EAGlC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,cAAc,EAAE,cAAc,CAC5B,uBAAuB,EACvB,wBAAwB,EACxB,wBAAwB,CACzB,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow, IExecuteContext,
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
export declare const QueryExpansionMethod: {
|
|
9
10
|
readonly MULTI_QUERY: "multi-query";
|
|
@@ -76,6 +77,7 @@ declare const outputSchema: {
|
|
|
76
77
|
};
|
|
77
78
|
export type QueryExpanderTaskInput = FromSchema<typeof inputSchema>;
|
|
78
79
|
export type QueryExpanderTaskOutput = FromSchema<typeof outputSchema>;
|
|
80
|
+
export type QueryExpanderTaskConfig = TaskConfig<QueryExpanderTaskInput>;
|
|
79
81
|
/**
|
|
80
82
|
* Task for expanding queries to improve retrieval coverage.
|
|
81
83
|
* Supports multiple expansion methods including multi-query, HyDE, and paraphrasing.
|
|
@@ -83,7 +85,7 @@ export type QueryExpanderTaskOutput = FromSchema<typeof outputSchema>;
|
|
|
83
85
|
* Note: HyDE and paraphrase methods require an LLM model.
|
|
84
86
|
* For now, this implements simple rule-based expansion.
|
|
85
87
|
*/
|
|
86
|
-
export declare class QueryExpanderTask extends Task<QueryExpanderTaskInput, QueryExpanderTaskOutput,
|
|
88
|
+
export declare class QueryExpanderTask extends Task<QueryExpanderTaskInput, QueryExpanderTaskOutput, QueryExpanderTaskConfig> {
|
|
87
89
|
static type: string;
|
|
88
90
|
static category: string;
|
|
89
91
|
static title: string;
|
|
@@ -114,12 +116,7 @@ export declare class QueryExpanderTask extends Task<QueryExpanderTaskInput, Quer
|
|
|
114
116
|
*/
|
|
115
117
|
private preserveCapitalization;
|
|
116
118
|
}
|
|
117
|
-
export declare const queryExpander: (input: {
|
|
118
|
-
method?: "hyde" | "multi-query" | "paraphrase" | "synonyms" | undefined;
|
|
119
|
-
model?: string | undefined;
|
|
120
|
-
numVariations?: number | undefined;
|
|
121
|
-
query: string;
|
|
122
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
119
|
+
export declare const queryExpander: (input: QueryExpanderTaskInput, config?: QueryExpanderTaskConfig) => Promise<{
|
|
123
120
|
count: number;
|
|
124
121
|
method: string;
|
|
125
122
|
originalQuery: string;
|
|
@@ -127,7 +124,7 @@ export declare const queryExpander: (input: {
|
|
|
127
124
|
}>;
|
|
128
125
|
declare module "@workglow/task-graph" {
|
|
129
126
|
interface Workflow {
|
|
130
|
-
queryExpander: CreateWorkflow<QueryExpanderTaskInput, QueryExpanderTaskOutput,
|
|
127
|
+
queryExpander: CreateWorkflow<QueryExpanderTaskInput, QueryExpanderTaskOutput, QueryExpanderTaskConfig>;
|
|
131
128
|
}
|
|
132
129
|
}
|
|
133
130
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"QueryExpanderTask.d.ts","sourceRoot":"","sources":["../../src/task/QueryExpanderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,
|
|
1
|
+
{"version":3,"file":"QueryExpanderTask.d.ts","sourceRoot":"","sources":["../../src/task/QueryExpanderTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,oBAAoB;aAC/B,WAAW,EAAE,aAAa;aAC1B,IAAI,EAAE,MAAM;aACZ,QAAQ,EAAE,UAAU;aACpB,UAAU,EAAE,YAAY;CAChB,CAAC;AAEX,MAAM,MAAM,oBAAoB,GAAG,CAAC,OAAO,oBAAoB,CAAC,CAAC,MAAM,OAAO,oBAAoB,CAAC,CAAC;AAEpG,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,8BAA8B;;iBAE7C,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,mCAAmC;qBAChD,OAAO;;iBAET,aAAa;qBACX,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,sBAAsB;qBAC7B,WAAW,EAAE,wCAAwC;qBACrD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;qBACX,OAAO,EAAE,CAAC;;iBAEZ,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,kEAAkE;;;;;CAKlD,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,KAAK;qBACH,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,4BAA4B;;iBAE3C,aAAa;qBACX,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,gBAAgB;qBACvB,WAAW,EAAE,0BAA0B;;iBAEzC,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,oCAAoC;;iBAEnD,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,6BAA6B;;;;;CAKb,CAAC;AAEpC,MAAM,MAAM,sBAAsB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACpE,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACtE,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,sBAAsB,CAAC,CAAC;AAEzE;;;;;;GAMG;AACH,qBAAa,iBAAkB,SAAQ,IAAI,CACzC,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,CACxB;IACC,OAAuB,IAAI,SAAuB;IAClD,OAAuB,QAAQ,SAAS;IACxC,OAAuB,KAAK,SAAoB;IAChD,OAAuB,WAAW,SAAkD;IACpF,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,sBAAsB,EAC7B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,uBAAuB,CAAC,CAgClC;IAED;;OAEG;IACH,OAAO,CAAC,mBAAmB;IAyC3B;;OAEG;IACH,OAAO,CAAC,aAAa;IAkBrB;;OAEG;IACH,OAAO,CAAC,gBAAgB;IA6CxB;;;OAGG;IACH,OAAO,CAAC,mBAAmB;IAoB3B;;OAEG;IACH,OAAO,CAAC,sBAAsB;CAM/B;AAED,eAAO,MAAM,aAAa,UAAW,sBAAsB,WAAW,uBAAuB;;;;;EAE5F,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,aAAa,EAAE,cAAc,CAC3B,sBAAsB,EACtB,uBAAuB,EACvB,uBAAuB,CACxB,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow, IExecuteContext,
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
declare const inputSchema: {
|
|
9
10
|
readonly type: "object";
|
|
@@ -170,6 +171,7 @@ declare const outputSchema: {
|
|
|
170
171
|
};
|
|
171
172
|
export type RerankerTaskInput = FromSchema<typeof inputSchema>;
|
|
172
173
|
export type RerankerTaskOutput = FromSchema<typeof outputSchema>;
|
|
174
|
+
export type RerankerTaskConfig = TaskConfig<RerankerTaskInput>;
|
|
173
175
|
/**
|
|
174
176
|
* Task for reranking retrieved chunks to improve relevance.
|
|
175
177
|
* Supports multiple reranking methods including cross-encoder models.
|
|
@@ -177,7 +179,7 @@ export type RerankerTaskOutput = FromSchema<typeof outputSchema>;
|
|
|
177
179
|
* Note: Cross-encoder reranking requires a model to be loaded.
|
|
178
180
|
* For now, this implements simple heuristic-based reranking.
|
|
179
181
|
*/
|
|
180
|
-
export declare class RerankerTask extends Task<RerankerTaskInput, RerankerTaskOutput,
|
|
182
|
+
export declare class RerankerTask extends Task<RerankerTaskInput, RerankerTaskOutput, RerankerTaskConfig> {
|
|
181
183
|
static type: string;
|
|
182
184
|
static category: string;
|
|
183
185
|
static title: string;
|
|
@@ -199,31 +201,7 @@ export declare class RerankerTask extends Task<RerankerTaskInput, RerankerTaskOu
|
|
|
199
201
|
*/
|
|
200
202
|
private reciprocalRankFusion;
|
|
201
203
|
}
|
|
202
|
-
export declare const reranker: (input: {
|
|
203
|
-
chunks: string[];
|
|
204
|
-
metadata?: {
|
|
205
|
-
[x: string]: unknown;
|
|
206
|
-
}[] | undefined;
|
|
207
|
-
method?: "cross-encoder" | "reciprocal-rank-fusion" | "simple" | undefined;
|
|
208
|
-
model?: string | {
|
|
209
|
-
[x: string]: unknown;
|
|
210
|
-
description?: string | undefined;
|
|
211
|
-
metadata?: {
|
|
212
|
-
[x: string]: unknown;
|
|
213
|
-
} | undefined;
|
|
214
|
-
model_id?: string | undefined;
|
|
215
|
-
provider: string;
|
|
216
|
-
provider_config: {
|
|
217
|
-
[x: string]: unknown;
|
|
218
|
-
credential_key?: string | undefined;
|
|
219
|
-
};
|
|
220
|
-
tasks?: string[] | undefined;
|
|
221
|
-
title?: string | undefined;
|
|
222
|
-
} | undefined;
|
|
223
|
-
query: string;
|
|
224
|
-
scores?: number[] | undefined;
|
|
225
|
-
topK?: number | undefined;
|
|
226
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
204
|
+
export declare const reranker: (input: RerankerTaskInput, config?: RerankerTaskConfig) => Promise<{
|
|
227
205
|
chunks: string[];
|
|
228
206
|
count: number;
|
|
229
207
|
metadata?: {
|
|
@@ -234,7 +212,7 @@ export declare const reranker: (input: {
|
|
|
234
212
|
}>;
|
|
235
213
|
declare module "@workglow/task-graph" {
|
|
236
214
|
interface Workflow {
|
|
237
|
-
reranker: CreateWorkflow<RerankerTaskInput, RerankerTaskOutput,
|
|
215
|
+
reranker: CreateWorkflow<RerankerTaskInput, RerankerTaskOutput, RerankerTaskConfig>;
|
|
238
216
|
}
|
|
239
217
|
}
|
|
240
218
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,
|
|
1
|
+
{"version":3,"file":"RerankerTask.d.ts","sourceRoot":"","sources":["../../src/task/RerankerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAMnE,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,qCAAqC;;iBAEpD,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,iCAAiC;;iBAEhD,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,gBAAgB;qBACvB,WAAW,EAAE,qCAAqC;;iBAEpD,QAAQ;qBACN,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,UAAU;yBACjB,WAAW,EAAE,yBAAyB;;qBAExC,KAAK,EAAE,UAAU;qBACjB,WAAW,EAAE,oCAAoC;;iBAEnD,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,iDAAiD;qBAC9D,OAAO,EAAE,CAAC;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,6BAA6B;qBAC1C,OAAO,EAAE,QAAQ;;iBAEnB,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAO0B,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,iBAAiB;qBACxB,WAAW,EAAE,+BAA+B;;iBAE9C,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,iBAAiB;qBACxB,WAAW,EAAE,sBAAsB;;iBAErC,QAAQ;qBACN,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,UAAU;yBACjB,WAAW,EAAE,yBAAyB;;qBAExC,KAAK,EAAE,UAAU;qBACjB,WAAW,EAAE,8BAA8B;;iBAE7C,eAAe;qBACb,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,qCAAqC;;iBAEpD,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,4BAA4B;;;;;CAKZ,CAAC;AAEpC,MAAM,MAAM,iBAAiB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAC/D,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACjE,MAAM,MAAM,kBAAkB,GAAG,UAAU,CAAC,iBAAiB,CAAC,CAAC;AAS/D;;;;;;GAMG;AACH,qBAAa,YAAa,SAAQ,IAAI,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC;IAC/F,OAAuB,IAAI,SAAkB;IAC7C,OAAuB,QAAQ,SAAS;IACxC,OAAuB,KAAK,SAAc;IAC1C,OAAuB,WAAW,SAAkD;IACpF,OAAuB,SAAS,UAAQ;IACxC,OAAO,CAAC,yBAAyB,CAAC,CAAgB;IAElD,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,iBAAiB,EACxB,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,kBAAkB,CAAC,CA0C7B;YAEa,kBAAkB;IAqChC,OAAO,CAAC,wBAAwB;IAqBhC;;OAEG;IACH,OAAO,CAAC,YAAY;IAuDpB;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAqB7B;AAED,eAAO,MAAM,QAAQ,UAAW,iBAAiB,WAAW,kBAAkB;;;;;;;;EAE7E,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,QAAQ,EAAE,cAAc,CAAC,iBAAiB,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;KACrF;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow, IExecuteContext,
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
declare const inputSchema: {
|
|
9
10
|
readonly type: "object";
|
|
@@ -62,11 +63,12 @@ declare const outputSchema: {
|
|
|
62
63
|
};
|
|
63
64
|
export type StructuralParserTaskInput = FromSchema<typeof inputSchema>;
|
|
64
65
|
export type StructuralParserTaskOutput = FromSchema<typeof outputSchema>;
|
|
66
|
+
export type StructuralParserTaskConfig = TaskConfig<StructuralParserTaskInput>;
|
|
65
67
|
/**
|
|
66
68
|
* Task for parsing documents into hierarchical tree structure
|
|
67
69
|
* Supports markdown and plain text with automatic format detection
|
|
68
70
|
*/
|
|
69
|
-
export declare class StructuralParserTask extends Task<StructuralParserTaskInput, StructuralParserTaskOutput,
|
|
71
|
+
export declare class StructuralParserTask extends Task<StructuralParserTaskInput, StructuralParserTaskOutput, StructuralParserTaskConfig> {
|
|
70
72
|
static type: string;
|
|
71
73
|
static category: string;
|
|
72
74
|
static title: string;
|
|
@@ -77,20 +79,14 @@ export declare class StructuralParserTask extends Task<StructuralParserTaskInput
|
|
|
77
79
|
execute(input: StructuralParserTaskInput, context: IExecuteContext): Promise<StructuralParserTaskOutput>;
|
|
78
80
|
private countNodes;
|
|
79
81
|
}
|
|
80
|
-
export declare const structuralParser: (input: {
|
|
81
|
-
doc_id?: string | undefined;
|
|
82
|
-
format?: "auto" | "markdown" | "text" | undefined;
|
|
83
|
-
sourceUri?: string | undefined;
|
|
84
|
-
text: string;
|
|
85
|
-
title: string;
|
|
86
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
82
|
+
export declare const structuralParser: (input: StructuralParserTaskInput, config?: StructuralParserTaskConfig) => Promise<{
|
|
87
83
|
doc_id: string;
|
|
88
84
|
documentTree: unknown;
|
|
89
85
|
nodeCount: number;
|
|
90
86
|
}>;
|
|
91
87
|
declare module "@workglow/task-graph" {
|
|
92
88
|
interface Workflow {
|
|
93
|
-
structuralParser: CreateWorkflow<StructuralParserTaskInput, StructuralParserTaskOutput,
|
|
89
|
+
structuralParser: CreateWorkflow<StructuralParserTaskInput, StructuralParserTaskOutput, StructuralParserTaskConfig>;
|
|
94
90
|
}
|
|
95
91
|
}
|
|
96
92
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StructuralParserTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuralParserTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,
|
|
1
|
+
{"version":3,"file":"StructuralParserTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuralParserTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AACvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAGnE,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,2BAA2B;;iBAE1C,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,OAAO;qBACd,WAAW,EAAE,gBAAgB;;iBAE/B,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,iDAAiD;qBAC9D,OAAO,EAAE,MAAM;;iBAEjB,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,8CAA8C;;iBAE7D,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,sCAAsC;;;;;CAKtB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,mCAAmC;;iBAElD,YAAY;qBACV,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,mCAAmC;;iBAElD,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,mCAAmC;;;;;CAKnB,CAAC;AAEpC,MAAM,MAAM,yBAAyB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AACvE,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACzE,MAAM,MAAM,0BAA0B,GAAG,UAAU,CAAC,yBAAyB,CAAC,CAAC;AAE/E;;;GAGG;AACH,qBAAa,oBAAqB,SAAQ,IAAI,CAC5C,yBAAyB,EACzB,0BAA0B,EAC1B,0BAA0B,CAC3B;IACC,OAAuB,IAAI,SAA0B;IACrD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAuB;IACnD,OAAuB,WAAW,SAAsD;IACxF,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,yBAAyB,EAChC,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,0BAA0B,CAAC,CAyBrC;IAED,OAAO,CAAC,UAAU;CASnB;AAED,eAAO,MAAM,gBAAgB,UACpB,yBAAyB,WACvB,0BAA0B;;;;EAGpC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,gBAAgB,EAAE,cAAc,CAC9B,yBAAyB,EACzB,0BAA0B,EAC1B,0BAA0B,CAC3B,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow
|
|
6
|
+
import { CreateWorkflow } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
import { StreamingAiTask } from "./base/StreamingAiTask";
|
|
9
10
|
export declare const StructuredGenerationInputSchema: {
|
|
@@ -114,7 +115,8 @@ export declare const StructuredGenerationOutputSchema: {
|
|
|
114
115
|
};
|
|
115
116
|
export type StructuredGenerationTaskInput = FromSchema<typeof StructuredGenerationInputSchema>;
|
|
116
117
|
export type StructuredGenerationTaskOutput = FromSchema<typeof StructuredGenerationOutputSchema>;
|
|
117
|
-
export
|
|
118
|
+
export type StructuredGenerationTaskConfig = TaskConfig<StructuredGenerationTaskInput>;
|
|
119
|
+
export declare class StructuredGenerationTask extends StreamingAiTask<StructuredGenerationTaskInput, StructuredGenerationTaskOutput, StructuredGenerationTaskConfig> {
|
|
118
120
|
static type: string;
|
|
119
121
|
static category: string;
|
|
120
122
|
static title: string;
|
|
@@ -125,36 +127,14 @@ export declare class StructuredGenerationTask extends StreamingAiTask<Structured
|
|
|
125
127
|
/**
|
|
126
128
|
* Task for generating structured JSON output using a language model
|
|
127
129
|
*/
|
|
128
|
-
export declare const structuredGeneration: (input: {
|
|
129
|
-
maxTokens?: number | undefined;
|
|
130
|
-
model: string | {
|
|
131
|
-
[x: string]: unknown;
|
|
132
|
-
description?: string | undefined;
|
|
133
|
-
metadata?: {
|
|
134
|
-
[x: string]: unknown;
|
|
135
|
-
} | undefined;
|
|
136
|
-
model_id?: string | undefined;
|
|
137
|
-
provider: string;
|
|
138
|
-
provider_config: {
|
|
139
|
-
[x: string]: unknown;
|
|
140
|
-
credential_key?: string | undefined;
|
|
141
|
-
};
|
|
142
|
-
tasks?: string[] | undefined;
|
|
143
|
-
title?: string | undefined;
|
|
144
|
-
};
|
|
145
|
-
outputSchema: {
|
|
146
|
-
[x: string]: unknown;
|
|
147
|
-
};
|
|
148
|
-
prompt: string;
|
|
149
|
-
temperature?: number | undefined;
|
|
150
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
130
|
+
export declare const structuredGeneration: (input: StructuredGenerationTaskInput, config?: StructuredGenerationTaskConfig) => Promise<{
|
|
151
131
|
object: {
|
|
152
132
|
[x: string]: unknown;
|
|
153
133
|
};
|
|
154
134
|
}>;
|
|
155
135
|
declare module "@workglow/task-graph" {
|
|
156
136
|
interface Workflow {
|
|
157
|
-
structuredGeneration: CreateWorkflow<StructuredGenerationTaskInput, StructuredGenerationTaskOutput,
|
|
137
|
+
structuredGeneration: CreateWorkflow<StructuredGenerationTaskInput, StructuredGenerationTaskOutput, StructuredGenerationTaskConfig>;
|
|
158
138
|
}
|
|
159
139
|
}
|
|
160
140
|
//# sourceMappingURL=StructuredGenerationTask.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"StructuredGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuredGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,UAAU,
|
|
1
|
+
{"version":3,"file":"StructuredGenerationTask.d.ts","sourceRoot":"","sources":["../../src/task/StructuredGenerationTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AAIzD,eAAO,MAAM,+BAA+B;mBACpC,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;iBACL,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,+CAA+C;;iBAE9D,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,qDAAqD;qBAClE,oBAAoB;;iBAEtB,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,IAAI;qBACb,YAAY,EAAE,eAAe;;iBAE/B,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;qBAClD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,gCAAgC;mBACrC,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,mBAAmB;qBAC1B,WAAW,EAAE,mEAAmE;qBAChF,UAAU,EAAE,QAAQ;qBACpB,qBAAqB;qBACrB,oBAAoB;;;;;CAKS,CAAC;AAEpC,MAAM,MAAM,6BAA6B,GAAG,UAAU,CAAC,OAAO,+BAA+B,CAAC,CAAC;AAC/F,MAAM,MAAM,8BAA8B,GAAG,UAAU,CAAC,OAAO,gCAAgC,CAAC,CAAC;AACjG,MAAM,MAAM,8BAA8B,GAAG,UAAU,CAAC,6BAA6B,CAAC,CAAC;AAEvF,qBAAa,wBAAyB,SAAQ,eAAe,CAC3D,6BAA6B,EAC7B,8BAA8B,EAC9B,8BAA8B,CAC/B;IACC,OAAuB,IAAI,SAA8B;IACzD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAA2B;IACvD,OAAuB,WAAW,SACyD;IAC3F,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;GAEG;AACH,eAAO,MAAM,oBAAoB,UACxB,6BAA6B,WAC3B,8BAA8B;;;;EAGxC,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,oBAAoB,EAAE,cAAc,CAClC,6BAA6B,EAC7B,8BAA8B,EAC9B,8BAA8B,CAC/B,CAAC;KACH;CACF"}
|
|
@@ -3,7 +3,8 @@
|
|
|
3
3
|
* Copyright 2025 Steven Roussey <sroussey@gmail.com>
|
|
4
4
|
* SPDX-License-Identifier: Apache-2.0
|
|
5
5
|
*/
|
|
6
|
-
import { CreateWorkflow, IExecuteContext,
|
|
6
|
+
import { CreateWorkflow, IExecuteContext, Task } from "@workglow/task-graph";
|
|
7
|
+
import type { TaskConfig } from "@workglow/task-graph";
|
|
7
8
|
import { DataPortSchema, FromSchema } from "@workglow/util/schema";
|
|
8
9
|
export declare const ChunkingStrategy: {
|
|
9
10
|
readonly FIXED: "fixed";
|
|
@@ -85,11 +86,12 @@ declare const outputSchema: {
|
|
|
85
86
|
};
|
|
86
87
|
export type TextChunkerTaskInput = FromSchema<typeof inputSchema>;
|
|
87
88
|
export type TextChunkerTaskOutput = FromSchema<typeof outputSchema>;
|
|
89
|
+
export type TextChunkerTaskConfig = TaskConfig<TextChunkerTaskInput>;
|
|
88
90
|
/**
|
|
89
91
|
* Task for chunking text into smaller segments with configurable strategies.
|
|
90
92
|
* Supports fixed-size, sentence-based, paragraph-based, and semantic chunking.
|
|
91
93
|
*/
|
|
92
|
-
export declare class TextChunkerTask extends Task<TextChunkerTaskInput, TextChunkerTaskOutput,
|
|
94
|
+
export declare class TextChunkerTask extends Task<TextChunkerTaskInput, TextChunkerTaskOutput, TextChunkerTaskConfig> {
|
|
93
95
|
static type: string;
|
|
94
96
|
static category: string;
|
|
95
97
|
static title: string;
|
|
@@ -111,12 +113,7 @@ export declare class TextChunkerTask extends Task<TextChunkerTaskInput, TextChun
|
|
|
111
113
|
*/
|
|
112
114
|
private chunkByParagraph;
|
|
113
115
|
}
|
|
114
|
-
export declare const textChunker: (input: {
|
|
115
|
-
chunkOverlap?: number | undefined;
|
|
116
|
-
chunkSize?: number | undefined;
|
|
117
|
-
strategy?: "fixed" | "paragraph" | "semantic" | "sentence" | undefined;
|
|
118
|
-
text: string;
|
|
119
|
-
}, config?: TaskConfig | undefined) => Promise<{
|
|
116
|
+
export declare const textChunker: (input: TextChunkerTaskInput, config?: TextChunkerTaskConfig) => Promise<{
|
|
120
117
|
chunks: string[];
|
|
121
118
|
metadata: {
|
|
122
119
|
endChar?: number | undefined;
|
|
@@ -127,7 +124,7 @@ export declare const textChunker: (input: {
|
|
|
127
124
|
}>;
|
|
128
125
|
declare module "@workglow/task-graph" {
|
|
129
126
|
interface Workflow {
|
|
130
|
-
textChunker: CreateWorkflow<TextChunkerTaskInput, TextChunkerTaskOutput,
|
|
127
|
+
textChunker: CreateWorkflow<TextChunkerTaskInput, TextChunkerTaskOutput, TextChunkerTaskConfig>;
|
|
131
128
|
}
|
|
132
129
|
}
|
|
133
130
|
export {};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,
|
|
1
|
+
{"version":3,"file":"TextChunkerTask.d.ts","sourceRoot":"","sources":["../../src/task/TextChunkerTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAE,eAAe,EAAE,IAAI,EAAY,MAAM,sBAAsB,CAAC;AAEvF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AAEnE,eAAO,MAAM,gBAAgB;aAC3B,KAAK,EAAE,OAAO;aACd,QAAQ,EAAE,UAAU;aACpB,SAAS,EAAE,WAAW;aACtB,QAAQ,EAAE,UAAU;CACZ,CAAC;AAEX,MAAM,MAAM,gBAAgB,GAAG,CAAC,OAAO,gBAAgB,CAAC,CAAC,MAAM,OAAO,gBAAgB,CAAC,CAAC;AAExF,QAAA,MAAM,WAAW;mBACT,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,mBAAmB;;iBAElC,SAAS;qBACP,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,GAAG;;iBAEd,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,gDAAgD;qBAC7D,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,EAAE;;iBAEb,QAAQ;qBACN,IAAI,EAAE,QAAQ;qBACd,IAAI;qBACJ,KAAK,EAAE,mBAAmB;qBAC1B,WAAW,EAAE,mCAAmC;qBAChD,OAAO;;;;;CAKsB,CAAC;AAEpC,QAAA,MAAM,YAAY;mBACV,QAAQ;;iBAEZ,MAAM;qBACJ,IAAI,EAAE,OAAO;qBACb,KAAK;yBAAI,IAAI,EAAE,QAAQ;;qBACvB,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,2BAA2B;;iBAE1C,QAAQ;qBACN,IAAI,EAAE,OAAO;qBACb,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,UAAU;6BACR,KAAK;iCAAI,IAAI,EAAE,QAAQ;;6BACvB,SAAS;iCAAI,IAAI,EAAE,QAAQ;;6BAC3B,OAAO;iCAAI,IAAI,EAAE,QAAQ;;6BACzB,MAAM;iCAAI,IAAI,EAAE,QAAQ;;;yBAE1B,oBAAoB;;qBAEtB,KAAK,EAAE,gBAAgB;qBACvB,WAAW,EAAE,yBAAyB;;;;;CAKT,CAAC;AAEpC,MAAM,MAAM,oBAAoB,GAAG,UAAU,CAAC,OAAO,WAAW,CAAC,CAAC;AAClE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AACpE,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AASrE;;;GAGG;AACH,qBAAa,eAAgB,SAAQ,IAAI,CACvC,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SACgD;IAClF,OAAuB,SAAS,UAAQ;IAExC,OAAuB,WAAW,IAAI,cAAc,CAEnD;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAEc,OAAO,CACpB,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,OAAO,CAAC,qBAAqB,CAAC,CAyBhC;IAED;;OAEG;IACH,OAAO,CAAC,UAAU;IA8BlB;;OAEG;IACH,OAAO,CAAC,eAAe;IAoFvB;;OAEG;IACH,OAAO,CAAC,gBAAgB;CAoEzB;AAED,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB;;;;;;;;EAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}
|