From 5ed7c4fbe1858b0eaa773af2dc1f4ccfce22b3c2 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 7 Feb 2024 20:47:29 -0300
Subject: [PATCH] Add nodeColors and nodeNames for models

---
 src/frontend/src/utils/styleUtils.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/utils/styleUtils.ts b/src/frontend/src/utils/styleUtils.ts
index 541ef5322..41ec2d047 100644
--- a/src/frontend/src/utils/styleUtils.ts
+++ b/src/frontend/src/utils/styleUtils.ts
@@ -183,6 +183,7 @@ export const gradients = [
 
 export const nodeColors: { [char: string]: string } = {
   prompts: "#4367BF",
+  models: "#AA2411",
   llms: "#6344BE",
   chains: "#FE7500",
   Document: "#7AAE42",
@@ -211,7 +212,8 @@ export const nodeColors: { [char: string]: string } = {
 
 export const nodeNames: { [char: string]: string } = {
   prompts: "Prompts",
-  llms: "LLMs",
+  models: "Language Models",
+  llms: "Model Settings",
   chains: "Chains",
   agents: "Agents",
   tools: "Tools",
@@ -281,6 +283,7 @@ export const nodeIconsLucide: iconsType = {
   WikipediaAPIWrapper: SvgWikipedia,
   chains: Link,
   memories: Cpu,
+  models: Lightbulb,
   llms: Lightbulb,
   prompts: TerminalSquare,
   tools: Wrench,