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 (
- Inspect Node -
+ +
{/*
{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}