diff --git a/chain-forge/src/EvaluatorNode.js b/chain-forge/src/EvaluatorNode.js
index a7841e8..72b9281 100644
--- a/chain-forge/src/EvaluatorNode.js
+++ b/chain-forge/src/EvaluatorNode.js
@@ -4,6 +4,7 @@ import useStore from './store';
import StatusIndicator from './StatusIndicatorComponent'
import NodeLabel from './NodeLabelComponent'
import AlertModal from './AlertModal'
+import { IconTerminal } from '@tabler/icons-react'
// Ace code editor
import AceEditor from "react-ace";
@@ -159,9 +160,10 @@ const EvaluatorNode = ({ data, id }) => {
onMouseLeave={handleMouseLeave}
>
-
+ onEdit={hideStatusIndicator}
+ icon={
} />
diff --git a/chain-forge/src/InspectorNode.js b/chain-forge/src/InspectorNode.js
index e00ebc3..f1b0271 100644
--- a/chain-forge/src/InspectorNode.js
+++ b/chain-forge/src/InspectorNode.js
@@ -1,6 +1,7 @@
import React, { useState } from 'react';
import { Handle } from 'react-flow-renderer';
import useStore from './store';
+import NodeLabel from './NodeLabelComponent'
const bucketResponsesByLLM = (responses) => {
let responses_by_llm = {};
@@ -116,8 +117,10 @@ const InspectorNode = ({ data, id }) => {
return (
{/*
{varSelects}
*/}
diff --git a/chain-forge/src/NodeLabelComponent.js b/chain-forge/src/NodeLabelComponent.js
index 5b93215..bcec710 100644
--- a/chain-forge/src/NodeLabelComponent.js
+++ b/chain-forge/src/NodeLabelComponent.js
@@ -2,7 +2,7 @@ import useStore from './store';
import { EditText } from 'react-edit-text';
import 'react-edit-text/dist/index.css';
-export default function NodeLabel({ title, nodeId, onEdit, onSave }) {
+export default function NodeLabel({ title, nodeId, icon, onEdit, onSave }) {
const setDataPropsForNode = useStore((state) => state.setDataPropsForNode);
const handleNodeLabelChange = (evt) => {
@@ -15,13 +15,14 @@ export default function NodeLabel({ title, nodeId, onEdit, onSave }) {
if (onEdit) onEdit();
};
- return (
+ return (<>
+ {icon ? (<>{icon} >) : <>>}
- );
+ >);
}
\ No newline at end of file
diff --git a/chain-forge/src/PromptNode.js b/chain-forge/src/PromptNode.js
index e7290f5..79be80c 100644
--- a/chain-forge/src/PromptNode.js
+++ b/chain-forge/src/PromptNode.js
@@ -291,7 +291,8 @@ const PromptNode = ({ data, id }) => {
+ onEdit={hideStatusIndicator}
+ icon={'💬'} />
diff --git a/chain-forge/src/TextFieldsNode.js b/chain-forge/src/TextFieldsNode.js
index 8c37287..5226bf8 100644
--- a/chain-forge/src/TextFieldsNode.js
+++ b/chain-forge/src/TextFieldsNode.js
@@ -1,5 +1,6 @@
import React, { useState, useRef, useEffect, useCallback } from 'react';
import { Handle } from 'react-flow-renderer';
+import { IconTextPlus } from '@tabler/icons-react'
import useStore from './store';
import NodeLabel from './NodeLabelComponent'
import TemplateHooks from './TemplateHooksComponent';
@@ -96,7 +97,7 @@ const TextFieldsNode = ({ data, id }) => {
return (
-
+ } />
{fields}
diff --git a/chain-forge/src/VisNode.js b/chain-forge/src/VisNode.js
index 80e14c2..fdec6e3 100644
--- a/chain-forge/src/VisNode.js
+++ b/chain-forge/src/VisNode.js
@@ -182,7 +182,8 @@ const VisNode = ({ data, id }) => {
>
+ nodeId={id}
+ icon={'📊'} />
{plotlyObj}