Skip to content

Commit 28c1ef9

Browse files
WIP: Add task readable name pipes
1 parent 5cd7093 commit 28c1ef9

3 files changed

Lines changed: 14 additions & 1 deletion

File tree

libs/envi/modeler/src/lib/create-envi-modeler-workflow.ts

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import { MCPTaskRegistry } from '@idl/mcp/tasks';
12
import { ENVIModelerEdge, ENVIModelerNode } from '@idl/types/envi/modeler';
23

34
import {
@@ -18,6 +19,7 @@ import { RemoveSingletonNodes } from './helpers/remove-singleton-nodes';
1819
export function CreateENVIModelerWorkflow(
1920
nodes: ENVIModelerNode[],
2021
edges: ENVIModelerEdge[],
22+
registry: MCPTaskRegistry,
2123
): Record<string, unknown> {
2224
/**
2325
* Preprocess nodes to simplify and automatically resolve errors
@@ -39,7 +41,7 @@ export function CreateENVIModelerWorkflow(
3941
LAYOUT_BASE_X,
4042
LAYOUT_BASE_Y,
4143
];
42-
return BuildNodeJSON(node, modelName, location);
44+
return BuildNodeJSON(node, modelName, location, registry);
4345
});
4446

4547
// Build edges array

libs/envi/modeler/src/lib/helpers/build-node-json.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import { MCPTaskRegistry } from '@idl/mcp/tasks';
12
import { ENVIModelerNode } from '@idl/types/mcp';
23

34
import { FIXED_DISPLAY_NAMES } from '../create-envi-modeler-workflow.interface';
@@ -7,6 +8,7 @@ export function BuildNodeJSON(
78
node: ENVIModelerNode,
89
modelName: string,
910
location: [number, number],
11+
registry: MCPTaskRegistry,
1012
): Record<string, unknown> {
1113
const base: Record<string, unknown> = {
1214
display_name:
@@ -93,6 +95,14 @@ export function BuildNodeJSON(
9395
envitask['revision'] = node.revision;
9496
}
9597
base['envitask'] = envitask;
98+
99+
// get display name
100+
const info = registry.getTaskDetail(node.task_name as string);
101+
102+
// set display name of task
103+
if (info.structure.meta.readableName) {
104+
base['display_name'] = info.structure.meta.readableName;
105+
}
96106
break;
97107
}
98108

libs/mcp/server-tools/src/lib/tools/envi/register-mcp-tool-create-envi-modeler-workflow.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,7 @@ export function RegisterMCPTool_CreateENVIModelerWorkflow(
212212
const modelJSON = CreateENVIModelerWorkflow(
213213
nodes as ENVIModelerNode[],
214214
edges as ENVIModelerEdge[],
215+
registry,
215216
);
216217
const modelContent = JSON.stringify(modelJSON, null, 4);
217218

0 commit comments

Comments
 (0)