jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Added colors to types

Browse source
This commit is contained in:
Lucas Oliveira 2023-12-04 16:35:06 -03:00
commit d19fdf5373
2 changed files with 11 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/frontend/src/CustomNodes/GenericNode/index.tsx
View file

@ -464,10 +464,10 @@ export default function GenericNode({
data={data} data={data}
color={ color={
nodeColors[ nodeColors[
types[data.node?.template[templateField].type!] data.node?.template[templateField].type!
] ?? ] ??
nodeColors[ nodeColors[
data.node?.template[templateField].type! types[data.node?.template[templateField].type!]
] ?? ] ??
nodeColors.unknown nodeColors.unknown
} }
@ -519,7 +519,13 @@ export default function GenericNode({
dataType: data.type, dataType: data.type,
})} })}
data={data} data={data}
color={nodeColors[types[data.type]] ?? nodeColors.unknown} color={
(data.node?.output_types &&
data.node.output_types.length > 0
? nodeColors[data.node.output_types[0]] ??
nodeColors[types[data.node.output_types[0]]]
: nodeColors[types[data.type]]) ?? nodeColors.unknown
}
title={ title={
data.node?.output_types && data.node.output_types.length > 0 data.node?.output_types && data.node.output_types.length > 0
? data.node.output_types.join("|") ? data.node.output_types.join("|")

2
src/frontend/src/utils/styleUtils.ts
View file

@ -174,6 +174,8 @@ export const nodeColors: { [char: string]: string } = {
prompts: "#4367BF", prompts: "#4367BF",
llms: "#6344BE", llms: "#6344BE",
chains: "#FE7500", chains: "#FE7500",
Document: "#7AAE42",
list: "#9AAE42",
agents: "#903BBE", agents: "#903BBE",
tools: "#FF3434", tools: "#FF3434",
memories: "#F5B85A", memories: "#F5B85A",