From a9c7fc0a6983c942b96b45f0158cbaecad11f74f Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 30 May 2023 00:38:14 -0300
Subject: [PATCH 01/59] =?UTF-8?q?=F0=9F=94=A7=20chore(config):=20add=20Vec?=
 =?UTF-8?q?torStoreToolkit=20to=20toolkits=20list=20=F0=9F=90=9B=20fix(bas?=
 =?UTF-8?q?e.py):=20remove=20deepcopy=20for=20VectorStore=20and=20VectorSt?=
 =?UTF-8?q?oreRouter=20agents=20=F0=9F=90=9B=20fix(nodes.py):=20remove=20d?=
 =?UTF-8?q?eepcopy=20for=20VectorStore=20and=20VectorStoreRouter=20agents?=
 =?UTF-8?q?=20=F0=9F=94=A7=20chore(loading.py):=20comment=20out=20unused?=
 =?UTF-8?q?=20code=20for=20loading=20toolkits=20=F0=9F=90=9B=20fix(toolkit?=
 =?UTF-8?q?s/base.py):=20add=20Tool=20to=20base=5Fclasses=20in=20get=5Fsig?=
 =?UTF-8?q?nature=20method=20The=20changes=20to=20the=20config=20file=20ad?=
 =?UTF-8?q?d=20the=20VectorStoreToolkit=20to=20the=20list=20of=20toolkits.?=
 =?UTF-8?q?=20The=20deepcopy=20for=20VectorStore=20and=20VectorStoreRouter?=
 =?UTF-8?q?=20agents=20was=20causing=20issues,=20so=20it=20was=20removed?=
 =?UTF-8?q?=20from=20the=20base.py=20and=20nodes.py=20files.=20The=20loadi?=
 =?UTF-8?q?ng.py=20file=20had=20some=20unused=20code=20for=20loading=20too?=
 =?UTF-8?q?lkits,=20so=20it=20was=20commented=20out.=20Finally,=20the=20ba?=
 =?UTF-8?q?se.py=20file=20had=20a=20bug=20where=20the=20Tool=20class=20was?=
 =?UTF-8?q?=20not=20being=20added=20to=20the=20base=5Fclasses=20list=20in?=
 =?UTF-8?q?=20the=20get=5Fsignature=20method,=20so=20it=20was=20added.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/config.yaml                |  1 +
 src/backend/langflow/graph/base.py              | 14 +-------------
 src/backend/langflow/graph/nodes.py             |  5 +----
 src/backend/langflow/interface/loading.py       |  7 +++++--
 src/backend/langflow/interface/toolkits/base.py | 15 +++++++++------
 5 files changed, 17 insertions(+), 25 deletions(-)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index 02b17cd85..b073ed544 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -74,6 +74,7 @@ toolkits:
   - JsonToolkit
   - VectorStoreInfo
   - VectorStoreRouterToolkit
+  - VectorStoreToolkit
 tools:
   - Search
   - PAL-MATH
diff --git a/src/backend/langflow/graph/base.py b/src/backend/langflow/graph/base.py
index 187d2983e..5b64885fb 100644
--- a/src/backend/langflow/graph/base.py
+++ b/src/backend/langflow/graph/base.py
@@ -212,19 +212,7 @@ class Node:
         if not self._built or force:
             self._build()
 
-        #! Deepcopy is breaking for vectorstores
-        if self.base_type in [
-            "vectorstores",
-            "VectorStoreRouterAgent",
-            "VectorStoreAgent",
-            "VectorStoreInfo",
-        ] or self.node_type in [
-            "VectorStoreInfo",
-            "VectorStoreRouterToolkit",
-            "SQLDatabase",
-        ]:
-            return self._built_object
-        return deepcopy(self._built_object)
+        return self._built_object
 
     def add_edge(self, edge: "Edge") -> None:
         self.edges.append(edge)
diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 189e40b5c..7d9b05366 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -14,7 +14,7 @@ class AgentNode(Node):
     def _set_tools_and_chains(self) -> None:
         for edge in self.edges:
             source_node = edge.source
-            if isinstance(source_node, ToolNode):
+            if isinstance(source_node, (ToolNode, ToolkitNode)):
                 self.tools.append(source_node)
             elif isinstance(source_node, ChainNode):
                 self.chains.append(source_node)
@@ -32,9 +32,6 @@ class AgentNode(Node):
 
             self._build()
 
-        #! Cannot deepcopy VectorStore, VectorStoreRouter, or SQL agents
-        if self.node_type in ["VectorStoreAgent", "VectorStoreRouterAgent", "SQLAgent"]:
-            return self._built_object
         return self._built_object
 
 
diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py
index cd6898a7f..d720c6b0c 100644
--- a/src/backend/langflow/interface/loading.py
+++ b/src/backend/langflow/interface/loading.py
@@ -101,8 +101,11 @@ def instantiate_tool(node_type, class_object, params):
 
 def instantiate_toolkit(node_type, class_object, params):
     loaded_toolkit = class_object(**params)
-    if toolkits_creator.has_create_function(node_type):
-        return load_toolkits_executor(node_type, loaded_toolkit, params)
+    # Commenting this out for now to use toolkits as normal tools
+    # if toolkits_creator.has_create_function(node_type):
+    #     return load_toolkits_executor(node_type, loaded_toolkit, params)
+    if isinstance(loaded_toolkit, BaseToolkit):
+        return loaded_toolkit.get_tools()
     return loaded_toolkit
 
 
diff --git a/src/backend/langflow/interface/toolkits/base.py b/src/backend/langflow/interface/toolkits/base.py
index cbe625f0d..9f01b2bb2 100644
--- a/src/backend/langflow/interface/toolkits/base.py
+++ b/src/backend/langflow/interface/toolkits/base.py
@@ -42,24 +42,27 @@ class ToolkitCreator(LangChainTypeCreator):
 
     def get_signature(self, name: str) -> Optional[Dict]:
         try:
-            return build_template_from_class(name, self.type_to_loader_dict)
+            template = build_template_from_class(name, self.type_to_loader_dict)
+            # add Tool to base_classes
+            if template:
+                template["base_classes"].append("Tool")
+            return template
         except ValueError as exc:
-            raise ValueError("Prompt not found") from exc
+            raise ValueError("Toolkit not found") from exc
         except AttributeError as exc:
-            logger.error(f"Prompt {name} not loaded: {exc}")
+            logger.error(f"Toolkit {name} not loaded: {exc}")
             return None
 
     def to_list(self) -> List[str]:
         return list(self.type_to_loader_dict.keys())
 
     def get_create_function(self, name: str) -> Callable:
-        if loader_name := self.create_functions.get(name, None):
-            # import loader
+        if loader_name := self.create_functions.get(name):
             return import_module(
                 f"from langchain.agents.agent_toolkits import {loader_name[0]}"
             )
         else:
-            raise ValueError("Loader not found")
+            raise ValueError("Toolkit not found")
 
     def has_create_function(self, name: str) -> bool:
         # check if the function list is not empty

From ad3bb997eed981c9426e0b76b16f84309b18ae74 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 30 May 2023 01:11:27 -0300
Subject: [PATCH 02/59] =?UTF-8?q?=F0=9F=90=9B=20fix(base.py):=20extend=20l?=
 =?UTF-8?q?ist=20only=20if=20key=20exists=20and=20is=20a=20list=20?=
 =?UTF-8?q?=F0=9F=90=9B=20fix(nodes.py):=20flatten=20list=20of=20tools=20i?=
 =?UTF-8?q?f=20it=20is=20a=20list=20of=20lists=20=F0=9F=90=9B=20fix(toolki?=
 =?UTF-8?q?ts/base.py):=20add=20"toolkit"=20check=20to=20avoid=20adding=20?=
 =?UTF-8?q?"Tool"=20to=20non-toolkit=20classes=20=F0=9F=93=9D=20docs(agent?=
 =?UTF-8?q?s.py):=20update=20node=20descriptions=20to=20reflect=20CSV=20an?=
 =?UTF-8?q?d=20zero=20shot=20agents=20The=20changes=20in=20base.py=20and?=
 =?UTF-8?q?=20nodes.py=20ensure=20that=20the=20code=20works=20as=20intende?=
 =?UTF-8?q?d=20and=20avoids=20errors=20when=20extending=20lists.=20The=20c?=
 =?UTF-8?q?hange=20in=20toolkits/base.py=20ensures=20that=20"Tool"=20is=20?=
 =?UTF-8?q?only=20added=20to=20classes=20that=20are=20toolkits.=20The=20ch?=
 =?UTF-8?q?anges=20in=20agents.py=20update=20the=20node=20descriptions=20t?=
 =?UTF-8?q?o=20reflect=20that=20the=20CSVAgentNode=20constructs=20a=20CSV?=
 =?UTF-8?q?=20agent=20and=20the=20InitializeAgentNode=20constructs=20a=20z?=
 =?UTF-8?q?ero=20shot=20agent.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/base.py                    | 6 ++++++
 src/backend/langflow/graph/nodes.py                   | 5 +++++
 src/backend/langflow/interface/toolkits/base.py       | 2 +-
 src/backend/langflow/template/frontend_node/agents.py | 4 ++--
 4 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/graph/base.py b/src/backend/langflow/graph/base.py
index 5b64885fb..08b255441 100644
--- a/src/backend/langflow/graph/base.py
+++ b/src/backend/langflow/graph/base.py
@@ -175,6 +175,12 @@ class Node:
                         # turn result which is a function into a coroutine
                         # so that it can be awaited
                         self.params["coroutine"] = sync_to_async(result)
+                if isinstance(result, list):
+                    # If the result is a list, then we need to extend the list
+                    # with the result but first check if the key exists
+                    # if it doesn't, then we need to create a new list
+                    if isinstance(self.params[key], list):
+                        self.params[key].extend(result)
 
                 self.params[key] = result
             elif isinstance(value, list) and all(
diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 7d9b05366..ea94e10b8 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -62,6 +62,11 @@ class PromptNode(Node):
                     if tools is not None
                     else []
                 )
+                # flatten the list of tools if it is a list of lists
+                # first check if it is a list
+                if isinstance(tools, list) and isinstance(tools[0], list):
+                    tools = [tool for sublist in tools for tool in sublist]
+
                 self.params["tools"] = tools
                 prompt_params = [
                     key
diff --git a/src/backend/langflow/interface/toolkits/base.py b/src/backend/langflow/interface/toolkits/base.py
index 9f01b2bb2..be2345c02 100644
--- a/src/backend/langflow/interface/toolkits/base.py
+++ b/src/backend/langflow/interface/toolkits/base.py
@@ -44,7 +44,7 @@ class ToolkitCreator(LangChainTypeCreator):
         try:
             template = build_template_from_class(name, self.type_to_loader_dict)
             # add Tool to base_classes
-            if template:
+            if "toolkit" in name.lower() and template:
                 template["base_classes"].append("Tool")
             return template
         except ValueError as exc:
diff --git a/src/backend/langflow/template/frontend_node/agents.py b/src/backend/langflow/template/frontend_node/agents.py
index e4fe40187..451dd7eca 100644
--- a/src/backend/langflow/template/frontend_node/agents.py
+++ b/src/backend/langflow/template/frontend_node/agents.py
@@ -146,7 +146,7 @@ class CSVAgentNode(FrontendNode):
             ),
         ],
     )
-    description: str = """Construct a json agent from a CSV and tools."""
+    description: str = """Construct a CSV agent from a CSV and tools."""
     base_classes: list[str] = ["AgentExecutor"]
 
     def to_dict(self):
@@ -194,7 +194,7 @@ class InitializeAgentNode(FrontendNode):
             ),
         ],
     )
-    description: str = """Construct a json agent from an LLM and tools."""
+    description: str = """Construct a zero shot agent from an LLM and tools."""
     base_classes: list[str] = ["AgentExecutor", "function"]
 
     def to_dict(self):

From 1a8d5561e9dbbecd636495ca3a75aecdaa2f3623 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 30 May 2023 01:15:48 -0300
Subject: [PATCH 03/59] fix

---
 src/backend/langflow/graph/nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index ea94e10b8..5f136b3c7 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -64,7 +64,7 @@ class PromptNode(Node):
                 )
                 # flatten the list of tools if it is a list of lists
                 # first check if it is a list
-                if isinstance(tools, list) and isinstance(tools[0], list):
+                if tools and isinstance(tools, list) and isinstance(tools[0], list):
                     tools = [tool for sublist in tools for tool in sublist]
 
                 self.params["tools"] = tools

From 4b6a8595df8e519a24fe1c064baf85c04ebe345b Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 30 May 2023 21:34:23 -0300
Subject: [PATCH 04/59] =?UTF-8?q?=F0=9F=90=9B=20fix(nodes.py):=20change=20?=
 =?UTF-8?q?type=20hint=20of=20tools=20list=20to=20include=20ToolkitNode=20?=
 =?UTF-8?q?The=20type=20hint=20of=20the=20tools=20list=20in=20the=20AgentN?=
 =?UTF-8?q?ode=20class=20has=20been=20updated=20to=20include=20the=20Toolk?=
 =?UTF-8?q?itNode=20class.=20This=20is=20because=20the=20tools=20list=20ca?=
 =?UTF-8?q?n=20now=20contain=20instances=20of=20the=20ToolkitNode=20class?=
 =?UTF-8?q?=20in=20addition=20to=20the=20ToolNode=20class.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/nodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 5f136b3c7..9db6260e9 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -8,7 +8,7 @@ class AgentNode(Node):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="agents")
 
-        self.tools: List[ToolNode] = []
+        self.tools: List[Union[ToolNode, ToolkitNode]] = []
         self.chains: List[ChainNode] = []
 
     def _set_tools_and_chains(self) -> None:

From 041748b2fb29a1641e889973baddf56941611fe9 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Tue, 30 May 2023 21:47:42 -0300
Subject: [PATCH 05/59] =?UTF-8?q?=F0=9F=94=A8=20refactor(nodes.py):=20extr?=
 =?UTF-8?q?act=20flatten=5Flist=20function=20to=20utils=20module=20and=20u?=
 =?UTF-8?q?se=20it=20in=20PromptNode.build=20method=20=F0=9F=90=9B=20fix(n?=
 =?UTF-8?q?odes.py):=20change=20tools=20parameter=20type=20hint=20in=20Pro?=
 =?UTF-8?q?mptNode.build=20method=20to=20accept=20a=20list=20of=20Union[To?=
 =?UTF-8?q?olNode,=20ToolkitNode]=20The=20flatten=5Flist=20function=20was?=
 =?UTF-8?q?=20extracted=20from=20the=20PromptNode.build=20method=20and=20m?=
 =?UTF-8?q?oved=20to=20the=20utils=20module=20to=20improve=20code=20reusab?=
 =?UTF-8?q?ility.=20The=20PromptNode.build=20method=20now=20uses=20the=20f?=
 =?UTF-8?q?latten=5Flist=20function=20to=20flatten=20the=20list=20of=20too?=
 =?UTF-8?q?ls=20if=20it=20is=20a=20list=20of=20lists.=20The=20tools=20para?=
 =?UTF-8?q?meter=20type=20hint=20was=20changed=20to=20accept=20a=20list=20?=
 =?UTF-8?q?of=20Union[ToolNode,=20ToolkitNode]=20to=20improve=20type=20saf?=
 =?UTF-8?q?ety.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/nodes.py | 62 ++++++++++++++---------------
 src/backend/langflow/graph/utils.py | 12 ++++++
 2 files changed, 41 insertions(+), 33 deletions(-)

diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/nodes.py
index 9db6260e9..21fe0f673 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/nodes.py
@@ -1,7 +1,12 @@
 from typing import Any, Dict, List, Optional, Union
 
 from langflow.graph.base import Node
-from langflow.graph.utils import extract_input_variables_from_prompt
+from langflow.graph.utils import extract_input_variables_from_prompt, flatten_list
+
+
+class ToolkitNode(Node):
+    def __init__(self, data: Dict):
+        super().__init__(data, base_type="toolkits")
 
 
 class AgentNode(Node):
@@ -47,7 +52,7 @@ class PromptNode(Node):
     def build(
         self,
         force: bool = False,
-        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
+        tools: Optional[List[Union[ToolNode, ToolkitNode]]] = None,
     ) -> Any:
         if not self._built or force:
             if (
@@ -65,8 +70,7 @@ class PromptNode(Node):
                 # flatten the list of tools if it is a list of lists
                 # first check if it is a list
                 if tools and isinstance(tools, list) and isinstance(tools[0], list):
-                    tools = [tool for sublist in tools for tool in sublist]
-
+                    tools = flatten_list(tools)
                 self.params["tools"] = tools
                 prompt_params = [
                     key
@@ -85,30 +89,6 @@ class PromptNode(Node):
         return self._built_object
 
 
-class ChainNode(Node):
-    def __init__(self, data: Dict):
-        super().__init__(data, base_type="chains")
-
-    def build(
-        self,
-        force: bool = False,
-        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
-    ) -> Any:
-        if not self._built or force:
-            # Check if the chain requires a PromptNode
-            for key, value in self.params.items():
-                if isinstance(value, PromptNode):
-                    # Build the PromptNode, passing the tools if available
-                    self.params[key] = value.build(tools=tools, force=force)
-
-            self._build()
-
-        #! Cannot deepcopy SQLDatabaseChain
-        if self.node_type in ["SQLDatabaseChain"]:
-            return self._built_object
-        return self._built_object
-
-
 class LLMNode(Node):
     built_node_type = None
     class_built_object = None
@@ -130,11 +110,6 @@ class LLMNode(Node):
         return self._built_object
 
 
-class ToolkitNode(Node):
-    def __init__(self, data: Dict):
-        super().__init__(data, base_type="toolkits")
-
-
 class FileToolNode(ToolNode):
     def __init__(self, data: Dict):
         super().__init__(data)
@@ -193,3 +168,24 @@ class TextSplitterNode(Node):
         if self._built_object:
             return f"""{self.node_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
         return f"{self.node_type}()"
+
+
+class ChainNode(Node):
+    def __init__(self, data: Dict):
+        super().__init__(data, base_type="chains")
+
+    def build(
+        self,
+        force: bool = False,
+        tools: Optional[List[Union[ToolNode, ToolkitNode]]] = None,
+    ) -> Any:
+        if not self._built or force:
+            # Check if the chain requires a PromptNode
+            for key, value in self.params.items():
+                if isinstance(value, PromptNode):
+                    # Build the PromptNode, passing the tools if available
+                    self.params[key] = value.build(tools=tools, force=force)
+
+            self._build()
+
+        return self._built_object
diff --git a/src/backend/langflow/graph/utils.py b/src/backend/langflow/graph/utils.py
index 6d56e933e..e22b27cf5 100644
--- a/src/backend/langflow/graph/utils.py
+++ b/src/backend/langflow/graph/utils.py
@@ -1,4 +1,5 @@
 import re
+from typing import Any, Union
 
 
 def validate_prompt(prompt: str):
@@ -17,3 +18,14 @@ def fix_prompt(prompt: str):
 def extract_input_variables_from_prompt(prompt: str) -> list[str]:
     """Extract input variables from prompt."""
     return re.findall(r"{(.*?)}", prompt)
+
+
+def flatten_list(list_of_lists: list[Union[list, Any]]) -> list:
+    """Flatten list of lists."""
+    new_list = []
+    for item in list_of_lists:
+        if isinstance(item, list):
+            new_list.extend(item)
+        else:
+            new_list.append(item)
+    return new_list

From 4160274b298a8ade902eccecc36268a80bb849dd Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Tue, 30 May 2023 22:15:01 -0300
Subject: [PATCH 06/59] Adding search nodes to ExtraSideBarComponent

---
 .../ExtraSidebarComponent/index.tsx           |   9 +-
 .../extraSidebarComponent/index.tsx           | 163 +++++++++++-------
 2 files changed, 107 insertions(+), 65 deletions(-)

diff --git a/src/frontend/src/components/ExtraSidebarComponent/index.tsx b/src/frontend/src/components/ExtraSidebarComponent/index.tsx
index f5acd5a23..b06a58d9c 100644
--- a/src/frontend/src/components/ExtraSidebarComponent/index.tsx
+++ b/src/frontend/src/components/ExtraSidebarComponent/index.tsx
@@ -1,6 +1,6 @@
 import { Disclosure } from "@headlessui/react";
 import { ChevronLeftIcon } from "@heroicons/react/24/outline";
-import { useContext } from "react";
+import { useContext, useState } from "react";
 import { Link } from "react-router-dom";
 import { classNames } from "../../utils";
 import { locationContext } from "../../contexts/locationContext";
@@ -13,6 +13,7 @@ export default function ExtraSidebar() {
     extraNavigation,
     extraComponent,
   } = useContext(locationContext);
+
   return (
     <>
       <aside
@@ -21,10 +22,8 @@ export default function ExtraSidebar() {
         } flex-shrink-0 flex overflow-hidden flex-col border-r dark:border-r-gray-700 transition-all duration-500`}
       >
         <div className="w-52 dark:bg-gray-800 border dark:border-gray-700  overflow-y-auto scrollbar-hide h-full flex flex-col items-start">
-          <div className="flex pt-1 px-4 justify-between align-middle w-full">
-            <span className="text-gray-900 dark:text-white py-[2px] font-medium ">
-              {extraNavigation.title}
-            </span>
+          <div className="flex px-4 justify-between align-middle w-full">
+            <span className="text-gray-900 dark:text-white py-[2px] font-medium "></span>
           </div>
           <div className="flex flex-grow flex-col w-full">
             {extraNavigation.options ? (
diff --git a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
index b88e13586..e8dbfbb6b 100644
--- a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
@@ -1,13 +1,20 @@
 import { Bars2Icon } from "@heroicons/react/24/outline";
 import DisclosureComponent from "../DisclosureComponent";
-import { nodeColors, nodeIcons, nodeNames } from "../../../../utils";
-import { useContext, useEffect, useState } from "react";
+import {
+  classNames,
+  nodeColors,
+  nodeIcons,
+  nodeNames,
+} from "../../../../utils";
+import { useContext, useEffect, useState, useRef } from "react";
 import { typesContext } from "../../../../contexts/typesContext";
 import { APIClassType, APIObjectType } from "../../../../types/api";
 import TooltipReact from "../../../../components/ReactTooltipComponent";
+import { MagnifyingGlassIcon } from "@heroicons/react/24/outline";
 
 export default function ExtraSidebar() {
   const { data } = useContext(typesContext);
+  const [dataFilter, setFilterData] = useState(data);
 
   function onDragStart(
     event: React.DragEvent<any>,
@@ -24,66 +31,102 @@ export default function ExtraSidebar() {
     event.dataTransfer.setData("json", JSON.stringify(data));
   }
 
+  function handleSearchInput(e: string) {
+    setFilterData((_) => {
+      let ret = {};
+      Object.keys(data).forEach((d: keyof APIObjectType, i) => {
+        ret[d] = {};
+        let keys = Object.keys(data[d]).filter((nd) =>
+          nd.toLowerCase().includes(e.toLowerCase())
+        );
+        keys.forEach((element) => {
+          ret[d][element] = data[d][element];
+        });
+      });
+      return ret;
+    });
+  }
+
   return (
-    <div className="mt-1 w-full">
-      {Object.keys(data)
-        .sort()
-        .map((d: keyof APIObjectType, i) => (
-          <DisclosureComponent
-            key={i}
-            button={{
-              title: nodeNames[d] ?? nodeNames.unknown,
-              Icon: nodeIcons[d] ?? nodeIcons.unknown,
-            }}
-          >
-            <div className="p-2 flex flex-col gap-2">
-              {Object.keys(data[d])
-                .sort()
-                .map((t: string, k) => (
-                  <TooltipReact
-                    selector={t}
-                    htmlContent={t}
-                    position="right"
-                    delayShow={1500}
-                    key={k}
-                  >
-                    <div key={k} data-tooltip-id={t}>
-                      <div
-                        draggable
-                        className={" cursor-grab border-l-8 rounded-l-md"}
-                        style={{
-                          borderLeftColor: nodeColors[d] ?? nodeColors.unknown,
-                        }}
-                        onDragStart={(event) =>
-                          onDragStart(event, {
-                            type: t,
-                            node: data[d][t],
-                          })
-                        }
-                        onDragEnd={() => {
-                          document.body.removeChild(
-                            document.getElementsByClassName(
-                              "cursor-grabbing"
-                            )[0]
-                          );
-                        }}
+    <>
+      <div className="relative mt-2 flex items-center mb-2 mx-2">
+        <input
+          type="text"
+          name="search"
+          id="search"
+          placeholder="Search nodes"
+          className="dark:text-white focus:outline-none block w-full rounded-md py-1.5 ps-3 pr-9 text-gray-900 shadow-sm ring-1 ring-inset ring-gray-300 placeholder:text-gray-400 sm:text-sm sm:leading-6 dark:ring-0 dark:bg-[#2d3747] dark:focus:outline-none"
+          onChange={(e) => {
+            handleSearchInput(e.target.value);
+          }}
+        />
+        <div className="absolute inset-y-0 right-0 flex py-1.5 pr-3 items-center">
+          <MagnifyingGlassIcon className="h-5 w-5 dark:text-white"></MagnifyingGlassIcon>
+        </div>
+      </div>
+
+      <div className="mt-1 w-full">
+        {Object.keys(dataFilter)
+          .sort()
+          .map((d: keyof APIObjectType, i) =>
+            Object.keys(dataFilter[d]).length > 0 ? (
+              <DisclosureComponent
+                key={i}
+                button={{
+                  title: nodeNames[d] ?? nodeNames.unknown,
+                  Icon: nodeIcons[d] ?? nodeIcons.unknown,
+                }}
+              >
+                <div className="p-2 flex flex-col gap-2">
+                  {Object.keys(dataFilter[d])
+                    .sort()
+                    .map((t: string, k) => (
+                      <TooltipReact
+                        selector={t}
+                        htmlContent={t}
+                        position="right"
+                        delayShow={1500}
+                        key={k}
                       >
-                        <div className="flex w-full justify-between text-sm px-3 py-1 bg-white dark:bg-gray-800 items-center border-dashed border-gray-400 dark:border-gray-600 border-l-0 rounded-md rounded-l-none border">
-                          <span className="text-black dark:text-white w-36 pr-1 truncate text-xs">
-                            {t}
-                          </span>
-                          <Bars2Icon className="w-4 h-6  text-gray-400 dark:text-gray-600" />
+                        <div key={k} data-tooltip-id={t}>
+                          <div
+                            draggable
+                            className={" cursor-grab border-l-8 rounded-l-md"}
+                            style={{
+                              borderLeftColor:
+                                nodeColors[d] ?? nodeColors.unknown,
+                            }}
+                            onDragStart={(event) =>
+                              onDragStart(event, {
+                                type: t,
+                                node: data[d][t],
+                              })
+                            }
+                            onDragEnd={() => {
+                              document.body.removeChild(
+                                document.getElementsByClassName(
+                                  "cursor-grabbing"
+                                )[0]
+                              );
+                            }}
+                          >
+                            <div className="flex w-full justify-between text-sm px-3 py-1 bg-white dark:bg-gray-800 items-center border-dashed border-gray-400 dark:border-gray-600 border-l-0 rounded-md rounded-l-none border">
+                              <span className="text-black dark:text-white w-36 pr-1 truncate text-xs">
+                                {t}
+                              </span>
+                              <Bars2Icon className="w-4 h-6  text-gray-400 dark:text-gray-600" />
+                            </div>
+                          </div>
                         </div>
-                      </div>
-                    </div>
-                  </TooltipReact>
-                ))}
-              {Object.keys(data[d]).length === 0 && (
-                <div className="text-gray-400 text-center">Coming soon</div>
-              )}
-            </div>
-          </DisclosureComponent>
-        ))}
-    </div>
+                      </TooltipReact>
+                    ))}
+                </div>
+              </DisclosureComponent>
+            ) : (
+              <div key={i}></div>
+            )
+          )}
+      </div>
+    </>
   );
 }

From a4312158fa260ce730be64b9871dace0e4c71167 Mon Sep 17 00:00:00 2001
From: Gabriel Almeida <gabriel@logspace.ai>
Date: Wed, 31 May 2023 01:04:40 -0300
Subject: [PATCH 07/59] =?UTF-8?q?=F0=9F=94=A7=20chore(pre-commit):=20add?=
 =?UTF-8?q?=20git=20add=20command=20to=20pre-commit=20hook=20The=20pre-com?=
 =?UTF-8?q?mit=20hook=20now=20adds=20the=20files=20that=20were=20formatted?=
 =?UTF-8?q?=20by=20the=20make=20format=20command=20to=20the=20staging=20ar?=
 =?UTF-8?q?ea.=20This=20ensures=20that=20the=20changes=20made=20by=20the?=
 =?UTF-8?q?=20formatting=20are=20included=20in=20the=20commit.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .githooks/pre-commit | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/.githooks/pre-commit b/.githooks/pre-commit
index f5008c586..ef67eaa37 100755
--- a/.githooks/pre-commit
+++ b/.githooks/pre-commit
@@ -1,2 +1,6 @@
+#!/bin/sh
 
-make format
\ No newline at end of file
+added_files=$(git diff --name-only --cached --diff-filter=d)
+
+make format
+git add ${added_files}
\ No newline at end of file

From 25d77eaf87b9e778b235626841e036f046fcf2e8 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 31 May 2023 11:44:12 -0300
Subject: [PATCH 08/59] =?UTF-8?q?=F0=9F=90=9B=20fix(custom.py):=20fix=20Js?=
 =?UTF-8?q?onAgent.from=5Ftoolkit=5Fand=5Fllm=20method=20to=20handle=20bot?=
 =?UTF-8?q?h=20list=20and=20JsonToolkit=20input=20=F0=9F=94=A5=20chore(tes?=
 =?UTF-8?q?t=5Fgraph.py):=20remove=20unused=20openapi=5Fgraph=20parameter?=
 =?UTF-8?q?=20from=20test=5Fbuild=20method=20The=20JsonAgent.from=5Ftoolki?=
 =?UTF-8?q?t=5Fand=5Fllm=20method=20was=20failing=20when=20a=20list=20was?=
 =?UTF-8?q?=20passed=20as=20input=20instead=20of=20a=20JsonToolkit=20objec?=
 =?UTF-8?q?t.=20The=20fix=20now=20handles=20both=20cases.=20The=20openapi?=
 =?UTF-8?q?=5Fgraph=20parameter=20was=20removed=20from=20the=20test=5Fbuil?=
 =?UTF-8?q?d=20method=20as=20it=20was=20unused.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/interface/agents/custom.py | 2 +-
 tests/test_graph.py                             | 3 +--
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/interface/agents/custom.py b/src/backend/langflow/interface/agents/custom.py
index 4654ef7cb..3aaa132d4 100644
--- a/src/backend/langflow/interface/agents/custom.py
+++ b/src/backend/langflow/interface/agents/custom.py
@@ -69,7 +69,7 @@ class JsonAgent(CustomAgentExecutor):
 
     @classmethod
     def from_toolkit_and_llm(cls, toolkit: JsonToolkit, llm: BaseLanguageModel):
-        tools = toolkit.get_tools()
+        tools = toolkit if isinstance(toolkit, list) else toolkit.get_tools()
         tool_names = {tool.name for tool in tools}
         prompt = ZeroShotAgent.create_prompt(
             tools,
diff --git a/tests/test_graph.py b/tests/test_graph.py
index a0f5945fc..b92457b81 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -237,11 +237,10 @@ def test_build_params(basic_graph):
     assert "memory" in root.params
 
 
-def test_build(basic_graph, complex_graph, openapi_graph):
+def test_build(basic_graph, complex_graph):
     """Test Node's build method"""
     assert_agent_was_built(basic_graph)
     assert_agent_was_built(complex_graph)
-    assert_agent_was_built(openapi_graph)
 
 
 def assert_agent_was_built(graph):

From d02d11213e0fd76f9f38f488e8b571b985de0302 Mon Sep 17 00:00:00 2001
From: Jeremy Naccache <jeremy.naccache@epitech.eu>
Date: Wed, 31 May 2023 17:00:18 +0200
Subject: [PATCH 09/59] fix(chatModal/index.ts) updated validation condition
 for required properties of nodes, to allow False values to be accepted

---
 src/frontend/src/modals/chatModal/index.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/modals/chatModal/index.tsx b/src/frontend/src/modals/chatModal/index.tsx
index c6a86daaf..46d62266b 100644
--- a/src/frontend/src/modals/chatModal/index.tsx
+++ b/src/frontend/src/modals/chatModal/index.tsx
@@ -290,7 +290,9 @@ export default function ChatModal({
         errors.concat(
           template[t].required &&
             template[t].show &&
-            (!template[t].value || template[t].value === "") &&
+            (template[t].value === undefined ||
+              template[t].value === null ||
+              template[t].value === "") &&
             !reactFlowInstance
               .getEdges()
               .some(

From dedbe3c906807f9f1ede4d4b941960b0b34e05dc Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 31 May 2023 15:40:20 -0300
Subject: [PATCH 10/59] =?UTF-8?q?=F0=9F=8E=A8=20refactor(complex=5Fexample?=
 =?UTF-8?q?.json):=20rename=20"PythonFunction"=20to=20"PythonFunctionTool"?=
 =?UTF-8?q?=20and=20add=20"description"=20and=20"name"=20fields=20The=20"P?=
 =?UTF-8?q?ythonFunction"=20type=20has=20been=20renamed=20to=20"PythonFunc?=
 =?UTF-8?q?tionTool"=20to=20better=20reflect=20its=20purpose.=20Additional?=
 =?UTF-8?q?ly,=20two=20new=20fields=20have=20been=20added:=20"description"?=
 =?UTF-8?q?=20and=20"name".=20These=20fields=20are=20required=20and=20allo?=
 =?UTF-8?q?w=20for=20a=20more=20detailed=20description=20of=20the=20tool?=
 =?UTF-8?q?=20and=20its=20name.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 tests/data/complex_example.json | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/tests/data/complex_example.json b/tests/data/complex_example.json
index 8d46a97d6..89a3b9324 100644
--- a/tests/data/complex_example.json
+++ b/tests/data/complex_example.json
@@ -197,7 +197,7 @@
                     "y": 136.29836646158452
                 },
                 "data": {
-                    "type": "PythonFunction",
+                    "type": "PythonFunctionTool",
                     "node": {
                         "template": {
                             "code": {
@@ -210,6 +210,26 @@
                                 "type": "str",
                                 "list": false
                             },
+                            "description": {
+                                "required": true,
+                                "placeholder": "",
+                                "show": true,
+                                "multiline": true,
+                                "value": "My description",
+                                "name": "description",
+                                "type": "str",
+                                "list": false
+                            },
+                            "name": {
+                                "required": true,
+                                "placeholder": "",
+                                "show": true,
+                                "multiline": true,
+                                "value": "My Tool",
+                                "name": "name",
+                                "type": "str",
+                                "list": false
+                            },
                             "_type": "python_function"
                         },
                         "description": "Python function to be executed.",

From be0984908146cc39ddef30bd1f1e27b67fcbb934 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 31 May 2023 15:41:16 -0300
Subject: [PATCH 11/59] =?UTF-8?q?=F0=9F=9A=80=20feat(langflow):=20rename?=
 =?UTF-8?q?=20PythonFunction=20to=20PythonFunctionTool=20for=20better=20se?=
 =?UTF-8?q?mantics=20=F0=9F=9A=80=20feat(langflow):=20add=20PythonFunction?=
 =?UTF-8?q?ToolNode=20to=20the=20frontend=20node=20tools=20=F0=9F=9A=80=20?=
 =?UTF-8?q?feat(langflow):=20add=20PythonFunctionTool=20to=20the=20custom?=
 =?UTF-8?q?=20tools=20=F0=9F=9A=80=20feat(langflow):=20add=20get=5Ffunctio?=
 =?UTF-8?q?n=20to=20importing=20utils=20to=20get=20the=20function=20from?=
 =?UTF-8?q?=20code=20=F0=9F=9A=80=20feat(langflow):=20add=20func=20paramet?=
 =?UTF-8?q?er=20to=20PythonFunctionTool=20to=20store=20the=20function=20?=
 =?UTF-8?q?=F0=9F=9A=80=20feat(langflow):=20add=20name=20and=20description?=
 =?UTF-8?q?=20parameters=20to=20PythonFunctionTool=20=F0=9F=9A=80=20feat(l?=
 =?UTF-8?q?angflow):=20update=20instantiate=5Ftool=20to=20use=20PythonFunc?=
 =?UTF-8?q?tionTool=20instead=20of=20PythonFunction=20=F0=9F=9A=80=20feat(?=
 =?UTF-8?q?langflow):=20update=20constants=20to=20use=20PythonFunctionTool?=
 =?UTF-8?q?=20instead=20of=20PythonFunction=20=F0=9F=9A=80=20feat(langflow?=
 =?UTF-8?q?):=20update=20custom.py=20to=20use=20PythonFunctionTool=20inste?=
 =?UTF-8?q?ad=20of=20PythonFunction=20=F0=9F=9A=80=20feat(langflow):=20upd?=
 =?UTF-8?q?ate=20loading.py=20to=20use=20get=5Ffunction=20and=20PythonFunc?=
 =?UTF-8?q?tionTool=20=F0=9F=9A=80=20feat(langflow):=20update=20utils.py?=
 =?UTF-8?q?=20to=20use=20get=5Ffunction=20=F0=9F=9A=80=20feat(langflow):?=
 =?UTF-8?q?=20update=20test=5Fcustom=5Ftypes.py=20to=20use=20get=5Ffunctio?=
 =?UTF-8?q?n=20and=20PythonFunctionTool=20=F0=9F=9A=80=20feat(langflow):?=
 =?UTF-8?q?=20update=20test=5Fgraph.py=20to=20use=20PythonFunctionTool=20i?=
 =?UTF-8?q?nstead=20of=20PythonFunction=20The=20changes=20rename=20PythonF?=
 =?UTF-8?q?unction=20to=20PythonFunctionTool=20for=20better=20semantics.?=
 =?UTF-8?q?=20The=20frontend=20node=20tools,=20custom=20tools,=20and=20con?=
 =?UTF-8?q?stants=20are=20updated=20to=20use=20PythonFunctionTool=20instea?=
 =?UTF-8?q?d=20of=20PythonFunction.=20The=20get=5Ffunction=20function=20is?=
 =?UTF-8?q?=20added=20to=20importing=20utils=20to=20get=20the=20function?=
 =?UTF-8?q?=20from=20code.=20The=20PythonFunctionTool=20is=20updated=20to?=
 =?UTF-8?q?=20store=20the=20function=20in=20the=20func=20parameter=20and?=
 =?UTF-8?q?=20to=20have=20name=20and=20description=20parameters.=20The=20i?=
 =?UTF-8?q?nstantiate=5Ftool,=20loading.py,=20and=20utils.py=20are=20updat?=
 =?UTF-8?q?ed=20to=20use=20get=5Ffunction=20and=20PythonFunctionTool.=20Th?=
 =?UTF-8?q?e=20test=5Fcustom=5Ftypes.py=20and=20test=5Fgraph.py=20are=20up?=
 =?UTF-8?q?dated=20to=20use=20PythonFunctionTool=20instead=20of=20PythonFu?=
 =?UTF-8?q?nction.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/config.yaml              |  2 +-
 src/backend/langflow/custom/customs.py        |  2 +-
 .../langflow/interface/importing/utils.py     |  8 +++++
 src/backend/langflow/interface/loading.py     | 12 +++----
 src/backend/langflow/interface/tools/base.py  |  3 +-
 .../langflow/interface/tools/constants.py     |  4 +--
 .../langflow/interface/tools/custom.py        | 22 +++++++------
 .../langflow/template/frontend_node/tools.py  | 32 ++++++++++++++++---
 tests/test_custom_types.py                    | 15 ++++++---
 tests/test_graph.py                           |  4 ++-
 10 files changed, 73 insertions(+), 31 deletions(-)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index 3b8554360..fe05eb406 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -79,7 +79,7 @@ tools:
   - Calculator
   - Serper Search
   - Tool
-  - PythonFunction
+  - PythonFunctionTool
   - JsonSpec
   - News API
   - TMDB API
diff --git a/src/backend/langflow/custom/customs.py b/src/backend/langflow/custom/customs.py
index ee266b0ee..f7a82e4a3 100644
--- a/src/backend/langflow/custom/customs.py
+++ b/src/backend/langflow/custom/customs.py
@@ -4,7 +4,7 @@ from langflow.template import frontend_node
 CUSTOM_NODES = {
     "prompts": {"ZeroShotPrompt": frontend_node.prompts.ZeroShotPromptNode()},
     "tools": {
-        "PythonFunction": frontend_node.tools.PythonFunctionNode(),
+        "PythonFunctionTool": frontend_node.tools.PythonFunctionToolNode(),
         "Tool": frontend_node.tools.ToolNode(),
     },
     "agents": {
diff --git a/src/backend/langflow/interface/importing/utils.py b/src/backend/langflow/interface/importing/utils.py
index d08e52999..f65376d48 100644
--- a/src/backend/langflow/interface/importing/utils.py
+++ b/src/backend/langflow/interface/importing/utils.py
@@ -9,6 +9,7 @@ from langchain.base_language import BaseLanguageModel
 from langchain.chains.base import Chain
 from langchain.chat_models.base import BaseChatModel
 from langchain.tools import BaseTool
+from langflow.utils import validate
 
 
 def import_module(module_path: str) -> Any:
@@ -147,3 +148,10 @@ def import_utility(utility: str) -> Any:
     if utility == "SQLDatabase":
         return import_class(f"langchain.sql_database.{utility}")
     return import_class(f"langchain.utilities.{utility}")
+
+
+def get_function(code):
+    """Get the function"""
+    function_name = validate.extract_function_name(code)
+
+    return validate.create_function(code, function_name)
diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py
index 69c697823..05ca348a7 100644
--- a/src/backend/langflow/interface/loading.py
+++ b/src/backend/langflow/interface/loading.py
@@ -20,12 +20,12 @@ from langchain.llms.loading import load_llm_from_config
 from pydantic import ValidationError
 
 from langflow.interface.agents.custom import CUSTOM_AGENTS
-from langflow.interface.importing.utils import import_by_type
+from langflow.interface.importing.utils import get_function, import_by_type
 from langflow.interface.run import fix_memory_inputs
 from langflow.interface.toolkits.base import toolkits_creator
 from langflow.interface.types import get_type_list
 from langflow.interface.utils import load_file_into_dict
-from langflow.utils import util, validate
+from langflow.utils import util
 
 
 def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
@@ -99,11 +99,9 @@ def instantiate_tool(node_type, class_object, params):
     if node_type == "JsonSpec":
         params["dict_"] = load_file_into_dict(params.pop("path"))
         return class_object(**params)
-    elif node_type == "PythonFunction":
-        function_string = params["code"]
-        if isinstance(function_string, str):
-            return validate.eval_function(function_string)
-        raise ValueError("Function should be a string")
+    elif node_type == "PythonFunctionTool":
+        params["func"] = get_function(params.get("code"))
+        return class_object(**params)
     elif node_type.lower() == "tool":
         return class_object(**params)
     return class_object(**params)
diff --git a/src/backend/langflow/interface/tools/base.py b/src/backend/langflow/interface/tools/base.py
index a8e7045c0..d6b114e4c 100644
--- a/src/backend/langflow/interface/tools/base.py
+++ b/src/backend/langflow/interface/tools/base.py
@@ -71,7 +71,8 @@ class ToolCreator(LangChainTypeCreator):
 
             for tool, tool_fcn in ALL_TOOLS_NAMES.items():
                 tool_params = get_tool_params(tool_fcn)
-                tool_name = tool_params.get("name", tool)
+
+                tool_name = tool_params.get("name") or tool
 
                 if tool_name in settings.tools or settings.dev:
                     if tool_name == "JsonSpec":
diff --git a/src/backend/langflow/interface/tools/constants.py b/src/backend/langflow/interface/tools/constants.py
index f939d55ad..31c75ec08 100644
--- a/src/backend/langflow/interface/tools/constants.py
+++ b/src/backend/langflow/interface/tools/constants.py
@@ -9,10 +9,10 @@ from langchain.agents.load_tools import (
 from langchain.tools.json.tool import JsonSpec
 
 from langflow.interface.importing.utils import import_class
-from langflow.interface.tools.custom import PythonFunction
+from langflow.interface.tools.custom import PythonFunctionTool
 
 FILE_TOOLS = {"JsonSpec": JsonSpec}
-CUSTOM_TOOLS = {"Tool": Tool, "PythonFunction": PythonFunction}
+CUSTOM_TOOLS = {"Tool": Tool, "PythonFunctionTool": PythonFunctionTool}
 
 OTHER_TOOLS = {tool: import_class(f"langchain.tools.{tool}") for tool in tools.__all__}
 
diff --git a/src/backend/langflow/interface/tools/custom.py b/src/backend/langflow/interface/tools/custom.py
index 4c641f388..b2d43565d 100644
--- a/src/backend/langflow/interface/tools/custom.py
+++ b/src/backend/langflow/interface/tools/custom.py
@@ -1,13 +1,14 @@
-from typing import Callable, Optional
+from typing import Optional
+from langflow.interface.importing.utils import get_function
 
 from pydantic import BaseModel, validator
 
 from langflow.utils import validate
+from langchain.agents.tools import Tool
 
 
 class Function(BaseModel):
     code: str
-    function: Optional[Callable] = None
     imports: Optional[str] = None
 
     # Eval code and store the function
@@ -24,14 +25,17 @@ class Function(BaseModel):
 
         return v
 
-    def get_function(self):
-        """Get the function"""
-        function_name = validate.extract_function_name(self.code)
 
-        return validate.create_function(self.code, function_name)
-
-
-class PythonFunction(Function):
+class PythonFunctionTool(Function, Tool):
     """Python function"""
 
+    name: str = "Custom Tool"
+    description: str
     code: str
+
+    def ___init__(self, name: str, description: str, code: str):
+        self.name = name
+        self.description = description
+        self.code = code
+        self.func = get_function(self.code)
+        super().__init__(name=name, description=description, func=self.func)
diff --git a/src/backend/langflow/template/frontend_node/tools.py b/src/backend/langflow/template/frontend_node/tools.py
index 2819be4d9..4e97fec8c 100644
--- a/src/backend/langflow/template/frontend_node/tools.py
+++ b/src/backend/langflow/template/frontend_node/tools.py
@@ -59,11 +59,33 @@ class ToolNode(FrontendNode):
         return super().to_dict()
 
 
-class PythonFunctionNode(FrontendNode):
-    name: str = "PythonFunction"
+class PythonFunctionToolNode(FrontendNode):
+    name: str = "PythonFunctionTool"
     template: Template = Template(
-        type_name="python_function",
+        type_name="PythonFunctionTool",
         fields=[
+            TemplateField(
+                field_type="str",
+                required=True,
+                placeholder="",
+                is_list=False,
+                show=True,
+                multiline=False,
+                value="",
+                name="name",
+                advanced=False,
+            ),
+            TemplateField(
+                field_type="str",
+                required=True,
+                placeholder="",
+                is_list=False,
+                show=True,
+                multiline=False,
+                value="",
+                name="description",
+                advanced=False,
+            ),
             TemplateField(
                 field_type="code",
                 required=True,
@@ -73,11 +95,11 @@ class PythonFunctionNode(FrontendNode):
                 value=DEFAULT_PYTHON_FUNCTION,
                 name="code",
                 advanced=False,
-            )
+            ),
         ],
     )
     description: str = "Python function to be executed."
-    base_classes: list[str] = ["function"]
+    base_classes: list[str] = ["Tool"]
 
     def to_dict(self):
         return super().to_dict()
diff --git a/tests/test_custom_types.py b/tests/test_custom_types.py
index 399450e2e..7503426ab 100644
--- a/tests/test_custom_types.py
+++ b/tests/test_custom_types.py
@@ -1,16 +1,23 @@
 # Test this:
+from langflow.interface.importing.utils import get_function
 import pytest
-from langflow.interface.tools.custom import PythonFunction
+from langflow.interface.tools.custom import PythonFunctionTool
 from langflow.utils import constants
 
 
 def test_python_function():
     """Test Python function"""
-    func = PythonFunction(code=constants.DEFAULT_PYTHON_FUNCTION)
-    assert func.get_function()("text") == "text"
+    code = constants.DEFAULT_PYTHON_FUNCTION
+    func = get_function(code)
+    func = PythonFunctionTool(name="Test", description="Testing", code=code, func=func)
+    assert func("text") == "text"
     # the tool decorator should raise an error if
     # the function is not str -> str
 
     # This raises ValidationError
     with pytest.raises(SyntaxError):
-        func = PythonFunction(code=pytest.CODE_WITH_SYNTAX_ERROR)
+        code = pytest.CODE_WITH_SYNTAX_ERROR
+        func = get_function(code)
+        func = PythonFunctionTool(
+            name="Test", description="Testing", code=code, func=func
+        )
diff --git a/tests/test_graph.py b/tests/test_graph.py
index a0f5945fc..7826870e8 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -156,7 +156,9 @@ def test_get_node_neighbors_complex(complex_graph):
     tool_neighbors = complex_graph.get_nodes_with_target(tool)
     assert tool_neighbors is not None
     # Check if there is a PythonFunction in the tool's neighbors
-    assert any("PythonFunction" in neighbor.data["type"] for neighbor in tool_neighbors)
+    assert any(
+        "PythonFunctionTool" in neighbor.data["type"] for neighbor in tool_neighbors
+    )
 
 
 def test_get_node(basic_graph):

From 893dc22fc41aa2dc4773fb05af38c520c0ea7abd Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 21:21:51 -0300
Subject: [PATCH 12/59] Change the icon to move just in the last message

---
 src/frontend/src/modals/chatModal/chatMessage/index.tsx | 5 ++++-
 src/frontend/src/modals/chatModal/index.tsx             | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/modals/chatModal/chatMessage/index.tsx b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
index 2db84adbc..85ebb61af 100644
--- a/src/frontend/src/modals/chatModal/chatMessage/index.tsx
+++ b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
@@ -16,10 +16,13 @@ import Convert from "ansi-to-html";
 export default function ChatMessage({
   chat,
   lockChat,
+  lastMessage
 }: {
   chat: ChatMessageType;
   lockChat: boolean;
+  lastMessage: boolean;
 }) {
+  
   const convert = new Convert({ newline: true });
   const [message, setMessage] = useState("");
   const imgRef = useRef(null);
@@ -48,7 +51,7 @@ export default function ChatMessage({
                 "absolute transition-opacity duration-500 scale-150 " +
                 (lockChat ? "opacity-100" : "opacity-0")
               }
-              src={AiIcon}
+              src={lastMessage ? AiIcon : AiIconStill}
             />
             <img
               className={
diff --git a/src/frontend/src/modals/chatModal/index.tsx b/src/frontend/src/modals/chatModal/index.tsx
index 46d62266b..1c17a21e0 100644
--- a/src/frontend/src/modals/chatModal/index.tsx
+++ b/src/frontend/src/modals/chatModal/index.tsx
@@ -416,7 +416,7 @@ export default function ChatModal({
                 >
                   {chatHistory.length > 0 ? (
                     chatHistory.map((c, i) => (
-                      <ChatMessage lockChat={lockChat} chat={c} key={i} />
+                      <ChatMessage lockChat={lockChat} chat={c} lastMessage={chatHistory.length-1 == i ? true : false} key={i} />
                     ))
                   ) : (
                     <div className="flex flex-col h-full text-center justify-center w-full items-center align-middle">

From 6332c4c0840f8e355bb862b09300bff3b924abcb Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 21:22:42 -0300
Subject: [PATCH 13/59] Changing the icon to move just in the last message on
 chat

---
 src/frontend/src/modals/chatModal/chatMessage/index.tsx | 3 +--
 src/frontend/src/modals/chatModal/index.tsx             | 7 ++++++-
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/frontend/src/modals/chatModal/chatMessage/index.tsx b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
index 85ebb61af..003c21694 100644
--- a/src/frontend/src/modals/chatModal/chatMessage/index.tsx
+++ b/src/frontend/src/modals/chatModal/chatMessage/index.tsx
@@ -16,13 +16,12 @@ import Convert from "ansi-to-html";
 export default function ChatMessage({
   chat,
   lockChat,
-  lastMessage
+  lastMessage,
 }: {
   chat: ChatMessageType;
   lockChat: boolean;
   lastMessage: boolean;
 }) {
-  
   const convert = new Convert({ newline: true });
   const [message, setMessage] = useState("");
   const imgRef = useRef(null);
diff --git a/src/frontend/src/modals/chatModal/index.tsx b/src/frontend/src/modals/chatModal/index.tsx
index 1c17a21e0..cf2b52aac 100644
--- a/src/frontend/src/modals/chatModal/index.tsx
+++ b/src/frontend/src/modals/chatModal/index.tsx
@@ -416,7 +416,12 @@ export default function ChatModal({
                 >
                   {chatHistory.length > 0 ? (
                     chatHistory.map((c, i) => (
-                      <ChatMessage lockChat={lockChat} chat={c} lastMessage={chatHistory.length-1 == i ? true : false} key={i} />
+                      <ChatMessage
+                        lockChat={lockChat}
+                        chat={c}
+                        lastMessage={chatHistory.length - 1 == i ? true : false}
+                        key={i}
+                      />
                     ))
                   ) : (
                     <div className="flex flex-col h-full text-center justify-center w-full items-center align-middle">

From 9e4f898f54bdc18f99a06313466906d9555306a0 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 21:38:06 -0300
Subject: [PATCH 14/59] Changing the mouse cursor on Canva's React Flow

---
 src/frontend/src/index.css | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 899fb1c87..84ede36f1 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -15,3 +15,11 @@ code {
 	font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
 		monospace;
 }
+
+/* 
+The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
+The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor.
+*/
+.react-flow__pane{
+	cursor: default;
+} 
\ No newline at end of file

From d9cae6b211a82a849ce03c42ad784b49b4e30641 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 22:00:03 -0300
Subject: [PATCH 15/59] fixing identation warnings

---
 src/frontend/src/index.css | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 84ede36f1..f4981616c 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -16,10 +16,8 @@ code {
 		monospace;
 }
 
-/* 
-The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
-The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor.
-*/
-.react-flow__pane{
+/*The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
+The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor.*/
+.react-flow__pane {
 	cursor: default;
-} 
\ No newline at end of file
+}
\ No newline at end of file

From 344b94f9b5d523e13edf6f4b61c4142af94ff24e Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 22:06:49 -0300
Subject: [PATCH 16/59] fixing warning identation

---
 src/frontend/src/index.css | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index f4981616c..32f378e63 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -16,8 +16,8 @@ code {
 		monospace;
 }
 
-/*The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
-The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor.*/
+/* The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
+The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor. */
 .react-flow__pane {
 	cursor: default;
-}
\ No newline at end of file
+  }

From f27d067609da99e7cd0f1db93b8f20601e7f83d8 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 22:16:16 -0300
Subject: [PATCH 17/59] fixing identation warnings spaces

---
 src/frontend/src/index.css | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 32f378e63..3fa44bc31 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -19,5 +19,5 @@ code {
 /* The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
 The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor. */
 .react-flow__pane {
-	cursor: default;
-  }
+    	cursor: default;
+}

From 2ee0244834c0ad56d5b15de169f318805886354e Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Wed, 31 May 2023 22:20:12 -0300
Subject: [PATCH 18/59] fixing identation warnings spaces - new

---
 src/frontend/src/index.css | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 3fa44bc31..67f44ca1c 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -19,5 +19,5 @@ code {
 /* The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
 The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor. */
 .react-flow__pane {
-    	cursor: default;
+  cursor: default;
 }

From 585e94285d89a63f63014015fd4f8f20a16a1fbe Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Thu, 1 Jun 2023 00:27:01 -0300
Subject: [PATCH 19/59] Adding title and icon of the parent menu on hover of
 the handle

---
 .../components/parameterComponent/index.tsx   | 37 ++++++++++++++++++-
 src/frontend/src/types/components/index.ts    |  1 +
 2 files changed, 37 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index 653248763..e26ae589f 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -15,6 +15,8 @@ import InputFileComponent from "../../../../components/inputFileComponent";
 import { TabsContext } from "../../../../contexts/tabsContext";
 import IntComponent from "../../../../components/intComponent";
 import PromptAreaComponent from "../../../../components/promptComponent";
+import { nodeNames, nodeIcons } from "../../../../utils";
+import React from "react";
 
 export default function ParameterComponent({
   left,
@@ -28,6 +30,8 @@ export default function ParameterComponent({
   required = false,
 }: ParameterComponentType) {
   const ref = useRef(null);
+  const refParent = useRef("");
+  const refParentIcon = useRef(null);
   const updateNodeInternals = useUpdateNodeInternals();
   const [position, setPosition] = useState(0);
   useEffect(() => {
@@ -48,6 +52,19 @@ export default function ParameterComponent({
   let disabled =
     reactFlowInstance?.getEdges().some((e) => e.targetHandle === id) ?? false;
   const { save } = useContext(TabsContext);
+  const [myData, setMyData] = useState(useContext(typesContext).data);
+
+  useEffect(() => {
+    Object.keys(myData).forEach((d) => {
+      let keys = Object.keys(myData[d]).filter(
+        (nd) => nd.toLowerCase() == data.type.toLowerCase()
+      );
+      if (keys.length > 0) {
+        refParent.current = d;
+        refParentIcon.current = nodeIcons[d];
+      }
+    });
+  }, []);
 
   return (
     <div
@@ -69,7 +86,25 @@ export default function ParameterComponent({
           type === "int") ? (
           <></>
         ) : (
-          <Tooltip title={tooltipTitle + (required ? " (required)" : "")}>
+          <Tooltip
+            title={
+              <>
+                <div className="flex center items-center">
+                  <div
+                    className="h-5 w-5"
+                    style={{
+                      color: color,
+                    }}
+                  >
+                    {React.createElement(refParentIcon.current)}
+                  </div>
+                  <span className="ms-2 items-center">
+                    {nodeNames[refParent?.current] ?? ""}
+                  </span>
+                </div>
+              </>
+            }
+          >
             <Handle
               type={left ? "target" : "source"}
               position={left ? Position.Left : Position.Right}
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index e7559023a..326834374 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -3,6 +3,7 @@ import {
   ReactElement,
   ReactFragment,
   ReactNode,
+  SVGProps,
 } from "react";
 import { NodeDataType } from "../flow/index";
 export type InputComponentType = {

From a9e1fb4eb729eb38ed03037677eb702865664a29 Mon Sep 17 00:00:00 2001
From: Deepankar Mahapatro <deepankar.mahapatro@jina.ai>
Date: Thu, 1 Jun 2023 09:58:25 +0530
Subject: [PATCH 20/59] build: push lcserve image as public

---
 Makefile       |  2 +-
 poetry.lock    | 10 ++++++----
 pyproject.toml |  2 +-
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/Makefile b/Makefile
index fdb09a13e..15337f65b 100644
--- a/Makefile
+++ b/Makefile
@@ -59,7 +59,7 @@ lcserve_push:
 	make build_frontend
 	@version=$$(poetry version --short); \
 	lc-serve push --app langflow.lcserve:app --app-dir . \
-		--image-name langflow --image-tag $${version} --verbose
+		--image-name langflow --image-tag $${version} --verbose --public
 
 lcserve_deploy:
 	@:$(if $(uses),,$(error `uses` is not set. Please run `make uses=... lcserve_deploy`))
diff --git a/poetry.lock b/poetry.lock
index a414aab16..d02b5952d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,4 +1,4 @@
-# This file is automatically @generated by Poetry 1.4.0 and should not be changed by hand.
+# This file is automatically @generated by Poetry 1.4.2 and should not be changed by hand.
 
 [[package]]
 name = "aiofiles"
@@ -1096,7 +1096,9 @@ files = [
     {file = "duckdb-0.8.0-cp311-cp311-macosx_10_9_universal2.whl", hash = "sha256:b2707096d6df4321044fcde2c9f04da632d11a8be60957fd09d49a42fae71a29"},
     {file = "duckdb-0.8.0-cp311-cp311-macosx_10_9_x86_64.whl", hash = "sha256:b27df1b70ae74d2c88efb5ffca8490954fdc678099509a9c4404ca30acc53426"},
     {file = "duckdb-0.8.0-cp311-cp311-macosx_11_0_arm64.whl", hash = "sha256:75a97c800271b52dd0f37696d074c50576dcb4b2750b6115932a98696a268070"},
+    {file = "duckdb-0.8.0-cp311-cp311-manylinux_2_17_i686.manylinux2014_i686.whl", hash = "sha256:804cac261a5e016506a6d67838a65d19b06a237f7949f1704f0e800eb708286a"},
     {file = "duckdb-0.8.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:c6b9abca7fa6713e1d031c18485343b4de99742c7e1b85c10718aa2f31a4e2c6"},
+    {file = "duckdb-0.8.0-cp311-cp311-musllinux_1_1_i686.whl", hash = "sha256:51aa6d606d49072abcfeb3be209eb559ac94c1b5e70f58ac3adbb94aca9cd69f"},
     {file = "duckdb-0.8.0-cp311-cp311-musllinux_1_1_x86_64.whl", hash = "sha256:7c8dc769aaf2be0a1c57995ca657e5b92c1c56fc8437edb720ca6cab571adf14"},
     {file = "duckdb-0.8.0-cp311-cp311-win32.whl", hash = "sha256:c4207d18b42387c4a035846d8878eb967070198be8ac26fd77797ce320d1a400"},
     {file = "duckdb-0.8.0-cp311-cp311-win_amd64.whl", hash = "sha256:0c392257547c20794c3072fcbca99a49ef0a49974005d755e93893e2b4875267"},
@@ -2374,13 +2376,13 @@ text-helpers = ["chardet (>=5.1.0,<6.0.0)"]
 
 [[package]]
 name = "langchain-serve"
-version = "0.0.38"
+version = "0.0.39"
 description = "Langchain Serve - serve your langchain apps on Jina AI Cloud."
 category = "main"
 optional = true
 python-versions = "*"
 files = [
-    {file = "langchain-serve-0.0.38.tar.gz", hash = "sha256:649b8e26eebe6b33960c081b388fb7118acbfdc00f97dd935a580ab88aca53d6"},
+    {file = "langchain-serve-0.0.39.tar.gz", hash = "sha256:9854cb65d4bc4003eec334e4b0c7ff7625984f595ea46e1aa60c443a1978f47d"},
 ]
 
 [package.dependencies]
@@ -6178,4 +6180,4 @@ deploy = ["langchain-serve"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.12"
-content-hash = "9ce165d2decf2190d7ce69be608872b3ed9abe705a276045623706d01665754b"
+content-hash = "ad6b2111e54888347e033f2df354220c8a00667375e19e02d129dabba50adafa"
diff --git a/pyproject.toml b/pyproject.toml
index 79d09b422..9c4928dd7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,7 +49,7 @@ psycopg2-binary = "^2.9.6"
 pyarrow = "^11.0.0"
 tiktoken = "^0.3.3"
 wikipedia = "^1.4.0"
-langchain-serve = { version = "^0.0.38", optional = true }
+langchain-serve = { version = "^0.0.39", optional = true }
 qdrant-client = "^1.2.0"
 websockets = "^11.0.3"
 weaviate-client = "^3.19.2"

From 2285304e095e77ca3bbc2e929c88b6bb0dcd14fa Mon Sep 17 00:00:00 2001
From: Deepankar Mahapatro <deepankar.mahapatro@jina.ai>
Date: Thu, 1 Jun 2023 10:56:39 +0530
Subject: [PATCH 21/59] build: update lcserve version to support public

---
 poetry.lock    | 6 +++---
 pyproject.toml | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index d02b5952d..5e53d5282 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2376,13 +2376,13 @@ text-helpers = ["chardet (>=5.1.0,<6.0.0)"]
 
 [[package]]
 name = "langchain-serve"
-version = "0.0.39"
+version = "0.0.40"
 description = "Langchain Serve - serve your langchain apps on Jina AI Cloud."
 category = "main"
 optional = true
 python-versions = "*"
 files = [
-    {file = "langchain-serve-0.0.39.tar.gz", hash = "sha256:9854cb65d4bc4003eec334e4b0c7ff7625984f595ea46e1aa60c443a1978f47d"},
+    {file = "langchain-serve-0.0.40.tar.gz", hash = "sha256:c60b173fcf0b682fbb70d34e8f485ce168e2229f55cb5c4ffbc26a5206af1c06"},
 ]
 
 [package.dependencies]
@@ -6180,4 +6180,4 @@ deploy = ["langchain-serve"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.12"
-content-hash = "ad6b2111e54888347e033f2df354220c8a00667375e19e02d129dabba50adafa"
+content-hash = "2bf357ad30f79c68751b34c991b4a73767ceb628657f4133228d4eb487d8a6fb"
diff --git a/pyproject.toml b/pyproject.toml
index 9c4928dd7..78104f327 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,7 +49,7 @@ psycopg2-binary = "^2.9.6"
 pyarrow = "^11.0.0"
 tiktoken = "^0.3.3"
 wikipedia = "^1.4.0"
-langchain-serve = { version = "^0.0.39", optional = true }
+langchain-serve = { version = ">0.0.39", optional = true }
 qdrant-client = "^1.2.0"
 websockets = "^11.0.3"
 weaviate-client = "^3.19.2"

From 5b5eea98955c6fb7ba0614b783fd1109a6eed7ce Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 1 Jun 2023 15:07:56 -0300
Subject: [PATCH 22/59] =?UTF-8?q?=F0=9F=9A=80=20feat(langflow):=20reorgani?=
 =?UTF-8?q?ze=20graph=20package=20to=20improve=20modularity=20and=20mainta?=
 =?UTF-8?q?inability=20The=20changes=20include:=20-=20Moved=20the=20`Edge`?=
 =?UTF-8?q?=20class=20to=20a=20new=20`edge`=20package=20-=20Moved=20the=20?=
 =?UTF-8?q?`Graph`=20class=20to=20a=20new=20`graph`=20package=20-=20Moved?=
 =?UTF-8?q?=20the=20`Node`=20class=20to=20a=20new=20`node`=20package=20-?=
 =?UTF-8?q?=20Moved=20the=20`VectorStoreNode`=20class=20to=20the=20`node/t?=
 =?UTF-8?q?ypes.py`=20module=20-=20Moved=20the=20`Edge`,=20`Graph`,=20and?=
 =?UTF-8?q?=20`Node`=20classes=20to=20their=20respective=20`base.py`=20mod?=
 =?UTF-8?q?ules=20-=20Added=20an=20`=5F=5Finit=5F=5F.py`=20file=20to=20eac?=
 =?UTF-8?q?h=20package=20to=20allow=20for=20importing=20of=20classes=20-?=
 =?UTF-8?q?=20Added=20a=20`constants.py`=20module=20to=20the=20`graph`=20p?=
 =?UTF-8?q?ackage=20to=20store=20constants=20used=20in=20the=20`Graph`=20c?=
 =?UTF-8?q?lass=20-=20Refactored=20the=20`Graph`=20class=20to=20use=20the?=
 =?UTF-8?q?=20new=20`Node`=20and=20`Edge`=20classes=20-=20Refactored=20the?=
 =?UTF-8?q?=20`Graph`=20class=20to=20use=20a=20dictionary=20to=20map=20nod?=
 =?UTF-8?q?e=20types=20to=20their=20respective=20classes=20-=20Refactored?=
 =?UTF-8?q?=20the=20`Graph`=20class=20to=20remove=20invalid=20nodes=20from?=
 =?UTF-8?q?=20the=20graph=20-=20Refactored=20the=20`Graph`=20class=20to=20?=
 =?UTF-8?q?handle=20the=20LLM=20node=20within=20the=20graph=20-=20Refactor?=
 =?UTF-8?q?ed=20the=20`Graph`=20class=20to=20build=20the=20nodes=20before?=
 =?UTF-8?q?=20building=20the=20edges=20-=20Refactored=20the=20`Graph`=20cl?=
 =?UTF-8?q?ass=20to=20use=20the=20`get=5Fnode`=20method=20to=20find=20node?=
 =?UTF-8?q?s=20by=20id=20-=20Refactored=20the=20`Graph`=20class=20to=20use?=
 =?UTF-8?q?=20the=20`get=5Fnode=5Fneighbors`=20method=20to=20find=20the=20?=
 =?UTF-8?q?neighbors=20of=20a=20node=20-=20Refactored=20the=20`Graph`=20cl?=
 =?UTF-8?q?ass=20to=20use=20the=20`get=5Fchildren=5Fby=5Fnode=5Ftype`=20me?=
 =?UTF-8?q?thod=20to=20find=20the=20children=20of=20a=20node=20based=20on?=
 =?UTF-8?q?=20the=20node=20type?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

These changes improve the modularity and maintainability of the `langflow` package by separating the classes into their respective packages and modules. The changes also make it easier to add new node types to the `Graph` class by using a dictionary to map node types to their respective classes.

🚀 feat(node): add Node class to represent a node in the graph
🚀 feat(constants.py): add DIRECT_TYPES constant to represent direct types in a node's template
The Node class represents a node in the graph and is responsible for parsing the data and building the module. The DIRECT_TYPES constant is a list of direct types in a node's template.

🚧 chore(types.py): add import statements for typing and Node classes
This commit adds import statements for the typing module and the Node class to the types.py file. This is necessary for the code to run properly as it uses these classes and modules.

🚧 chore(loading.py): remove unnecessary import statement
This commit removes an unnecessary import statement from the loading.py file. The import statement was causing a circular import error and was not needed for the code to run properly.

🚧 chore(run.py): update import statement for Graph class
This commit updates the import statement for the Graph class in the run.py file. The import statement was outdated and was causing an import error.

🚧 chore(conftest.py): update import statement for Graph class
This commit updates the import statement for the Graph class in the conftest.py file. The import statement was outdated and was causing an import error.

🚧 chore(test_graph.py): update import statements for Node and Edge classes
This commit updates the import statements for the Node and Edge classes in the test_graph.py file. The import statements were outdated and were causing import errors.
---
 src/backend/langflow/api/validate.py          |  2 +-
 src/backend/langflow/graph/__init__.py        | 37 ++++++++-
 src/backend/langflow/graph/edge/__init__.py   |  0
 src/backend/langflow/graph/edge/base.py       | 52 ++++++++++++
 src/backend/langflow/graph/graph/__init__.py  |  0
 .../graph/{graph.py => graph/base.py}         | 81 ++++++++-----------
 src/backend/langflow/graph/graph/constants.py | 49 +++++++++++
 src/backend/langflow/graph/node/__init__.py   |  0
 src/backend/langflow/graph/{ => node}/base.py | 73 +++--------------
 .../langflow/graph/{ => node}/constants.py    |  0
 .../graph/{nodes.py => node/types.py}         |  2 +-
 src/backend/langflow/interface/loading.py     |  2 +-
 src/backend/langflow/interface/run.py         |  2 +-
 tests/conftest.py                             |  2 +-
 tests/test_graph.py                           |  6 +-
 15 files changed, 189 insertions(+), 119 deletions(-)
 create mode 100644 src/backend/langflow/graph/edge/__init__.py
 create mode 100644 src/backend/langflow/graph/edge/base.py
 create mode 100644 src/backend/langflow/graph/graph/__init__.py
 rename src/backend/langflow/graph/{graph.py => graph/base.py} (64%)
 create mode 100644 src/backend/langflow/graph/graph/constants.py
 create mode 100644 src/backend/langflow/graph/node/__init__.py
 rename src/backend/langflow/graph/{ => node}/base.py (81%)
 rename src/backend/langflow/graph/{ => node}/constants.py (100%)
 rename src/backend/langflow/graph/{nodes.py => node/types.py} (99%)

diff --git a/src/backend/langflow/api/validate.py b/src/backend/langflow/api/validate.py
index 0e2a7752c..53a7ee350 100644
--- a/src/backend/langflow/api/validate.py
+++ b/src/backend/langflow/api/validate.py
@@ -9,7 +9,7 @@ from langflow.api.base import (
     PromptValidationResponse,
     validate_prompt,
 )
-from langflow.graph.nodes import VectorStoreNode
+from langflow.graph.node.types import VectorStoreNode
 from langflow.interface.run import build_graph
 from langflow.utils.logger import logger
 from langflow.utils.validate import validate_code
diff --git a/src/backend/langflow/graph/__init__.py b/src/backend/langflow/graph/__init__.py
index 097b7a695..44859da02 100644
--- a/src/backend/langflow/graph/__init__.py
+++ b/src/backend/langflow/graph/__init__.py
@@ -1,4 +1,35 @@
-from langflow.graph.base import Edge, Node
-from langflow.graph.graph import Graph
+from langflow.graph.edge.base import Edge
+from langflow.graph.graph.base import Graph
+from langflow.graph.node.base import Node
+from langflow.graph.node.types import (
+    AgentNode,
+    ChainNode,
+    DocumentLoaderNode,
+    EmbeddingNode,
+    LLMNode,
+    MemoryNode,
+    PromptNode,
+    TextSplitterNode,
+    ToolNode,
+    ToolkitNode,
+    VectorStoreNode,
+    WrapperNode,
+)
 
-__all__ = ["Graph", "Node", "Edge"]
+__all__ = [
+    "Graph",
+    "Node",
+    "Edge",
+    "AgentNode",
+    "ChainNode",
+    "DocumentLoaderNode",
+    "EmbeddingNode",
+    "LLMNode",
+    "MemoryNode",
+    "PromptNode",
+    "TextSplitterNode",
+    "ToolNode",
+    "ToolkitNode",
+    "VectorStoreNode",
+    "WrapperNode",
+]
diff --git a/src/backend/langflow/graph/edge/__init__.py b/src/backend/langflow/graph/edge/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/graph/edge/base.py b/src/backend/langflow/graph/edge/base.py
new file mode 100644
index 000000000..2bf5a1ba4
--- /dev/null
+++ b/src/backend/langflow/graph/edge/base.py
@@ -0,0 +1,52 @@
+from langflow.utils.logger import logger
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langflow.graph.node.base import Node
+
+
+class Edge:
+    def __init__(self, source: "Node", target: "Node"):
+        self.source: "Node" = source
+        self.target: "Node" = target
+        self.validate_edge()
+
+    def validate_edge(self) -> None:
+        # Validate that the outputs of the source node are valid inputs
+        # for the target node
+        self.source_types = self.source.output
+        self.target_reqs = self.target.required_inputs + self.target.optional_inputs
+        # Both lists contain strings and sometimes a string contains the value we are
+        # looking for e.g. comgin_out=["Chain"] and target_reqs=["LLMChain"]
+        # so we need to check if any of the strings in source_types is in target_reqs
+        self.valid = any(
+            output in target_req
+            for output in self.source_types
+            for target_req in self.target_reqs
+        )
+        # Get what type of input the target node is expecting
+
+        self.matched_type = next(
+            (
+                output
+                for output in self.source_types
+                for target_req in self.target_reqs
+                if output in target_req
+            ),
+            None,
+        )
+        no_matched_type = self.matched_type is None
+        if no_matched_type:
+            logger.debug(self.source_types)
+            logger.debug(self.target_reqs)
+        if no_matched_type:
+            raise ValueError(
+                f"Edge between {self.source.node_type} and {self.target.node_type} "
+                f"has no matched type"
+            )
+
+    def __repr__(self) -> str:
+        return (
+            f"Edge(source={self.source.id}, target={self.target.id}, valid={self.valid}"
+            f", matched_type={self.matched_type})"
+        )
diff --git a/src/backend/langflow/graph/graph/__init__.py b/src/backend/langflow/graph/graph/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/graph/graph.py b/src/backend/langflow/graph/graph/base.py
similarity index 64%
rename from src/backend/langflow/graph/graph.py
rename to src/backend/langflow/graph/graph/base.py
index b289d5c31..3ba67837f 100644
--- a/src/backend/langflow/graph/graph.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -1,38 +1,20 @@
 from typing import Dict, List, Type, Union
 
-from langflow.graph.base import Edge, Node
-from langflow.graph.nodes import (
-    AgentNode,
-    ChainNode,
-    DocumentLoaderNode,
-    EmbeddingNode,
+from langflow.graph.edge.base import Edge
+from langflow.graph.graph.constants import NODE_TYPE_MAP
+from langflow.graph.node.base import Node
+from langflow.graph.node.types import (
     FileToolNode,
     LLMNode,
-    MemoryNode,
-    PromptNode,
-    TextSplitterNode,
     ToolkitNode,
-    ToolNode,
-    VectorStoreNode,
-    WrapperNode,
 )
-from langflow.interface.agents.base import agent_creator
-from langflow.interface.chains.base import chain_creator
-from langflow.interface.document_loaders.base import documentloader_creator
-from langflow.interface.embeddings.base import embedding_creator
-from langflow.interface.llms.base import llm_creator
-from langflow.interface.memories.base import memory_creator
-from langflow.interface.prompts.base import prompt_creator
-from langflow.interface.text_splitters.base import textsplitter_creator
-from langflow.interface.toolkits.base import toolkits_creator
-from langflow.interface.tools.base import tool_creator
 from langflow.interface.tools.constants import FILE_TOOLS
-from langflow.interface.vector_store.base import vectorstore_creator
-from langflow.interface.wrappers.base import wrapper_creator
 from langflow.utils import payload
 
 
 class Graph:
+    """A class representing a graph of nodes and edges."""
+
     def __init__(
         self,
         nodes: List[Dict[str, Union[str, Dict[str, Union[str, List[str]]]]]],
@@ -43,6 +25,7 @@ class Graph:
         self._build_graph()
 
     def _build_graph(self) -> None:
+        """Builds the graph from the nodes and edges."""
         self.nodes = self._build_nodes()
         self.edges = self._build_edges()
         for edge in self.edges:
@@ -51,17 +34,25 @@ class Graph:
 
         # This is a hack to make sure that the LLM node is sent to
         # the toolkit node
+        self._build_node_params()
+        # remove invalid nodes
+        self._remove_invalid_nodes()
+
+    def _build_node_params(self) -> None:
+        """Identifies and handles the LLM node within the graph."""
         llm_node = None
         for node in self.nodes:
             node._build_params()
-
             if isinstance(node, LLMNode):
                 llm_node = node
 
-        for node in self.nodes:
-            if isinstance(node, ToolkitNode):
-                node.params["llm"] = llm_node
-        # remove invalid nodes
+        if llm_node:
+            for node in self.nodes:
+                if isinstance(node, ToolkitNode):
+                    node.params["llm"] = llm_node
+
+    def _remove_invalid_nodes(self) -> None:
+        """Removes invalid nodes from the graph."""
         self.nodes = [
             node
             for node in self.nodes
@@ -70,19 +61,23 @@ class Graph:
         ]
 
     def _validate_node(self, node: Node) -> bool:
+        """Validates a node."""
         # All nodes that do not have edges are invalid
         return len(node.edges) > 0
 
     def get_node(self, node_id: str) -> Union[None, Node]:
+        """Returns a node by id."""
         return next((node for node in self.nodes if node.id == node_id), None)
 
     def get_nodes_with_target(self, node: Node) -> List[Node]:
+        """Returns the nodes connected to a node."""
         connected_nodes: List[Node] = [
             edge.source for edge in self.edges if edge.target == node
         ]
         return connected_nodes
 
     def build(self) -> List[Node]:
+        """Builds the graph."""
         # Get root node
         root_node = payload.get_root_node(self)
         if root_node is None:
@@ -90,6 +85,7 @@ class Graph:
         return root_node.build()
 
     def get_node_neighbors(self, node: Node) -> Dict[Node, int]:
+        """Returns the neighbors of a node."""
         neighbors: Dict[Node, int] = {}
         for edge in self.edges:
             if edge.source == node:
@@ -105,6 +101,7 @@ class Graph:
         return neighbors
 
     def _build_edges(self) -> List[Edge]:
+        """Builds the edges of the graph."""
         # Edge takes two nodes as arguments, so we need to build the nodes first
         # and then build the edges
         # if we can't find a node, we raise an error
@@ -121,30 +118,15 @@ class Graph:
         return edges
 
     def _get_node_class(self, node_type: str, node_lc_type: str) -> Type[Node]:
-        node_type_map: Dict[str, Type[Node]] = {
-            **{t: PromptNode for t in prompt_creator.to_list()},
-            **{t: AgentNode for t in agent_creator.to_list()},
-            **{t: ChainNode for t in chain_creator.to_list()},
-            **{t: ToolNode for t in tool_creator.to_list()},
-            **{t: ToolkitNode for t in toolkits_creator.to_list()},
-            **{t: WrapperNode for t in wrapper_creator.to_list()},
-            **{t: LLMNode for t in llm_creator.to_list()},
-            **{t: MemoryNode for t in memory_creator.to_list()},
-            **{t: EmbeddingNode for t in embedding_creator.to_list()},
-            **{t: VectorStoreNode for t in vectorstore_creator.to_list()},
-            **{t: DocumentLoaderNode for t in documentloader_creator.to_list()},
-            **{t: TextSplitterNode for t in textsplitter_creator.to_list()},
-        }
-
+        """Returns the node class based on the node type."""
         if node_type in FILE_TOOLS:
             return FileToolNode
-        if node_type in node_type_map:
-            return node_type_map[node_type]
-        if node_lc_type in node_type_map:
-            return node_type_map[node_lc_type]
-        return Node
+        if node_type in NODE_TYPE_MAP:
+            return NODE_TYPE_MAP[node_type]
+        return NODE_TYPE_MAP[node_lc_type] if node_lc_type in NODE_TYPE_MAP else Node
 
     def _build_nodes(self) -> List[Node]:
+        """Builds the nodes of the graph."""
         nodes: List[Node] = []
         for node in self._nodes:
             node_data = node["data"]
@@ -157,6 +139,7 @@ class Graph:
         return nodes
 
     def get_children_by_node_type(self, node: Node, node_type: str) -> List[Node]:
+        """Returns the children of a node based on the node type."""
         children = []
         node_types = [node.data["type"]]
         if "node" in node.data:
diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py
new file mode 100644
index 000000000..f5bc9b8e3
--- /dev/null
+++ b/src/backend/langflow/graph/graph/constants.py
@@ -0,0 +1,49 @@
+from langflow.graph.node.base import Node
+from langflow.graph.node.types import (
+    AgentNode,
+    ChainNode,
+    DocumentLoaderNode,
+    EmbeddingNode,
+    LLMNode,
+    MemoryNode,
+    PromptNode,
+    TextSplitterNode,
+    ToolNode,
+    ToolkitNode,
+    VectorStoreNode,
+    WrapperNode,
+)
+from langflow.interface.agents.base import agent_creator
+from langflow.interface.chains.base import chain_creator
+from langflow.interface.document_loaders.base import documentloader_creator
+from langflow.interface.embeddings.base import embedding_creator
+from langflow.interface.llms.base import llm_creator
+from langflow.interface.memories.base import memory_creator
+from langflow.interface.prompts.base import prompt_creator
+from langflow.interface.text_splitters.base import textsplitter_creator
+from langflow.interface.toolkits.base import toolkits_creator
+from langflow.interface.tools.base import tool_creator
+from langflow.interface.vector_store.base import vectorstore_creator
+from langflow.interface.wrappers.base import wrapper_creator
+
+
+from typing import Dict, Type
+
+
+DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"]
+
+
+NODE_TYPE_MAP: Dict[str, Type[Node]] = {
+    **{t: PromptNode for t in prompt_creator.to_list()},
+    **{t: AgentNode for t in agent_creator.to_list()},
+    **{t: ChainNode for t in chain_creator.to_list()},
+    **{t: ToolNode for t in tool_creator.to_list()},
+    **{t: ToolkitNode for t in toolkits_creator.to_list()},
+    **{t: WrapperNode for t in wrapper_creator.to_list()},
+    **{t: LLMNode for t in llm_creator.to_list()},
+    **{t: MemoryNode for t in memory_creator.to_list()},
+    **{t: EmbeddingNode for t in embedding_creator.to_list()},
+    **{t: VectorStoreNode for t in vectorstore_creator.to_list()},
+    **{t: DocumentLoaderNode for t in documentloader_creator.to_list()},
+    **{t: TextSplitterNode for t in textsplitter_creator.to_list()},
+}
diff --git a/src/backend/langflow/graph/node/__init__.py b/src/backend/langflow/graph/node/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/graph/base.py b/src/backend/langflow/graph/node/base.py
similarity index 81%
rename from src/backend/langflow/graph/base.py
rename to src/backend/langflow/graph/node/base.py
index cc5e2902b..5076deb9c 100644
--- a/src/backend/langflow/graph/base.py
+++ b/src/backend/langflow/graph/node/base.py
@@ -1,27 +1,27 @@
-# Description: Graph class for building a graph of nodes and edges
-# Insights:
-#   - Defer prompts building to the last moment or when they have all the tools
-#   - Build each inner agent first, then build the outer agent
-
-import contextlib
-import inspect
-import types
-import warnings
-from typing import Any, Dict, List, Optional
-
 from langflow.cache import base as cache_utils
-from langflow.graph.constants import DIRECT_TYPES
+from langflow.graph.node.constants import DIRECT_TYPES
 from langflow.interface import loading
 from langflow.interface.listing import ALL_TYPES_DICT
 from langflow.utils.logger import logger
 from langflow.utils.util import sync_to_async
 
 
+import contextlib
+import inspect
+import types
+import warnings
+from typing import Any, Dict, List, Optional
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from langflow.graph.edge.base import Edge
+
+
 class Node:
     def __init__(self, data: Dict, base_type: Optional[str] = None) -> None:
         self.id: str = data["id"]
         self._data = data
-        self.edges: List[Edge] = []
+        self.edges: List["Edge"] = []
         self.base_type: Optional[str] = base_type
         self._parse_data()
         self._built_object = None
@@ -227,50 +227,3 @@ class Node:
 
     def _built_object_repr(self):
         return repr(self._built_object)
-
-
-class Edge:
-    def __init__(self, source: "Node", target: "Node"):
-        self.source: "Node" = source
-        self.target: "Node" = target
-        self.validate_edge()
-
-    def validate_edge(self) -> None:
-        # Validate that the outputs of the source node are valid inputs
-        # for the target node
-        self.source_types = self.source.output
-        self.target_reqs = self.target.required_inputs + self.target.optional_inputs
-        # Both lists contain strings and sometimes a string contains the value we are
-        # looking for e.g. comgin_out=["Chain"] and target_reqs=["LLMChain"]
-        # so we need to check if any of the strings in source_types is in target_reqs
-        self.valid = any(
-            output in target_req
-            for output in self.source_types
-            for target_req in self.target_reqs
-        )
-        # Get what type of input the target node is expecting
-
-        self.matched_type = next(
-            (
-                output
-                for output in self.source_types
-                for target_req in self.target_reqs
-                if output in target_req
-            ),
-            None,
-        )
-        no_matched_type = self.matched_type is None
-        if no_matched_type:
-            logger.debug(self.source_types)
-            logger.debug(self.target_reqs)
-        if no_matched_type:
-            raise ValueError(
-                f"Edge between {self.source.node_type} and {self.target.node_type} "
-                f"has no matched type"
-            )
-
-    def __repr__(self) -> str:
-        return (
-            f"Edge(source={self.source.id}, target={self.target.id}, valid={self.valid}"
-            f", matched_type={self.matched_type})"
-        )
diff --git a/src/backend/langflow/graph/constants.py b/src/backend/langflow/graph/node/constants.py
similarity index 100%
rename from src/backend/langflow/graph/constants.py
rename to src/backend/langflow/graph/node/constants.py
diff --git a/src/backend/langflow/graph/nodes.py b/src/backend/langflow/graph/node/types.py
similarity index 99%
rename from src/backend/langflow/graph/nodes.py
rename to src/backend/langflow/graph/node/types.py
index 189e40b5c..9b25fd6ee 100644
--- a/src/backend/langflow/graph/nodes.py
+++ b/src/backend/langflow/graph/node/types.py
@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 
-from langflow.graph.base import Node
+from langflow.graph.node.base import Node
 from langflow.graph.utils import extract_input_variables_from_prompt
 
 
diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py
index 69c697823..a3799be16 100644
--- a/src/backend/langflow/interface/loading.py
+++ b/src/backend/langflow/interface/loading.py
@@ -12,6 +12,7 @@ from langchain.agents.load_tools import (
     _LLM_TOOLS,
 )
 from langchain.agents.loading import load_agent_from_config
+from langflow.graph import Graph
 from langchain.agents.tools import Tool
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.base import BaseCallbackManager
@@ -164,7 +165,6 @@ def instantiate_utility(node_type, class_object, params):
 def load_flow_from_json(path: str, build=True):
     """Load flow from json file"""
     # This is done to avoid circular imports
-    from langflow.graph import Graph
 
     with open(path, "r", encoding="utf-8") as f:
         flow_graph = json.load(f)
diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py
index d24b6a0dc..c2483416f 100644
--- a/src/backend/langflow/interface/run.py
+++ b/src/backend/langflow/interface/run.py
@@ -6,7 +6,7 @@ from langchain.schema import AgentAction
 
 from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler  # type: ignore
 from langflow.cache.base import compute_dict_hash, load_cache, memoize_dict
-from langflow.graph.graph import Graph
+from langflow.graph import Graph
 from langflow.utils.logger import logger
 
 
diff --git a/tests/conftest.py b/tests/conftest.py
index 870c48a32..d0af2ad84 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -2,6 +2,7 @@ import json
 from pathlib import Path
 from typing import AsyncGenerator
 
+from langflow.graph.graph.base import Graph
 import pytest
 from fastapi.testclient import TestClient
 from httpx import AsyncClient
@@ -46,7 +47,6 @@ def client():
 
 def get_graph(_type="basic"):
     """Get a graph from a json file"""
-    from langflow.graph.graph import Graph
 
     if _type == "basic":
         path = pytest.BASIC_EXAMPLE_PATH
diff --git a/tests/test_graph.py b/tests/test_graph.py
index a0f5945fc..cdbe0ba93 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -1,10 +1,12 @@
 from typing import Type, Union
+from langflow.graph.edge.base import Edge
+from langflow.graph.node.base import Node
 
 import pytest
 from langchain.chains.base import Chain
 from langchain.llms.fake import FakeListLLM
-from langflow.graph import Edge, Graph, Node
-from langflow.graph.nodes import (
+from langflow.graph import Graph
+from langflow.graph.node.types import (
     AgentNode,
     ChainNode,
     FileToolNode,

From 0cef4bd818aa9206c11ad793432cbfda598b35cc Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 1 Jun 2023 15:37:11 -0300
Subject: [PATCH 23/59] refactor: change Node to Vertex

---
 src/backend/langflow/api/validate.py          |  4 +-
 src/backend/langflow/graph/__init__.py        | 54 ++++++++--------
 src/backend/langflow/graph/edge/base.py       | 10 +--
 src/backend/langflow/graph/graph/base.py      | 56 ++++++++--------
 src/backend/langflow/graph/graph/constants.py | 54 ++++++++--------
 .../graph/{node => vertex}/__init__.py        |  0
 .../langflow/graph/{node => vertex}/base.py   | 26 ++++----
 .../graph/{node => vertex}/constants.py       |  0
 .../langflow/graph/{node => vertex}/types.py  | 64 ++++++++++---------
 tests/test_graph.py                           | 54 ++++++++--------
 10 files changed, 164 insertions(+), 158 deletions(-)
 rename src/backend/langflow/graph/{node => vertex}/__init__.py (100%)
 rename src/backend/langflow/graph/{node => vertex}/base.py (92%)
 rename src/backend/langflow/graph/{node => vertex}/constants.py (100%)
 rename src/backend/langflow/graph/{node => vertex}/types.py (78%)

diff --git a/src/backend/langflow/api/validate.py b/src/backend/langflow/api/validate.py
index 53a7ee350..e90e554f0 100644
--- a/src/backend/langflow/api/validate.py
+++ b/src/backend/langflow/api/validate.py
@@ -9,7 +9,7 @@ from langflow.api.base import (
     PromptValidationResponse,
     validate_prompt,
 )
-from langflow.graph.node.types import VectorStoreNode
+from langflow.graph.vertex.types import VectorStoreVertex
 from langflow.interface.run import build_graph
 from langflow.utils.logger import logger
 from langflow.utils.validate import validate_code
@@ -49,7 +49,7 @@ def post_validate_node(node_id: str, data: dict):
         node = graph.get_node(node_id)
         if node is None:
             raise ValueError(f"Node {node_id} not found")
-        if not isinstance(node, VectorStoreNode):
+        if not isinstance(node, VectorStoreVertex):
             node.build()
         return json.dumps({"valid": True, "params": str(node._built_object_repr())})
     except Exception as e:
diff --git a/src/backend/langflow/graph/__init__.py b/src/backend/langflow/graph/__init__.py
index 44859da02..a68e844ee 100644
--- a/src/backend/langflow/graph/__init__.py
+++ b/src/backend/langflow/graph/__init__.py
@@ -1,35 +1,35 @@
 from langflow.graph.edge.base import Edge
 from langflow.graph.graph.base import Graph
-from langflow.graph.node.base import Node
-from langflow.graph.node.types import (
-    AgentNode,
-    ChainNode,
-    DocumentLoaderNode,
-    EmbeddingNode,
-    LLMNode,
-    MemoryNode,
-    PromptNode,
-    TextSplitterNode,
-    ToolNode,
-    ToolkitNode,
-    VectorStoreNode,
-    WrapperNode,
+from langflow.graph.vertex.base import Vertex
+from langflow.graph.vertex.types import (
+    AgentVertex,
+    ChainVertex,
+    DocumentLoaderVertex,
+    EmbeddingVertex,
+    LLMVertex,
+    MemoryVertex,
+    PromptVertex,
+    TextSplitterVertex,
+    ToolVertex,
+    ToolkitVertex,
+    VectorStoreVertex,
+    WrapperVertex,
 )
 
 __all__ = [
     "Graph",
-    "Node",
+    "Vertex",
     "Edge",
-    "AgentNode",
-    "ChainNode",
-    "DocumentLoaderNode",
-    "EmbeddingNode",
-    "LLMNode",
-    "MemoryNode",
-    "PromptNode",
-    "TextSplitterNode",
-    "ToolNode",
-    "ToolkitNode",
-    "VectorStoreNode",
-    "WrapperNode",
+    "AgentVertex",
+    "ChainVertex",
+    "DocumentLoaderVertex",
+    "EmbeddingVertex",
+    "LLMVertex",
+    "MemoryVertex",
+    "PromptVertex",
+    "TextSplitterVertex",
+    "ToolVertex",
+    "ToolkitVertex",
+    "VectorStoreVertex",
+    "WrapperVertex",
 ]
diff --git a/src/backend/langflow/graph/edge/base.py b/src/backend/langflow/graph/edge/base.py
index 2bf5a1ba4..08f084a5c 100644
--- a/src/backend/langflow/graph/edge/base.py
+++ b/src/backend/langflow/graph/edge/base.py
@@ -2,13 +2,13 @@ from langflow.utils.logger import logger
 from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
-    from langflow.graph.node.base import Node
+    from langflow.graph.vertex.base import Vertex
 
 
 class Edge:
-    def __init__(self, source: "Node", target: "Node"):
-        self.source: "Node" = source
-        self.target: "Node" = target
+    def __init__(self, source: "Vertex", target: "Vertex"):
+        self.source: "Vertex" = source
+        self.target: "Vertex" = target
         self.validate_edge()
 
     def validate_edge(self) -> None:
@@ -41,7 +41,7 @@ class Edge:
             logger.debug(self.target_reqs)
         if no_matched_type:
             raise ValueError(
-                f"Edge between {self.source.node_type} and {self.target.node_type} "
+                f"Edge between {self.source.vertex_type} and {self.target.vertex_type} "
                 f"has no matched type"
             )
 
diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py
index 3ba67837f..020f539ec 100644
--- a/src/backend/langflow/graph/graph/base.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -1,12 +1,12 @@
 from typing import Dict, List, Type, Union
 
 from langflow.graph.edge.base import Edge
-from langflow.graph.graph.constants import NODE_TYPE_MAP
-from langflow.graph.node.base import Node
-from langflow.graph.node.types import (
-    FileToolNode,
-    LLMNode,
-    ToolkitNode,
+from langflow.graph.graph.constants import VERTEX_TYPE_MAP
+from langflow.graph.vertex.base import Vertex
+from langflow.graph.vertex.types import (
+    FileToolVertex,
+    LLMVertex,
+    ToolkitVertex,
 )
 from langflow.interface.tools.constants import FILE_TOOLS
 from langflow.utils import payload
@@ -26,7 +26,7 @@ class Graph:
 
     def _build_graph(self) -> None:
         """Builds the graph from the nodes and edges."""
-        self.nodes = self._build_nodes()
+        self.nodes = self._build_vertices()
         self.edges = self._build_edges()
         for edge in self.edges:
             edge.source.add_edge(edge)
@@ -43,12 +43,12 @@ class Graph:
         llm_node = None
         for node in self.nodes:
             node._build_params()
-            if isinstance(node, LLMNode):
+            if isinstance(node, LLMVertex):
                 llm_node = node
 
         if llm_node:
             for node in self.nodes:
-                if isinstance(node, ToolkitNode):
+                if isinstance(node, ToolkitVertex):
                     node.params["llm"] = llm_node
 
     def _remove_invalid_nodes(self) -> None:
@@ -60,23 +60,23 @@ class Graph:
             or (len(self.nodes) == 1 and len(self.edges) == 0)
         ]
 
-    def _validate_node(self, node: Node) -> bool:
+    def _validate_node(self, node: Vertex) -> bool:
         """Validates a node."""
         # All nodes that do not have edges are invalid
         return len(node.edges) > 0
 
-    def get_node(self, node_id: str) -> Union[None, Node]:
+    def get_node(self, node_id: str) -> Union[None, Vertex]:
         """Returns a node by id."""
         return next((node for node in self.nodes if node.id == node_id), None)
 
-    def get_nodes_with_target(self, node: Node) -> List[Node]:
+    def get_nodes_with_target(self, node: Vertex) -> List[Vertex]:
         """Returns the nodes connected to a node."""
-        connected_nodes: List[Node] = [
+        connected_nodes: List[Vertex] = [
             edge.source for edge in self.edges if edge.target == node
         ]
         return connected_nodes
 
-    def build(self) -> List[Node]:
+    def build(self) -> List[Vertex]:
         """Builds the graph."""
         # Get root node
         root_node = payload.get_root_node(self)
@@ -84,9 +84,9 @@ class Graph:
             raise ValueError("No root node found")
         return root_node.build()
 
-    def get_node_neighbors(self, node: Node) -> Dict[Node, int]:
+    def get_node_neighbors(self, node: Vertex) -> Dict[Vertex, int]:
         """Returns the neighbors of a node."""
-        neighbors: Dict[Node, int] = {}
+        neighbors: Dict[Vertex, int] = {}
         for edge in self.edges:
             if edge.source == node:
                 neighbor = edge.target
@@ -117,28 +117,30 @@ class Graph:
             edges.append(Edge(source, target))
         return edges
 
-    def _get_node_class(self, node_type: str, node_lc_type: str) -> Type[Node]:
+    def _get_vertex_class(self, node_type: str, node_lc_type: str) -> Type[Vertex]:
         """Returns the node class based on the node type."""
         if node_type in FILE_TOOLS:
-            return FileToolNode
-        if node_type in NODE_TYPE_MAP:
-            return NODE_TYPE_MAP[node_type]
-        return NODE_TYPE_MAP[node_lc_type] if node_lc_type in NODE_TYPE_MAP else Node
+            return FileToolVertex
+        if node_type in VERTEX_TYPE_MAP:
+            return VERTEX_TYPE_MAP[node_type]
+        return (
+            VERTEX_TYPE_MAP[node_lc_type] if node_lc_type in VERTEX_TYPE_MAP else Vertex
+        )
 
-    def _build_nodes(self) -> List[Node]:
-        """Builds the nodes of the graph."""
-        nodes: List[Node] = []
+    def _build_vertices(self) -> List[Vertex]:
+        """Builds the vertices of the graph."""
+        nodes: List[Vertex] = []
         for node in self._nodes:
             node_data = node["data"]
             node_type: str = node_data["type"]  # type: ignore
             node_lc_type: str = node_data["node"]["template"]["_type"]  # type: ignore
 
-            NodeClass = self._get_node_class(node_type, node_lc_type)
-            nodes.append(NodeClass(node))
+            VertexClass = self._get_vertex_class(node_type, node_lc_type)
+            nodes.append(VertexClass(node))
 
         return nodes
 
-    def get_children_by_node_type(self, node: Node, node_type: str) -> List[Node]:
+    def get_children_by_node_type(self, node: Vertex, node_type: str) -> List[Vertex]:
         """Returns the children of a node based on the node type."""
         children = []
         node_types = [node.data["type"]]
diff --git a/src/backend/langflow/graph/graph/constants.py b/src/backend/langflow/graph/graph/constants.py
index f5bc9b8e3..ff1317d39 100644
--- a/src/backend/langflow/graph/graph/constants.py
+++ b/src/backend/langflow/graph/graph/constants.py
@@ -1,17 +1,17 @@
-from langflow.graph.node.base import Node
-from langflow.graph.node.types import (
-    AgentNode,
-    ChainNode,
-    DocumentLoaderNode,
-    EmbeddingNode,
-    LLMNode,
-    MemoryNode,
-    PromptNode,
-    TextSplitterNode,
-    ToolNode,
-    ToolkitNode,
-    VectorStoreNode,
-    WrapperNode,
+from langflow.graph.vertex.base import Vertex
+from langflow.graph.vertex.types import (
+    AgentVertex,
+    ChainVertex,
+    DocumentLoaderVertex,
+    EmbeddingVertex,
+    LLMVertex,
+    MemoryVertex,
+    PromptVertex,
+    TextSplitterVertex,
+    ToolVertex,
+    ToolkitVertex,
+    VectorStoreVertex,
+    WrapperVertex,
 )
 from langflow.interface.agents.base import agent_creator
 from langflow.interface.chains.base import chain_creator
@@ -33,17 +33,17 @@ from typing import Dict, Type
 DIRECT_TYPES = ["str", "bool", "code", "int", "float", "Any", "prompt"]
 
 
-NODE_TYPE_MAP: Dict[str, Type[Node]] = {
-    **{t: PromptNode for t in prompt_creator.to_list()},
-    **{t: AgentNode for t in agent_creator.to_list()},
-    **{t: ChainNode for t in chain_creator.to_list()},
-    **{t: ToolNode for t in tool_creator.to_list()},
-    **{t: ToolkitNode for t in toolkits_creator.to_list()},
-    **{t: WrapperNode for t in wrapper_creator.to_list()},
-    **{t: LLMNode for t in llm_creator.to_list()},
-    **{t: MemoryNode for t in memory_creator.to_list()},
-    **{t: EmbeddingNode for t in embedding_creator.to_list()},
-    **{t: VectorStoreNode for t in vectorstore_creator.to_list()},
-    **{t: DocumentLoaderNode for t in documentloader_creator.to_list()},
-    **{t: TextSplitterNode for t in textsplitter_creator.to_list()},
+VERTEX_TYPE_MAP: Dict[str, Type[Vertex]] = {
+    **{t: PromptVertex for t in prompt_creator.to_list()},
+    **{t: AgentVertex for t in agent_creator.to_list()},
+    **{t: ChainVertex for t in chain_creator.to_list()},
+    **{t: ToolVertex for t in tool_creator.to_list()},
+    **{t: ToolkitVertex for t in toolkits_creator.to_list()},
+    **{t: WrapperVertex for t in wrapper_creator.to_list()},
+    **{t: LLMVertex for t in llm_creator.to_list()},
+    **{t: MemoryVertex for t in memory_creator.to_list()},
+    **{t: EmbeddingVertex for t in embedding_creator.to_list()},
+    **{t: VectorStoreVertex for t in vectorstore_creator.to_list()},
+    **{t: DocumentLoaderVertex for t in documentloader_creator.to_list()},
+    **{t: TextSplitterVertex for t in textsplitter_creator.to_list()},
 }
diff --git a/src/backend/langflow/graph/node/__init__.py b/src/backend/langflow/graph/vertex/__init__.py
similarity index 100%
rename from src/backend/langflow/graph/node/__init__.py
rename to src/backend/langflow/graph/vertex/__init__.py
diff --git a/src/backend/langflow/graph/node/base.py b/src/backend/langflow/graph/vertex/base.py
similarity index 92%
rename from src/backend/langflow/graph/node/base.py
rename to src/backend/langflow/graph/vertex/base.py
index 5076deb9c..4593e0a40 100644
--- a/src/backend/langflow/graph/node/base.py
+++ b/src/backend/langflow/graph/vertex/base.py
@@ -1,5 +1,5 @@
 from langflow.cache import base as cache_utils
-from langflow.graph.node.constants import DIRECT_TYPES
+from langflow.graph.vertex.constants import DIRECT_TYPES
 from langflow.interface import loading
 from langflow.interface.listing import ALL_TYPES_DICT
 from langflow.utils.logger import logger
@@ -17,7 +17,7 @@ if TYPE_CHECKING:
     from langflow.graph.edge.base import Edge
 
 
-class Node:
+class Vertex:
     def __init__(self, data: Dict, base_type: Optional[str] = None) -> None:
         self.id: str = data["id"]
         self._data = data
@@ -48,12 +48,12 @@ class Node:
         ]
 
         template_dict = self.data["node"]["template"]
-        self.node_type = (
+        self.vertex_type = (
             self.data["type"] if "Tool" not in self.output else template_dict["_type"]
         )
         if self.base_type is None:
             for base_type, value in ALL_TYPES_DICT.items():
-                if self.node_type in value:
+                if self.vertex_type in value:
                     self.base_type = base_type
                     break
 
@@ -113,7 +113,7 @@ class Node:
                 if value["required"] and not edges:
                     # If a required parameter is not found, raise an error
                     raise ValueError(
-                        f"Required input {key} for module {self.node_type} not found"
+                        f"Required input {key} for module {self.vertex_type} not found"
                     )
                 elif value["list"]:
                     # If this is a list parameter, append all sources to a list
@@ -128,7 +128,7 @@ class Node:
                 # so we need to check if value has value
                 new_value = value.get("value")
                 if new_value is None:
-                    warnings.warn(f"Value for {key} in {self.node_type} is None. ")
+                    warnings.warn(f"Value for {key} in {self.vertex_type} is None. ")
                 if value.get("type") == "int":
                     with contextlib.suppress(TypeError, ValueError):
                         new_value = int(new_value)  # type: ignore
@@ -148,12 +148,12 @@ class Node:
         # and continue
         # Another aspect is that the node_type is the class that we need to import
         # and instantiate with these built params
-        logger.debug(f"Building {self.node_type}")
+        logger.debug(f"Building {self.vertex_type}")
         # Build each node in the params dict
         for key, value in self.params.copy().items():
             # Check if Node or list of Nodes and not self
             # to avoid recursion
-            if isinstance(value, Node):
+            if isinstance(value, Vertex):
                 if value == self:
                     del self.params[key]
                     continue
@@ -177,7 +177,7 @@ class Node:
 
                 self.params[key] = result
             elif isinstance(value, list) and all(
-                isinstance(node, Node) for node in value
+                isinstance(node, Vertex) for node in value
             ):
                 self.params[key] = []
                 for node in value:
@@ -193,17 +193,17 @@ class Node:
 
         try:
             self._built_object = loading.instantiate_class(
-                node_type=self.node_type,
+                node_type=self.vertex_type,
                 base_type=self.base_type,
                 params=self.params,
             )
         except Exception as exc:
             raise ValueError(
-                f"Error building node {self.node_type}: {str(exc)}"
+                f"Error building node {self.vertex_type}: {str(exc)}"
             ) from exc
 
         if self._built_object is None:
-            raise ValueError(f"Node type {self.node_type} not found")
+            raise ValueError(f"Node type {self.vertex_type} not found")
 
         self._built = True
 
@@ -220,7 +220,7 @@ class Node:
         return f"Node(id={self.id}, data={self.data})"
 
     def __eq__(self, __o: object) -> bool:
-        return self.id == __o.id if isinstance(__o, Node) else False
+        return self.id == __o.id if isinstance(__o, Vertex) else False
 
     def __hash__(self) -> int:
         return id(self)
diff --git a/src/backend/langflow/graph/node/constants.py b/src/backend/langflow/graph/vertex/constants.py
similarity index 100%
rename from src/backend/langflow/graph/node/constants.py
rename to src/backend/langflow/graph/vertex/constants.py
diff --git a/src/backend/langflow/graph/node/types.py b/src/backend/langflow/graph/vertex/types.py
similarity index 78%
rename from src/backend/langflow/graph/node/types.py
rename to src/backend/langflow/graph/vertex/types.py
index 9b25fd6ee..4a3290c13 100644
--- a/src/backend/langflow/graph/node/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -1,22 +1,22 @@
 from typing import Any, Dict, List, Optional, Union
 
-from langflow.graph.node.base import Node
+from langflow.graph.vertex.base import Vertex
 from langflow.graph.utils import extract_input_variables_from_prompt
 
 
-class AgentNode(Node):
+class AgentVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="agents")
 
-        self.tools: List[ToolNode] = []
-        self.chains: List[ChainNode] = []
+        self.tools: List[ToolVertex] = []
+        self.chains: List[ChainVertex] = []
 
     def _set_tools_and_chains(self) -> None:
         for edge in self.edges:
             source_node = edge.source
-            if isinstance(source_node, ToolNode):
+            if isinstance(source_node, ToolVertex):
                 self.tools.append(source_node)
-            elif isinstance(source_node, ChainNode):
+            elif isinstance(source_node, ChainVertex):
                 self.chains.append(source_node)
 
     def build(self, force: bool = False) -> Any:
@@ -33,24 +33,28 @@ class AgentNode(Node):
             self._build()
 
         #! Cannot deepcopy VectorStore, VectorStoreRouter, or SQL agents
-        if self.node_type in ["VectorStoreAgent", "VectorStoreRouterAgent", "SQLAgent"]:
+        if self.vertex_type in [
+            "VectorStoreAgent",
+            "VectorStoreRouterAgent",
+            "SQLAgent",
+        ]:
             return self._built_object
         return self._built_object
 
 
-class ToolNode(Node):
+class ToolVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="tools")
 
 
-class PromptNode(Node):
+class PromptVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="prompts")
 
     def build(
         self,
         force: bool = False,
-        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
+        tools: Optional[Union[List[Vertex], List[ToolVertex]]] = None,
     ) -> Any:
         if not self._built or force:
             if (
@@ -59,7 +63,7 @@ class PromptNode(Node):
             ):
                 self.params["input_variables"] = []
             # Check if it is a ZeroShotPrompt and needs a tool
-            if "ShotPrompt" in self.node_type:
+            if "ShotPrompt" in self.vertex_type:
                 tools = (
                     [tool_node.build() for tool_node in tools]
                     if tools is not None
@@ -83,31 +87,31 @@ class PromptNode(Node):
         return self._built_object
 
 
-class ChainNode(Node):
+class ChainVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="chains")
 
     def build(
         self,
         force: bool = False,
-        tools: Optional[Union[List[Node], List[ToolNode]]] = None,
+        tools: Optional[Union[List[Vertex], List[ToolVertex]]] = None,
     ) -> Any:
         if not self._built or force:
             # Check if the chain requires a PromptNode
             for key, value in self.params.items():
-                if isinstance(value, PromptNode):
+                if isinstance(value, PromptVertex):
                     # Build the PromptNode, passing the tools if available
                     self.params[key] = value.build(tools=tools, force=force)
 
             self._build()
 
         #! Cannot deepcopy SQLDatabaseChain
-        if self.node_type in ["SQLDatabaseChain"]:
+        if self.vertex_type in ["SQLDatabaseChain"]:
             return self._built_object
         return self._built_object
 
 
-class LLMNode(Node):
+class LLMVertex(Vertex):
     built_node_type = None
     class_built_object = None
 
@@ -117,28 +121,28 @@ class LLMNode(Node):
     def build(self, force: bool = False) -> Any:
         # LLM is different because some models might take up too much memory
         # or time to load. So we only load them when we need them.ß
-        if self.node_type == self.built_node_type:
+        if self.vertex_type == self.built_node_type:
             return self.class_built_object
         if not self._built or force:
             self._build()
-            self.built_node_type = self.node_type
+            self.built_node_type = self.vertex_type
             self.class_built_object = self._built_object
         # Avoid deepcopying the LLM
         # that are loaded from a file
         return self._built_object
 
 
-class ToolkitNode(Node):
+class ToolkitVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="toolkits")
 
 
-class FileToolNode(ToolNode):
+class FileToolVertex(ToolVertex):
     def __init__(self, data: Dict):
         super().__init__(data)
 
 
-class WrapperNode(Node):
+class WrapperVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="wrappers")
 
@@ -150,7 +154,7 @@ class WrapperNode(Node):
         return self._built_object
 
 
-class DocumentLoaderNode(Node):
+class DocumentLoaderVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="documentloaders")
 
@@ -158,17 +162,17 @@ class DocumentLoaderNode(Node):
         # This built_object is a list of documents. Maybe we should
         # show how many documents are in the list?
         if self._built_object:
-            return f"""{self.node_type}({len(self._built_object)} documents)
+            return f"""{self.vertex_type}({len(self._built_object)} documents)
             Documents: {self._built_object[:3]}..."""
-        return f"{self.node_type}()"
+        return f"{self.vertex_type}()"
 
 
-class EmbeddingNode(Node):
+class EmbeddingVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="embeddings")
 
 
-class VectorStoreNode(Node):
+class VectorStoreVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="vectorstores")
 
@@ -176,12 +180,12 @@ class VectorStoreNode(Node):
         return "Vector stores can take time to build. It will build on the first query."
 
 
-class MemoryNode(Node):
+class MemoryVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="memory")
 
 
-class TextSplitterNode(Node):
+class TextSplitterVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="textsplitters")
 
@@ -189,5 +193,5 @@ class TextSplitterNode(Node):
         # This built_object is a list of documents. Maybe we should
         # show how many documents are in the list?
         if self._built_object:
-            return f"""{self.node_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
-        return f"{self.node_type}()"
+            return f"""{self.vertex_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
+        return f"{self.vertex_type}()"
diff --git a/tests/test_graph.py b/tests/test_graph.py
index cdbe0ba93..c7b5ddf0c 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -1,20 +1,20 @@
 from typing import Type, Union
 from langflow.graph.edge.base import Edge
-from langflow.graph.node.base import Node
+from langflow.graph.vertex.base import Vertex
 
 import pytest
 from langchain.chains.base import Chain
 from langchain.llms.fake import FakeListLLM
 from langflow.graph import Graph
-from langflow.graph.node.types import (
-    AgentNode,
-    ChainNode,
-    FileToolNode,
-    LLMNode,
-    PromptNode,
-    ToolkitNode,
-    ToolNode,
-    WrapperNode,
+from langflow.graph.vertex.types import (
+    AgentVertex,
+    ChainVertex,
+    FileToolVertex,
+    LLMVertex,
+    PromptVertex,
+    ToolkitVertex,
+    ToolVertex,
+    WrapperVertex,
 )
 from langflow.interface.run import get_result_and_thought
 from langflow.utils.payload import get_root_node
@@ -25,7 +25,7 @@ from langflow.utils.payload import get_root_node
 # BASIC_EXAMPLE_PATH, COMPLEX_EXAMPLE_PATH, OPENAPI_EXAMPLE_PATH
 
 
-def get_node_by_type(graph, node_type: Type[Node]) -> Union[Node, None]:
+def get_node_by_type(graph, node_type: Type[Vertex]) -> Union[Vertex, None]:
     """Get a node by type"""
     return next((node for node in graph.nodes if isinstance(node, node_type)), None)
 
@@ -35,7 +35,7 @@ def test_graph_structure(basic_graph):
     assert len(basic_graph.nodes) > 0
     assert len(basic_graph.edges) > 0
     for node in basic_graph.nodes:
-        assert isinstance(node, Node)
+        assert isinstance(node, Vertex)
     for edge in basic_graph.edges:
         assert isinstance(edge, Edge)
         assert edge.source in basic_graph.nodes
@@ -165,7 +165,7 @@ def test_get_node(basic_graph):
     """Test getting a single node"""
     node_id = basic_graph.nodes[0].id
     node = basic_graph.get_node(node_id)
-    assert isinstance(node, Node)
+    assert isinstance(node, Vertex)
     assert node.id == node_id
 
 
@@ -174,7 +174,7 @@ def test_build_nodes(basic_graph):
 
     assert len(basic_graph.nodes) == len(basic_graph._nodes)
     for node in basic_graph.nodes:
-        assert isinstance(node, Node)
+        assert isinstance(node, Vertex)
 
 
 def test_build_edges(basic_graph):
@@ -182,8 +182,8 @@ def test_build_edges(basic_graph):
     assert len(basic_graph.edges) == len(basic_graph._edges)
     for edge in basic_graph.edges:
         assert isinstance(edge, Edge)
-        assert isinstance(edge.source, Node)
-        assert isinstance(edge.target, Node)
+        assert isinstance(edge.source, Vertex)
+        assert isinstance(edge.target, Vertex)
 
 
 def test_get_root_node(basic_graph, complex_graph):
@@ -191,13 +191,13 @@ def test_get_root_node(basic_graph, complex_graph):
     assert isinstance(basic_graph, Graph)
     root = get_root_node(basic_graph)
     assert root is not None
-    assert isinstance(root, Node)
+    assert isinstance(root, Vertex)
     assert root.data["type"] == "TimeTravelGuideChain"
     # For complex example, the root node is a ZeroShotAgent too
     assert isinstance(complex_graph, Graph)
     root = get_root_node(complex_graph)
     assert root is not None
-    assert isinstance(root, Node)
+    assert isinstance(root, Vertex)
     assert root.data["type"] == "ZeroShotAgent"
 
 
@@ -257,14 +257,14 @@ def assert_agent_was_built(graph):
 
 
 def test_agent_node_build(complex_graph):
-    agent_node = get_node_by_type(complex_graph, AgentNode)
+    agent_node = get_node_by_type(complex_graph, AgentVertex)
     assert agent_node is not None
     built_object = agent_node.build()
     assert built_object is not None
 
 
 def test_tool_node_build(complex_graph):
-    tool_node = get_node_by_type(complex_graph, ToolNode)
+    tool_node = get_node_by_type(complex_graph, ToolVertex)
     assert tool_node is not None
     built_object = tool_node.build()
     assert built_object is not None
@@ -272,7 +272,7 @@ def test_tool_node_build(complex_graph):
 
 
 def test_chain_node_build(complex_graph):
-    chain_node = get_node_by_type(complex_graph, ChainNode)
+    chain_node = get_node_by_type(complex_graph, ChainVertex)
     assert chain_node is not None
     built_object = chain_node.build()
     assert built_object is not None
@@ -280,7 +280,7 @@ def test_chain_node_build(complex_graph):
 
 
 def test_prompt_node_build(complex_graph):
-    prompt_node = get_node_by_type(complex_graph, PromptNode)
+    prompt_node = get_node_by_type(complex_graph, PromptVertex)
     assert prompt_node is not None
     built_object = prompt_node.build()
     assert built_object is not None
@@ -288,7 +288,7 @@ def test_prompt_node_build(complex_graph):
 
 
 def test_llm_node_build(basic_graph):
-    llm_node = get_node_by_type(basic_graph, LLMNode)
+    llm_node = get_node_by_type(basic_graph, LLMVertex)
     assert llm_node is not None
     built_object = llm_node.build()
     assert built_object is not None
@@ -296,7 +296,7 @@ def test_llm_node_build(basic_graph):
 
 
 def test_toolkit_node_build(openapi_graph):
-    toolkit_node = get_node_by_type(openapi_graph, ToolkitNode)
+    toolkit_node = get_node_by_type(openapi_graph, ToolkitVertex)
     assert toolkit_node is not None
     built_object = toolkit_node.build()
     assert built_object is not None
@@ -304,7 +304,7 @@ def test_toolkit_node_build(openapi_graph):
 
 
 def test_file_tool_node_build(openapi_graph):
-    file_tool_node = get_node_by_type(openapi_graph, FileToolNode)
+    file_tool_node = get_node_by_type(openapi_graph, FileToolVertex)
     assert file_tool_node is not None
     built_object = file_tool_node.build()
     assert built_object is not None
@@ -312,7 +312,7 @@ def test_file_tool_node_build(openapi_graph):
 
 
 def test_wrapper_node_build(openapi_graph):
-    wrapper_node = get_node_by_type(openapi_graph, WrapperNode)
+    wrapper_node = get_node_by_type(openapi_graph, WrapperVertex)
     assert wrapper_node is not None
     built_object = wrapper_node.build()
     assert built_object is not None
@@ -327,7 +327,7 @@ def test_get_result_and_thought(basic_graph):
     message = "Hello"
     # Find the node that is an LLMNode and change the
     # _built_object to a FakeListLLM
-    llm_node = get_node_by_type(basic_graph, LLMNode)
+    llm_node = get_node_by_type(basic_graph, LLMVertex)
     assert llm_node is not None
     llm_node._built_object = FakeListLLM(responses=responses)
     llm_node._built = True

From da5b15fa6899e9a461d72b6d6806a2575f183914 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Thu, 1 Jun 2023 17:47:27 -0300
Subject: [PATCH 24/59] Bringing the menu options opened on search

---
 .../pages/FlowPage/components/DisclosureComponent/index.tsx  | 5 +++--
 .../FlowPage/components/extraSidebarComponent/index.tsx      | 3 +++
 src/frontend/src/types/components/index.ts                   | 1 +
 3 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/pages/FlowPage/components/DisclosureComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/DisclosureComponent/index.tsx
index 31bf78156..6878ad3c4 100644
--- a/src/frontend/src/pages/FlowPage/components/DisclosureComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/DisclosureComponent/index.tsx
@@ -5,6 +5,7 @@ import { DisclosureComponentType } from "../../../../types/components";
 export default function DisclosureComponent({
   button: { title, Icon, buttons = [] },
   children,
+  openDisc,
 }: DisclosureComponentType) {
   return (
     <Disclosure as="div" key={title}>
@@ -27,14 +28,14 @@ export default function DisclosureComponent({
                 <div>
                   <ChevronRightIcon
                     className={`${
-                      open ? "rotate-90 transform" : ""
+                      open || openDisc ? "rotate-90 transform" : ""
                     } h-4 w-4 text-gray-800 dark:text-white`}
                   />
                 </div>
               </div>
             </Disclosure.Button>
           </div>
-          <Disclosure.Panel as="div" className="-mt-px">
+          <Disclosure.Panel as="div" className="-mt-px" static={openDisc}>
             {children}
           </Disclosure.Panel>
         </>
diff --git a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
index e8dbfbb6b..deeed6c70 100644
--- a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
@@ -15,6 +15,7 @@ import { MagnifyingGlassIcon } from "@heroicons/react/24/outline";
 export default function ExtraSidebar() {
   const { data } = useContext(typesContext);
   const [dataFilter, setFilterData] = useState(data);
+  const [search, setSearch] = useState("");
 
   function onDragStart(
     event: React.DragEvent<any>,
@@ -58,6 +59,7 @@ export default function ExtraSidebar() {
           className="dark:text-white focus:outline-none block w-full rounded-md py-1.5 ps-3 pr-9 text-gray-900 shadow-sm ring-1 ring-inset ring-gray-300 placeholder:text-gray-400 sm:text-sm sm:leading-6 dark:ring-0 dark:bg-[#2d3747] dark:focus:outline-none"
           onChange={(e) => {
             handleSearchInput(e.target.value);
+            setSearch(e.target.value);
           }}
         />
         <div className="absolute inset-y-0 right-0 flex py-1.5 pr-3 items-center">
@@ -71,6 +73,7 @@ export default function ExtraSidebar() {
           .map((d: keyof APIObjectType, i) =>
             Object.keys(dataFilter[d]).length > 0 ? (
               <DisclosureComponent
+                openDisc={search.length == 0 ? false : true}
                 key={i}
                 button={{
                   title: nodeNames[d] ?? nodeNames.unknown,
diff --git a/src/frontend/src/types/components/index.ts b/src/frontend/src/types/components/index.ts
index e7559023a..267a239fc 100644
--- a/src/frontend/src/types/components/index.ts
+++ b/src/frontend/src/types/components/index.ts
@@ -56,6 +56,7 @@ export type FileComponentType = {
 
 export type DisclosureComponentType = {
   children: ReactNode;
+  openDisc: boolean;
   button: {
     title: string;
     Icon: ForwardRefExoticComponent<React.SVGProps<SVGSVGElement>>;

From 76bbf6247502bf67fa7575d78902e613544f3e6c Mon Sep 17 00:00:00 2001
From: Jacob Lee <jacob.lee@tempus.com>
Date: Thu, 1 Jun 2023 18:04:56 -0500
Subject: [PATCH 25/59] Surface an error message when a flow has an unknown
 node type.

This improves the user experience for issues like #283.
---
 src/frontend/src/contexts/tabsContext.tsx | 38 +++++++++++++----------
 1 file changed, 21 insertions(+), 17 deletions(-)

diff --git a/src/frontend/src/contexts/tabsContext.tsx b/src/frontend/src/contexts/tabsContext.tsx
index 1d7ad43d5..acc7011bc 100644
--- a/src/frontend/src/contexts/tabsContext.tsx
+++ b/src/frontend/src/contexts/tabsContext.tsx
@@ -47,7 +47,7 @@ export const TabsContext = createContext<TabsContextType>(
 );
 
 export function TabsProvider({ children }: { children: ReactNode }) {
-  const { setNoticeData } = useContext(alertContext);
+  const { setErrorData, setNoticeData } = useContext(alertContext);
   const [tabIndex, setTabIndex] = useState(0);
   const [flows, setFlows] = useState<Array<FlowType>>([]);
   const [id, setId] = useState(uuidv4());
@@ -95,25 +95,25 @@ export function TabsProvider({ children }: { children: ReactNode }) {
           edge.style = { stroke: "#555555" };
         });
         flow.data.nodes.forEach((node) => {
-          if (Object.keys(templates[node.data.type]["template"]).length > 0) {
-            node.data.node.base_classes =
-              templates[node.data.type]["base_classes"];
+          const template = templates[node.data.type];
+          if (!template) {
+            setErrorData({ title: `Unknown node type: ${node.data.type}` });
+            return;
+          }
+          if (Object.keys(template["template"]).length > 0) {
+            node.data.node.base_classes = template["base_classes"];
             flow.data.edges.forEach((edge) => {
               if (edge.source === node.id) {
                 edge.sourceHandle = edge.sourceHandle
                   .split("|")
                   .slice(0, 2)
-                  .concat(templates[node.data.type]["base_classes"])
+                  .concat(template["base_classes"])
                   .join("|");
               }
             });
-            node.data.node.description =
-              templates[node.data.type]["description"];
+            node.data.node.description = template["description"];
             node.data.node.template = updateTemplate(
-              templates[node.data.type][
-                "template"
-              ] as unknown as APITemplateType,
-
+              template["template"] as unknown as APITemplateType,
               node.data.node.template as APITemplateType
             );
           }
@@ -316,21 +316,25 @@ export function TabsProvider({ children }: { children: ReactNode }) {
         edge.animated = edge.targetHandle.split("|")[0] === "Text";
       });
       data.nodes.forEach((node) => {
-        if (Object.keys(templates[node.data.type]["template"]).length > 0) {
-          node.data.node.base_classes =
-            templates[node.data.type]["base_classes"];
+        const template = templates[node.data.type];
+        if (!template) {
+          setErrorData({ title: `Unknown node type: ${node.data.type}` });
+          return;
+        }
+        if (Object.keys(template["template"]).length > 0) {
+          node.data.node.base_classes = template["base_classes"];
           flow.data.edges.forEach((edge) => {
             if (edge.source === node.id) {
               edge.sourceHandle = edge.sourceHandle
                 .split("|")
                 .slice(0, 2)
-                .concat(templates[node.data.type]["base_classes"])
+                .concat(template["base_classes"])
                 .join("|");
             }
           });
-          node.data.node.description = templates[node.data.type]["description"];
+          node.data.node.description = template["description"];
           node.data.node.template = updateTemplate(
-            templates[node.data.type]["template"] as unknown as APITemplateType,
+            template["template"] as unknown as APITemplateType,
             node.data.node.template as APITemplateType
           );
         }

From 8a049187df86a524aeb3cfd8260fb17042874986 Mon Sep 17 00:00:00 2001
From: Jim White <jim@ifcx.org>
Date: Thu, 1 Jun 2023 22:55:02 -0700
Subject: [PATCH 26/59] Add Anthropic to LLMs including an icon.

---
 poetry.lock                                   | 369 ++++--------------
 pyproject.toml                                |   1 +
 src/backend/langflow/config.yaml              |   1 +
 .../src/icons/Anthropic/anthropic.svg         |  19 +
 .../src/icons/Anthropic/anthropic_box.svg     |  11 +
 src/frontend/src/icons/Anthropic/index.tsx    |   9 +
 src/frontend/src/utils.ts                     |   2 +
 7 files changed, 121 insertions(+), 291 deletions(-)
 create mode 100644 src/frontend/src/icons/Anthropic/anthropic.svg
 create mode 100644 src/frontend/src/icons/Anthropic/anthropic_box.svg
 create mode 100644 src/frontend/src/icons/Anthropic/index.tsx

diff --git a/poetry.lock b/poetry.lock
index 5e53d5282..fde4fcc7d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1,10 +1,9 @@
-# This file is automatically @generated by Poetry 1.4.2 and should not be changed by hand.
+# This file is automatically @generated by Poetry 1.5.1 and should not be changed by hand.
 
 [[package]]
 name = "aiofiles"
 version = "23.1.0"
 description = "File support for asyncio."
-category = "main"
 optional = false
 python-versions = ">=3.7,<4.0"
 files = [
@@ -16,7 +15,6 @@ files = [
 name = "aiohttp"
 version = "3.8.4"
 description = "Async http client/server framework (asyncio)"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -125,7 +123,6 @@ speedups = ["Brotli", "aiodns", "cchardet"]
 name = "aiosignal"
 version = "1.3.1"
 description = "aiosignal: a list of registered asynchronous callbacks"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -140,7 +137,6 @@ frozenlist = ">=1.1.0"
 name = "aiostream"
 version = "0.4.5"
 description = "Generator-based operators for asynchronous iteration"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -148,11 +144,30 @@ files = [
     {file = "aiostream-0.4.5.tar.gz", hash = "sha256:3ecbf87085230fbcd9605c32ca20c4fb41af02c71d076eab246ea22e35947d88"},
 ]
 
+[[package]]
+name = "anthropic"
+version = "0.2.9"
+description = "Library for accessing the anthropic API"
+optional = false
+python-versions = ">=3.8"
+files = [
+    {file = "anthropic-0.2.9-py3-none-any.whl", hash = "sha256:e7cce215cf6c446de29280deb31b07b5587993d48e84850eaad3fc69bd1fec0a"},
+    {file = "anthropic-0.2.9.tar.gz", hash = "sha256:2d44564d362cced6e8e662366e4de7f94dcdc6cb61346a5e528359b0afc1f2f3"},
+]
+
+[package.dependencies]
+aiohttp = "*"
+httpx = "*"
+requests = "*"
+tokenizers = "*"
+
+[package.extras]
+dev = ["black (>=22.3.0)", "pytest"]
+
 [[package]]
 name = "anyio"
 version = "3.7.0"
 description = "High level compatibility layer for multiple asynchronous event loop implementations"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -174,7 +189,6 @@ trio = ["trio (<0.22)"]
 name = "appnope"
 version = "0.1.3"
 description = "Disable App Nap on macOS >= 10.9"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -186,7 +200,6 @@ files = [
 name = "argilla"
 version = "1.3.2"
 description = "Open-source tool for exploring, labeling, and monitoring data for NLP projects."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -214,7 +227,6 @@ server = ["Deprecated (>=1.2.0,<1.3.0)", "PyYAML (>=5.4.1,<6.1.0)", "aiofiles (>
 name = "asgiref"
 version = "3.7.2"
 description = "ASGI specs, helper code, and adapters"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -232,7 +244,6 @@ tests = ["mypy (>=0.800)", "pytest", "pytest-asyncio"]
 name = "asttokens"
 version = "2.2.1"
 description = "Annotate AST trees with source code positions"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -250,7 +261,6 @@ test = ["astroid", "pytest"]
 name = "async-timeout"
 version = "4.0.2"
 description = "Timeout context manager for asyncio programs"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -262,7 +272,6 @@ files = [
 name = "attrs"
 version = "23.1.0"
 description = "Classes Without Boilerplate"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -281,7 +290,6 @@ tests-no-zope = ["cloudpickle", "hypothesis", "mypy (>=1.1.1)", "pympler", "pyte
 name = "authlib"
 version = "1.2.0"
 description = "The ultimate Python library in building OAuth and OpenID Connect servers and clients."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -296,7 +304,6 @@ cryptography = ">=3.2"
 name = "backcall"
 version = "0.2.0"
 description = "Specifications for callback functions passed in to an API"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -308,7 +315,6 @@ files = [
 name = "backoff"
 version = "2.2.1"
 description = "Function decoration for backoff and retry"
-category = "main"
 optional = false
 python-versions = ">=3.7,<4.0"
 files = [
@@ -320,7 +326,6 @@ files = [
 name = "beautifulsoup4"
 version = "4.12.2"
 description = "Screen-scraping library"
-category = "main"
 optional = false
 python-versions = ">=3.6.0"
 files = [
@@ -339,7 +344,6 @@ lxml = ["lxml"]
 name = "black"
 version = "23.3.0"
 description = "The uncompromising code formatter."
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -389,7 +393,6 @@ uvloop = ["uvloop (>=0.15.2)"]
 name = "cachetools"
 version = "5.3.1"
 description = "Extensible memoizing collections and decorators"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -401,7 +404,6 @@ files = [
 name = "certifi"
 version = "2023.5.7"
 description = "Python package for providing Mozilla's CA Bundle."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -413,7 +415,6 @@ files = [
 name = "cffi"
 version = "1.15.1"
 description = "Foreign Function Interface for Python calling C code."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -490,7 +491,6 @@ pycparser = "*"
 name = "chardet"
 version = "5.1.0"
 description = "Universal encoding detector for Python 3"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -502,7 +502,6 @@ files = [
 name = "charset-normalizer"
 version = "3.1.0"
 description = "The Real First Universal Charset Detector. Open, modern and actively maintained alternative to Chardet."
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -587,7 +586,6 @@ files = [
 name = "chromadb"
 version = "0.3.25"
 description = "Chroma."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -616,7 +614,6 @@ uvicorn = {version = ">=0.18.3", extras = ["standard"]}
 name = "click"
 version = "8.1.3"
 description = "Composable command line interface toolkit"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -631,7 +628,6 @@ colorama = {version = "*", markers = "platform_system == \"Windows\""}
 name = "clickhouse-connect"
 version = "0.5.25"
 description = "ClickHouse core driver, SqlAlchemy, and Superset libraries"
-category = "main"
 optional = false
 python-versions = "~=3.7"
 files = [
@@ -719,14 +715,13 @@ superset = ["apache-superset (>=1.4.1)"]
 
 [[package]]
 name = "cohere"
-version = "4.6.0"
+version = "4.6.1"
 description = ""
-category = "main"
 optional = false
 python-versions = ">=3.7,<4.0"
 files = [
-    {file = "cohere-4.6.0-py3-none-any.whl", hash = "sha256:fc60fa73a2d96bdb9f70da4a290d3ede320b74ac01a24c229011049d7cb3511f"},
-    {file = "cohere-4.6.0.tar.gz", hash = "sha256:43218a0a40f6fc023e068732994fb631ce5d160a0bc9f9a3a22524b5932f34ea"},
+    {file = "cohere-4.6.1-py3-none-any.whl", hash = "sha256:bee44741e91b22f49a36c9c4f014e2a26eb752251fb860176a0462c6a1ae4e79"},
+    {file = "cohere-4.6.1.tar.gz", hash = "sha256:69f0cf0a6202f89de04358028e68b33aa7f8e11ea64fb4f5a28904632d2aa672"},
 ]
 
 [package.dependencies]
@@ -738,7 +733,6 @@ requests = ">=2.0,<3.0"
 name = "colorama"
 version = "0.4.6"
 description = "Cross-platform colored terminal text."
-category = "main"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,!=3.6.*,>=2.7"
 files = [
@@ -750,7 +744,6 @@ files = [
 name = "coloredlogs"
 version = "15.0.1"
 description = "Colored terminal output for Python's logging module"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
 files = [
@@ -768,7 +761,6 @@ cron = ["capturer (>=2.4)"]
 name = "comm"
 version = "0.1.3"
 description = "Jupyter Python Comm implementation, for usage in ipykernel, xeus-python etc."
-category = "dev"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -788,7 +780,6 @@ typing = ["mypy (>=0.990)"]
 name = "coverage"
 version = "7.2.7"
 description = "Code coverage measurement for Python"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -862,31 +853,30 @@ toml = ["tomli"]
 
 [[package]]
 name = "cryptography"
-version = "40.0.2"
+version = "41.0.1"
 description = "cryptography is a package which provides cryptographic recipes and primitives to Python developers."
-category = "main"
 optional = false
-python-versions = ">=3.6"
+python-versions = ">=3.7"
 files = [
-    {file = "cryptography-40.0.2-cp36-abi3-macosx_10_12_universal2.whl", hash = "sha256:8f79b5ff5ad9d3218afb1e7e20ea74da5f76943ee5edb7f76e56ec5161ec782b"},
-    {file = "cryptography-40.0.2-cp36-abi3-macosx_10_12_x86_64.whl", hash = "sha256:05dc219433b14046c476f6f09d7636b92a1c3e5808b9a6536adf4932b3b2c440"},
-    {file = "cryptography-40.0.2-cp36-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:4df2af28d7bedc84fe45bd49bc35d710aede676e2a4cb7fc6d103a2adc8afe4d"},
-    {file = "cryptography-40.0.2-cp36-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:0dcca15d3a19a66e63662dc8d30f8036b07be851a8680eda92d079868f106288"},
-    {file = "cryptography-40.0.2-cp36-abi3-manylinux_2_28_aarch64.whl", hash = "sha256:a04386fb7bc85fab9cd51b6308633a3c271e3d0d3eae917eebab2fac6219b6d2"},
-    {file = "cryptography-40.0.2-cp36-abi3-manylinux_2_28_x86_64.whl", hash = "sha256:adc0d980fd2760c9e5de537c28935cc32b9353baaf28e0814df417619c6c8c3b"},
-    {file = "cryptography-40.0.2-cp36-abi3-musllinux_1_1_aarch64.whl", hash = "sha256:d5a1bd0e9e2031465761dfa920c16b0065ad77321d8a8c1f5ee331021fda65e9"},
-    {file = "cryptography-40.0.2-cp36-abi3-musllinux_1_1_x86_64.whl", hash = "sha256:a95f4802d49faa6a674242e25bfeea6fc2acd915b5e5e29ac90a32b1139cae1c"},
-    {file = "cryptography-40.0.2-cp36-abi3-win32.whl", hash = "sha256:aecbb1592b0188e030cb01f82d12556cf72e218280f621deed7d806afd2113f9"},
-    {file = "cryptography-40.0.2-cp36-abi3-win_amd64.whl", hash = "sha256:b12794f01d4cacfbd3177b9042198f3af1c856eedd0a98f10f141385c809a14b"},
-    {file = "cryptography-40.0.2-pp38-pypy38_pp73-macosx_10_12_x86_64.whl", hash = "sha256:142bae539ef28a1c76794cca7f49729e7c54423f615cfd9b0b1fa90ebe53244b"},
-    {file = "cryptography-40.0.2-pp38-pypy38_pp73-manylinux_2_28_aarch64.whl", hash = "sha256:956ba8701b4ffe91ba59665ed170a2ebbdc6fc0e40de5f6059195d9f2b33ca0e"},
-    {file = "cryptography-40.0.2-pp38-pypy38_pp73-manylinux_2_28_x86_64.whl", hash = "sha256:4f01c9863da784558165f5d4d916093737a75203a5c5286fde60e503e4276c7a"},
-    {file = "cryptography-40.0.2-pp38-pypy38_pp73-win_amd64.whl", hash = "sha256:3daf9b114213f8ba460b829a02896789751626a2a4e7a43a28ee77c04b5e4958"},
-    {file = "cryptography-40.0.2-pp39-pypy39_pp73-macosx_10_12_x86_64.whl", hash = "sha256:48f388d0d153350f378c7f7b41497a54ff1513c816bcbbcafe5b829e59b9ce5b"},
-    {file = "cryptography-40.0.2-pp39-pypy39_pp73-manylinux_2_28_aarch64.whl", hash = "sha256:c0764e72b36a3dc065c155e5b22f93df465da9c39af65516fe04ed3c68c92636"},
-    {file = "cryptography-40.0.2-pp39-pypy39_pp73-manylinux_2_28_x86_64.whl", hash = "sha256:cbaba590180cba88cb99a5f76f90808a624f18b169b90a4abb40c1fd8c19420e"},
-    {file = "cryptography-40.0.2-pp39-pypy39_pp73-win_amd64.whl", hash = "sha256:7a38250f433cd41df7fcb763caa3ee9362777fdb4dc642b9a349721d2bf47404"},
-    {file = "cryptography-40.0.2.tar.gz", hash = "sha256:c33c0d32b8594fa647d2e01dbccc303478e16fdd7cf98652d5b3ed11aa5e5c99"},
+    {file = "cryptography-41.0.1-cp37-abi3-macosx_10_12_universal2.whl", hash = "sha256:f73bff05db2a3e5974a6fd248af2566134d8981fd7ab012e5dd4ddb1d9a70699"},
+    {file = "cryptography-41.0.1-cp37-abi3-macosx_10_12_x86_64.whl", hash = "sha256:1a5472d40c8f8e91ff7a3d8ac6dfa363d8e3138b961529c996f3e2df0c7a411a"},
+    {file = "cryptography-41.0.1-cp37-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:7fa01527046ca5facdf973eef2535a27fec4cb651e4daec4d043ef63f6ecd4ca"},
+    {file = "cryptography-41.0.1-cp37-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:b46e37db3cc267b4dea1f56da7346c9727e1209aa98487179ee8ebed09d21e43"},
+    {file = "cryptography-41.0.1-cp37-abi3-manylinux_2_28_aarch64.whl", hash = "sha256:d198820aba55660b4d74f7b5fd1f17db3aa5eb3e6893b0a41b75e84e4f9e0e4b"},
+    {file = "cryptography-41.0.1-cp37-abi3-manylinux_2_28_x86_64.whl", hash = "sha256:948224d76c4b6457349d47c0c98657557f429b4e93057cf5a2f71d603e2fc3a3"},
+    {file = "cryptography-41.0.1-cp37-abi3-musllinux_1_1_aarch64.whl", hash = "sha256:059e348f9a3c1950937e1b5d7ba1f8e968508ab181e75fc32b879452f08356db"},
+    {file = "cryptography-41.0.1-cp37-abi3-musllinux_1_1_x86_64.whl", hash = "sha256:b4ceb5324b998ce2003bc17d519080b4ec8d5b7b70794cbd2836101406a9be31"},
+    {file = "cryptography-41.0.1-cp37-abi3-win32.whl", hash = "sha256:8f4ab7021127a9b4323537300a2acfb450124b2def3756f64dc3a3d2160ee4b5"},
+    {file = "cryptography-41.0.1-cp37-abi3-win_amd64.whl", hash = "sha256:1fee5aacc7367487b4e22484d3c7e547992ed726d14864ee33c0176ae43b0d7c"},
+    {file = "cryptography-41.0.1-pp38-pypy38_pp73-macosx_10_12_x86_64.whl", hash = "sha256:9a6c7a3c87d595608a39980ebaa04d5a37f94024c9f24eb7d10262b92f739ddb"},
+    {file = "cryptography-41.0.1-pp38-pypy38_pp73-manylinux_2_28_aarch64.whl", hash = "sha256:5d092fdfedaec4cbbffbf98cddc915ba145313a6fdaab83c6e67f4e6c218e6f3"},
+    {file = "cryptography-41.0.1-pp38-pypy38_pp73-manylinux_2_28_x86_64.whl", hash = "sha256:1a8e6c2de6fbbcc5e14fd27fb24414507cb3333198ea9ab1258d916f00bc3039"},
+    {file = "cryptography-41.0.1-pp38-pypy38_pp73-win_amd64.whl", hash = "sha256:cb33ccf15e89f7ed89b235cff9d49e2e62c6c981a6061c9c8bb47ed7951190bc"},
+    {file = "cryptography-41.0.1-pp39-pypy39_pp73-macosx_10_12_x86_64.whl", hash = "sha256:5f0ff6e18d13a3de56f609dd1fd11470918f770c6bd5d00d632076c727d35485"},
+    {file = "cryptography-41.0.1-pp39-pypy39_pp73-manylinux_2_28_aarch64.whl", hash = "sha256:7bfc55a5eae8b86a287747053140ba221afc65eb06207bedf6e019b8934b477c"},
+    {file = "cryptography-41.0.1-pp39-pypy39_pp73-manylinux_2_28_x86_64.whl", hash = "sha256:eb8163f5e549a22888c18b0d53d6bb62a20510060a22fd5a995ec8a05268df8a"},
+    {file = "cryptography-41.0.1-pp39-pypy39_pp73-win_amd64.whl", hash = "sha256:8dde71c4169ec5ccc1087bb7521d54251c016f126f922ab2dfe6649170a3b8c5"},
+    {file = "cryptography-41.0.1.tar.gz", hash = "sha256:d34579085401d3f49762d2f7d6634d6b6c2ae1242202e860f4d26b046e3a1006"},
 ]
 
 [package.dependencies]
@@ -895,23 +885,22 @@ cffi = ">=1.12"
 [package.extras]
 docs = ["sphinx (>=5.3.0)", "sphinx-rtd-theme (>=1.1.1)"]
 docstest = ["pyenchant (>=1.6.11)", "sphinxcontrib-spelling (>=4.0.1)", "twine (>=1.12.0)"]
-pep8test = ["black", "check-manifest", "mypy", "ruff"]
-sdist = ["setuptools-rust (>=0.11.4)"]
+nox = ["nox"]
+pep8test = ["black", "check-sdist", "mypy", "ruff"]
+sdist = ["build"]
 ssh = ["bcrypt (>=3.1.5)"]
-test = ["iso8601", "pretend", "pytest (>=6.2.0)", "pytest-benchmark", "pytest-cov", "pytest-shard (>=0.1.2)", "pytest-subtests", "pytest-xdist"]
+test = ["pretend", "pytest (>=6.2.0)", "pytest-benchmark", "pytest-cov", "pytest-xdist"]
 test-randomorder = ["pytest-randomly"]
-tox = ["tox"]
 
 [[package]]
 name = "ctransformers"
-version = "0.2.2"
+version = "0.2.4"
 description = "Python bindings for the Transformer models implemented in C/C++ using GGML library."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
-    {file = "ctransformers-0.2.2-py3-none-any.whl", hash = "sha256:bf682dd0293dd87911c9a9a1169a4873ff55baebc16d465c6029c77f11b18cf6"},
-    {file = "ctransformers-0.2.2.tar.gz", hash = "sha256:1fc36b3fde36d9fd3cb69e48993315bb1f5f54ae552720eb909dc4b3a131c743"},
+    {file = "ctransformers-0.2.4-py3-none-any.whl", hash = "sha256:0476764b50863ab5032f75c5fefbfb8bebe19e5aadfa103bc9e0706edfc530be"},
+    {file = "ctransformers-0.2.4.tar.gz", hash = "sha256:e569260cee73b3ffb24aa1473a56539f75bfbca54a89c83844b3fbcb952f00a8"},
 ]
 
 [package.dependencies]
@@ -924,7 +913,6 @@ tests = ["pytest"]
 name = "dataclasses-json"
 version = "0.5.7"
 description = "Easily serialize dataclasses to and from JSON"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -944,7 +932,6 @@ dev = ["flake8", "hypothesis", "ipython", "mypy (>=0.710)", "portray", "pytest (
 name = "debugpy"
 version = "1.6.7"
 description = "An implementation of the Debug Adapter Protocol for Python"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -972,7 +959,6 @@ files = [
 name = "decorator"
 version = "5.1.1"
 description = "Decorators for Humans"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -984,7 +970,6 @@ files = [
 name = "deprecated"
 version = "1.2.14"
 description = "Python @deprecated decorator to deprecate old python classes, functions or methods."
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
 files = [
@@ -1002,7 +987,6 @@ dev = ["PyTest", "PyTest-Cov", "bump2version (<1)", "sphinx (<2)", "tox"]
 name = "dill"
 version = "0.3.6"
 description = "serialize all of python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1017,7 +1001,6 @@ graph = ["objgraph (>=1.7.2)"]
 name = "docarray"
 version = "0.21.0"
 description = "The data structure for unstructured data"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1044,14 +1027,13 @@ weaviate = ["weaviate-client (>=3.9.0,<3.10.0)"]
 
 [[package]]
 name = "docker"
-version = "6.1.2"
+version = "6.1.3"
 description = "A Python library for the Docker Engine API."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "docker-6.1.2-py3-none-any.whl", hash = "sha256:134cd828f84543cbf8e594ff81ca90c38288df3c0a559794c12f2e4b634ea19e"},
-    {file = "docker-6.1.2.tar.gz", hash = "sha256:dcc088adc2ec4e7cfc594e275d8bd2c9738c56c808de97476939ef67db5af8c2"},
+    {file = "docker-6.1.3-py3-none-any.whl", hash = "sha256:aecd2277b8bf8e506e484f6ab7aec39abe0038e29fa4a6d3ba86c3fe01844ed9"},
+    {file = "docker-6.1.3.tar.gz", hash = "sha256:aa6d17830045ba5ef0168d5eaa34d37beeb113948c413affe1d5991fc11f9a20"},
 ]
 
 [package.dependencies]
@@ -1068,7 +1050,6 @@ ssh = ["paramiko (>=2.4.3)"]
 name = "docstring-parser"
 version = "0.15"
 description = "Parse Python docstrings in reST, Google and Numpydoc format"
-category = "main"
 optional = false
 python-versions = ">=3.6,<4.0"
 files = [
@@ -1080,7 +1061,6 @@ files = [
 name = "duckdb"
 version = "0.8.0"
 description = "DuckDB embedded database"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1137,7 +1117,6 @@ files = [
 name = "ecdsa"
 version = "0.18.0"
 description = "ECDSA cryptographic signature library (pure python)"
-category = "main"
 optional = false
 python-versions = ">=2.6, !=3.0.*, !=3.1.*, !=3.2.*"
 files = [
@@ -1156,7 +1135,6 @@ gmpy2 = ["gmpy2"]
 name = "et-xmlfile"
 version = "1.1.0"
 description = "An implementation of lxml.xmlfile for the standard library"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -1168,7 +1146,6 @@ files = [
 name = "exceptiongroup"
 version = "1.1.1"
 description = "Backport of PEP 654 (exception groups)"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1183,7 +1160,6 @@ test = ["pytest (>=6)"]
 name = "executing"
 version = "1.2.0"
 description = "Get the currently executing AST node of a frame, and other information"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -1198,7 +1174,6 @@ tests = ["asttokens", "littleutils", "pytest", "rich"]
 name = "fake-useragent"
 version = "1.1.3"
 description = "Up-to-date simple useragent faker with real world database"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1213,7 +1188,6 @@ importlib-resources = {version = ">=5.0", markers = "python_version < \"3.10\""}
 name = "fastapi"
 version = "0.95.2"
 description = "FastAPI framework, high performance, easy to learn, fast to code, ready for production"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1235,7 +1209,6 @@ test = ["anyio[trio] (>=3.2.1,<4.0.0)", "black (==23.1.0)", "coverage[toml] (>=6
 name = "filelock"
 version = "3.12.0"
 description = "A platform independent file lock."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1251,7 +1224,6 @@ testing = ["covdefaults (>=2.3)", "coverage (>=7.2.3)", "diff-cover (>=7.5)", "p
 name = "flatbuffers"
 version = "23.5.26"
 description = "The FlatBuffers serialization format for Python"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1263,7 +1235,6 @@ files = [
 name = "frozenlist"
 version = "1.3.3"
 description = "A list-like structure which implements collections.abc.MutableSequence"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1347,7 +1318,6 @@ files = [
 name = "google-api-core"
 version = "2.11.0"
 description = "Google API client core library"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1370,7 +1340,6 @@ grpcio-gcp = ["grpcio-gcp (>=0.2.2,<1.0dev)"]
 name = "google-api-python-client"
 version = "2.88.0"
 description = "Google API Client Library for Python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1379,7 +1348,7 @@ files = [
 ]
 
 [package.dependencies]
-google-api-core = ">=1.31.5,<2.0.0 || >2.3.0,<3.0.0dev"
+google-api-core = ">=1.31.5,<2.0.dev0 || >2.3.0,<3.0.0dev"
 google-auth = ">=1.19.0,<3.0.0dev"
 google-auth-httplib2 = ">=0.1.0"
 httplib2 = ">=0.15.0,<1dev"
@@ -1387,14 +1356,13 @@ uritemplate = ">=3.0.1,<5"
 
 [[package]]
 name = "google-auth"
-version = "2.19.0"
+version = "2.19.1"
 description = "Google Authentication Library"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
-    {file = "google-auth-2.19.0.tar.gz", hash = "sha256:f39d528077ac540793dd3c22a8706178f157642a67d874db25c640b7fead277e"},
-    {file = "google_auth-2.19.0-py2.py3-none-any.whl", hash = "sha256:be617bfaf77774008e9d177573f782e109188c8a64ae6e744285df5cea3e7df6"},
+    {file = "google-auth-2.19.1.tar.gz", hash = "sha256:a9cfa88b3e16196845e64a3658eb953992129d13ac7337b064c6546f77c17183"},
+    {file = "google_auth-2.19.1-py2.py3-none-any.whl", hash = "sha256:ea165e014c7cbd496558796b627c271aa8c18b4cba79dc1cc962b24c5efdfb85"},
 ]
 
 [package.dependencies]
@@ -1415,7 +1383,6 @@ requests = ["requests (>=2.20.0,<3.0.0dev)"]
 name = "google-auth-httplib2"
 version = "0.1.0"
 description = "Google Authentication Library: httplib2 transport"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1432,7 +1399,6 @@ six = "*"
 name = "google-search-results"
 version = "2.4.2"
 description = "Scrape and search localized results from Google, Bing, Baidu, Yahoo, Yandex, Ebay, Homedepot, youtube at scale using SerpApi.com"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -1446,7 +1412,6 @@ requests = "*"
 name = "googleapis-common-protos"
 version = "1.59.0"
 description = "Common protobufs used in Google APIs"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1464,7 +1429,6 @@ grpc = ["grpcio (>=1.44.0,<2.0.0dev)"]
 name = "greenlet"
 version = "2.0.2"
 description = "Lightweight in-process concurrent programming"
-category = "main"
 optional = false
 python-versions = ">=2.7,!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*"
 files = [
@@ -1538,7 +1502,6 @@ test = ["objgraph", "psutil"]
 name = "grpcio"
 version = "1.47.5"
 description = "HTTP/2-based RPC framework"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -1600,7 +1563,6 @@ protobuf = ["grpcio-tools (>=1.47.5)"]
 name = "grpcio-health-checking"
 version = "1.47.5"
 description = "Standard Health Checking Service for gRPC"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -1616,7 +1578,6 @@ protobuf = ">=3.12.0"
 name = "grpcio-reflection"
 version = "1.47.5"
 description = "Standard Protobuf Reflection Service for gRPC"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -1632,7 +1593,6 @@ protobuf = ">=3.12.0"
 name = "grpcio-tools"
 version = "1.47.5"
 description = "Protobuf code generator for gRPC"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -1693,7 +1653,6 @@ setuptools = "*"
 name = "gunicorn"
 version = "20.1.0"
 description = "WSGI HTTP Server for UNIX"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -1714,7 +1673,6 @@ tornado = ["tornado (>=0.2)"]
 name = "h11"
 version = "0.14.0"
 description = "A pure-Python, bring-your-own-I/O implementation of HTTP/1.1"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1726,7 +1684,6 @@ files = [
 name = "h2"
 version = "4.1.0"
 description = "HTTP/2 State-Machine based protocol implementation"
-category = "main"
 optional = false
 python-versions = ">=3.6.1"
 files = [
@@ -1742,7 +1699,6 @@ hyperframe = ">=6.0,<7"
 name = "hnswlib"
 version = "0.7.0"
 description = "hnswlib"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -1756,7 +1712,6 @@ numpy = "*"
 name = "hpack"
 version = "4.0.0"
 description = "Pure-Python HPACK header compression"
-category = "main"
 optional = false
 python-versions = ">=3.6.1"
 files = [
@@ -1768,7 +1723,6 @@ files = [
 name = "httpcore"
 version = "0.16.3"
 description = "A minimal low-level HTTP client."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1780,17 +1734,16 @@ files = [
 anyio = ">=3.0,<5.0"
 certifi = "*"
 h11 = ">=0.13,<0.15"
-sniffio = ">=1.0.0,<2.0.0"
+sniffio = "==1.*"
 
 [package.extras]
 http2 = ["h2 (>=3,<5)"]
-socks = ["socksio (>=1.0.0,<2.0.0)"]
+socks = ["socksio (==1.*)"]
 
 [[package]]
 name = "httplib2"
 version = "0.22.0"
 description = "A comprehensive HTTP client library."
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
 files = [
@@ -1805,7 +1758,6 @@ pyparsing = {version = ">=2.4.2,<3.0.0 || >3.0.0,<3.0.1 || >3.0.1,<3.0.2 || >3.0
 name = "httptools"
 version = "0.5.0"
 description = "A collection of framework independent HTTP protocol utils."
-category = "main"
 optional = false
 python-versions = ">=3.5.0"
 files = [
@@ -1859,7 +1811,6 @@ test = ["Cython (>=0.29.24,<0.30.0)"]
 name = "httpx"
 version = "0.23.3"
 description = "The next generation HTTP client."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1876,15 +1827,14 @@ sniffio = "*"
 
 [package.extras]
 brotli = ["brotli", "brotlicffi"]
-cli = ["click (>=8.0.0,<9.0.0)", "pygments (>=2.0.0,<3.0.0)", "rich (>=10,<13)"]
+cli = ["click (==8.*)", "pygments (==2.*)", "rich (>=10,<13)"]
 http2 = ["h2 (>=3,<5)"]
-socks = ["socksio (>=1.0.0,<2.0.0)"]
+socks = ["socksio (==1.*)"]
 
 [[package]]
 name = "huggingface-hub"
 version = "0.13.4"
 description = "Client library to download and publish models, datasets and other repos on the huggingface.co hub"
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -1915,7 +1865,6 @@ typing = ["types-PyYAML", "types-requests", "types-simplejson", "types-toml", "t
 name = "humanfriendly"
 version = "10.0"
 description = "Human friendly output for text interfaces using Python"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
 files = [
@@ -1930,7 +1879,6 @@ pyreadline3 = {version = "*", markers = "sys_platform == \"win32\" and python_ve
 name = "hyperframe"
 version = "6.0.1"
 description = "HTTP/2 framing layer for Python"
-category = "main"
 optional = false
 python-versions = ">=3.6.1"
 files = [
@@ -1942,7 +1890,6 @@ files = [
 name = "idna"
 version = "3.4"
 description = "Internationalized Domain Names in Applications (IDNA)"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -1954,7 +1901,6 @@ files = [
 name = "importlib-metadata"
 version = "6.0.1"
 description = "Read metadata from Python packages"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1974,7 +1920,6 @@ testing = ["flake8 (<5)", "flufl.flake8", "importlib-resources (>=1.3)", "packag
 name = "importlib-resources"
 version = "5.12.0"
 description = "Read resources from Python packages"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -1993,7 +1938,6 @@ testing = ["flake8 (<5)", "pytest (>=6)", "pytest-black (>=0.3.7)", "pytest-chec
 name = "iniconfig"
 version = "2.0.0"
 description = "brain-dead simple config-ini parsing"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2005,7 +1949,6 @@ files = [
 name = "ipykernel"
 version = "6.23.1"
 description = "IPython Kernel for Jupyter"
-category = "dev"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -2019,7 +1962,7 @@ comm = ">=0.1.1"
 debugpy = ">=1.6.5"
 ipython = ">=7.23.1"
 jupyter-client = ">=6.1.12"
-jupyter-core = ">=4.12,<5.0.0 || >=5.1.0"
+jupyter-core = ">=4.12,<5.0.dev0 || >=5.1.dev0"
 matplotlib-inline = ">=0.1"
 nest-asyncio = "*"
 packaging = "*"
@@ -2039,7 +1982,6 @@ test = ["flaky", "ipyparallel", "pre-commit", "pytest (>=7.0)", "pytest-asyncio"
 name = "ipython"
 version = "8.13.2"
 description = "IPython: Productive Interactive Computing"
-category = "dev"
 optional = false
 python-versions = ">=3.9"
 files = [
@@ -2079,7 +2021,6 @@ test-extra = ["curio", "matplotlib (!=3.2.0)", "nbformat", "numpy (>=1.21)", "pa
 name = "jcloud"
 version = "0.2.10"
 description = "Simplify deploying and managing Jina projects on Jina Cloud"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -2102,7 +2043,6 @@ test = ["black (==22.3.0)", "jina (>=3.7.0)", "mock", "pytest", "pytest-asyncio"
 name = "jedi"
 version = "0.18.2"
 description = "An autocompletion tool for Python that can be used for text editors."
-category = "dev"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -2122,7 +2062,6 @@ testing = ["Django (<3.1)", "attrs", "colorama", "docopt", "pytest (<7.0.0)"]
 name = "jina"
 version = "3.15.2"
 description = "Build multimodal AI services via cloud native technologies · Neural Search · Generative AI · MLOps"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -2240,7 +2179,6 @@ websockets = ["websockets"]
 name = "jina-hubble-sdk"
 version = "0.38.0"
 description = "SDK for Hubble API at Jina AI."
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -2266,7 +2204,6 @@ full = ["aiohttp", "black (==22.3.0)", "docker", "filelock", "flake8 (==4.0.1)",
 name = "jinja2"
 version = "3.1.2"
 description = "A very fast and expressive template engine."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2284,7 +2221,6 @@ i18n = ["Babel (>=2.7)"]
 name = "joblib"
 version = "1.2.0"
 description = "Lightweight pipelining with Python functions"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2296,7 +2232,6 @@ files = [
 name = "jupyter-client"
 version = "8.2.0"
 description = "Jupyter protocol implementation and client libraries"
-category = "dev"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -2306,7 +2241,7 @@ files = [
 
 [package.dependencies]
 importlib-metadata = {version = ">=4.8.3", markers = "python_version < \"3.10\""}
-jupyter-core = ">=4.12,<5.0.0 || >=5.1.0"
+jupyter-core = ">=4.12,<5.0.dev0 || >=5.1.dev0"
 python-dateutil = ">=2.8.2"
 pyzmq = ">=23.0"
 tornado = ">=6.2"
@@ -2320,7 +2255,6 @@ test = ["coverage", "ipykernel (>=6.14)", "mypy", "paramiko", "pre-commit", "pyt
 name = "jupyter-core"
 version = "5.3.0"
 description = "Jupyter core package. A base package on which Jupyter projects rely."
-category = "dev"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -2341,7 +2275,6 @@ test = ["ipykernel", "pre-commit", "pytest", "pytest-cov", "pytest-timeout"]
 name = "langchain"
 version = "0.0.186"
 description = "Building applications with LLMs through composability"
-category = "main"
 optional = false
 python-versions = ">=3.8.1,<4.0"
 files = [
@@ -2376,13 +2309,12 @@ text-helpers = ["chardet (>=5.1.0,<6.0.0)"]
 
 [[package]]
 name = "langchain-serve"
-version = "0.0.40"
+version = "0.0.41"
 description = "Langchain Serve - serve your langchain apps on Jina AI Cloud."
-category = "main"
 optional = true
 python-versions = "*"
 files = [
-    {file = "langchain-serve-0.0.40.tar.gz", hash = "sha256:c60b173fcf0b682fbb70d34e8f485ce168e2229f55cb5c4ffbc26a5206af1c06"},
+    {file = "langchain-serve-0.0.41.tar.gz", hash = "sha256:fcf0d3ac9e48b5b24825ae2e1d8a383795c5744b18ef61cb74a99020a9e5d46a"},
 ]
 
 [package.dependencies]
@@ -2405,7 +2337,6 @@ test = ["psutil", "pytest", "pytest-asyncio"]
 name = "linkify-it-py"
 version = "2.0.2"
 description = "Links recognition library with FULL unicode support."
-category = "main"
 optional = true
 python-versions = ">=3.7"
 files = [
@@ -2426,7 +2357,6 @@ test = ["coverage", "pytest", "pytest-cov"]
 name = "llama-cpp-python"
 version = "0.1.55"
 description = "A Python wrapper for llama.cpp"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2440,7 +2370,6 @@ typing-extensions = ">=4.5.0,<5.0.0"
 name = "lxml"
 version = "4.9.2"
 description = "Powerful and Pythonic XML processing library combining libxml2/libxslt with the ElementTree API."
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, != 3.4.*"
 files = [
@@ -2533,7 +2462,6 @@ source = ["Cython (>=0.29.7)"]
 name = "lz4"
 version = "4.3.2"
 description = "LZ4 Bindings for Python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2583,7 +2511,6 @@ tests = ["psutil", "pytest (!=3.3.0)", "pytest-cov"]
 name = "markdown"
 version = "3.4.3"
 description = "Python implementation of John Gruber's Markdown."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2601,7 +2528,6 @@ testing = ["coverage", "pyyaml"]
 name = "markdown-it-py"
 version = "2.2.0"
 description = "Python port of markdown-it. Markdown parsing, done right!"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2628,7 +2554,6 @@ testing = ["coverage", "pytest", "pytest-cov", "pytest-regressions"]
 name = "markupsafe"
 version = "2.1.2"
 description = "Safely add untrusted strings to HTML/XML markup."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2688,7 +2613,6 @@ files = [
 name = "marshmallow"
 version = "3.19.0"
 description = "A lightweight library for converting complex datatypes to and from native Python datatypes."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2709,7 +2633,6 @@ tests = ["pytest", "pytz", "simplejson"]
 name = "marshmallow-enum"
 version = "1.5.1"
 description = "Enum field for Marshmallow"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -2724,7 +2647,6 @@ marshmallow = ">=2.0.0"
 name = "matplotlib-inline"
 version = "0.1.6"
 description = "Inline Matplotlib backend for Jupyter"
-category = "dev"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -2739,7 +2661,6 @@ traitlets = "*"
 name = "mdit-py-plugins"
 version = "0.3.5"
 description = "Collection of plugins for markdown-it-py"
-category = "main"
 optional = true
 python-versions = ">=3.7"
 files = [
@@ -2759,7 +2680,6 @@ testing = ["coverage", "pytest", "pytest-cov", "pytest-regressions"]
 name = "mdurl"
 version = "0.1.2"
 description = "Markdown URL utilities"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2771,7 +2691,6 @@ files = [
 name = "monotonic"
 version = "1.6"
 description = "An implementation of time.monotonic() for Python 2 & < 3.3"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -2783,7 +2702,6 @@ files = [
 name = "mpmath"
 version = "1.3.0"
 description = "Python library for arbitrary-precision floating-point arithmetic"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -2801,7 +2719,6 @@ tests = ["pytest (>=4.6)"]
 name = "msg-parser"
 version = "1.2.0"
 description = "This module enables reading, parsing and converting Microsoft Outlook MSG E-Mail files."
-category = "main"
 optional = false
 python-versions = ">=3.4"
 files = [
@@ -2819,7 +2736,6 @@ rtf = ["compressed-rtf (>=1.0.5)"]
 name = "multidict"
 version = "6.0.4"
 description = "multidict implementation"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2903,7 +2819,6 @@ files = [
 name = "mypy"
 version = "1.3.0"
 description = "Optional static typing for Python"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -2950,7 +2865,6 @@ reports = ["lxml"]
 name = "mypy-extensions"
 version = "1.0.0"
 description = "Type system extensions for programs checked with the mypy type checker."
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -2962,7 +2876,6 @@ files = [
 name = "nest-asyncio"
 version = "1.5.6"
 description = "Patch asyncio to allow nested event loops"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -2974,7 +2887,6 @@ files = [
 name = "networkx"
 version = "3.1"
 description = "Python package for creating and manipulating graphs and networks"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -2993,7 +2905,6 @@ test = ["codecov (>=2.1)", "pytest (>=7.2)", "pytest-cov (>=4.0)"]
 name = "nltk"
 version = "3.8.1"
 description = "Natural Language Toolkit"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3019,7 +2930,6 @@ twitter = ["twython"]
 name = "numexpr"
 version = "2.8.4"
 description = "Fast numerical expression evaluator for NumPy"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3062,7 +2972,6 @@ numpy = ">=1.13.3"
 name = "numpy"
 version = "1.23.5"
 description = "NumPy is the fundamental package for array computing with Python."
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -3100,7 +3009,6 @@ files = [
 name = "olefile"
 version = "0.46"
 description = "Python package to parse, read and write Microsoft OLE2 files (Structured Storage or Compound Document, Microsoft Office)"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
 files = [
@@ -3111,7 +3019,6 @@ files = [
 name = "onnxruntime"
 version = "1.15.0"
 description = "ONNX Runtime is a runtime accelerator for Machine Learning models"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -3153,7 +3060,6 @@ sympy = "*"
 name = "openai"
 version = "0.27.7"
 description = "Python client library for the OpenAI API"
-category = "main"
 optional = false
 python-versions = ">=3.7.1"
 files = [
@@ -3168,7 +3074,7 @@ tqdm = "*"
 
 [package.extras]
 datalib = ["numpy", "openpyxl (>=3.0.7)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1.0.11)"]
-dev = ["black (>=21.6b0,<22.0)", "pytest (>=6.0.0,<7.0.0)", "pytest-asyncio", "pytest-mock"]
+dev = ["black (>=21.6b0,<22.0)", "pytest (==6.*)", "pytest-asyncio", "pytest-mock"]
 embeddings = ["matplotlib", "numpy", "openpyxl (>=3.0.7)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1.0.11)", "plotly", "scikit-learn (>=1.0.2)", "scipy", "tenacity (>=8.0.1)"]
 wandb = ["numpy", "openpyxl (>=3.0.7)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1.0.11)", "wandb"]
 
@@ -3176,7 +3082,6 @@ wandb = ["numpy", "openpyxl (>=3.0.7)", "pandas (>=1.2.3)", "pandas-stubs (>=1.1
 name = "openapi-schema-pydantic"
 version = "1.2.4"
 description = "OpenAPI (v3) specification schema as pydantic class"
-category = "main"
 optional = false
 python-versions = ">=3.6.1"
 files = [
@@ -3191,7 +3096,6 @@ pydantic = ">=1.8.2"
 name = "openpyxl"
 version = "3.1.2"
 description = "A Python library to read/write Excel 2010 xlsx/xlsm files"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3206,7 +3110,6 @@ et-xmlfile = "*"
 name = "opentelemetry-api"
 version = "1.18.0"
 description = "OpenTelemetry Python API"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3223,7 +3126,6 @@ setuptools = ">=16.0"
 name = "opentelemetry-exporter-otlp"
 version = "1.18.0"
 description = "OpenTelemetry Collector Exporters"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3239,7 +3141,6 @@ opentelemetry-exporter-otlp-proto-http = "1.18.0"
 name = "opentelemetry-exporter-otlp-proto-common"
 version = "1.18.0"
 description = "OpenTelemetry Protobuf encoding"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3254,7 +3155,6 @@ opentelemetry-proto = "1.18.0"
 name = "opentelemetry-exporter-otlp-proto-grpc"
 version = "1.18.0"
 description = "OpenTelemetry Collector Protobuf over gRPC Exporter"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3279,7 +3179,6 @@ test = ["pytest-grpc"]
 name = "opentelemetry-exporter-otlp-proto-http"
 version = "1.18.0"
 description = "OpenTelemetry Collector Protobuf over HTTP Exporter"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3304,7 +3203,6 @@ test = ["responses (==0.22.0)"]
 name = "opentelemetry-exporter-prometheus"
 version = "1.12.0rc1"
 description = "Prometheus Metric Exporter for OpenTelemetry"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3321,7 +3219,6 @@ prometheus-client = ">=0.5.0,<1.0.0"
 name = "opentelemetry-instrumentation"
 version = "0.39b0"
 description = "Instrumentation Tools & Auto Instrumentation for OpenTelemetry Python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3338,7 +3235,6 @@ wrapt = ">=1.0.0,<2.0.0"
 name = "opentelemetry-instrumentation-aiohttp-client"
 version = "0.39b0"
 description = "OpenTelemetry aiohttp client instrumentation"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3361,7 +3257,6 @@ test = ["opentelemetry-instrumentation-aiohttp-client[instruments]"]
 name = "opentelemetry-instrumentation-asgi"
 version = "0.39b0"
 description = "ASGI instrumentation for OpenTelemetry"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3384,7 +3279,6 @@ test = ["opentelemetry-instrumentation-asgi[instruments]", "opentelemetry-test-u
 name = "opentelemetry-instrumentation-fastapi"
 version = "0.39b0"
 description = "OpenTelemetry FastAPI Instrumentation"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3407,7 +3301,6 @@ test = ["httpx (>=0.22,<1.0)", "opentelemetry-instrumentation-fastapi[instrument
 name = "opentelemetry-instrumentation-grpc"
 version = "0.39b0"
 description = "OpenTelemetry gRPC instrumentation"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3430,7 +3323,6 @@ test = ["opentelemetry-instrumentation-grpc[instruments]", "opentelemetry-sdk (>
 name = "opentelemetry-proto"
 version = "1.18.0"
 description = "OpenTelemetry Python Proto"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3445,7 +3337,6 @@ protobuf = ">=3.19,<5.0"
 name = "opentelemetry-sdk"
 version = "1.18.0"
 description = "OpenTelemetry Python SDK"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3463,7 +3354,6 @@ typing-extensions = ">=3.7.4"
 name = "opentelemetry-semantic-conventions"
 version = "0.39b0"
 description = "OpenTelemetry Semantic Conventions"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3475,7 +3365,6 @@ files = [
 name = "opentelemetry-util-http"
 version = "0.39b0"
 description = "Web util for OpenTelemetry"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3487,7 +3376,6 @@ files = [
 name = "overrides"
 version = "7.3.1"
 description = "A decorator to automatically detect mismatch when overriding a method."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3499,7 +3387,6 @@ files = [
 name = "packaging"
 version = "23.1"
 description = "Core utilities for Python packages"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3511,7 +3398,6 @@ files = [
 name = "pandas"
 version = "1.5.3"
 description = "Powerful data structures for data analysis, time series, and statistics"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -3560,7 +3446,6 @@ test = ["hypothesis (>=5.5.3)", "pytest (>=6.0)", "pytest-xdist (>=1.31)"]
 name = "pandas-stubs"
 version = "2.0.1.230501"
 description = "Type annotations for pandas"
-category = "dev"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -3575,7 +3460,6 @@ types-pytz = ">=2022.1.1"
 name = "parso"
 version = "0.8.3"
 description = "A Python Parser"
-category = "dev"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3591,7 +3475,6 @@ testing = ["docopt", "pytest (<6.0.0)"]
 name = "pathspec"
 version = "0.11.1"
 description = "Utility library for gitignore style pattern matching of file paths."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3603,7 +3486,6 @@ files = [
 name = "pexpect"
 version = "4.8.0"
 description = "Pexpect allows easy control of interactive console applications."
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -3618,7 +3500,6 @@ ptyprocess = ">=0.5"
 name = "pickleshare"
 version = "0.7.5"
 description = "Tiny 'shelve'-like database with concurrency support"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -3630,7 +3511,6 @@ files = [
 name = "pillow"
 version = "9.5.0"
 description = "Python Imaging Library (Fork)"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3710,7 +3590,6 @@ tests = ["check-manifest", "coverage", "defusedxml", "markdown2", "olefile", "pa
 name = "platformdirs"
 version = "3.5.1"
 description = "A small Python package for determining appropriate platform-specific dirs, e.g. a \"user data dir\"."
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3726,7 +3605,6 @@ test = ["appdirs (==1.4.4)", "covdefaults (>=2.3)", "pytest (>=7.3.1)", "pytest-
 name = "pluggy"
 version = "1.0.0"
 description = "plugin and hook calling mechanisms for python"
-category = "dev"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3742,7 +3620,6 @@ testing = ["pytest", "pytest-benchmark"]
 name = "portalocker"
 version = "2.7.0"
 description = "Wraps the portalocker recipe for easy usage"
-category = "main"
 optional = false
 python-versions = ">=3.5"
 files = [
@@ -3762,7 +3639,6 @@ tests = ["pytest (>=5.4.1)", "pytest-cov (>=2.8.1)", "pytest-mypy (>=0.8.0)", "p
 name = "posthog"
 version = "3.0.1"
 description = "Integrate PostHog into any python application."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -3786,7 +3662,6 @@ test = ["coverage", "flake8", "freezegun (==0.3.15)", "mock (>=2.0.0)", "pylint"
 name = "prometheus-client"
 version = "0.17.0"
 description = "Python client for the Prometheus monitoring system."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3801,7 +3676,6 @@ twisted = ["twisted"]
 name = "prompt-toolkit"
 version = "3.0.38"
 description = "Library for building powerful interactive command lines in Python"
-category = "dev"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -3816,7 +3690,6 @@ wcwidth = "*"
 name = "protobuf"
 version = "3.20.3"
 description = "Protocol Buffers"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -3848,7 +3721,6 @@ files = [
 name = "psutil"
 version = "5.9.5"
 description = "Cross-platform lib for process and system monitoring in Python."
-category = "dev"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
 files = [
@@ -3875,7 +3747,6 @@ test = ["enum34", "ipaddress", "mock", "pywin32", "wmi"]
 name = "psycopg2-binary"
 version = "2.9.6"
 description = "psycopg2 - Python-PostgreSQL Database Adapter"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -3947,7 +3818,6 @@ files = [
 name = "ptyprocess"
 version = "0.7.0"
 description = "Run a subprocess in a pseudo terminal"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -3959,7 +3829,6 @@ files = [
 name = "pure-eval"
 version = "0.2.2"
 description = "Safely evaluate AST nodes without side effects"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -3974,7 +3843,6 @@ tests = ["pytest"]
 name = "pyarrow"
 version = "11.0.0"
 description = "Python library for Apache Arrow"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4012,7 +3880,6 @@ numpy = ">=1.16.6"
 name = "pyasn1"
 version = "0.5.0"
 description = "Pure-Python implementation of ASN.1 types and DER/BER/CER codecs (X.208)"
-category = "main"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,>=2.7"
 files = [
@@ -4024,7 +3891,6 @@ files = [
 name = "pyasn1-modules"
 version = "0.3.0"
 description = "A collection of ASN.1-based protocols modules"
-category = "main"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,!=3.5.*,>=2.7"
 files = [
@@ -4039,7 +3905,6 @@ pyasn1 = ">=0.4.6,<0.6.0"
 name = "pycparser"
 version = "2.21"
 description = "C parser in Python"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*"
 files = [
@@ -4051,7 +3916,6 @@ files = [
 name = "pydantic"
 version = "1.10.8"
 description = "Data validation and settings management using python type hints"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4104,7 +3968,6 @@ email = ["email-validator (>=1.0.3)"]
 name = "pygments"
 version = "2.15.1"
 description = "Pygments is a syntax highlighting package written in Python."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4119,7 +3982,6 @@ plugins = ["importlib-metadata"]
 name = "pypandoc"
 version = "1.11"
 description = "Thin wrapper for pandoc."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -4131,7 +3993,6 @@ files = [
 name = "pyparsing"
 version = "3.0.9"
 description = "pyparsing module - Classes and methods to define and execute parsing grammars"
-category = "main"
 optional = false
 python-versions = ">=3.6.8"
 files = [
@@ -4146,7 +4007,6 @@ diagrams = ["jinja2", "railroad-diagrams"]
 name = "pypdf"
 version = "3.9.0"
 description = "A pure-python PDF library capable of splitting, merging, cropping, and transforming PDF files"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -4168,7 +4028,6 @@ image = ["Pillow"]
 name = "pyreadline3"
 version = "3.4.1"
 description = "A python implementation of GNU readline."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4180,7 +4039,6 @@ files = [
 name = "pysrt"
 version = "1.1.2"
 description = "SubRip (.srt) subtitle parser and writer"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4194,7 +4052,6 @@ chardet = "*"
 name = "pytest"
 version = "7.3.1"
 description = "pytest: simple powerful testing with Python"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4217,7 +4074,6 @@ testing = ["argcomplete", "attrs (>=19.2.0)", "hypothesis (>=3.56)", "mock", "no
 name = "pytest-cov"
 version = "4.1.0"
 description = "Pytest plugin for measuring coverage."
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4236,7 +4092,6 @@ testing = ["fields", "hunter", "process-tests", "pytest-xdist", "six", "virtuale
 name = "python-dateutil"
 version = "2.8.2"
 description = "Extensions to the standard Python datetime module"
-category = "main"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,>=2.7"
 files = [
@@ -4251,7 +4106,6 @@ six = ">=1.5"
 name = "python-docx"
 version = "0.8.11"
 description = "Create and update Microsoft Word .docx files."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4265,7 +4119,6 @@ lxml = ">=2.3.2"
 name = "python-dotenv"
 version = "1.0.0"
 description = "Read key-value pairs from a .env file and set them as environment variables"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -4280,7 +4133,6 @@ cli = ["click (>=5.0)"]
 name = "python-jose"
 version = "3.3.0"
 description = "JOSE implementation in Python"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4302,7 +4154,6 @@ pycryptodome = ["pyasn1", "pycryptodome (>=3.3.1,<4.0.0)"]
 name = "python-magic"
 version = "0.4.27"
 description = "File type identification using libmagic"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*"
 files = [
@@ -4314,7 +4165,6 @@ files = [
 name = "python-multipart"
 version = "0.0.6"
 description = "A streaming multipart parser for Python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4329,7 +4179,6 @@ dev = ["atomicwrites (==1.2.1)", "attrs (==19.2.0)", "coverage (==6.5.0)", "hatc
 name = "python-pptx"
 version = "0.6.21"
 description = "Generate and manipulate Open XML PowerPoint (.pptx) files"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4345,7 +4194,6 @@ XlsxWriter = ">=0.5.7"
 name = "pytz"
 version = "2023.3"
 description = "World timezone definitions, modern and historical"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4357,7 +4205,6 @@ files = [
 name = "pywin32"
 version = "306"
 description = "Python for Window Extensions"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4381,7 +4228,6 @@ files = [
 name = "pyyaml"
 version = "6.0"
 description = "YAML parser and emitter for Python"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -4431,7 +4277,6 @@ files = [
 name = "pyzmq"
 version = "25.1.0"
 description = "Python bindings for 0MQ"
-category = "dev"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -4521,7 +4366,6 @@ cffi = {version = "*", markers = "implementation_name == \"pypy\""}
 name = "qdrant-client"
 version = "1.2.0"
 description = "Client library for the Qdrant vector search engine"
-category = "main"
 optional = false
 python-versions = ">=3.7,<3.12"
 files = [
@@ -4543,7 +4387,6 @@ urllib3 = ">=1.26.14,<2.0.0"
 name = "regex"
 version = "2023.5.5"
 description = "Alternative regular expression module, to replace re."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -4641,7 +4484,6 @@ files = [
 name = "requests"
 version = "2.28.2"
 description = "Python HTTP for Humans."
-category = "main"
 optional = false
 python-versions = ">=3.7, <4"
 files = [
@@ -4663,7 +4505,6 @@ use-chardet-on-py3 = ["chardet (>=3.0.2,<6)"]
 name = "rfc3986"
 version = "1.5.0"
 description = "Validating URI References per RFC 3986"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4679,14 +4520,13 @@ idna2008 = ["idna"]
 
 [[package]]
 name = "rich"
-version = "13.3.5"
+version = "13.4.1"
 description = "Render rich text, tables, progress bars, syntax highlighting, markdown and more to the terminal"
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
-    {file = "rich-13.3.5-py3-none-any.whl", hash = "sha256:69cdf53799e63f38b95b9bf9c875f8c90e78dd62b2f00c13a911c7a3b9fa4704"},
-    {file = "rich-13.3.5.tar.gz", hash = "sha256:2d11b9b8dd03868f09b4fffadc84a6a8cda574e40dc90821bd845720ebb8e89c"},
+    {file = "rich-13.4.1-py3-none-any.whl", hash = "sha256:d204aadb50b936bf6b1a695385429d192bc1fdaf3e8b907e8e26f4c4e4b5bf75"},
+    {file = "rich-13.4.1.tar.gz", hash = "sha256:76f6b65ea7e5c5d924ba80e322231d7cb5b5981aa60bfc1e694f1bc097fe6fe1"},
 ]
 
 [package.dependencies]
@@ -4700,7 +4540,6 @@ jupyter = ["ipywidgets (>=7.5.1,<9)"]
 name = "rsa"
 version = "4.9"
 description = "Pure-Python RSA implementation"
-category = "main"
 optional = false
 python-versions = ">=3.6,<4"
 files = [
@@ -4715,7 +4554,6 @@ pyasn1 = ">=0.1.3"
 name = "ruff"
 version = "0.0.254"
 description = "An extremely fast Python linter, written in Rust."
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4742,7 +4580,6 @@ files = [
 name = "scikit-learn"
 version = "1.2.2"
 description = "A set of python modules for machine learning and data mining"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -4785,7 +4622,6 @@ tests = ["black (>=22.3.0)", "flake8 (>=3.8.2)", "matplotlib (>=3.1.3)", "mypy (
 name = "scipy"
 version = "1.10.1"
 description = "Fundamental algorithms for scientific computing in Python"
-category = "main"
 optional = false
 python-versions = "<3.12,>=3.8"
 files = [
@@ -4824,7 +4660,6 @@ test = ["asv", "gmpy2", "mpmath", "pooch", "pytest", "pytest-cov", "pytest-timeo
 name = "sentence-transformers"
 version = "2.2.2"
 description = "Multilingual text embeddings"
-category = "main"
 optional = false
 python-versions = ">=3.6.0"
 files = [
@@ -4847,7 +4682,6 @@ transformers = ">=4.6.0,<5.0.0"
 name = "sentencepiece"
 version = "0.1.99"
 description = "SentencePiece python wrapper"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -4902,7 +4736,6 @@ files = [
 name = "setuptools"
 version = "67.8.0"
 description = "Easily download, build, install, upgrade, and uninstall Python packages"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4919,7 +4752,6 @@ testing-integration = ["build[virtualenv]", "filelock (>=3.4.0)", "jaraco.envs (
 name = "six"
 version = "1.16.0"
 description = "Python 2 and 3 compatibility utilities"
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*"
 files = [
@@ -4931,7 +4763,6 @@ files = [
 name = "sniffio"
 version = "1.3.0"
 description = "Sniff out which async library your code is running under"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4943,7 +4774,6 @@ files = [
 name = "soupsieve"
 version = "2.4.1"
 description = "A modern CSS selector implementation for Beautiful Soup."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -4955,7 +4785,6 @@ files = [
 name = "sqlalchemy"
 version = "2.0.15"
 description = "Database Abstraction Library"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5003,7 +4832,7 @@ files = [
 ]
 
 [package.dependencies]
-greenlet = {version = "!=0.4.17", markers = "platform_machine == \"aarch64\" or platform_machine == \"ppc64le\" or platform_machine == \"x86_64\" or platform_machine == \"amd64\" or platform_machine == \"AMD64\" or platform_machine == \"win32\" or platform_machine == \"WIN32\""}
+greenlet = {version = "!=0.4.17", markers = "platform_machine == \"win32\" or platform_machine == \"WIN32\" or platform_machine == \"AMD64\" or platform_machine == \"amd64\" or platform_machine == \"x86_64\" or platform_machine == \"ppc64le\" or platform_machine == \"aarch64\""}
 typing-extensions = ">=4.2.0"
 
 [package.extras]
@@ -5033,7 +4862,6 @@ sqlcipher = ["sqlcipher3-binary"]
 name = "stack-data"
 version = "0.6.2"
 description = "Extract data from python stack frames and tracebacks for informative displays"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5053,7 +4881,6 @@ tests = ["cython", "littleutils", "pygments", "pytest", "typeguard"]
 name = "starlette"
 version = "0.27.0"
 description = "The little ASGI library that shines."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5072,7 +4899,6 @@ full = ["httpx (>=0.22.0)", "itsdangerous", "jinja2", "python-multipart", "pyyam
 name = "sympy"
 version = "1.12"
 description = "Computer algebra system (CAS) in Python"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -5087,7 +4913,6 @@ mpmath = ">=0.19"
 name = "tenacity"
 version = "8.2.2"
 description = "Retry code until it succeeds"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -5100,14 +4925,13 @@ doc = ["reno", "sphinx", "tornado (>=4.5)"]
 
 [[package]]
 name = "textual"
-version = "0.26.0"
+version = "0.27.0"
 description = "Modern Text User Interface framework"
-category = "main"
 optional = true
 python-versions = ">=3.7,<4.0"
 files = [
-    {file = "textual-0.26.0-py3-none-any.whl", hash = "sha256:1efd04e9f61b3e95fd1c65436d3262f99e3f86cdeb524d13045bb551eb615c02"},
-    {file = "textual-0.26.0.tar.gz", hash = "sha256:78094c83017d2836b726513abdf434cc034a0e68cc45e63b3b056c9b8b7fa673"},
+    {file = "textual-0.27.0-py3-none-any.whl", hash = "sha256:dc45eaf7da330686c56d6f76f59d05fd216ce6aad90fa44ee269881efc622151"},
+    {file = "textual-0.27.0.tar.gz", hash = "sha256:8bdcb09dc35a706ef939b1276ccfdec10eaaee6147b41cb7587cf33298a8dd33"},
 ]
 
 [package.dependencies]
@@ -5123,7 +4947,6 @@ dev = ["aiohttp (>=3.8.1)", "click (>=8.1.2)", "msgpack (>=1.0.3)"]
 name = "threadpoolctl"
 version = "3.1.0"
 description = "threadpoolctl"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -5135,7 +4958,6 @@ files = [
 name = "tiktoken"
 version = "0.3.3"
 description = "tiktoken is a fast BPE tokeniser for use with OpenAI's models"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -5181,7 +5003,6 @@ blobfile = ["blobfile (>=2)"]
 name = "tokenizers"
 version = "0.13.3"
 description = "Fast and Customizable Tokenizers"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -5236,7 +5057,6 @@ testing = ["black (==22.3)", "datasets", "numpy", "pytest", "requests"]
 name = "toml"
 version = "0.10.2"
 description = "Python Library for Tom's Obvious, Minimal Language"
-category = "main"
 optional = true
 python-versions = ">=2.6, !=3.0.*, !=3.1.*, !=3.2.*"
 files = [
@@ -5248,7 +5068,6 @@ files = [
 name = "tomli"
 version = "2.0.1"
 description = "A lil' TOML parser"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5260,7 +5079,6 @@ files = [
 name = "torch"
 version = "2.0.1"
 description = "Tensors and Dynamic neural networks in Python with strong GPU acceleration"
-category = "main"
 optional = false
 python-versions = ">=3.8.0"
 files = [
@@ -5300,7 +5118,6 @@ opt-einsum = ["opt-einsum (>=3.3)"]
 name = "torchvision"
 version = "0.15.2"
 description = "image and video datasets and models for torch deep learning"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -5328,7 +5145,7 @@ files = [
 
 [package.dependencies]
 numpy = "*"
-pillow = ">=5.3.0,<8.3.0 || >=8.4.0"
+pillow = ">=5.3.0,<8.3.dev0 || >=8.4.dev0"
 requests = "*"
 torch = "2.0.1"
 
@@ -5339,7 +5156,6 @@ scipy = ["scipy"]
 name = "tornado"
 version = "6.3.2"
 description = "Tornado is a Python web framework and asynchronous networking library, originally developed at FriendFeed."
-category = "dev"
 optional = false
 python-versions = ">= 3.8"
 files = [
@@ -5360,7 +5176,6 @@ files = [
 name = "tqdm"
 version = "4.65.0"
 description = "Fast, Extensible Progress Meter"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5381,7 +5196,6 @@ telegram = ["requests"]
 name = "traitlets"
 version = "5.9.0"
 description = "Traitlets Python configuration system"
-category = "dev"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5397,7 +5211,6 @@ test = ["argcomplete (>=2.0)", "pre-commit", "pytest", "pytest-mock"]
 name = "transformers"
 version = "4.29.0"
 description = "State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow"
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -5465,7 +5278,6 @@ vision = ["Pillow"]
 name = "typer"
 version = "0.7.0"
 description = "Typer, build great CLIs. Easy to code. Based on Python type hints."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -5486,7 +5298,6 @@ test = ["black (>=22.3.0,<23.0.0)", "coverage (>=6.2,<7.0)", "isort (>=5.0.6,<6.
 name = "types-pillow"
 version = "9.5.0.4"
 description = "Typing stubs for Pillow"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5498,7 +5309,6 @@ files = [
 name = "types-pytz"
 version = "2023.3.0.0"
 description = "Typing stubs for pytz"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5510,7 +5320,6 @@ files = [
 name = "types-pyyaml"
 version = "6.0.12.10"
 description = "Typing stubs for PyYAML"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -5522,7 +5331,6 @@ files = [
 name = "types-requests"
 version = "2.31.0.1"
 description = "Typing stubs for requests"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5537,7 +5345,6 @@ types-urllib3 = "*"
 name = "types-urllib3"
 version = "1.26.25.13"
 description = "Typing stubs for urllib3"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5549,7 +5356,6 @@ files = [
 name = "typing-extensions"
 version = "4.5.0"
 description = "Backported and Experimental Type Hints for Python 3.7+"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5561,7 +5367,6 @@ files = [
 name = "typing-inspect"
 version = "0.8.0"
 description = "Runtime inspection utilities for typing module."
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -5577,7 +5382,6 @@ typing-extensions = ">=3.7.4"
 name = "uc-micro-py"
 version = "1.0.2"
 description = "Micro subset of unicode data files for linkify-it-py projects."
-category = "main"
 optional = true
 python-versions = ">=3.7"
 files = [
@@ -5592,7 +5396,6 @@ test = ["coverage", "pytest", "pytest-cov"]
 name = "unstructured"
 version = "0.5.13"
 description = "A library that prepares raw documents for downstream ML tasks."
-category = "main"
 optional = false
 python-versions = ">=3.7.0"
 files = [
@@ -5632,7 +5435,6 @@ wikipedia = ["wikipedia"]
 name = "uritemplate"
 version = "4.1.1"
 description = "Implementation of RFC 6570 URI Templates"
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -5644,7 +5446,6 @@ files = [
 name = "urllib3"
 version = "1.26.16"
 description = "HTTP library with thread-safe connection pooling, file post, and more."
-category = "main"
 optional = false
 python-versions = ">=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*, !=3.4.*, !=3.5.*"
 files = [
@@ -5661,7 +5462,6 @@ socks = ["PySocks (>=1.5.6,!=1.5.7,<2.0)"]
 name = "uvicorn"
 version = "0.20.0"
 description = "The lightning-fast ASGI server."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5676,7 +5476,7 @@ h11 = ">=0.8"
 httptools = {version = ">=0.5.0", optional = true, markers = "extra == \"standard\""}
 python-dotenv = {version = ">=0.13", optional = true, markers = "extra == \"standard\""}
 pyyaml = {version = ">=5.1", optional = true, markers = "extra == \"standard\""}
-uvloop = {version = ">=0.14.0,<0.15.0 || >0.15.0,<0.15.1 || >0.15.1", optional = true, markers = "sys_platform != \"win32\" and sys_platform != \"cygwin\" and platform_python_implementation != \"PyPy\" and extra == \"standard\""}
+uvloop = {version = ">=0.14.0,<0.15.0 || >0.15.0,<0.15.1 || >0.15.1", optional = true, markers = "(sys_platform != \"win32\" and sys_platform != \"cygwin\") and platform_python_implementation != \"PyPy\" and extra == \"standard\""}
 watchfiles = {version = ">=0.13", optional = true, markers = "extra == \"standard\""}
 websockets = {version = ">=10.4", optional = true, markers = "extra == \"standard\""}
 
@@ -5687,7 +5487,6 @@ standard = ["colorama (>=0.4)", "httptools (>=0.5.0)", "python-dotenv (>=0.13)",
 name = "uvloop"
 version = "0.17.0"
 description = "Fast implementation of asyncio event loop on top of libuv"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5732,7 +5531,6 @@ test = ["Cython (>=0.29.32,<0.30.0)", "aiohttp", "flake8 (>=3.9.2,<3.10.0)", "my
 name = "validators"
 version = "0.20.0"
 description = "Python Data Validation for Humans™."
-category = "main"
 optional = false
 python-versions = ">=3.4"
 files = [
@@ -5749,7 +5547,6 @@ test = ["flake8 (>=2.4.0)", "isort (>=4.2.2)", "pytest (>=2.2.3)"]
 name = "watchfiles"
 version = "0.19.0"
 description = "Simple, modern and high performance file watching and code reload in python."
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5784,7 +5581,6 @@ anyio = ">=3.0.0"
 name = "wcwidth"
 version = "0.2.6"
 description = "Measures the displayed width of unicode strings in a terminal"
-category = "dev"
 optional = false
 python-versions = "*"
 files = [
@@ -5796,7 +5592,6 @@ files = [
 name = "weaviate-client"
 version = "3.19.2"
 description = "A python native weaviate client"
-category = "main"
 optional = false
 python-versions = ">=3.8"
 files = [
@@ -5817,7 +5612,6 @@ grpc = ["grpcio", "grpcio-tools"]
 name = "websocket-client"
 version = "1.5.2"
 description = "WebSocket client for Python with low level API options"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5834,7 +5628,6 @@ test = ["websockets"]
 name = "websockets"
 version = "11.0.3"
 description = "An implementation of the WebSocket Protocol (RFC 6455 & 7692)"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -5914,7 +5707,6 @@ files = [
 name = "wikipedia"
 version = "1.4.0"
 description = "Wikipedia API for Python"
-category = "main"
 optional = false
 python-versions = "*"
 files = [
@@ -5929,7 +5721,6 @@ requests = ">=2.0.0,<3.0.0"
 name = "wrapt"
 version = "1.14.1"
 description = "Module for decorators, wrappers and monkey patching."
-category = "main"
 optional = false
 python-versions = "!=3.0.*,!=3.1.*,!=3.2.*,!=3.3.*,!=3.4.*,>=2.7"
 files = [
@@ -6003,7 +5794,6 @@ files = [
 name = "xlsxwriter"
 version = "3.1.2"
 description = "A Python module for creating Excel XLSX files."
-category = "main"
 optional = false
 python-versions = ">=3.6"
 files = [
@@ -6015,7 +5805,6 @@ files = [
 name = "yarl"
 version = "1.9.2"
 description = "Yet another URL library"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -6103,7 +5892,6 @@ multidict = ">=4.0"
 name = "zipp"
 version = "3.15.0"
 description = "Backport of pathlib-compatible object wrapper for zip files"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -6119,7 +5907,6 @@ testing = ["big-O", "flake8 (<5)", "jaraco.functools", "jaraco.itertools", "more
 name = "zstandard"
 version = "0.21.0"
 description = "Zstandard bindings for Python"
-category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
@@ -6180,4 +5967,4 @@ deploy = ["langchain-serve"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.12"
-content-hash = "2bf357ad30f79c68751b34c991b4a73767ceb628657f4133228d4eb487d8a6fb"
+content-hash = "7c5d90d962ae4ee5d725de2b5986d09f5aba60d68b327e0f96653dbba113d10e"
diff --git a/pyproject.toml b/pyproject.toml
index 78104f327..15e6ea6ff 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -57,6 +57,7 @@ jina = "3.15.2"
 sentence-transformers = "^2.2.2"
 ctransformers = "^0.2.2"
 cohere = "^4.6.0"
+anthropic = "^0.2.9"
 
 
 [tool.poetry.group.dev.dependencies]
diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index 4060d1f3e..f2833ae26 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -55,6 +55,7 @@ llms:
   - LlamaCpp
   - CTransformers
   - Cohere
+  - Anthropic
 memories:
   - ConversationBufferMemory
   - ConversationSummaryMemory
diff --git a/src/frontend/src/icons/Anthropic/anthropic.svg b/src/frontend/src/icons/Anthropic/anthropic.svg
new file mode 100644
index 000000000..1cb8c0185
--- /dev/null
+++ b/src/frontend/src/icons/Anthropic/anthropic.svg
@@ -0,0 +1,19 @@
+<?xml version="1.0" encoding="utf-8"?>
+<svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="-85.949 86.888 308.4 138.858" width="308.4" height="138.858">
+  <defs>
+    <style type="text/css">
+	.st0{fill:#1F1F1E;}
+    </style>
+  </defs>
+<path class="st0" d="M19.9,260.5h21.9v52.6h14v-52.6h21.9v-12.6H19.9V260.5z M-4.7,293.5L-34,247.9h-15.8v65.1h13.5v-45.6L-7,313.1
+	H8.8v-65.1H-4.7V293.5z M133.5,273.9h-30.7v-26h-14v65.1h14v-26.6h30.7v26.6h14v-65.1h-14V273.9z M-100.9,247.9l-26,65.1h14.5
+	l5.3-13.7h27.2l5.3,13.7h14.5l-26-65.1H-100.9z M-102.4,287.3l8.9-22.9l8.9,22.9H-102.4z M263.1,246.8c-18.8,0-32.1,14-32.1,33.8
+	c0,19.6,13.3,33.6,32.1,33.6c18.7,0,31.9-14,31.9-33.6C295,260.8,281.8,246.8,263.1,246.8z M263.1,301.2c-11,0-17.7-7.8-17.7-20.6
+	c0-12.9,6.7-20.8,17.7-20.8c10.9,0,17.5,7.8,17.5,20.8C280.6,293.3,274,301.2,263.1,301.2z M458.2,291.2c-2.4,6.3-7.3,10-13.9,10
+	c-11,0-17.7-7.8-17.7-20.6c0-12.9,6.7-20.8,17.7-20.8c6.6,0,11.4,3.6,13.9,10H473c-3.6-14-14.5-23-28.7-23
+	c-18.8,0-32.1,14-32.1,33.8c0,19.6,13.3,33.6,32.1,33.6c14.2,0,25.1-9.1,28.8-23H458.2z M369.3,247.9l26,65.1h14.2l-26-65.1H369.3z
+	 M339.8,247.9H308v65.1h14v-23.6h17.9c14.8,0,23.8-7.8,23.8-20.8C363.7,255.7,354.6,247.9,339.8,247.9z M339.2,276.9H322v-16.4h17.2
+	c6.9,0,10.5,2.8,10.5,8.2C349.7,274.1,346.1,276.9,339.2,276.9z M220.3,267.8c0-12.3-9-19.8-23.8-19.8h-31.8v65.1h14v-25.5h15.5
+	l14,25.5h15.4l-15.5-27.4C215.8,282.6,220.3,276.4,220.3,267.8z M178.6,260.5h17.2c6.9,0,10.5,2.5,10.5,7.3c0,4.7-3.6,7.3-10.5,7.3
+	h-17.2V260.5z"/>
+</svg>
\ No newline at end of file
diff --git a/src/frontend/src/icons/Anthropic/anthropic_box.svg b/src/frontend/src/icons/Anthropic/anthropic_box.svg
new file mode 100644
index 000000000..fa9923ed7
--- /dev/null
+++ b/src/frontend/src/icons/Anthropic/anthropic_box.svg
@@ -0,0 +1,11 @@
+<?xml version="1.0" encoding="utf-8"?>
+<svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="-126.9 247.9 207.161 212.728" width="207.161" height="212.728">
+  <defs>
+    <style type="text/css">
+	.st0{fill:#1F1F1E;}
+</style>
+  </defs>
+  <path class="st0" d="M 19.9 260.5 L 41.8 260.5 L 41.8 313.1 L 55.8 313.1 L 55.8 260.5 L 77.7 260.5 L 77.7 247.9 L 19.9 247.9 L 19.9 260.5 Z M -4.7 293.5 L -34 247.9 L -49.8 247.9 L -49.8 313 L -36.3 313 L -36.3 267.4 L -7 313.1 L 8.8 313.1 L 8.8 248 L -4.7 248 L -4.7 293.5 Z M -100.9 247.9 L -126.9 313 L -112.4 313 L -107.1 299.3 L -79.9 299.3 L -74.6 313 L -60.1 313 L -86.1 247.9 L -100.9 247.9 Z M -102.4 287.3 L -93.5 264.4 L -84.6 287.3 L -102.4 287.3 Z"></path>
+  <path class="st0" d="M 38.246 437.628 C 35.846 443.928 30.946 447.628 24.346 447.628 C 13.346 447.628 6.646 439.828 6.646 427.028 C 6.646 414.128 13.346 406.228 24.346 406.228 C 30.946 406.228 35.746 409.828 38.246 416.228 L 53.046 416.228 C 49.446 402.228 38.546 393.228 24.346 393.228 C 5.546 393.228 -7.754 407.228 -7.754 427.028 C -7.754 446.628 5.546 460.628 24.346 460.628 C 38.546 460.628 49.446 451.528 53.146 437.628 L 38.246 437.628 Z M -50.654 394.328 L -24.654 459.428 L -10.454 459.428 L -36.454 394.328 L -50.654 394.328 Z M -80.154 394.328 L -111.954 394.328 L -111.954 459.428 L -97.954 459.428 L -97.954 435.828 L -80.054 435.828 C -65.254 435.828 -56.254 428.028 -56.254 415.028 C -56.254 402.128 -65.354 394.328 -80.154 394.328 Z M -80.754 423.328 L -97.954 423.328 L -97.954 406.928 L -80.754 406.928 C -73.854 406.928 -70.254 409.728 -70.254 415.128 C -70.254 420.528 -73.854 423.328 -80.754 423.328 Z"></path>
+  <path class="st0" d="M -81.239 347.704 L -111.939 347.704 L -111.939 321.704 L -125.939 321.704 L -125.939 386.804 L -111.939 386.804 L -111.939 360.204 L -81.239 360.204 L -81.239 386.804 L -67.239 386.804 L -67.239 321.704 L -81.239 321.704 L -81.239 347.704 Z M 48.361 320.604 C 29.561 320.604 16.261 334.604 16.261 354.404 C 16.261 374.004 29.561 388.004 48.361 388.004 C 67.061 388.004 80.261 374.004 80.261 354.404 C 80.261 334.604 67.061 320.604 48.361 320.604 Z M 48.361 375.004 C 37.361 375.004 30.661 367.204 30.661 354.404 C 30.661 341.504 37.361 333.604 48.361 333.604 C 59.261 333.604 65.861 341.404 65.861 354.404 C 65.861 367.104 59.261 375.004 48.361 375.004 Z M 5.561 341.604 C 5.561 329.304 -3.439 321.804 -18.239 321.804 L -50.039 321.804 L -50.039 386.904 L -36.039 386.904 L -36.039 361.404 L -20.539 361.404 L -6.539 386.904 L 8.861 386.904 L -6.639 359.504 C 1.061 356.404 5.561 350.204 5.561 341.604 Z M -36.139 334.304 L -18.939 334.304 C -12.039 334.304 -8.439 336.804 -8.439 341.604 C -8.439 346.304 -12.039 348.904 -18.939 348.904 L -36.139 348.904 L -36.139 334.304 Z"></path>
+</svg>
\ No newline at end of file
diff --git a/src/frontend/src/icons/Anthropic/index.tsx b/src/frontend/src/icons/Anthropic/index.tsx
new file mode 100644
index 000000000..4cdf8f910
--- /dev/null
+++ b/src/frontend/src/icons/Anthropic/index.tsx
@@ -0,0 +1,9 @@
+import React, { forwardRef } from "react";
+import { ReactComponent as AnthropicSVG } from "./anthropic_box.svg";
+
+export const AnthropicIcon = forwardRef<
+  SVGSVGElement,
+  React.PropsWithChildren<{}>
+>((props, ref) => {
+  return <AnthropicSVG ref={ref} {...props} />;
+});
diff --git a/src/frontend/src/utils.ts b/src/frontend/src/utils.ts
index 559080f3f..0e7289dd0 100644
--- a/src/frontend/src/utils.ts
+++ b/src/frontend/src/utils.ts
@@ -21,6 +21,7 @@ import { FlowType, NodeType } from "./types/flow";
 import { APITemplateType, TemplateVariableType } from "./types/api";
 import _ from "lodash";
 import { ChromaIcon } from "./icons/ChromaIcon";
+import { AnthropicIcon } from "./icons/Anthropic";
 import { AirbyteIcon } from "./icons/Airbyte";
 import { AzIcon } from "./icons/AzLogo";
 import { BingIcon } from "./icons/Bing";
@@ -153,6 +154,7 @@ export const nodeIcons: {
   AirbyteJSONLoader: AirbyteIcon,
   // SerpAPIWrapper: SerperIcon,
   // AZLyricsLoader: AzIcon,
+  Anthropic: AnthropicIcon,
   BingSearchAPIWrapper: BingIcon,
   BingSearchRun: BingIcon,
   Cohere: CohereIcon,

From 4f152bc7efe9174e6adb3280d45a1eab29fcb661 Mon Sep 17 00:00:00 2001
From: Jim White <jim@ifcx.org>
Date: Thu, 1 Jun 2023 23:28:44 -0700
Subject: [PATCH 27/59] Fix view for regular Anthropic logo

---
 src/frontend/src/icons/Anthropic/anthropic.svg | 14 ++------------
 1 file changed, 2 insertions(+), 12 deletions(-)

diff --git a/src/frontend/src/icons/Anthropic/anthropic.svg b/src/frontend/src/icons/Anthropic/anthropic.svg
index 1cb8c0185..67ae02ea5 100644
--- a/src/frontend/src/icons/Anthropic/anthropic.svg
+++ b/src/frontend/src/icons/Anthropic/anthropic.svg
@@ -1,19 +1,9 @@
 <?xml version="1.0" encoding="utf-8"?>
-<svg version="1.1" xmlns="http://www.w3.org/2000/svg" viewBox="-85.949 86.888 308.4 138.858" width="308.4" height="138.858">
+<svg version="1.1" xmlns="http://www.w3.org/2000/svg"  viewBox="-170.333 113.047 600 67.4" width="600" height="67.4">
   <defs>
     <style type="text/css">
 	.st0{fill:#1F1F1E;}
     </style>
   </defs>
-<path class="st0" d="M19.9,260.5h21.9v52.6h14v-52.6h21.9v-12.6H19.9V260.5z M-4.7,293.5L-34,247.9h-15.8v65.1h13.5v-45.6L-7,313.1
-	H8.8v-65.1H-4.7V293.5z M133.5,273.9h-30.7v-26h-14v65.1h14v-26.6h30.7v26.6h14v-65.1h-14V273.9z M-100.9,247.9l-26,65.1h14.5
-	l5.3-13.7h27.2l5.3,13.7h14.5l-26-65.1H-100.9z M-102.4,287.3l8.9-22.9l8.9,22.9H-102.4z M263.1,246.8c-18.8,0-32.1,14-32.1,33.8
-	c0,19.6,13.3,33.6,32.1,33.6c18.7,0,31.9-14,31.9-33.6C295,260.8,281.8,246.8,263.1,246.8z M263.1,301.2c-11,0-17.7-7.8-17.7-20.6
-	c0-12.9,6.7-20.8,17.7-20.8c10.9,0,17.5,7.8,17.5,20.8C280.6,293.3,274,301.2,263.1,301.2z M458.2,291.2c-2.4,6.3-7.3,10-13.9,10
-	c-11,0-17.7-7.8-17.7-20.6c0-12.9,6.7-20.8,17.7-20.8c6.6,0,11.4,3.6,13.9,10H473c-3.6-14-14.5-23-28.7-23
-	c-18.8,0-32.1,14-32.1,33.8c0,19.6,13.3,33.6,32.1,33.6c14.2,0,25.1-9.1,28.8-23H458.2z M369.3,247.9l26,65.1h14.2l-26-65.1H369.3z
-	 M339.8,247.9H308v65.1h14v-23.6h17.9c14.8,0,23.8-7.8,23.8-20.8C363.7,255.7,354.6,247.9,339.8,247.9z M339.2,276.9H322v-16.4h17.2
-	c6.9,0,10.5,2.8,10.5,8.2C349.7,274.1,346.1,276.9,339.2,276.9z M220.3,267.8c0-12.3-9-19.8-23.8-19.8h-31.8v65.1h14v-25.5h15.5
-	l14,25.5h15.4l-15.5-27.4C215.8,282.6,220.3,276.4,220.3,267.8z M178.6,260.5h17.2c6.9,0,10.5,2.5,10.5,7.3c0,4.7-3.6,7.3-10.5,7.3
-	h-17.2V260.5z"/>
+  <path class="st0" d="M -23.533 126.747 L -1.633 126.747 L -1.633 179.347 L 12.367 179.347 L 12.367 126.747 L 34.267 126.747 L 34.267 114.147 L -23.533 114.147 L -23.533 126.747 Z M -48.133 159.747 L -77.433 114.147 L -93.233 114.147 L -93.233 179.247 L -79.733 179.247 L -79.733 133.647 L -50.433 179.347 L -34.633 179.347 L -34.633 114.247 L -48.133 114.247 L -48.133 159.747 Z M 90.067 140.147 L 59.367 140.147 L 59.367 114.147 L 45.367 114.147 L 45.367 179.247 L 59.367 179.247 L 59.367 152.647 L 90.067 152.647 L 90.067 179.247 L 104.067 179.247 L 104.067 114.147 L 90.067 114.147 L 90.067 140.147 Z M -144.333 114.147 L -170.333 179.247 L -155.833 179.247 L -150.533 165.547 L -123.333 165.547 L -118.033 179.247 L -103.533 179.247 L -129.533 114.147 L -144.333 114.147 Z M -145.833 153.547 L -136.933 130.647 L -128.033 153.547 L -145.833 153.547 Z M 219.667 113.047 C 200.867 113.047 187.567 127.047 187.567 146.847 C 187.567 166.447 200.867 180.447 219.667 180.447 C 238.367 180.447 251.567 166.447 251.567 146.847 C 251.567 127.047 238.367 113.047 219.667 113.047 Z M 219.667 167.447 C 208.667 167.447 201.967 159.647 201.967 146.847 C 201.967 133.947 208.667 126.047 219.667 126.047 C 230.567 126.047 237.167 133.847 237.167 146.847 C 237.167 159.547 230.567 167.447 219.667 167.447 Z M 414.767 157.447 C 412.367 163.747 407.467 167.447 400.867 167.447 C 389.867 167.447 383.167 159.647 383.167 146.847 C 383.167 133.947 389.867 126.047 400.867 126.047 C 407.467 126.047 412.267 129.647 414.767 136.047 L 429.567 136.047 C 425.967 122.047 415.067 113.047 400.867 113.047 C 382.067 113.047 368.767 127.047 368.767 146.847 C 368.767 166.447 382.067 180.447 400.867 180.447 C 415.067 180.447 425.967 171.347 429.667 157.447 L 414.767 157.447 Z M 325.867 114.147 L 351.867 179.247 L 366.067 179.247 L 340.067 114.147 L 325.867 114.147 Z M 296.367 114.147 L 264.567 114.147 L 264.567 179.247 L 278.567 179.247 L 278.567 155.647 L 296.467 155.647 C 311.267 155.647 320.267 147.847 320.267 134.847 C 320.267 121.947 311.167 114.147 296.367 114.147 Z M 295.767 143.147 L 278.567 143.147 L 278.567 126.747 L 295.767 126.747 C 302.667 126.747 306.267 129.547 306.267 134.947 C 306.267 140.347 302.667 143.147 295.767 143.147 Z M 176.867 134.047 C 176.867 121.747 167.867 114.247 153.067 114.247 L 121.267 114.247 L 121.267 179.347 L 135.267 179.347 L 135.267 153.847 L 150.767 153.847 L 164.767 179.347 L 180.167 179.347 L 164.667 151.947 C 172.367 148.847 176.867 142.647 176.867 134.047 Z M 135.167 126.747 L 152.367 126.747 C 159.267 126.747 162.867 129.247 162.867 134.047 C 162.867 138.747 159.267 141.347 152.367 141.347 L 135.167 141.347 L 135.167 126.747 Z"></path>
 </svg>
\ No newline at end of file

From 7d49319696c1d0c4191772a473fe882b0cc9446c Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Fri, 2 Jun 2023 11:23:55 -0300
Subject: [PATCH 28/59] saving darkMode on localStorage

---
 src/frontend/src/contexts/darkContext.tsx                    | 5 ++++-
 .../pages/FlowPage/components/tabsManagerComponent/index.tsx | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/frontend/src/contexts/darkContext.tsx b/src/frontend/src/contexts/darkContext.tsx
index f612fe381..2a76d0e53 100644
--- a/src/frontend/src/contexts/darkContext.tsx
+++ b/src/frontend/src/contexts/darkContext.tsx
@@ -13,13 +13,16 @@ const initialValue = {
 export const darkContext = createContext<darkContextType>(initialValue);
 
 export function DarkProvider({ children }) {
-  const [dark, setDark] = useState(false);
+  const [dark, setDark] = useState(
+    JSON.parse(window.localStorage.getItem("isDark")) ?? false
+  );
   useEffect(() => {
     if (dark) {
       document.getElementById("body").classList.add("dark");
     } else {
       document.getElementById("body").classList.remove("dark");
     }
+    window.localStorage.setItem("isDark", dark.toString());
   }, [dark]);
   return (
     <darkContext.Provider
diff --git a/src/frontend/src/pages/FlowPage/components/tabsManagerComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/tabsManagerComponent/index.tsx
index 278fe6756..816247911 100644
--- a/src/frontend/src/pages/FlowPage/components/tabsManagerComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/tabsManagerComponent/index.tsx
@@ -116,7 +116,7 @@ export default function TabsManagerComponent() {
           </button>
         </div>
       </div>
-      <div className="w-full h-full">
+      <div className="w-full h-full dark:bg-gray-800">
         <ReactFlowProvider>
           {flows[tabIndex] ? (
             <FlowPage flow={flows[tabIndex]}></FlowPage>

From b7286afd323983f88c1085c40294e516c25be525 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 2 Jun 2023 12:14:03 -0300
Subject: [PATCH 29/59] =?UTF-8?q?=F0=9F=8E=A8=20style(types.py):=20improve?=
 =?UTF-8?q?=20readability=20of=20return=20statement=20in=20TextSplitterVer?=
 =?UTF-8?q?tex=20The=20return=20statement=20in=20TextSplitterVertex=20was?=
 =?UTF-8?q?=20improved=20to=20be=20more=20readable=20by=20adding=20a=20new?=
 =?UTF-8?q?=20line=20before=20the=20Documents=20field.=20This=20makes=20it?=
 =?UTF-8?q?=20easier=20to=20read=20and=20understand=20the=20output=20of=20?=
 =?UTF-8?q?the=20function.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/vertex/types.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index 4a3290c13..7d61f2393 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -193,5 +193,6 @@ class TextSplitterVertex(Vertex):
         # This built_object is a list of documents. Maybe we should
         # show how many documents are in the list?
         if self._built_object:
-            return f"""{self.vertex_type}({len(self._built_object)} documents)\nDocuments: {self._built_object[:3]}..."""
+            return f"""{self.vertex_type}({len(self._built_object)} documents)
+            \nDocuments: {self._built_object[:3]}..."""
         return f"{self.vertex_type}()"

From 534f47f1532ffc183a0d9d8c15cc5db830a0ac5b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 2 Jun 2023 14:14:47 -0300
Subject: [PATCH 30/59] =?UTF-8?q?=F0=9F=94=A8=20refactor(types.py):=20impo?=
 =?UTF-8?q?rt=20flatten=5Flist=20function=20from=20graph.utils=20module=20?=
 =?UTF-8?q?This=20commit=20simply=20imports=20the=20flatten=5Flist=20funct?=
 =?UTF-8?q?ion=20from=20the=20graph.utils=20module=20to=20be=20used=20in?=
 =?UTF-8?q?=20the=20AgentVertex=20class.=20This=20improves=20the=20readabi?=
 =?UTF-8?q?lity=20of=20the=20code=20and=20reduces=20the=20number=20of=20li?=
 =?UTF-8?q?nes=20of=20code.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/vertex/types.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index 5b4e01ede..0b0d0923f 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -1,15 +1,15 @@
 from typing import Any, Dict, List, Optional, Union
 
 from langflow.graph.vertex.base import Vertex
-from langflow.graph.utils import extract_input_variables_from_prompt
+from langflow.graph.utils import extract_input_variables_from_prompt, flatten_list
 
 
 class AgentVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="agents")
 
-        self.tools: List[Union[ToolNode, ToolkitNode]] = []
-        self.chains: List[ChainNode] = []
+        self.tools: List[Union[ToolVertex, ToolkitVertex]] = []
+        self.chains: List[ChainVertex] = []
 
     def _set_tools_and_chains(self) -> None:
         for edge in self.edges:
@@ -94,10 +94,10 @@ class ChainVertex(Vertex):
         tools: Optional[Union[List[Vertex], List[ToolVertex]]] = None,
     ) -> Any:
         if not self._built or force:
-            # Check if the chain requires a PromptNode
+            # Check if the chain requires a PromptVertex
             for key, value in self.params.items():
                 if isinstance(value, PromptVertex):
-                    # Build the PromptNode, passing the tools if available
+                    # Build the PromptVertex, passing the tools if available
                     self.params[key] = value.build(tools=tools, force=force)
 
             self._build()

From 430043098dd280143b271168a7c87e52af2a1ac6 Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Fri, 2 Jun 2023 14:15:06 -0300
Subject: [PATCH 31/59] fix dropdown value with wrong color

---
 src/frontend/src/components/dropdownComponent/index.tsx | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/frontend/src/components/dropdownComponent/index.tsx b/src/frontend/src/components/dropdownComponent/index.tsx
index eebbe1554..0a214eb4e 100644
--- a/src/frontend/src/components/dropdownComponent/index.tsx
+++ b/src/frontend/src/components/dropdownComponent/index.tsx
@@ -25,7 +25,9 @@ export default function Dropdown({
           <>
             <div className="relative mt-1 w-full">
               <Listbox.Button className="relative w-full cursor-default rounded-md border border-gray-300 bg-white dark:bg-gray-900 py-2 pl-3 pr-10 text-left shadow-sm focus:border-indigo-500 focus:outline-none focus:ring-1 focus:ring-indigo-500 sm:text-sm">
-                <span className="block truncate w-full">{internalValue}</span>
+                <span className="block truncate w-full dark:text-gray-300">
+                  {internalValue}
+                </span>
                 <span className="pointer-events-none absolute inset-y-0 right-0 flex items-center pr-2">
                   <ChevronUpDownIcon
                     className="h-5 w-5 text-gray-400"

From df3cdb90b7d13c4c38008f0aa8fbb17bf504ebef Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Fri, 2 Jun 2023 14:21:38 -0300
Subject: [PATCH 32/59] =?UTF-8?q?=F0=9F=94=A8=20refactor(types.py):=20reor?=
 =?UTF-8?q?der=20class=20definitions=20to=20match=20the=20order=20of=20the?=
 =?UTF-8?q?ir=20usage=20in=20the=20code=20The=20order=20of=20the=20class?=
 =?UTF-8?q?=20definitions=20in=20the=20file=20has=20been=20changed=20to=20?=
 =?UTF-8?q?match=20the=20order=20of=20their=20usage=20in=20the=20code.=20T?=
 =?UTF-8?q?his=20improves=20the=20readability=20of=20the=20code=20and=20ma?=
 =?UTF-8?q?kes=20it=20easier=20to=20understand=20the=20relationships=20bet?=
 =?UTF-8?q?ween=20the=20classes.=20No=20functionality=20has=20been=20chang?=
 =?UTF-8?q?ed.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/vertex/types.py | 137 ++++++++++-----------
 1 file changed, 67 insertions(+), 70 deletions(-)

diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index 0b0d0923f..b81e72439 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -8,13 +8,13 @@ class AgentVertex(Vertex):
     def __init__(self, data: Dict):
         super().__init__(data, base_type="agents")
 
-        self.tools: List[Union[ToolVertex, ToolkitVertex]] = []
+        self.tools: List[Union[ToolkitVertex, ToolVertex]] = []
         self.chains: List[ChainVertex] = []
 
     def _set_tools_and_chains(self) -> None:
         for edge in self.edges:
             source_node = edge.source
-            if isinstance(source_node, ToolVertex):
+            if isinstance(source_node, (ToolVertex, ToolkitVertex)):
                 self.tools.append(source_node)
             elif isinstance(source_node, ChainVertex):
                 self.chains.append(source_node)
@@ -40,74 +40,6 @@ class ToolVertex(Vertex):
         super().__init__(data, base_type="tools")
 
 
-class PromptVertex(Vertex):
-    def __init__(self, data: Dict):
-        super().__init__(data, base_type="prompts")
-
-    def build(
-        self,
-        force: bool = False,
-        tools: Optional[Union[List[Vertex], List[ToolVertex]]] = None,
-    ) -> Any:
-        if not self._built or force:
-            if (
-                "input_variables" not in self.params
-                or self.params["input_variables"] is None
-            ):
-                self.params["input_variables"] = []
-            # Check if it is a ZeroShotPrompt and needs a tool
-            if "ShotPrompt" in self.vertex_type:
-                tools = (
-                    [tool_node.build() for tool_node in tools]
-                    if tools is not None
-                    else []
-                )
-                # flatten the list of tools if it is a list of lists
-                # first check if it is a list
-                if tools and isinstance(tools, list) and isinstance(tools[0], list):
-                    tools = flatten_list(tools)
-                self.params["tools"] = tools
-                prompt_params = [
-                    key
-                    for key, value in self.params.items()
-                    if isinstance(value, str) and key != "format_instructions"
-                ]
-            else:
-                prompt_params = ["template"]
-            for param in prompt_params:
-                prompt_text = self.params[param]
-                variables = extract_input_variables_from_prompt(prompt_text)
-                self.params["input_variables"].extend(variables)
-            self.params["input_variables"] = list(set(self.params["input_variables"]))
-
-            self._build()
-        return self._built_object
-
-
-class ChainVertex(Vertex):
-    def __init__(self, data: Dict):
-        super().__init__(data, base_type="chains")
-
-    def build(
-        self,
-        force: bool = False,
-        tools: Optional[Union[List[Vertex], List[ToolVertex]]] = None,
-    ) -> Any:
-        if not self._built or force:
-            # Check if the chain requires a PromptVertex
-            for key, value in self.params.items():
-                if isinstance(value, PromptVertex):
-                    # Build the PromptVertex, passing the tools if available
-                    self.params[key] = value.build(tools=tools, force=force)
-
-            self._build()
-
-        #! Cannot deepcopy SQLDatabaseChain
-        if self.vertex_type in ["SQLDatabaseChain"]:
-            return self._built_object
-        return self._built_object
-
-
 class LLMVertex(Vertex):
     built_node_type = None
     class_built_object = None
@@ -193,3 +125,68 @@ class TextSplitterVertex(Vertex):
             return f"""{self.vertex_type}({len(self._built_object)} documents)
             \nDocuments: {self._built_object[:3]}..."""
         return f"{self.vertex_type}()"
+
+
+class ChainVertex(Vertex):
+    def __init__(self, data: Dict):
+        super().__init__(data, base_type="chains")
+
+    def build(
+        self,
+        force: bool = False,
+        tools: Optional[List[Union[ToolkitVertex, ToolVertex]]] = None,
+    ) -> Any:
+        if not self._built or force:
+            # Check if the chain requires a PromptVertex
+            for key, value in self.params.items():
+                if isinstance(value, PromptVertex):
+                    # Build the PromptVertex, passing the tools if available
+                    self.params[key] = value.build(tools=tools, force=force)
+
+            self._build()
+
+        return self._built_object
+
+
+class PromptVertex(Vertex):
+    def __init__(self, data: Dict):
+        super().__init__(data, base_type="prompts")
+
+    def build(
+        self,
+        force: bool = False,
+        tools: Optional[List[Union[ToolkitVertex, ToolVertex]]] = None,
+    ) -> Any:
+        if not self._built or force:
+            if (
+                "input_variables" not in self.params
+                or self.params["input_variables"] is None
+            ):
+                self.params["input_variables"] = []
+            # Check if it is a ZeroShotPrompt and needs a tool
+            if "ShotPrompt" in self.vertex_type:
+                tools = (
+                    [tool_node.build() for tool_node in tools]
+                    if tools is not None
+                    else []
+                )
+                # flatten the list of tools if it is a list of lists
+                # first check if it is a list
+                if tools and isinstance(tools, list) and isinstance(tools[0], list):
+                    tools = flatten_list(tools)
+                self.params["tools"] = tools
+                prompt_params = [
+                    key
+                    for key, value in self.params.items()
+                    if isinstance(value, str) and key != "format_instructions"
+                ]
+            else:
+                prompt_params = ["template"]
+            for param in prompt_params:
+                prompt_text = self.params[param]
+                variables = extract_input_variables_from_prompt(prompt_text)
+                self.params["input_variables"].extend(variables)
+            self.params["input_variables"] = list(set(self.params["input_variables"]))
+
+            self._build()
+        return self._built_object

From 07afc08129a2fed81c452e243be931102ea250a5 Mon Sep 17 00:00:00 2001
From: Jim White <jim@ifcx.org>
Date: Fri, 2 Jun 2023 12:08:57 -0700
Subject: [PATCH 33/59] Add model names for Anthropic and add ChatAnthropic
 (preferred over the deprecated Anthropic LLM).

---
 src/backend/langflow/config.yaml                   |  1 +
 src/backend/langflow/interface/custom_lists.py     |  2 ++
 .../langflow/template/frontend_node/base.py        |  3 +++
 src/backend/langflow/utils/constants.py            | 14 ++++++++++++++
 src/backend/langflow/utils/util.py                 |  4 +++-
 src/frontend/src/utils.ts                          |  1 +
 6 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index f2833ae26..329d3321e 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -56,6 +56,7 @@ llms:
   - CTransformers
   - Cohere
   - Anthropic
+  - ChatAnthropic
 memories:
   - ConversationBufferMemory
   - ConversationSummaryMemory
diff --git a/src/backend/langflow/interface/custom_lists.py b/src/backend/langflow/interface/custom_lists.py
index 0fea838b6..a944363ae 100644
--- a/src/backend/langflow/interface/custom_lists.py
+++ b/src/backend/langflow/interface/custom_lists.py
@@ -11,12 +11,14 @@ from langchain import (
     text_splitter,
 )
 from langchain.agents import agent_toolkits
+from langchain.chat_models import ChatAnthropic
 from langchain.chat_models import ChatOpenAI
 
 from langflow.interface.importing.utils import import_class
 
 ## LLMs
 llm_type_to_cls_dict = llms.type_to_cls_dict
+llm_type_to_cls_dict["anthropic-chat"] = ChatAnthropic  # type: ignore
 llm_type_to_cls_dict["openai-chat"] = ChatOpenAI  # type: ignore
 
 ## Chains
diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py
index a64195813..a97c7b8b0 100644
--- a/src/backend/langflow/template/frontend_node/base.py
+++ b/src/backend/langflow/template/frontend_node/base.py
@@ -125,6 +125,9 @@ class FrontendNode(BaseModel):
         elif name == "ChatOpenAI" and key == "model_name":
             field.options = constants.CHAT_OPENAI_MODELS
             field.is_list = True
+        elif (name == "Anthropic" or name == "ChatAnthropic") and key == "model_name":
+            field.options = constants.ANTHROPIC_MODELS
+            field.is_list = True
         if "api_key" in key and "OpenAI" in str(name):
             field.display_name = "OpenAI API Key"
             field.required = False
diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py
index 2d101ab98..d41cd7dfc 100644
--- a/src/backend/langflow/utils/constants.py
+++ b/src/backend/langflow/utils/constants.py
@@ -7,6 +7,20 @@ OPENAI_MODELS = [
 ]
 CHAT_OPENAI_MODELS = ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k"]
 
+ANTHROPIC_MODELS = [
+    "claude-v1",  # largest model, ideal for a wide range of more complex tasks.
+    "claude-v1-100k",  # An enhanced version of claude-v1 with a 100,000 token (roughly 75,000 word) context window.
+    "claude-instant-v1",  # A smaller model with far lower latency, sampling at roughly 40 words/sec!
+    "claude-instant-v1-100k",  # An enhanced version of claude-instant-v1 with a 100,000 token context window that retains its performance.
+    # Specific sub-versions of the above models:
+    "claude-v1.3",  # Compared to claude-v1.2, it's more robust against red-team inputs, better at precise instruction-following, better at code, and better and non-English dialogue and writing.
+    "claude-v1.3-100k",  # An enhanced version of claude-v1.3 with a 100,000 token (roughly 75,000 word) context window.
+    "claude-v1.2",  # An improved version of claude-v1. It is slightly improved at general helpfulness, instruction following, coding, and other tasks. It is also considerably better with non-English languages. This model also has the ability to role play (in harmless ways) more consistently, and it defaults to writing somewhat longer and more thorough responses.
+    "claude-v1.0",  # An earlier version of claude-v1.
+    "claude-instant-v1.1",  # latest version of claude-instant-v1. It is better than claude-instant-v1.0 at a wide variety of tasks including writing, coding, and instruction following.
+    "claude-instant-v1.1-100k",  # An enhanced version of claude-instant-v1.1 with a 100,000 token context window that retains its lightning fast 40 word/sec performance.
+    "claude-instant-v1.0",  # An earlier version of claude-instant-v1.
+]
 
 DEFAULT_PYTHON_FUNCTION = """
 def python_function(text: str) -> str:
diff --git a/src/backend/langflow/utils/util.py b/src/backend/langflow/utils/util.py
index 293d31154..f4e4927d8 100644
--- a/src/backend/langflow/utils/util.py
+++ b/src/backend/langflow/utils/util.py
@@ -302,7 +302,9 @@ def format_dict(d, name: Optional[str] = None):
         elif name == "ChatOpenAI" and key == "model_name":
             value["options"] = constants.CHAT_OPENAI_MODELS
             value["list"] = True
-
+        elif (name == "Anthropic" or name == "ChatAnthropic") and key == "model_name":
+            value["options"] = constants.ANTHROPIC_MODELS
+            value["list"] = True
     return d
 
 
diff --git a/src/frontend/src/utils.ts b/src/frontend/src/utils.ts
index 0e7289dd0..119383667 100644
--- a/src/frontend/src/utils.ts
+++ b/src/frontend/src/utils.ts
@@ -155,6 +155,7 @@ export const nodeIcons: {
   // SerpAPIWrapper: SerperIcon,
   // AZLyricsLoader: AzIcon,
   Anthropic: AnthropicIcon,
+  ChatAnthropic: AnthropicIcon,
   BingSearchAPIWrapper: BingIcon,
   BingSearchRun: BingIcon,
   Cohere: CohereIcon,

From cce41189e9a2ba66c560c1c5bd57172cc32ff9d7 Mon Sep 17 00:00:00 2001
From: Jim White <jim@ifcx.org>
Date: Fri, 2 Jun 2023 12:18:38 -0700
Subject: [PATCH 34/59] Trim too long comments about Anthropic models.

---
 src/backend/langflow/utils/constants.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/backend/langflow/utils/constants.py b/src/backend/langflow/utils/constants.py
index d41cd7dfc..1b6bbdcc3 100644
--- a/src/backend/langflow/utils/constants.py
+++ b/src/backend/langflow/utils/constants.py
@@ -11,14 +11,14 @@ ANTHROPIC_MODELS = [
     "claude-v1",  # largest model, ideal for a wide range of more complex tasks.
     "claude-v1-100k",  # An enhanced version of claude-v1 with a 100,000 token (roughly 75,000 word) context window.
     "claude-instant-v1",  # A smaller model with far lower latency, sampling at roughly 40 words/sec!
-    "claude-instant-v1-100k",  # An enhanced version of claude-instant-v1 with a 100,000 token context window that retains its performance.
+    "claude-instant-v1-100k",  # Like claude-instant-v1 with a 100,000 token context window but retains its performance.
     # Specific sub-versions of the above models:
-    "claude-v1.3",  # Compared to claude-v1.2, it's more robust against red-team inputs, better at precise instruction-following, better at code, and better and non-English dialogue and writing.
+    "claude-v1.3",  # Vs claude-v1.2: better instruction-following, code, and non-English dialogue and writing.
     "claude-v1.3-100k",  # An enhanced version of claude-v1.3 with a 100,000 token (roughly 75,000 word) context window.
-    "claude-v1.2",  # An improved version of claude-v1. It is slightly improved at general helpfulness, instruction following, coding, and other tasks. It is also considerably better with non-English languages. This model also has the ability to role play (in harmless ways) more consistently, and it defaults to writing somewhat longer and more thorough responses.
+    "claude-v1.2",  # Vs claude-v1.1: small adv in general helpfulness, instruction following, coding, and other tasks.
     "claude-v1.0",  # An earlier version of claude-v1.
-    "claude-instant-v1.1",  # latest version of claude-instant-v1. It is better than claude-instant-v1.0 at a wide variety of tasks including writing, coding, and instruction following.
-    "claude-instant-v1.1-100k",  # An enhanced version of claude-instant-v1.1 with a 100,000 token context window that retains its lightning fast 40 word/sec performance.
+    "claude-instant-v1.1",  # Latest version of claude-instant-v1. Better than claude-instant-v1.0 at most tasks.
+    "claude-instant-v1.1-100k",  # Version of claude-instant-v1.1 with a 100K token context window.
     "claude-instant-v1.0",  # An earlier version of claude-instant-v1.
 ]
 

From 02cb94709d2e6b4a82f7334199ab746c55e104b4 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Fri, 2 Jun 2023 17:15:59 -0300
Subject: [PATCH 35/59] Adding shad tooltip and grouping by class on edges

---
 src/frontend/package-lock.json                | 467 +++++++++++++++++-
 src/frontend/package.json                     |   6 +
 .../components/parameterComponent/index.tsx   |  85 ++--
 .../src/CustomNodes/GenericNode/index.tsx     |  19 +-
 .../ReactTooltipComponent/index.tsx           |   4 +-
 .../components/ShadTooltipComponent/index.tsx |  25 +
 src/frontend/src/components/ui/tooltip.tsx    |  29 ++
 src/frontend/src/index.css                    |  84 ++++
 .../extraSidebarComponent/index.tsx           |  14 +-
 src/frontend/src/utils.ts                     |  61 +++
 src/frontend/tailwind.config.js               |  85 +++-
 src/frontend/tsconfig.json                    |   3 +-
 12 files changed, 812 insertions(+), 70 deletions(-)
 create mode 100644 src/frontend/src/components/ShadTooltipComponent/index.tsx
 create mode 100644 src/frontend/src/components/ui/tooltip.tsx

diff --git a/src/frontend/package-lock.json b/src/frontend/package-lock.json
index 374236168..9000c133f 100644
--- a/src/frontend/package-lock.json
+++ b/src/frontend/package-lock.json
@@ -13,6 +13,7 @@
                 "@headlessui/react": "^1.7.10",
                 "@heroicons/react": "^2.0.15",
                 "@mui/material": "^5.11.9",
+                "@radix-ui/react-tooltip": "^1.0.6",
                 "@tabler/icons-react": "^2.18.0",
                 "@tailwindcss/forms": "^0.5.3",
                 "@tailwindcss/line-clamp": "^0.4.4",
@@ -20,7 +21,10 @@
                 "ansi-to-html": "^0.7.2",
                 "axios": "^1.3.2",
                 "base64-js": "^1.5.1",
+                "class-variance-authority": "^0.6.0",
+                "clsx": "^1.2.1",
                 "lodash": "^4.17.21",
+                "lucide-react": "^0.233.0",
                 "react": "^18.2.0",
                 "react-ace": "^10.1.0",
                 "react-cookie": "^4.1.1",
@@ -37,6 +41,8 @@
                 "rehype-mathjax": "^4.0.2",
                 "remark-gfm": "^3.0.1",
                 "remark-math": "^5.1.1",
+                "tailwind-merge": "^1.13.0",
+                "tailwindcss-animate": "^1.0.5",
                 "uuid": "^9.0.0",
                 "vite-plugin-svgr": "^3.2.0",
                 "web-vitals": "^2.1.4"
@@ -911,6 +917,18 @@
                 "@floating-ui/core": "^1.2.6"
             }
         },
+        "node_modules/@floating-ui/react-dom": {
+            "version": "2.0.0",
+            "resolved": "https://registry.npmjs.org/@floating-ui/react-dom/-/react-dom-2.0.0.tgz",
+            "integrity": "sha512-Ke0oU3SeuABC2C4OFu2mSAwHIP5WUiV98O9YWoHV4Q5aT6E9k06DV0Khi5uYspR8xmmBk08t8ZDcz3TR3ARkEg==",
+            "dependencies": {
+                "@floating-ui/dom": "^1.2.7"
+            },
+            "peerDependencies": {
+                "react": ">=16.8.0",
+                "react-dom": ">=16.8.0"
+            }
+        },
         "node_modules/@headlessui/react": {
             "version": "1.7.10",
             "resolved": "https://registry.npmjs.org/@headlessui/react/-/react-1.7.10.tgz",
@@ -1274,6 +1292,407 @@
                 "url": "https://opencollective.com/popperjs"
             }
         },
+        "node_modules/@radix-ui/primitive": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/primitive/-/primitive-1.0.1.tgz",
+            "integrity": "sha512-yQ8oGX2GVsEYMWGxcovu1uGWPCxV5BFfeeYxqPmuAzUyLT9qmaMXSAhXpb0WrspIeqYzdJpkh2vHModJPgRIaw==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            }
+        },
+        "node_modules/@radix-ui/react-arrow": {
+            "version": "1.0.3",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-arrow/-/react-arrow-1.0.3.tgz",
+            "integrity": "sha512-wSP+pHsB/jQRaL6voubsQ/ZlrGBHHrOjmBnr19hxYgtS0WvAFwZhK2WP/YY5yF9uKECCEEDGxuLxq1NBK51wFA==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-primitive": "1.0.3"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-compose-refs": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-compose-refs/-/react-compose-refs-1.0.1.tgz",
+            "integrity": "sha512-fDSBgd44FKHa1FRMU59qBMPFcl2PZE+2nmqunj+BWFyYYjnhIDWL2ItDs3rrbJDQOtzt5nIebLCQc4QRfz6LJw==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-context": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-context/-/react-context-1.0.1.tgz",
+            "integrity": "sha512-ebbrdFoYTcuZ0v4wG5tedGnp9tzcV8awzsxYph7gXUyvnNLuTIcCk1q17JEbnVhXAKG9oX3KtchwiMIAYp9NLg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-dismissable-layer": {
+            "version": "1.0.4",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-dismissable-layer/-/react-dismissable-layer-1.0.4.tgz",
+            "integrity": "sha512-7UpBa/RKMoHJYjie1gkF1DlK8l1fdU/VKDpoS3rCCo8YBJR294GwcEHyxHw72yvphJ7ld0AXEcSLAzY2F/WyCg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/primitive": "1.0.1",
+                "@radix-ui/react-compose-refs": "1.0.1",
+                "@radix-ui/react-primitive": "1.0.3",
+                "@radix-ui/react-use-callback-ref": "1.0.1",
+                "@radix-ui/react-use-escape-keydown": "1.0.3"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-id": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-id/-/react-id-1.0.1.tgz",
+            "integrity": "sha512-tI7sT/kqYp8p96yGWY1OAnLHrqDgzHefRBKQ2YAkBS5ja7QLcZ9Z/uY7bEjPUatf8RomoXM8/1sMj1IJaE5UzQ==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-use-layout-effect": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-popper": {
+            "version": "1.1.2",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-popper/-/react-popper-1.1.2.tgz",
+            "integrity": "sha512-1CnGGfFi/bbqtJZZ0P/NQY20xdG3E0LALJaLUEoKwPLwl6PPPfbeiCqMVQnhoFRAxjJj4RpBRJzDmUgsex2tSg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@floating-ui/react-dom": "^2.0.0",
+                "@radix-ui/react-arrow": "1.0.3",
+                "@radix-ui/react-compose-refs": "1.0.1",
+                "@radix-ui/react-context": "1.0.1",
+                "@radix-ui/react-primitive": "1.0.3",
+                "@radix-ui/react-use-callback-ref": "1.0.1",
+                "@radix-ui/react-use-layout-effect": "1.0.1",
+                "@radix-ui/react-use-rect": "1.0.1",
+                "@radix-ui/react-use-size": "1.0.1",
+                "@radix-ui/rect": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-portal": {
+            "version": "1.0.3",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-portal/-/react-portal-1.0.3.tgz",
+            "integrity": "sha512-xLYZeHrWoPmA5mEKEfZZevoVRK/Q43GfzRXkWV6qawIWWK8t6ifIiLQdd7rmQ4Vk1bmI21XhqF9BN3jWf+phpA==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-primitive": "1.0.3"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-presence": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-presence/-/react-presence-1.0.1.tgz",
+            "integrity": "sha512-UXLW4UAbIY5ZjcvzjfRFo5gxva8QirC9hF7wRE4U5gz+TP0DbRk+//qyuAQ1McDxBt1xNMBTaciFGvEmJvAZCg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-compose-refs": "1.0.1",
+                "@radix-ui/react-use-layout-effect": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-primitive": {
+            "version": "1.0.3",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-primitive/-/react-primitive-1.0.3.tgz",
+            "integrity": "sha512-yi58uVyoAcK/Nq1inRY56ZSjKypBNKTa/1mcL8qdl6oJeEaDbOldlzrGn7P6Q3Id5d+SYNGc5AJgc4vGhjs5+g==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-slot": "1.0.2"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-slot": {
+            "version": "1.0.2",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-slot/-/react-slot-1.0.2.tgz",
+            "integrity": "sha512-YeTpuq4deV+6DusvVUW4ivBgnkHwECUu0BiN43L5UCDFgdhsRUWAghhTF5MbvNTPzmiFOx90asDSUjWuCNapwg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-compose-refs": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-tooltip": {
+            "version": "1.0.6",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-tooltip/-/react-tooltip-1.0.6.tgz",
+            "integrity": "sha512-DmNFOiwEc2UDigsYj6clJENma58OelxD24O4IODoZ+3sQc3Zb+L8w1EP+y9laTuKCLAysPw4fD6/v0j4KNV8rg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/primitive": "1.0.1",
+                "@radix-ui/react-compose-refs": "1.0.1",
+                "@radix-ui/react-context": "1.0.1",
+                "@radix-ui/react-dismissable-layer": "1.0.4",
+                "@radix-ui/react-id": "1.0.1",
+                "@radix-ui/react-popper": "1.1.2",
+                "@radix-ui/react-portal": "1.0.3",
+                "@radix-ui/react-presence": "1.0.1",
+                "@radix-ui/react-primitive": "1.0.3",
+                "@radix-ui/react-slot": "1.0.2",
+                "@radix-ui/react-use-controllable-state": "1.0.1",
+                "@radix-ui/react-visually-hidden": "1.0.3"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-callback-ref": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-callback-ref/-/react-use-callback-ref-1.0.1.tgz",
+            "integrity": "sha512-D94LjX4Sp0xJFVaoQOd3OO9k7tpBYNOXdVhkltUbGv2Qb9OXdrg/CpsjlZv7ia14Sylv398LswWBVVu5nqKzAQ==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-controllable-state": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-controllable-state/-/react-use-controllable-state-1.0.1.tgz",
+            "integrity": "sha512-Svl5GY5FQeN758fWKrjM6Qb7asvXeiZltlT4U2gVfl8Gx5UAv2sMR0LWo8yhsIZh2oQ0eFdZ59aoOOMV7b47VA==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-use-callback-ref": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-escape-keydown": {
+            "version": "1.0.3",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-escape-keydown/-/react-use-escape-keydown-1.0.3.tgz",
+            "integrity": "sha512-vyL82j40hcFicA+M4Ex7hVkB9vHgSse1ZWomAqV2Je3RleKGO5iM8KMOEtfoSB0PnIelMd2lATjTGMYqN5ylTg==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-use-callback-ref": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-layout-effect": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-layout-effect/-/react-use-layout-effect-1.0.1.tgz",
+            "integrity": "sha512-v/5RegiJWYdoCvMnITBkNNx6bCj20fiaJnWtRkU18yITptraXjffz5Qbn05uOiQnOvi+dbkznkoaMltz1GnszQ==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-rect": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-rect/-/react-use-rect-1.0.1.tgz",
+            "integrity": "sha512-Cq5DLuSiuYVKNU8orzJMbl15TXilTnJKUCltMVQg53BQOF1/C5toAaGrowkgksdBQ9H+SRL23g0HDmg9tvmxXw==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/rect": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-use-size": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-use-size/-/react-use-size-1.0.1.tgz",
+            "integrity": "sha512-ibay+VqrgcaI6veAojjofPATwledXiSmX+C0KrBk/xgpX9rBzPV3OsfwlhQdUOFbh+LKQorLYT+xTXW9V8yd0g==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-use-layout-effect": "1.0.1"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "react": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/react-visually-hidden": {
+            "version": "1.0.3",
+            "resolved": "https://registry.npmjs.org/@radix-ui/react-visually-hidden/-/react-visually-hidden-1.0.3.tgz",
+            "integrity": "sha512-D4w41yN5YRKtu464TLnByKzMDG/JlMPHtfZgQAu9v6mNakUqGUI9vUrfQKz8NK41VMm/xbZbh76NUTVtIYqOMA==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10",
+                "@radix-ui/react-primitive": "1.0.3"
+            },
+            "peerDependencies": {
+                "@types/react": "*",
+                "@types/react-dom": "*",
+                "react": "^16.8 || ^17.0 || ^18.0",
+                "react-dom": "^16.8 || ^17.0 || ^18.0"
+            },
+            "peerDependenciesMeta": {
+                "@types/react": {
+                    "optional": true
+                },
+                "@types/react-dom": {
+                    "optional": true
+                }
+            }
+        },
+        "node_modules/@radix-ui/rect": {
+            "version": "1.0.1",
+            "resolved": "https://registry.npmjs.org/@radix-ui/rect/-/rect-1.0.1.tgz",
+            "integrity": "sha512-fyrgCaedtvMg9NK3en0pnOYJdtfwxUcNolezkNPUsoX57X8oQk+NkqcvzHXD2uKNij6GXmWU9NDru2IWjrO4BQ==",
+            "dependencies": {
+                "@babel/runtime": "^7.13.10"
+            }
+        },
         "node_modules/@reactflow/background": {
             "version": "11.1.7",
             "resolved": "https://registry.npmjs.org/@reactflow/background/-/background-11.1.7.tgz",
@@ -2422,7 +2841,7 @@
             "version": "18.2.4",
             "resolved": "https://registry.npmjs.org/@types/react-dom/-/react-dom-18.2.4.tgz",
             "integrity": "sha512-G2mHoTMTL4yoydITgOGwWdWMVd8sNgyEP85xVmMKAPUBwQWm9wBPQUmvbeF4V3WBY1P7mmL4BkjQ0SqUpf1snw==",
-            "dev": true,
+            "devOptional": true,
             "dependencies": {
                 "@types/react": "*"
             }
@@ -2947,6 +3366,25 @@
                 "node": ">= 6"
             }
         },
+        "node_modules/class-variance-authority": {
+            "version": "0.6.0",
+            "resolved": "https://registry.npmjs.org/class-variance-authority/-/class-variance-authority-0.6.0.tgz",
+            "integrity": "sha512-qdRDgfjx3GRb9fpwpSvn+YaidnT7IUJNe4wt5/SWwM+PmUwJUhQRk/8zAyNro0PmVfmen2635UboTjIBXXxy5A==",
+            "dependencies": {
+                "clsx": "1.2.1"
+            },
+            "funding": {
+                "url": "https://joebell.co.uk"
+            },
+            "peerDependencies": {
+                "typescript": ">= 4.5.5 < 6"
+            },
+            "peerDependenciesMeta": {
+                "typescript": {
+                    "optional": true
+                }
+            }
+        },
         "node_modules/classcat": {
             "version": "5.0.4",
             "resolved": "https://registry.npmjs.org/classcat/-/classcat-5.0.4.tgz",
@@ -4728,6 +5166,14 @@
                 "yallist": "^3.0.2"
             }
         },
+        "node_modules/lucide-react": {
+            "version": "0.233.0",
+            "resolved": "https://registry.npmjs.org/lucide-react/-/lucide-react-0.233.0.tgz",
+            "integrity": "sha512-r0jMHF0vPDq2wBbZ0B3rtIcBjDyWDKpHu+vAjD2OHn2WLUr3HN5IHovtO0EMgQXuSI7YrMZbjsEZWC2uBHr8nQ==",
+            "peerDependencies": {
+                "react": "^16.5.1 || ^17.0.0 || ^18.0.0"
+            }
+        },
         "node_modules/lz-string": {
             "version": "1.4.4",
             "resolved": "https://registry.npmjs.org/lz-string/-/lz-string-1.4.4.tgz",
@@ -7001,6 +7447,15 @@
             "resolved": "https://registry.npmjs.org/symbol-tree/-/symbol-tree-3.2.4.tgz",
             "integrity": "sha512-9QNk5KwDF+Bvz+PyObkmSYjI5ksVUYtjW7AU22r2NKcfLJcXp96hkDWU3+XndOsUb+AQ9QhfzfCT2O+CNWT5Tw=="
         },
+        "node_modules/tailwind-merge": {
+            "version": "1.13.0",
+            "resolved": "https://registry.npmjs.org/tailwind-merge/-/tailwind-merge-1.13.0.tgz",
+            "integrity": "sha512-mUTmDbcU+IhOvJ0c42eLQ/nRkvolTqfpVaVQRSxfJAv9TabS6Y2zW/1wKpKLdKzyL3Gh8j6NTLl6MWNmvOM6kA==",
+            "funding": {
+                "type": "github",
+                "url": "https://github.com/sponsors/dcastil"
+            }
+        },
         "node_modules/tailwindcss": {
             "version": "3.3.2",
             "resolved": "https://registry.npmjs.org/tailwindcss/-/tailwindcss-3.3.2.tgz",
@@ -7038,6 +7493,14 @@
                 "node": ">=14.0.0"
             }
         },
+        "node_modules/tailwindcss-animate": {
+            "version": "1.0.5",
+            "resolved": "https://registry.npmjs.org/tailwindcss-animate/-/tailwindcss-animate-1.0.5.tgz",
+            "integrity": "sha512-UU3qrOJ4lFQABY+MVADmBm+0KW3xZyhMdRvejwtXqYOL7YjHYxmuREFAZdmVG5LPe5E9CAst846SLC4j5I3dcw==",
+            "peerDependencies": {
+                "tailwindcss": ">=3.0.0 || insiders"
+            }
+        },
         "node_modules/terser": {
             "version": "5.16.3",
             "resolved": "https://registry.npmjs.org/terser/-/terser-5.16.3.tgz",
@@ -7156,7 +7619,7 @@
             "version": "5.0.4",
             "resolved": "https://registry.npmjs.org/typescript/-/typescript-5.0.4.tgz",
             "integrity": "sha512-cW9T5W9xY37cc+jfEnaUvX91foxtHkza3Nw3wkoF4sSlKn0MONdkdEndig/qPBWXNkmplh3NzayQzCiHM4/hqw==",
-            "dev": true,
+            "devOptional": true,
             "bin": {
                 "tsc": "bin/tsc",
                 "tsserver": "bin/tsserver"
diff --git a/src/frontend/package.json b/src/frontend/package.json
index bd846ce19..c7af46608 100644
--- a/src/frontend/package.json
+++ b/src/frontend/package.json
@@ -8,6 +8,7 @@
         "@headlessui/react": "^1.7.10",
         "@heroicons/react": "^2.0.15",
         "@mui/material": "^5.11.9",
+        "@radix-ui/react-tooltip": "^1.0.6",
         "@tabler/icons-react": "^2.18.0",
         "@tailwindcss/forms": "^0.5.3",
         "@tailwindcss/line-clamp": "^0.4.4",
@@ -15,7 +16,10 @@
         "ansi-to-html": "^0.7.2",
         "axios": "^1.3.2",
         "base64-js": "^1.5.1",
+        "class-variance-authority": "^0.6.0",
+        "clsx": "^1.2.1",
         "lodash": "^4.17.21",
+        "lucide-react": "^0.233.0",
         "react": "^18.2.0",
         "react-ace": "^10.1.0",
         "react-cookie": "^4.1.1",
@@ -32,6 +36,8 @@
         "rehype-mathjax": "^4.0.2",
         "remark-gfm": "^3.0.1",
         "remark-math": "^5.1.1",
+        "tailwind-merge": "^1.13.0",
+        "tailwindcss-animate": "^1.0.5",
         "uuid": "^9.0.0",
         "vite-plugin-svgr": "^3.2.0",
         "web-vitals": "^2.1.4"
diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index e26ae589f..d6b8781a4 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -1,6 +1,11 @@
 import { Handle, Position, useUpdateNodeInternals } from "reactflow";
 import Tooltip from "../../../../components/TooltipComponent";
-import { classNames, isValidConnection } from "../../../../utils";
+import {
+  classNames,
+  groupByFamily,
+  isValidConnection,
+  toFirstUpperCase,
+} from "../../../../utils";
 import { useContext, useEffect, useRef, useState } from "react";
 import InputComponent from "../../../../components/inputComponent";
 import ToggleComponent from "../../../../components/toggleComponent";
@@ -17,6 +22,8 @@ import IntComponent from "../../../../components/intComponent";
 import PromptAreaComponent from "../../../../components/promptComponent";
 import { nodeNames, nodeIcons } from "../../../../utils";
 import React from "react";
+import { nodeColors } from "../../../../utils";
+import ShadTooltip from "../../../../components/ShadTooltipComponent";
 
 export default function ParameterComponent({
   left,
@@ -30,8 +37,7 @@ export default function ParameterComponent({
   required = false,
 }: ParameterComponentType) {
   const ref = useRef(null);
-  const refParent = useRef("");
-  const refParentIcon = useRef(null);
+  const refHtml = useRef(null);
   const updateNodeInternals = useUpdateNodeInternals();
   const [position, setPosition] = useState(0);
   useEffect(() => {
@@ -55,16 +61,45 @@ export default function ParameterComponent({
   const [myData, setMyData] = useState(useContext(typesContext).data);
 
   useEffect(() => {
-    Object.keys(myData).forEach((d) => {
-      let keys = Object.keys(myData[d]).filter(
-        (nd) => nd.toLowerCase() == data.type.toLowerCase()
-      );
-      if (keys.length > 0) {
-        refParent.current = d;
-        refParentIcon.current = nodeIcons[d];
-      }
-    });
-  }, []);
+    const groupedObj = groupByFamily(myData, tooltipTitle);
+
+    refHtml.current = groupedObj.map((item, i) => (
+      <span
+        key={item}
+        className={classNames(
+          i > 0 ? "items-center flex mt-3" : "items-center flex"
+        )}
+      >
+        <div
+          className="h-5 w-5"
+          style={{
+            color: nodeColors[item.family],
+          }}
+        >
+          {React.createElement(nodeIcons[item.family])}
+        </div>
+        <span className="ps-2 text-gray-950">
+          {nodeNames[item.family] ?? ""}{" "}
+          <span className={classNames(left ? "hidden" : "")}>
+            {" "}
+            -&nbsp;
+            {item.type.split(", ").length > 2
+              ? item.type.split(", ").map((el, i) => (
+                  <>
+                    <span key={el}>
+                      {i == item.type.split(", ").length - 1
+                        ? el
+                        : (el += `, `)}
+                    </span>
+                    {i % 2 == 0 && i > 0 && <br></br>}
+                  </>
+                ))
+              : item.type}
+          </span>
+        </span>
+      </span>
+    ));
+  }, [tooltipTitle]);
 
   return (
     <div
@@ -86,24 +121,10 @@ export default function ParameterComponent({
           type === "int") ? (
           <></>
         ) : (
-          <Tooltip
-            title={
-              <>
-                <div className="flex center items-center">
-                  <div
-                    className="h-5 w-5"
-                    style={{
-                      color: color,
-                    }}
-                  >
-                    {React.createElement(refParentIcon.current)}
-                  </div>
-                  <span className="ms-2 items-center">
-                    {nodeNames[refParent?.current] ?? ""}
-                  </span>
-                </div>
-              </>
-            }
+          <ShadTooltip
+            delayDuration={1500}
+            content={refHtml.current}
+            side={left ? "left" : "right"}
           >
             <Handle
               type={left ? "target" : "source"}
@@ -121,7 +142,7 @@ export default function ParameterComponent({
                 top: position,
               }}
             ></Handle>
-          </Tooltip>
+          </ShadTooltip>
         )}
 
         {left === true &&
diff --git a/src/frontend/src/CustomNodes/GenericNode/index.tsx b/src/frontend/src/CustomNodes/GenericNode/index.tsx
index 1a7f93ae5..79a241160 100644
--- a/src/frontend/src/CustomNodes/GenericNode/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/index.tsx
@@ -28,8 +28,8 @@ import NodeModal from "../../modals/NodeModal";
 import { useCallback } from "react";
 import { TabsContext } from "../../contexts/tabsContext";
 import { debounce } from "../../utils";
-import TooltipReact from "../../components/ReactTooltipComponent";
 import Tooltip from "../../components/TooltipComponent";
+import ShadTooltip from "../../components/ShadTooltipComponent";
 export default function GenericNode({
   data,
   selected,
@@ -115,14 +115,9 @@ export default function GenericNode({
             }}
           />
           <div className="ml-2 truncate">
-            <TooltipReact
-              delayShow={1000}
-              selector={`node-selector-${data.type}`}
-              htmlContent={data.type}
-              position="top"
-            >
+            <ShadTooltip delayDuration={1500} content={data.type}>
               <div className="ml-2 truncate">{data.type}</div>
-            </TooltipReact>
+            </ShadTooltip>
           </div>
         </div>
         <div className="flex gap-3">
@@ -253,11 +248,7 @@ export default function GenericNode({
                         : toTitleCase(t)
                     }
                     name={t}
-                    tooltipTitle={
-                      "Type: " +
-                      data.node.template[t].type +
-                      (data.node.template[t].list ? " list" : "")
-                    }
+                    tooltipTitle={data.node.template[t].type}
                     required={data.node.template[t].required}
                     id={data.node.template[t].type + "|" + t + "|" + data.id}
                     left={true}
@@ -283,7 +274,7 @@ export default function GenericNode({
             data={data}
             color={nodeColors[types[data.type]] ?? nodeColors.unknown}
             title={data.type}
-            tooltipTitle={`Type: ${data.node.base_classes.join(" | ")}`}
+            tooltipTitle={`${data.node.base_classes.join("\n")}`}
             id={[data.type, data.id, ...data.node.base_classes].join("|")}
             type={data.node.base_classes.join("|")}
             left={false}
diff --git a/src/frontend/src/components/ReactTooltipComponent/index.tsx b/src/frontend/src/components/ReactTooltipComponent/index.tsx
index aa736c212..cb2a54f7c 100644
--- a/src/frontend/src/components/ReactTooltipComponent/index.tsx
+++ b/src/frontend/src/components/ReactTooltipComponent/index.tsx
@@ -37,13 +37,15 @@ const TooltipReact: FC<TooltipProps> = ({
         id={selector}
         content={content}
         className={classNames(
-          "!bg-white !text-xs !font-normal !text-gray-700 !shadow-md !opacity-100 z-20",
+          "!bg-white !text-xs !font-normal !text-gray-700 !shadow-md !opacity-100 z-[9999]",
           className
         )}
         place={position}
         clickable={clickable}
         isOpen={disabled ? false : undefined}
         delayShow={delayShow}
+        positionStrategy="absolute"
+        float={true}
       >
         {htmlContent && htmlContent}
       </ReactTooltip>
diff --git a/src/frontend/src/components/ShadTooltipComponent/index.tsx b/src/frontend/src/components/ShadTooltipComponent/index.tsx
new file mode 100644
index 000000000..a360f3ff0
--- /dev/null
+++ b/src/frontend/src/components/ShadTooltipComponent/index.tsx
@@ -0,0 +1,25 @@
+import {
+  Tooltip,
+  TooltipContent,
+  TooltipProvider,
+  TooltipTrigger,
+} from "../ui/tooltip";
+
+const ShadTooltip = (props) => {
+  return (
+    <TooltipProvider>
+      <Tooltip delayDuration={props.delayDuration}>
+        <TooltipTrigger asChild>{props.children}</TooltipTrigger>
+        <TooltipContent
+          side={props.side}
+          avoidCollisions={false}
+          sticky="always"
+        >
+          {props.content}
+        </TooltipContent>
+      </Tooltip>
+    </TooltipProvider>
+  );
+};
+
+export default ShadTooltip;
diff --git a/src/frontend/src/components/ui/tooltip.tsx b/src/frontend/src/components/ui/tooltip.tsx
new file mode 100644
index 000000000..b2ec0f701
--- /dev/null
+++ b/src/frontend/src/components/ui/tooltip.tsx
@@ -0,0 +1,29 @@
+"use client";
+
+import * as React from "react";
+import * as TooltipPrimitive from "@radix-ui/react-tooltip";
+import { cn } from "../../utils";
+
+const TooltipProvider = TooltipPrimitive.Provider;
+
+const Tooltip = TooltipPrimitive.Root;
+
+const TooltipTrigger = TooltipPrimitive.Trigger;
+
+const TooltipContent = React.forwardRef<
+  React.ElementRef<typeof TooltipPrimitive.Content>,
+  React.ComponentPropsWithoutRef<typeof TooltipPrimitive.Content>
+>(({ className, sideOffset = 4, ...props }, ref) => (
+  <TooltipPrimitive.Content
+    ref={ref}
+    sideOffset={sideOffset}
+    className={cn(
+      "z-50 overflow-hidden rounded-md border bg-popover px-3 py-1.5 text-sm text-popover-foreground shadow-md animate-in fade-in-50 data-[side=bottom]:slide-in-from-top-1 data-[side=left]:slide-in-from-right-1 data-[side=right]:slide-in-from-left-1 data-[side=top]:slide-in-from-bottom-1",
+      className
+    )}
+    {...props}
+  />
+));
+TooltipContent.displayName = TooltipPrimitive.Content.displayName;
+
+export { Tooltip, TooltipTrigger, TooltipContent, TooltipProvider };
diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 899fb1c87..8809ecc3a 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -2,6 +2,84 @@
 @tailwind components;
 @tailwind utilities;
 
+@layer base {
+	:root {
+	  --background: 0 0% 100%;
+	  --foreground: 222.2 47.4% 11.2%;
+   
+	  --muted: 210 40% 96.1%;
+	  --muted-foreground: 215.4 16.3% 46.9%;
+   
+	  --popover: 0 0% 100%;
+	  --popover-foreground: 222.2 47.4% 11.2%;
+   
+	  --card: 0 0% 100%;
+	  --card-foreground: 222.2 47.4% 11.2%;
+   
+	  --border: 214.3 31.8% 91.4%;
+	  --input: 214.3 31.8% 91.4%;
+   
+	  --primary: 222.2 47.4% 11.2%;
+	  --primary-foreground: 210 40% 98%;
+   
+	  --secondary: 210 40% 96.1%;
+	  --secondary-foreground: 222.2 47.4% 11.2%;
+   
+	  --accent: 210 40% 96.1%;
+	  --accent-foreground: 222.2 47.4% 11.2%;
+   
+	  --destructive: 0 100% 50%;
+	  --destructive-foreground: 210 40% 98%;
+   
+	  --ring: 215 20.2% 65.1%;
+   
+	  --radius: 0.5rem;
+	}
+   
+	.dark {
+	  --background: 224 71% 4%;
+	  --foreground: 213 31% 91%;
+   
+	  --muted: 223 47% 11%;
+	  --muted-foreground: 215.4 16.3% 56.9%;
+   
+	  --popover: 224 71% 4%;
+	  --popover-foreground: 215 20.2% 65.1%;
+   
+	  --card: 224 71% 4%;
+	  --card-foreground: 213 31% 91%;
+   
+	  --border: 216 34% 17%;
+	  --input: 216 34% 17%;
+   
+	  --primary: 210 40% 98%;
+	  --primary-foreground: 222.2 47.4% 1.2%;
+   
+	  --secondary: 222.2 47.4% 11.2%;
+	  --secondary-foreground: 210 40% 98%;
+   
+	  --accent: 216 34% 17%;
+	  --accent-foreground: 210 40% 98%;
+   
+	  --destructive: 0 63% 31%;
+	  --destructive-foreground: 210 40% 98%;
+   
+	  --ring: 216 34% 17%;
+   
+	  --radius: 0.5rem;
+	}
+  }
+   
+  @layer base {
+	* {
+	  @apply border-border;
+	}
+	body {
+	  @apply bg-background text-foreground;
+	  font-feature-settings: "rlig" 1, "calt" 1;
+	}
+  }
+
 body {
 	margin: 0;
 	font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "Roboto", "Oxygen",
@@ -15,3 +93,9 @@ code {
 	font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
 		monospace;
 }
+
+/* The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
+The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor. */
+.react-flow__pane {
+  cursor: default;
+}
\ No newline at end of file
diff --git a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
index e8dbfbb6b..2a8ef1217 100644
--- a/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/extraSidebarComponent/index.tsx
@@ -9,8 +9,8 @@ import {
 import { useContext, useEffect, useState, useRef } from "react";
 import { typesContext } from "../../../../contexts/typesContext";
 import { APIClassType, APIObjectType } from "../../../../types/api";
-import TooltipReact from "../../../../components/ReactTooltipComponent";
 import { MagnifyingGlassIcon } from "@heroicons/react/24/outline";
+import ShadTooltip from "../../../../components/ShadTooltipComponent";
 
 export default function ExtraSidebar() {
   const { data } = useContext(typesContext);
@@ -81,12 +81,10 @@ export default function ExtraSidebar() {
                   {Object.keys(dataFilter[d])
                     .sort()
                     .map((t: string, k) => (
-                      <TooltipReact
-                        selector={t}
-                        htmlContent={t}
-                        position="right"
-                        delayShow={1500}
-                        key={k}
+                      <ShadTooltip
+                        content={t}
+                        delayDuration={1500}
+                        side="right"
                       >
                         <div key={k} data-tooltip-id={t}>
                           <div
@@ -118,7 +116,7 @@ export default function ExtraSidebar() {
                             </div>
                           </div>
                         </div>
-                      </TooltipReact>
+                      </ShadTooltip>
                     ))}
                 </div>
               </DisclosureComponent>
diff --git a/src/frontend/src/utils.ts b/src/frontend/src/utils.ts
index 559080f3f..4e1ea4f47 100644
--- a/src/frontend/src/utils.ts
+++ b/src/frontend/src/utils.ts
@@ -47,6 +47,12 @@ import { WolframIcon } from "./icons/Wolfram";
 import { WordIcon } from "./icons/Word";
 import { SerperIcon } from "./icons/Serper";
 import { v4 as uuidv4 } from "uuid";
+import { clsx, type ClassValue } from "clsx";
+import { twMerge } from "tailwind-merge";
+
+export function cn(...inputs: ClassValue[]) {
+  return twMerge(clsx(inputs));
+}
 
 export function classNames(...classes: Array<string>) {
   return classes.filter(Boolean).join(" ");
@@ -632,3 +638,58 @@ export function updateIds(newFlow, getNodeId) {
       e.targetHandle;
   });
 }
+
+export function groupByFamily(data, baseClasses) {
+  let arrOfParent: string[] = [];
+  let arrOfType: { family: string; type: string }[] = [];
+
+  Object.keys(data).map((d) => {
+    Object.keys(data[d]).map((n) => {
+      if (
+        data[d][n].base_classes.some((r) => baseClasses.split("\n").includes(r))
+      ) {
+        arrOfParent.push(d);
+      }
+    });
+  });
+
+  let uniq = arrOfParent.filter(
+    (item, index) => arrOfParent.indexOf(item) === index
+  );
+
+  Object.keys(data).map((d) => {
+    Object.keys(data[d]).map((n) => {
+      baseClasses.split("\n").forEach((tol) => {
+        data[d][n].base_classes.forEach((data) => {
+          if (tol == data) {
+            arrOfType.push({
+              family: d,
+              type: data,
+            });
+          }
+        });
+      });
+    });
+  });
+
+  let groupedBy = arrOfType.filter((object, index, self) => {
+    const foundIndex = self.findIndex(
+      (o) => o.family === object.family && o.type === object.type
+    );
+    return foundIndex === index;
+  });
+
+  let groupedObj = groupedBy.reduce((result, item) => {
+    const existingGroup = result.find((group) => group.family === item.family);
+
+    if (existingGroup) {
+      existingGroup.type += `, ${item.type}`;
+    } else {
+      result.push({ family: item.family, type: item.type });
+    }
+
+    return result;
+  }, []);
+
+  return groupedObj;
+}
diff --git a/src/frontend/tailwind.config.js b/src/frontend/tailwind.config.js
index 8df8b8c1e..5644a21bc 100644
--- a/src/frontend/tailwind.config.js
+++ b/src/frontend/tailwind.config.js
@@ -1,11 +1,83 @@
 /** @type {import('tailwindcss').Config} */
+const { fontFamily } = require("tailwindcss/defaultTheme")
+
 import plugin from "tailwindcss/plugin";
 module.exports = {
   content: ["./index.html", "./src/**/*.{js,ts,tsx,jsx}"],
   darkMode: "class",
   important: true,
   theme: {
+    container: {
+      center: true,
+      padding: "2rem",
+      screens: {
+        "2xl": "1400px",
+      },
+    },
     extend: {
+      colors: {
+        border: "hsl(var(--border))",
+        input: "hsl(var(--input))",
+        ring: "hsl(var(--ring))",
+        background: "hsl(var(--background))",
+        foreground: "hsl(var(--foreground))",
+        primary: {
+          DEFAULT: "hsl(var(--primary))",
+          foreground: "hsl(var(--primary-foreground))",
+        },
+        secondary: {
+          DEFAULT: "hsl(var(--secondary))",
+          foreground: "hsl(var(--secondary-foreground))",
+        },
+        destructive: {
+          DEFAULT: "hsl(var(--destructive))",
+          foreground: "hsl(var(--destructive-foreground))",
+        },
+        muted: {
+          DEFAULT: "hsl(var(--muted))",
+          foreground: "hsl(var(--muted-foreground))",
+        },
+        accent: {
+          DEFAULT: "hsl(var(--accent))",
+          foreground: "hsl(var(--accent-foreground))",
+        },
+        popover: {
+          DEFAULT: "hsl(var(--popover))",
+          foreground: "hsl(var(--popover-foreground))",
+        },
+        card: {
+          DEFAULT: "hsl(var(--card))",
+          foreground: "hsl(var(--card-foreground))",
+        },
+      },
+      borderRadius: {
+        lg: `var(--radius)`,
+        md: `calc(var(--radius) - 2px)`,
+        sm: "calc(var(--radius) - 4px)",
+      },
+      fontFamily: {
+        sans: ["var(--font-sans)", ...fontFamily.sans],
+      },
+      keyframes: {
+        "accordion-down": {
+          from: { height: 0 },
+          to: { height: "var(--radix-accordion-content-height)" },
+        },
+        "accordion-up": {
+          from: { height: "var(--radix-accordion-content-height)" },
+          to: { height: 0 },
+        },
+        pulseGreen: {
+          "0%": { boxShadow: "0 0 0 0 rgba(72, 187, 120, 0.7)" },
+          "100%": { boxShadow: "0 0 0 10px rgba(72, 187, 120, 0)" },
+        },
+      },
+      animation: {
+        "accordion-down": "accordion-down 0.2s ease-out",
+        "accordion-up": "accordion-up 0.2s ease-out",
+        "pulse-green": "pulseGreen 1s linear",
+        'spin-once': 'spin 1s linear 0.7'
+      },
       borderColor: {
         "red-outline": "rgba(255, 0, 0, 0.8)",
         "green-outline": "rgba(72, 187, 120, 0.7)",
@@ -14,17 +86,6 @@ module.exports = {
         "red-outline": "0 0 5px rgba(255, 0, 0, 0.5)",
         "green-outline": "0 0 5px rgba(72, 187, 120, 0.7)",
       },
-
-      animation: {
-        "pulse-green": "pulseGreen 1s linear",
-        'spin-once': 'spin 1s linear 0.7'
-      },
-      keyframes: {
-        pulseGreen: {
-          "0%": { boxShadow: "0 0 0 0 rgba(72, 187, 120, 0.7)" },
-          "100%": { boxShadow: "0 0 0 10px rgba(72, 187, 120, 0)" },
-        },
-      },
     },
   },
   plugins: [
@@ -96,4 +157,4 @@ module.exports = {
     }),
     require("@tailwindcss/typography"),
   ],
-};
+};
\ No newline at end of file
diff --git a/src/frontend/tsconfig.json b/src/frontend/tsconfig.json
index 3a05105de..2ed005aff 100644
--- a/src/frontend/tsconfig.json
+++ b/src/frontend/tsconfig.json
@@ -19,7 +19,8 @@
     "isolatedModules": true,
     "noEmit": true,
     "jsx": "react-jsx",
-    "noImplicitAny": false
+    "noImplicitAny": false,
+    "baseUrl": "."
   },
   "include": [
     "src"

From f0ca7eeac8b5cbb37a23d1daf3e20293595228e1 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Fri, 2 Jun 2023 17:29:22 -0300
Subject: [PATCH 36/59] formatting with codacy

---
 src/frontend/src/index.css | 92 +++++++++++++++++---------------------
 1 file changed, 40 insertions(+), 52 deletions(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 962e3c025..7c41b934e 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -6,70 +6,50 @@
 	:root {
 	  --background: 0 0% 100%;
 	  --foreground: 222.2 47.4% 11.2%;
-   
 	  --muted: 210 40% 96.1%;
 	  --muted-foreground: 215.4 16.3% 46.9%;
-   
 	  --popover: 0 0% 100%;
 	  --popover-foreground: 222.2 47.4% 11.2%;
-   
 	  --card: 0 0% 100%;
 	  --card-foreground: 222.2 47.4% 11.2%;
-   
 	  --border: 214.3 31.8% 91.4%;
 	  --input: 214.3 31.8% 91.4%;
-   
 	  --primary: 222.2 47.4% 11.2%;
 	  --primary-foreground: 210 40% 98%;
-   
 	  --secondary: 210 40% 96.1%;
 	  --secondary-foreground: 222.2 47.4% 11.2%;
-   
 	  --accent: 210 40% 96.1%;
 	  --accent-foreground: 222.2 47.4% 11.2%;
-   
 	  --destructive: 0 100% 50%;
 	  --destructive-foreground: 210 40% 98%;
-   
 	  --ring: 215 20.2% 65.1%;
-   
 	  --radius: 0.5rem;
 	}
-   
+  
 	.dark {
-	  --background: 224 71% 4%;
-	  --foreground: 213 31% 91%;
-   
-	  --muted: 223 47% 11%;
-	  --muted-foreground: 215.4 16.3% 56.9%;
-   
-	  --popover: 224 71% 4%;
-	  --popover-foreground: 215 20.2% 65.1%;
-   
-	  --card: 224 71% 4%;
-	  --card-foreground: 213 31% 91%;
-   
-	  --border: 216 34% 17%;
-	  --input: 216 34% 17%;
-   
-	  --primary: 210 40% 98%;
-	  --primary-foreground: 222.2 47.4% 1.2%;
-   
-	  --secondary: 222.2 47.4% 11.2%;
-	  --secondary-foreground: 210 40% 98%;
-   
-	  --accent: 216 34% 17%;
-	  --accent-foreground: 210 40% 98%;
-   
-	  --destructive: 0 63% 31%;
-	  --destructive-foreground: 210 40% 98%;
-   
-	  --ring: 216 34% 17%;
-   
-	  --radius: 0.5rem;
+	  -background: 224 71% 4%;
+	  -foreground: 213 31% 91%;
+	  -muted: 223 47% 11%;
+	  -muted-foreground: 215.4 16.3% 56.9%;
+	  -popover: 224 71% 4%;
+	  -popover-foreground: 215 20.2% 65.1%;
+	  -card: 224 71% 4%;
+	  -card-foreground: 213 31% 91%;
+	  -border: 216 34% 17%;
+	  -input: 216 34% 17%;
+	  -primary: 210 40% 98%;
+	  -primary-foreground: 222.2 47.4% 1.2%;
+	  -secondary: 222.2 47.4% 11.2%;
+	  -secondary-foreground: 210 40% 98%;
+	  -accent: 216 34% 17%;
+	  -accent-foreground: 210 40% 98%;
+	  -destructive: 0 63% 31%;
+	  -destructive-foreground: 210 40% 98%;
+	  -ring: 216 34% 17%;
+	  -radius: 0.5rem;
 	}
   }
-   
+  
   @layer base {
 	* {
 	  @apply border-border;
@@ -80,22 +60,30 @@
 	}
   }
 
+@layer base {
+  * {
+    @apply border-border;
+  }
+  body {
+    @apply bg-background text-foreground;
+    font-feature-settings: "rlig" 1, "calt" 1;
+  }
+}
+
 body {
-	margin: 0;
-	font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "Roboto", "Oxygen",
-		"Ubuntu", "Cantarell", "Fira Sans", "Droid Sans", "Helvetica Neue",
-		sans-serif;
-	-webkit-font-smoothing: antialiased;
-	-moz-osx-font-smoothing: grayscale;
+  margin: 0;
+  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", "Roboto", "Oxygen",
+    "Ubuntu", "Cantarell", "Fira Sans", "Droid Sans", "Helvetica Neue",
+    sans-serif;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
 }
 
 code {
-	font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
-		monospace;
+  font-family: source-code-pro, Menlo, Monaco, Consolas, "Courier New",
+    monospace;
 }
 
-/* The style below sets the cursor property of the element with the class .react-flow__pane to the default cursor.
-The cursor: default; property value restores the browser's default cursor style for the targeted element. By applying this style, the element will no longer have a custom cursor appearance such as "grab" or any other custom cursor defined elsewhere in the application. Instead, it will revert to the default cursor style determined by the browser, typically an arrow-shaped cursor. */
 .react-flow__pane {
   cursor: default;
 }

From c8665ef42241b8b529568754d7d7b9199fdae9a9 Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Fri, 2 Jun 2023 17:35:30 -0300
Subject: [PATCH 37/59] Fixing css formatting codacy

---
 src/frontend/src/index.css | 86 +++++++++++++++++++-------------------
 1 file changed, 43 insertions(+), 43 deletions(-)

diff --git a/src/frontend/src/index.css b/src/frontend/src/index.css
index 7c41b934e..14580c658 100644
--- a/src/frontend/src/index.css
+++ b/src/frontend/src/index.css
@@ -3,52 +3,52 @@
 @tailwind utilities;
 
 @layer base {
-	:root {
-	  --background: 0 0% 100%;
-	  --foreground: 222.2 47.4% 11.2%;
-	  --muted: 210 40% 96.1%;
-	  --muted-foreground: 215.4 16.3% 46.9%;
-	  --popover: 0 0% 100%;
-	  --popover-foreground: 222.2 47.4% 11.2%;
-	  --card: 0 0% 100%;
-	  --card-foreground: 222.2 47.4% 11.2%;
-	  --border: 214.3 31.8% 91.4%;
-	  --input: 214.3 31.8% 91.4%;
-	  --primary: 222.2 47.4% 11.2%;
-	  --primary-foreground: 210 40% 98%;
-	  --secondary: 210 40% 96.1%;
-	  --secondary-foreground: 222.2 47.4% 11.2%;
-	  --accent: 210 40% 96.1%;
-	  --accent-foreground: 222.2 47.4% 11.2%;
-	  --destructive: 0 100% 50%;
-	  --destructive-foreground: 210 40% 98%;
-	  --ring: 215 20.2% 65.1%;
-	  --radius: 0.5rem;
+  :root {
+    --background: 0 0% 100%;
+    --foreground: 222.2 47.4% 11.2%;
+    --muted: 210 40% 96.1%;
+    --muted-foreground: 215.4 16.3% 46.9%;
+    --popover: 0 0% 100%;
+    --popover-foreground: 222.2 47.4% 11.2%;
+    --card: 0 0% 100%;
+    --card-foreground: 222.2 47.4% 11.2%;
+    --border: 214.3 31.8% 91.4%;
+    --input: 214.3 31.8% 91.4%;
+    --primary: 222.2 47.4% 11.2%;
+    --primary-foreground: 210 40% 98%;
+    --secondary: 210 40% 96.1%;
+    --secondary-foreground: 222.2 47.4% 11.2%;
+    --accent: 210 40% 96.1%;
+    --accent-foreground: 222.2 47.4% 11.2%;
+    --destructive: 0 100% 50%;
+    --destructive-foreground: 210 40% 98%;
+    --ring: 215 20.2% 65.1%;
+    --radius: 0.5rem;
 	}
   
-	.dark {
-	  -background: 224 71% 4%;
-	  -foreground: 213 31% 91%;
-	  -muted: 223 47% 11%;
-	  -muted-foreground: 215.4 16.3% 56.9%;
-	  -popover: 224 71% 4%;
-	  -popover-foreground: 215 20.2% 65.1%;
-	  -card: 224 71% 4%;
-	  -card-foreground: 213 31% 91%;
-	  -border: 216 34% 17%;
-	  -input: 216 34% 17%;
-	  -primary: 210 40% 98%;
-	  -primary-foreground: 222.2 47.4% 1.2%;
-	  -secondary: 222.2 47.4% 11.2%;
-	  -secondary-foreground: 210 40% 98%;
-	  -accent: 216 34% 17%;
-	  -accent-foreground: 210 40% 98%;
-	  -destructive: 0 63% 31%;
-	  -destructive-foreground: 210 40% 98%;
-	  -ring: 216 34% 17%;
-	  -radius: 0.5rem;
+  .dark {
+    -background: 224 71% 4%;
+    -foreground: 213 31% 91%;
+    -muted: 223 47% 11%;
+    -muted-foreground: 215.4 16.3% 56.9%;
+    -popover: 224 71% 4%;
+    -popover-foreground: 215 20.2% 65.1%;
+    -card: 224 71% 4%;
+    -card-foreground: 213 31% 91%;
+    -border: 216 34% 17%;
+    -input: 216 34% 17%;
+    -primary: 210 40% 98%;
+    -primary-foreground: 222.2 47.4% 1.2%;
+    -secondary: 222.2 47.4% 11.2%;
+    -secondary-foreground: 210 40% 98%;
+    -accent: 216 34% 17%;
+    -accent-foreground: 210 40% 98%;
+    -destructive: 0 63% 31%;
+    -destructive-foreground: 210 40% 98%;
+    -ring: 216 34% 17%;
+    -radius: 0.5rem;
 	}
-  }
+}
   
   @layer base {
 	* {

From 1bd3303c52bcf8befd7da89abdb396ba39b14b8b Mon Sep 17 00:00:00 2001
From: Cristhian Zanforlin Lousa <cristhian.lousa@gmail.com>
Date: Fri, 2 Jun 2023 17:49:53 -0300
Subject: [PATCH 38/59] Change the delay to open to instant

---
 .../GenericNode/components/parameterComponent/index.tsx         | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
index d6b8781a4..dc77c2877 100644
--- a/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
+++ b/src/frontend/src/CustomNodes/GenericNode/components/parameterComponent/index.tsx
@@ -122,7 +122,7 @@ export default function ParameterComponent({
           <></>
         ) : (
           <ShadTooltip
-            delayDuration={1500}
+            delayDuration={0}
             content={refHtml.current}
             side={left ? "left" : "right"}
           >

From eeee895f6ad3c9cb3d73b54097568ee39bda636c Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Sun, 4 Jun 2023 21:42:14 -0300
Subject: [PATCH 39/59] =?UTF-8?q?=F0=9F=94=92=20chore(pyproject.toml):=20u?=
 =?UTF-8?q?pdate=20fastapi=20dependency=20to=20version=200.96.0=20FastAPI?=
 =?UTF-8?q?=20dependency=20has=20been=20updated=20to=20version=200.96.0=20?=
 =?UTF-8?q?to=20ensure=20that=20the=20application=20is=20using=20the=20lat?=
 =?UTF-8?q?est=20version=20of=20the=20framework.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 poetry.lock    | 8 ++++----
 pyproject.toml | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index 69619220a..58498c4d1 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1265,14 +1265,14 @@ importlib-resources = {version = ">=5.0", markers = "python_version < \"3.10\""}
 
 [[package]]
 name = "fastapi"
-version = "0.95.2"
+version = "0.96.0"
 description = "FastAPI framework, high performance, easy to learn, fast to code, ready for production"
 category = "main"
 optional = false
 python-versions = ">=3.7"
 files = [
-    {file = "fastapi-0.95.2-py3-none-any.whl", hash = "sha256:d374dbc4ef2ad9b803899bd3360d34c534adc574546e25314ab72c0c4411749f"},
-    {file = "fastapi-0.95.2.tar.gz", hash = "sha256:4d9d3e8c71c73f11874bcf5e33626258d143252e329a01002f767306c64fb982"},
+    {file = "fastapi-0.96.0-py3-none-any.whl", hash = "sha256:b8e11fe81e81eab4e1504209917338e0b80f783878a42c2b99467e5e1019a1e9"},
+    {file = "fastapi-0.96.0.tar.gz", hash = "sha256:71232d47c2787446991c81c41c249f8a16238d52d779c0e6b43927d3773dbe3c"},
 ]
 
 [package.dependencies]
@@ -6234,4 +6234,4 @@ deploy = ["langchain-serve"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.12"
-content-hash = "9dbfdba31c45b1d79c6b487cb4e5e9cb43252e2f2f52fe625648162bf4cd4d9a"
+content-hash = "4cac7dea0c1222711ba7eed82d5716d5e361d454edb6e0299b387fbb115d2c3d"
diff --git a/pyproject.toml b/pyproject.toml
index 780aa1b90..cde7ddca5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,7 +22,7 @@ langflow = "langflow.__main__:main"
 
 [tool.poetry.dependencies]
 python = ">=3.9,<3.12"
-fastapi = "^0.95.0"
+fastapi = "^0.96.0"
 uvicorn = "^0.20.0"
 beautifulsoup4 = "^4.11.2"
 google-search-results = "^2.4.1"

From 5b28bbb7953f96af64f4b7c5fd020f1923956343 Mon Sep 17 00:00:00 2001
From: Jacob Lee <jacob.lee@tempus.com>
Date: Mon, 5 Jun 2023 08:56:44 -0500
Subject: [PATCH 40/59] Initial support for Azure LLM nodes.

There are still some rough edges due to underlying langchain and
openai API limitations, e.g. hwchase17/langchain#3769 and
openai/openai-python#411. Notably, you can't use the Azure and
non-Azure node types in the same server, since there's global openai
configuration needed to choose between the two. So it's probably best
to still leave the Azure node types commented out in the default
config. But with this PR, if you uncomment those nodes and start the
server with OPENAI_API_TYPE=azure, you will have working Azure nodes.
---
 src/backend/langflow/config.yaml              |  1 +
 .../langflow/interface/custom_lists.py        |  3 +-
 .../langflow/template/frontend_node/llms.py   | 22 +++++-
 tests/test_llms_template.py                   | 74 +++++++++++++++++++
 4 files changed, 98 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index 963c29549..effeeb001 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -51,6 +51,7 @@ embeddings:
 llms:
   - OpenAI
   # - AzureOpenAI
+  # - AzureChatOpenAI
   - ChatOpenAI
   - LlamaCpp
   - CTransformers
diff --git a/src/backend/langflow/interface/custom_lists.py b/src/backend/langflow/interface/custom_lists.py
index a944363ae..34bc0103e 100644
--- a/src/backend/langflow/interface/custom_lists.py
+++ b/src/backend/langflow/interface/custom_lists.py
@@ -11,14 +11,15 @@ from langchain import (
     text_splitter,
 )
 from langchain.agents import agent_toolkits
+from langchain.chat_models import AzureChatOpenAI, ChatOpenAI
 from langchain.chat_models import ChatAnthropic
-from langchain.chat_models import ChatOpenAI
 
 from langflow.interface.importing.utils import import_class
 
 ## LLMs
 llm_type_to_cls_dict = llms.type_to_cls_dict
 llm_type_to_cls_dict["anthropic-chat"] = ChatAnthropic  # type: ignore
+llm_type_to_cls_dict["azure-chat"] = AzureChatOpenAI  # type: ignore
 llm_type_to_cls_dict["openai-chat"] = ChatOpenAI  # type: ignore
 
 ## Chains
diff --git a/src/backend/langflow/template/frontend_node/llms.py b/src/backend/langflow/template/frontend_node/llms.py
index 272770e2e..39e82422f 100644
--- a/src/backend/langflow/template/frontend_node/llms.py
+++ b/src/backend/langflow/template/frontend_node/llms.py
@@ -12,6 +12,18 @@ class LLMFrontendNode(FrontendNode):
                 field.name.title().replace("Openai", "OpenAI").replace("_", " ")
             ).replace("Api", "API")
 
+    @staticmethod
+    def format_azure_field(field: TemplateField):
+        if field.name == "model_name":
+            field.show = False  # Azure uses deployment_name instead of model_name.
+        if field.name == "openai_api_type":
+            field.show = False
+            field.password = False
+            field.value = "azure"
+        if field.name == "openai_api_version":
+            field.password = False
+            field.value = "2023-03-15-preview"
+
     @staticmethod
     def format_field(field: TemplateField, name: Optional[str] = None) -> None:
         display_names_dict = {
@@ -43,8 +55,16 @@ class LLMFrontendNode(FrontendNode):
             field.field_type = "code"
             field.advanced = True
             field.show = True
-        elif field.name in ["model_name", "temperature", "model_file", "model_type"]:
+        elif field.name in [
+            "model_name",
+            "temperature",
+            "model_file",
+            "model_type",
+            "deployment_name",
+        ]:
             field.advanced = False
             field.show = True
 
         LLMFrontendNode.format_openai_field(field)
+        if "azure" in name.lower():
+            LLMFrontendNode.format_azure_field(field)
diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index ccf2f6388..f54b452f1 100644
--- a/tests/test_llms_template.py
+++ b/tests/test_llms_template.py
@@ -482,3 +482,77 @@ def test_chat_open_ai(client: TestClient):
         "ChatOpenAI",
         "BaseLanguageModel",
     }
+
+
+def test_azure_open_ai(client: TestClient):
+    response = client.get("/all")
+    assert response.status_code == 200
+    json_response = response.json()
+    language_models = json_response["llms"]
+
+    model = language_models["AzureOpenAI"]
+    template = model["template"]
+
+    assert template["model_name"].show is False
+    assert template["deployment_name"] == {
+        "required": False,
+        "placeholder": "",
+        "show": True,
+        "multiline": False,
+        "value": "",
+        "password": False,
+        "name": "deployment_name",
+        "advanced": False,
+        "type": "str",
+        "list": False,
+    }
+
+
+def test_azure_chat_open_ai(client: TestClient):
+    response = client.get("/all")
+    assert response.status_code == 200
+    json_response = response.json()
+    language_models = json_response["llms"]
+
+    model = language_models["AzureChatOpenAI"]
+    template = model["template"]
+
+    assert template["model_name"].show is False
+    assert template["deployment_name"] == {
+        "required": False,
+        "placeholder": "",
+        "show": True,
+        "multiline": False,
+        "value": "",
+        "password": False,
+        "name": "deployment_name",
+        "advanced": False,
+        "type": "str",
+        "list": False,
+    }
+    assert template["openai_api_type"] == {
+        "required": False,
+        "placeholder": "",
+        "show": False,
+        "multiline": False,
+        "value": "azure",
+        "password": False,
+        "name": "openai_api_type",
+        "display_name": "OpenAI API Type",
+        "advanced": False,
+        "type": "str",
+        "list": False,
+    }
+    assert template["openai_api_version"] == {
+        "required": False,
+        "placeholder": "",
+        "show": True,
+        "multiline": False,
+        "value": "2023-03-15-preview",
+        "password": False,
+        "name": "openai_api_version",
+        "display_name": "OpenAI API Version",
+        "advanced": False,
+        "type": "str",
+        "list": False,
+    }

From 5a103f0f6f0b5dedb7dd38b81d65e84d011c10ce Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Mon, 5 Jun 2023 13:48:17 -0300
Subject: [PATCH 41/59] =?UTF-8?q?=F0=9F=94=A8=20refactor(base.py):=20extra?=
 =?UTF-8?q?ct=20methods=20to=20handle=20model-specific=20and=20API=20key-s?=
 =?UTF-8?q?pecific=20field=20values=20This=20commit=20refactors=20the=20Fr?=
 =?UTF-8?q?ontendNode=20class=20by=20extracting=20two=20methods=20to=20han?=
 =?UTF-8?q?dle=20specific=20field=20values=20related=20to=20models=20and?=
 =?UTF-8?q?=20API=20keys.=20The=20=5Fhandle=5Fmodel=5Fspecific=5Ffield=5Fv?=
 =?UTF-8?q?alues=20method=20handles=20the=20options=20and=20is=5Flist=20pr?=
 =?UTF-8?q?operties=20for=20fields=20related=20to=20models,=20while=20the?=
 =?UTF-8?q?=20=5Fhandle=5Fapi=5Fkey=5Fspecific=5Ffield=5Fvalues=20method?=
 =?UTF-8?q?=20handles=20the=20display=5Fname=20and=20required=20properties?=
 =?UTF-8?q?=20for=20fields=20related=20to=20API=20keys.=20This=20improves?=
 =?UTF-8?q?=20the=20readability=20and=20maintainability=20of=20the=20code.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../langflow/template/frontend_node/base.py   | 33 +++++++++++++------
 1 file changed, 23 insertions(+), 10 deletions(-)

diff --git a/src/backend/langflow/template/frontend_node/base.py b/src/backend/langflow/template/frontend_node/base.py
index a97c7b8b0..6d00cead0 100644
--- a/src/backend/langflow/template/frontend_node/base.py
+++ b/src/backend/langflow/template/frontend_node/base.py
@@ -117,17 +117,30 @@ class FrontendNode(BaseModel):
     ) -> None:
         """Handles specific field values for certain fields."""
         if key == "headers":
-            field.value = """{'Authorization':
-            'Bearer <token>'}"""
-        if name == "OpenAI" and key == "model_name":
-            field.options = constants.OPENAI_MODELS
-            field.is_list = True
-        elif name == "ChatOpenAI" and key == "model_name":
-            field.options = constants.CHAT_OPENAI_MODELS
-            field.is_list = True
-        elif (name == "Anthropic" or name == "ChatAnthropic") and key == "model_name":
-            field.options = constants.ANTHROPIC_MODELS
+            field.value = """{'Authorization': 'Bearer <token>'}"""
+        FrontendNode._handle_model_specific_field_values(field, key, name)
+        FrontendNode._handle_api_key_specific_field_values(field, key, name)
+
+    @staticmethod
+    def _handle_model_specific_field_values(
+        field: TemplateField, key: str, name: Optional[str] = None
+    ) -> None:
+        """Handles specific field values related to models."""
+        model_dict = {
+            "OpenAI": constants.OPENAI_MODELS,
+            "ChatOpenAI": constants.CHAT_OPENAI_MODELS,
+            "Anthropic": constants.ANTHROPIC_MODELS,
+            "ChatAnthropic": constants.ANTHROPIC_MODELS,
+        }
+        if name in model_dict and key == "model_name":
+            field.options = model_dict[name]
             field.is_list = True
+
+    @staticmethod
+    def _handle_api_key_specific_field_values(
+        field: TemplateField, key: str, name: Optional[str] = None
+    ) -> None:
+        """Handles specific field values related to API keys."""
         if "api_key" in key and "OpenAI" in str(name):
             field.display_name = "OpenAI API Key"
             field.required = False

From 79c677fb090a975a15f42fdea21432206640dddf Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 09:58:30 -0300
Subject: [PATCH 42/59] =?UTF-8?q?=F0=9F=9A=80=20feat(pyproject.toml):=20ad?=
 =?UTF-8?q?d=20pytest=20configuration=20options=20Added=20pytest=20configu?=
 =?UTF-8?q?ration=20options=20to=20the=20pyproject.toml=20file.=20The=20mi?=
 =?UTF-8?q?nimum=20version=20of=20pytest=20is=20set=20to=206.0,=20the=20'-?=
 =?UTF-8?q?ra'=20option=20is=20added=20to=20addopts=20to=20show=20all=20te?=
 =?UTF-8?q?st=20results,=20testpaths=20are=20set=20to=20include=20both=20'?=
 =?UTF-8?q?tests'=20and=20'integration'=20directories,=20console=20output?=
 =?UTF-8?q?=20style=20is=20set=20to=20'progress',=20and=20DeprecationWarni?=
 =?UTF-8?q?ng=20is=20ignored.=20log=5Fcli=20is=20set=20to=20true=20to=20en?=
 =?UTF-8?q?able=20logging=20of=20pytest=20output=20to=20the=20console.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pyproject.toml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index cde7ddca5..87843198b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -78,6 +78,15 @@ types-pillow = "^9.5.0.2"
 [tool.poetry.extras]
 deploy = ["langchain-serve"]
 
+[tool.pytest.ini_options]
+minversion = "6.0"
+addopts = "-ra"
+testpaths = ["tests", "integration"]
+console_output_style = "progress"
+filterwarnings = ["ignore::DeprecationWarning"]
+log_cli = true
+
+
 [tool.ruff]
 line-length = 120
 

From 3342e03a2cb75bcbcc36060b5204bee89ba773af Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 09:59:37 -0300
Subject: [PATCH 43/59] =?UTF-8?q?=F0=9F=94=80=20refactor(langflow):=20move?=
 =?UTF-8?q?=20routers=20to=20a=20single=20file=20and=20add=20health=20chec?=
 =?UTF-8?q?k=20endpoint=20The=20routers=20for=20the=20langflow=20API=20hav?=
 =?UTF-8?q?e=20been=20moved=20to=20a=20single=20file=20for=20better=20orga?=
 =?UTF-8?q?nization=20and=20maintainability.=20The=20routers=20have=20been?=
 =?UTF-8?q?=20imported=20and=20included=20in=20the=20main.py=20file=20usin?=
 =?UTF-8?q?g=20the=20new=20file.=20A=20new=20health=20check=20endpoint=20h?=
 =?UTF-8?q?as=20been=20added=20to=20the=20API=20to=20check=20the=20status?=
 =?UTF-8?q?=20of=20the=20application.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/__init__.py |  2 +-
 src/backend/langflow/main.py     | 12 ++++++------
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/backend/langflow/__init__.py b/src/backend/langflow/__init__.py
index 35fe814d2..17b1d940c 100644
--- a/src/backend/langflow/__init__.py
+++ b/src/backend/langflow/__init__.py
@@ -1,4 +1,4 @@
 from langflow.cache import cache_manager
-from langflow.interface.loading import load_flow_from_json
+from langflow.processing.process import load_flow_from_json
 
 __all__ = ["load_flow_from_json", "cache_manager"]
diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py
index 56cc32e46..de39d8750 100644
--- a/src/backend/langflow/main.py
+++ b/src/backend/langflow/main.py
@@ -1,9 +1,7 @@
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 
-from langflow.api.chat import router as chat_router
-from langflow.api.endpoints import router as endpoints_router
-from langflow.api.validate import router as validate_router
+from langflow.api import router
 
 
 def create_app():
@@ -14,6 +12,10 @@ def create_app():
         "*",
     ]
 
+    @app.get("/health")
+    def get_health():
+        return {"status": "OK"}
+
     app.add_middleware(
         CORSMiddleware,
         allow_origins=origins,
@@ -22,9 +24,7 @@ def create_app():
         allow_headers=["*"],
     )
 
-    app.include_router(endpoints_router)
-    app.include_router(validate_router)
-    app.include_router(chat_router)
+    app.include_router(router)
     return app
 
 

From ac42e8a66c1ce0c31c732cdcbb35980aecb96c69 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:00:15 -0300
Subject: [PATCH 44/59] chore: remove refactored files

---
 src/backend/langflow/api/base.py         |  84 ---------
 src/backend/langflow/api/callback.py     |  32 ----
 src/backend/langflow/api/chat.py         |  26 ---
 src/backend/langflow/api/chat_manager.py | 223 -----------------------
 src/backend/langflow/api/endpoints.py    |  47 -----
 src/backend/langflow/api/schemas.py      |  70 -------
 src/backend/langflow/api/validate.py     |  57 ------
 7 files changed, 539 deletions(-)
 delete mode 100644 src/backend/langflow/api/base.py
 delete mode 100644 src/backend/langflow/api/callback.py
 delete mode 100644 src/backend/langflow/api/chat.py
 delete mode 100644 src/backend/langflow/api/chat_manager.py
 delete mode 100644 src/backend/langflow/api/endpoints.py
 delete mode 100644 src/backend/langflow/api/schemas.py
 delete mode 100644 src/backend/langflow/api/validate.py

diff --git a/src/backend/langflow/api/base.py b/src/backend/langflow/api/base.py
deleted file mode 100644
index 8cddc52e4..000000000
--- a/src/backend/langflow/api/base.py
+++ /dev/null
@@ -1,84 +0,0 @@
-from pydantic import BaseModel, validator
-
-from langflow.graph.utils import extract_input_variables_from_prompt
-
-
-class CacheResponse(BaseModel):
-    data: dict
-
-
-class Code(BaseModel):
-    code: str
-
-
-class Prompt(BaseModel):
-    template: str
-
-
-# Build ValidationResponse class for {"imports": {"errors": []}, "function": {"errors": []}}
-class CodeValidationResponse(BaseModel):
-    imports: dict
-    function: dict
-
-    @validator("imports")
-    def validate_imports(cls, v):
-        return v or {"errors": []}
-
-    @validator("function")
-    def validate_function(cls, v):
-        return v or {"errors": []}
-
-
-class PromptValidationResponse(BaseModel):
-    input_variables: list
-
-
-INVALID_CHARACTERS = {
-    " ",
-    ",",
-    ".",
-    ":",
-    ";",
-    "!",
-    "?",
-    "/",
-    "\\",
-    "(",
-    ")",
-    "[",
-    "]",
-    "{",
-    "}",
-}
-
-
-def validate_prompt(template: str):
-    input_variables = extract_input_variables_from_prompt(template)
-
-    # Check if there are invalid characters in the input_variables
-    input_variables = check_input_variables(input_variables)
-
-    return PromptValidationResponse(input_variables=input_variables)
-
-
-def check_input_variables(input_variables: list):
-    invalid_chars = []
-    fixed_variables = []
-    for variable in input_variables:
-        new_var = variable
-        for char in INVALID_CHARACTERS:
-            if char in variable:
-                invalid_chars.append(char)
-                new_var = new_var.replace(char, "")
-        fixed_variables.append(new_var)
-        if new_var != variable:
-            input_variables.remove(variable)
-            input_variables.append(new_var)
-    # If any of the input_variables is not in the fixed_variables, then it means that
-    # there are invalid characters in the input_variables
-    if any(var not in fixed_variables for var in input_variables):
-        raise ValueError(
-            f"Invalid input variables: {input_variables}. Please, use something like {fixed_variables} instead."
-        )
-
-    return input_variables
diff --git a/src/backend/langflow/api/callback.py b/src/backend/langflow/api/callback.py
deleted file mode 100644
index d63e107c4..000000000
--- a/src/backend/langflow/api/callback.py
+++ /dev/null
@@ -1,32 +0,0 @@
-import asyncio
-from typing import Any
-
-from langchain.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
-
-from langflow.api.schemas import ChatResponse
-
-
-# https://github.com/hwchase17/chat-langchain/blob/master/callback.py
-class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
-    """Callback handler for streaming LLM responses."""
-
-    def __init__(self, websocket):
-        self.websocket = websocket
-
-    async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
-        resp = ChatResponse(message=token, type="stream", intermediate_steps="")
-        await self.websocket.send_json(resp.dict())
-
-
-class StreamingLLMCallbackHandler(BaseCallbackHandler):
-    """Callback handler for streaming LLM responses."""
-
-    def __init__(self, websocket):
-        self.websocket = websocket
-
-    def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
-        resp = ChatResponse(message=token, type="stream", intermediate_steps="")
-
-        loop = asyncio.get_event_loop()
-        coroutine = self.websocket.send_json(resp.dict())
-        asyncio.run_coroutine_threadsafe(coroutine, loop)
diff --git a/src/backend/langflow/api/chat.py b/src/backend/langflow/api/chat.py
deleted file mode 100644
index 4afa6c22f..000000000
--- a/src/backend/langflow/api/chat.py
+++ /dev/null
@@ -1,26 +0,0 @@
-from fastapi import (
-    APIRouter,
-    WebSocket,
-    WebSocketDisconnect,
-    WebSocketException,
-    status,
-)
-
-from langflow.api.chat_manager import ChatManager
-from langflow.utils.logger import logger
-
-router = APIRouter()
-chat_manager = ChatManager()
-
-
-@router.websocket("/chat/{client_id}")
-async def websocket_endpoint(client_id: str, websocket: WebSocket):
-    """Websocket endpoint for chat."""
-    try:
-        await chat_manager.handle_websocket(client_id, websocket)
-    except WebSocketException as exc:
-        logger.error(exc)
-        await websocket.close(code=status.WS_1011_INTERNAL_ERROR, reason=str(exc))
-    except WebSocketDisconnect as exc:
-        logger.error(exc)
-        await websocket.close(code=status.WS_1000_NORMAL_CLOSURE, reason=str(exc))
diff --git a/src/backend/langflow/api/chat_manager.py b/src/backend/langflow/api/chat_manager.py
deleted file mode 100644
index 8b1c7a621..000000000
--- a/src/backend/langflow/api/chat_manager.py
+++ /dev/null
@@ -1,223 +0,0 @@
-import asyncio
-import json
-from collections import defaultdict
-from typing import Dict, List
-
-from fastapi import WebSocket, status
-
-from langflow.api.schemas import ChatMessage, ChatResponse, FileResponse
-from langflow.cache import cache_manager
-from langflow.cache.manager import Subject
-from langflow.interface.run import (
-    get_result_and_steps,
-    load_or_build_langchain_object,
-)
-from langflow.interface.utils import pil_to_base64, try_setting_streaming_options
-from langflow.utils.logger import logger
-
-
-class ChatHistory(Subject):
-    def __init__(self):
-        super().__init__()
-        self.history: Dict[str, List[ChatMessage]] = defaultdict(list)
-
-    def add_message(self, client_id: str, message: ChatMessage):
-        """Add a message to the chat history."""
-
-        self.history[client_id].append(message)
-
-        if not isinstance(message, FileResponse):
-            self.notify()
-
-    def get_history(self, client_id: str, filter_messages=True) -> List[ChatMessage]:
-        """Get the chat history for a client."""
-        if history := self.history.get(client_id, []):
-            if filter_messages:
-                return [msg for msg in history if msg.type not in ["start", "stream"]]
-            return history
-        else:
-            return []
-
-    def empty_history(self, client_id: str):
-        """Empty the chat history for a client."""
-        self.history[client_id] = []
-
-
-class ChatManager:
-    def __init__(self):
-        self.active_connections: Dict[str, WebSocket] = {}
-        self.chat_history = ChatHistory()
-        self.cache_manager = cache_manager
-        self.cache_manager.attach(self.update)
-
-    def on_chat_history_update(self):
-        """Send the last chat message to the client."""
-        client_id = self.cache_manager.current_client_id
-        if client_id in self.active_connections:
-            chat_response = self.chat_history.get_history(
-                client_id, filter_messages=False
-            )[-1]
-            if chat_response.is_bot:
-                # Process FileResponse
-                if isinstance(chat_response, FileResponse):
-                    # If data_type is pandas, convert to csv
-                    if chat_response.data_type == "pandas":
-                        chat_response.data = chat_response.data.to_csv()
-                    elif chat_response.data_type == "image":
-                        # Base64 encode the image
-                        chat_response.data = pil_to_base64(chat_response.data)
-                # get event loop
-                loop = asyncio.get_event_loop()
-
-                coroutine = self.send_json(client_id, chat_response)
-                asyncio.run_coroutine_threadsafe(coroutine, loop)
-
-    def update(self):
-        if self.cache_manager.current_client_id in self.active_connections:
-            self.last_cached_object_dict = self.cache_manager.get_last()
-            # Add a new ChatResponse with the data
-            chat_response = FileResponse(
-                message=None,
-                type="file",
-                data=self.last_cached_object_dict["obj"],
-                data_type=self.last_cached_object_dict["type"],
-            )
-
-            self.chat_history.add_message(
-                self.cache_manager.current_client_id, chat_response
-            )
-
-    async def connect(self, client_id: str, websocket: WebSocket):
-        await websocket.accept()
-        self.active_connections[client_id] = websocket
-
-    def disconnect(self, client_id: str):
-        self.active_connections.pop(client_id, None)
-
-    async def send_message(self, client_id: str, message: str):
-        websocket = self.active_connections[client_id]
-        await websocket.send_text(message)
-
-    async def send_json(self, client_id: str, message: ChatMessage):
-        websocket = self.active_connections[client_id]
-        await websocket.send_json(message.dict())
-
-    async def process_message(self, client_id: str, payload: Dict):
-        # Process the graph data and chat message
-        chat_message = payload.pop("message", "")
-        chat_message = ChatMessage(message=chat_message)
-        self.chat_history.add_message(client_id, chat_message)
-
-        graph_data = payload
-        start_resp = ChatResponse(message=None, type="start", intermediate_steps="")
-        await self.send_json(client_id, start_resp)
-
-        is_first_message = len(self.chat_history.get_history(client_id=client_id)) <= 1
-        # Generate result and thought
-        try:
-            logger.debug("Generating result and thought")
-
-            result, intermediate_steps = await process_graph(
-                graph_data=graph_data,
-                is_first_message=is_first_message,
-                chat_message=chat_message,
-                websocket=self.active_connections[client_id],
-            )
-        except Exception as e:
-            # Log stack trace
-            logger.exception(e)
-            self.chat_history.empty_history(client_id)
-            raise e
-        # Send a response back to the frontend, if needed
-        intermediate_steps = intermediate_steps or ""
-        history = self.chat_history.get_history(client_id, filter_messages=False)
-        file_responses = []
-        if history:
-            # Iterate backwards through the history
-            for msg in reversed(history):
-                if isinstance(msg, FileResponse):
-                    if msg.data_type == "image":
-                        # Base64 encode the image
-                        msg.data = pil_to_base64(msg.data)
-                    file_responses.append(msg)
-                if msg.type == "start":
-                    break
-
-        response = ChatResponse(
-            message=result,
-            intermediate_steps=intermediate_steps.strip(),
-            type="end",
-            files=file_responses,
-        )
-        await self.send_json(client_id, response)
-        self.chat_history.add_message(client_id, response)
-
-    async def handle_websocket(self, client_id: str, websocket: WebSocket):
-        await self.connect(client_id, websocket)
-
-        try:
-            chat_history = self.chat_history.get_history(client_id)
-            # iterate and make BaseModel into dict
-            chat_history = [chat.dict() for chat in chat_history]
-            await websocket.send_json(chat_history)
-
-            while True:
-                json_payload = await websocket.receive_json()
-                try:
-                    payload = json.loads(json_payload)
-                except TypeError:
-                    payload = json_payload
-                if "clear_history" in payload:
-                    self.chat_history.history[client_id] = []
-                    continue
-
-                with self.cache_manager.set_client_id(client_id):
-                    await self.process_message(client_id, payload)
-
-        except Exception as e:
-            # Handle any exceptions that might occur
-            logger.exception(e)
-            # send a message to the client
-            await self.active_connections[client_id].close(
-                code=status.WS_1011_INTERNAL_ERROR, reason=str(e)[:120]
-            )
-            self.disconnect(client_id)
-        finally:
-            try:
-                connection = self.active_connections.get(client_id)
-                if connection:
-                    await connection.close(code=1000, reason="Client disconnected")
-                    self.disconnect(client_id)
-            except Exception as e:
-                logger.exception(e)
-            self.disconnect(client_id)
-
-
-async def process_graph(
-    graph_data: Dict,
-    is_first_message: bool,
-    chat_message: ChatMessage,
-    websocket: WebSocket,
-):
-    langchain_object = load_or_build_langchain_object(graph_data, is_first_message)
-    langchain_object = try_setting_streaming_options(langchain_object, websocket)
-    logger.debug("Loaded langchain object")
-
-    if langchain_object is None:
-        # Raise user facing error
-        raise ValueError(
-            "There was an error loading the langchain_object. Please, check all the nodes and try again."
-        )
-
-    # Generate result and thought
-    try:
-        logger.debug("Generating result and thought")
-        result, intermediate_steps = await get_result_and_steps(
-            langchain_object, chat_message.message or "", websocket=websocket
-        )
-        logger.debug("Generated result and intermediate_steps")
-        return result, intermediate_steps
-    except Exception as e:
-        # Log stack trace
-        logger.exception(e)
-        raise e
diff --git a/src/backend/langflow/api/endpoints.py b/src/backend/langflow/api/endpoints.py
deleted file mode 100644
index 021a81ca8..000000000
--- a/src/backend/langflow/api/endpoints.py
+++ /dev/null
@@ -1,47 +0,0 @@
-import logging
-from importlib.metadata import version
-
-from fastapi import APIRouter, HTTPException
-
-from langflow.api.schemas import (
-    ExportedFlow,
-    GraphData,
-    PredictRequest,
-    PredictResponse,
-)
-from langflow.interface.run import process_graph_cached
-from langflow.interface.types import build_langchain_types_dict
-
-# build router
-router = APIRouter()
-logger = logging.getLogger(__name__)
-
-
-@router.get("/all")
-def get_all():
-    return build_langchain_types_dict()
-
-
-@router.post("/predict", response_model=PredictResponse)
-async def get_load(predict_request: PredictRequest):
-    try:
-        exported_flow: ExportedFlow = predict_request.exported_flow
-        graph_data: GraphData = exported_flow.data
-        data = graph_data.dict()
-        response = process_graph_cached(data, predict_request.message)
-        return PredictResponse(result=response.get("result", ""))
-    except Exception as e:
-        # Log stack trace
-        logger.exception(e)
-        raise HTTPException(status_code=500, detail=str(e)) from e
-
-
-# get endpoint to return version of langflow
-@router.get("/version")
-def get_version():
-    return {"version": version("langflow")}
-
-
-@router.get("/health")
-def get_health():
-    return {"status": "OK"}
diff --git a/src/backend/langflow/api/schemas.py b/src/backend/langflow/api/schemas.py
deleted file mode 100644
index f73b0642d..000000000
--- a/src/backend/langflow/api/schemas.py
+++ /dev/null
@@ -1,70 +0,0 @@
-from typing import Any, Dict, List, Union
-
-from pydantic import BaseModel, validator
-
-
-class GraphData(BaseModel):
-    """Data inside the exported flow."""
-
-    nodes: List[Dict[str, Any]]
-    edges: List[Dict[str, Any]]
-
-
-class ExportedFlow(BaseModel):
-    """Exported flow from LangFlow."""
-
-    description: str
-    name: str
-    id: str
-    data: GraphData
-
-
-class PredictRequest(BaseModel):
-    """Predict request schema."""
-
-    message: str
-    exported_flow: ExportedFlow
-
-
-class PredictResponse(BaseModel):
-    """Predict response schema."""
-
-    result: str
-
-
-class ChatMessage(BaseModel):
-    """Chat message schema."""
-
-    is_bot: bool = False
-    message: Union[str, None] = None
-    type: str = "human"
-
-
-class ChatResponse(ChatMessage):
-    """Chat response schema."""
-
-    intermediate_steps: str
-    type: str
-    is_bot: bool = True
-    files: list = []
-
-    @validator("type")
-    def validate_message_type(cls, v):
-        if v not in ["start", "stream", "end", "error", "info", "file"]:
-            raise ValueError("type must be start, stream, end, error, info, or file")
-        return v
-
-
-class FileResponse(ChatMessage):
-    """File response schema."""
-
-    data: Any
-    data_type: str
-    type: str = "file"
-    is_bot: bool = True
-
-    @validator("data_type")
-    def validate_data_type(cls, v):
-        if v not in ["image", "csv"]:
-            raise ValueError("data_type must be image or csv")
-        return v
diff --git a/src/backend/langflow/api/validate.py b/src/backend/langflow/api/validate.py
deleted file mode 100644
index e90e554f0..000000000
--- a/src/backend/langflow/api/validate.py
+++ /dev/null
@@ -1,57 +0,0 @@
-import json
-
-from fastapi import APIRouter, HTTPException
-
-from langflow.api.base import (
-    Code,
-    CodeValidationResponse,
-    Prompt,
-    PromptValidationResponse,
-    validate_prompt,
-)
-from langflow.graph.vertex.types import VectorStoreVertex
-from langflow.interface.run import build_graph
-from langflow.utils.logger import logger
-from langflow.utils.validate import validate_code
-
-# build router
-router = APIRouter(prefix="/validate", tags=["validate"])
-
-
-@router.post("/code", status_code=200, response_model=CodeValidationResponse)
-def post_validate_code(code: Code):
-    try:
-        errors = validate_code(code.code)
-        return CodeValidationResponse(
-            imports=errors.get("imports", {}),
-            function=errors.get("function", {}),
-        )
-    except Exception as e:
-        return HTTPException(status_code=500, detail=str(e))
-
-
-@router.post("/prompt", status_code=200, response_model=PromptValidationResponse)
-def post_validate_prompt(prompt: Prompt):
-    try:
-        return validate_prompt(prompt.template)
-    except Exception as e:
-        logger.exception(e)
-        raise HTTPException(status_code=500, detail=str(e)) from e
-
-
-# validate node
-@router.post("/node/{node_id}", status_code=200)
-def post_validate_node(node_id: str, data: dict):
-    try:
-        # build graph
-        graph = build_graph(data)
-        # validate node
-        node = graph.get_node(node_id)
-        if node is None:
-            raise ValueError(f"Node {node_id} not found")
-        if not isinstance(node, VectorStoreVertex):
-            node.build()
-        return json.dumps({"valid": True, "params": str(node._built_object_repr())})
-    except Exception as e:
-        logger.exception(e)
-        return json.dumps({"valid": False, "params": str(e)})

From bdbb4a81279f535538cc55128b358674c5ec7a45 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:00:38 -0300
Subject: [PATCH 45/59] =?UTF-8?q?=F0=9F=9A=80=20feat(api):=20add=20version?=
 =?UTF-8?q?ing=20to=20the=20API=20and=20restructure=20the=20router=20The?=
 =?UTF-8?q?=20API=20now=20has=20versioning,=20with=20the=20prefix=20"/api/?=
 =?UTF-8?q?v1".=20The=20router=20has=20been=20restructured=20to=20include?=
 =?UTF-8?q?=20the=20chat,=20endpoints,=20and=20validate=20routers.=20This?=
 =?UTF-8?q?=20improves=20the=20organization=20of=20the=20code=20and=20make?=
 =?UTF-8?q?s=20it=20easier=20to=20add=20new=20routers=20in=20the=20future.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/api/__init__.py    | 3 +++
 src/backend/langflow/api/router.py      | 8 ++++++++
 src/backend/langflow/api/v1/__init__.py | 5 +++++
 3 files changed, 16 insertions(+)
 create mode 100644 src/backend/langflow/api/router.py
 create mode 100644 src/backend/langflow/api/v1/__init__.py

diff --git a/src/backend/langflow/api/__init__.py b/src/backend/langflow/api/__init__.py
index e69de29bb..f887c47e1 100644
--- a/src/backend/langflow/api/__init__.py
+++ b/src/backend/langflow/api/__init__.py
@@ -0,0 +1,3 @@
+from langflow.api.router import router
+
+__all__ = ["router"]
diff --git a/src/backend/langflow/api/router.py b/src/backend/langflow/api/router.py
new file mode 100644
index 000000000..23b5aa1c5
--- /dev/null
+++ b/src/backend/langflow/api/router.py
@@ -0,0 +1,8 @@
+# Router for base api
+from fastapi import APIRouter
+from langflow.api.v1 import chat_router, endpoints_router, validate_router
+
+router = APIRouter(prefix="/api/v1", tags=["api"])
+router.include_router(chat_router)
+router.include_router(endpoints_router)
+router.include_router(validate_router)
diff --git a/src/backend/langflow/api/v1/__init__.py b/src/backend/langflow/api/v1/__init__.py
new file mode 100644
index 000000000..d835b4535
--- /dev/null
+++ b/src/backend/langflow/api/v1/__init__.py
@@ -0,0 +1,5 @@
+from langflow.api.v1.endpoints import router as endpoints_router
+from langflow.api.v1.validate import router as validate_router
+from langflow.api.v1.chat import router as chat_router
+
+__all__ = ["chat_router", "endpoints_router", "validate_router"]

From 3e5878ddc282557315f532605357ff782af13d64 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:01:18 -0300
Subject: [PATCH 46/59] =?UTF-8?q?=F0=9F=8E=89=20feat(langflow):=20add=20ne?=
 =?UTF-8?q?w=20files=20base.py=20and=20callback.py=20The=20base.py=20file?=
 =?UTF-8?q?=20contains=20the=20following=20classes=20and=20functions:=20-?=
 =?UTF-8?q?=20CacheResponse:=20a=20pydantic=20BaseModel=20that=20represent?=
 =?UTF-8?q?s=20a=20response=20containing=20a=20dictionary=20of=20data=20-?=
 =?UTF-8?q?=20Code:=20a=20pydantic=20BaseModel=20that=20represents=20a=20c?=
 =?UTF-8?q?ode=20string=20-=20Prompt:=20a=20pydantic=20BaseModel=20that=20?=
 =?UTF-8?q?represents=20a=20prompt=20template=20string=20-=20CodeValidatio?=
 =?UTF-8?q?nResponse:=20a=20pydantic=20BaseModel=20that=20represents=20a?=
 =?UTF-8?q?=20response=20containing=20the=20validation=20results=20of=20co?=
 =?UTF-8?q?de=20-=20PromptValidationResponse:=20a=20pydantic=20BaseModel?=
 =?UTF-8?q?=20that=20represents=20a=20response=20containing=20the=20valida?=
 =?UTF-8?q?tion=20results=20of=20a=20prompt=20-=20validate=5Fprompt:=20a?=
 =?UTF-8?q?=20function=20that=20validates=20a=20prompt=20template=20string?=
 =?UTF-8?q?=20and=20returns=20a=20PromptValidationResponse=20object=20-=20?=
 =?UTF-8?q?check=5Finput=5Fvariables:=20a=20function=20that=20checks=20if?=
 =?UTF-8?q?=20input=20variables=20contain=20invalid=20characters=20and=20r?=
 =?UTF-8?q?eturns=20a=20list=20of=20fixed=20input=20variables?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

The callback.py file contains the following classes:
- AsyncStreamingLLMCallbackHandler: an AsyncCallbackHandler that handles streaming LLM responses asynchronously
- StreamingLLMCallbackHandler: a BaseCallbackHandler that handles streaming LLM responses

These files were added to provide support for Langflow's backend API.
---
 src/backend/langflow/api/v1/base.py     | 84 +++++++++++++++++++++++++
 src/backend/langflow/api/v1/callback.py | 32 ++++++++++
 2 files changed, 116 insertions(+)
 create mode 100644 src/backend/langflow/api/v1/base.py
 create mode 100644 src/backend/langflow/api/v1/callback.py

diff --git a/src/backend/langflow/api/v1/base.py b/src/backend/langflow/api/v1/base.py
new file mode 100644
index 000000000..6941bedf3
--- /dev/null
+++ b/src/backend/langflow/api/v1/base.py
@@ -0,0 +1,84 @@
+from pydantic import BaseModel, validator
+
+from langflow.interface.utils import extract_input_variables_from_prompt
+
+
+class CacheResponse(BaseModel):
+    data: dict
+
+
+class Code(BaseModel):
+    code: str
+
+
+class Prompt(BaseModel):
+    template: str
+
+
+# Build ValidationResponse class for {"imports": {"errors": []}, "function": {"errors": []}}
+class CodeValidationResponse(BaseModel):
+    imports: dict
+    function: dict
+
+    @validator("imports")
+    def validate_imports(cls, v):
+        return v or {"errors": []}
+
+    @validator("function")
+    def validate_function(cls, v):
+        return v or {"errors": []}
+
+
+class PromptValidationResponse(BaseModel):
+    input_variables: list
+
+
+INVALID_CHARACTERS = {
+    " ",
+    ",",
+    ".",
+    ":",
+    ";",
+    "!",
+    "?",
+    "/",
+    "\\",
+    "(",
+    ")",
+    "[",
+    "]",
+    "{",
+    "}",
+}
+
+
+def validate_prompt(template: str):
+    input_variables = extract_input_variables_from_prompt(template)
+
+    # Check if there are invalid characters in the input_variables
+    input_variables = check_input_variables(input_variables)
+
+    return PromptValidationResponse(input_variables=input_variables)
+
+
+def check_input_variables(input_variables: list):
+    invalid_chars = []
+    fixed_variables = []
+    for variable in input_variables:
+        new_var = variable
+        for char in INVALID_CHARACTERS:
+            if char in variable:
+                invalid_chars.append(char)
+                new_var = new_var.replace(char, "")
+        fixed_variables.append(new_var)
+        if new_var != variable:
+            input_variables.remove(variable)
+            input_variables.append(new_var)
+    # If any of the input_variables is not in the fixed_variables, then it means that
+    # there are invalid characters in the input_variables
+    if any(var not in fixed_variables for var in input_variables):
+        raise ValueError(
+            f"Invalid input variables: {input_variables}. Please, use something like {fixed_variables} instead."
+        )
+
+    return input_variables
diff --git a/src/backend/langflow/api/v1/callback.py b/src/backend/langflow/api/v1/callback.py
new file mode 100644
index 000000000..b58393d7b
--- /dev/null
+++ b/src/backend/langflow/api/v1/callback.py
@@ -0,0 +1,32 @@
+import asyncio
+from typing import Any
+
+from langchain.callbacks.base import AsyncCallbackHandler, BaseCallbackHandler
+
+from langflow.api.v1.schemas import ChatResponse
+
+
+# https://github.com/hwchase17/chat-langchain/blob/master/callback.py
+class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
+    """Callback handler for streaming LLM responses."""
+
+    def __init__(self, websocket):
+        self.websocket = websocket
+
+    async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
+        resp = ChatResponse(message=token, type="stream", intermediate_steps="")
+        await self.websocket.send_json(resp.dict())
+
+
+class StreamingLLMCallbackHandler(BaseCallbackHandler):
+    """Callback handler for streaming LLM responses."""
+
+    def __init__(self, websocket):
+        self.websocket = websocket
+
+    def on_llm_new_token(self, token: str, **kwargs: Any) -> None:
+        resp = ChatResponse(message=token, type="stream", intermediate_steps="")
+
+        loop = asyncio.get_event_loop()
+        coroutine = self.websocket.send_json(resp.dict())
+        asyncio.run_coroutine_threadsafe(coroutine, loop)

From 2bfe93e0b8e44ff82785889e7589928d2cb8799b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:01:44 -0300
Subject: [PATCH 47/59] =?UTF-8?q?=F0=9F=9A=80=20feat(langflow):=20add=20ne?=
 =?UTF-8?q?w=20API=20endpoints=20for=20chat,=20validation,=20and=20version?=
 =?UTF-8?q?=20This=20commit=20adds=20new=20API=20endpoints=20for=20chat,?=
 =?UTF-8?q?=20validation,=20and=20version.=20The=20chat=20endpoint=20is=20?=
 =?UTF-8?q?a=20websocket=20endpoint=20for=20chat.=20The=20validation=20end?=
 =?UTF-8?q?point=20has=20three=20sub-endpoints=20for=20validating=20code,?=
 =?UTF-8?q?=20prompt,=20and=20node.=20The=20version=20endpoint=20returns?=
 =?UTF-8?q?=20the=20version=20of=20LangFlow.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/api/v1/chat.py      | 26 +++++++++
 src/backend/langflow/api/v1/endpoints.py | 44 +++++++++++++++
 src/backend/langflow/api/v1/schemas.py   | 70 ++++++++++++++++++++++++
 src/backend/langflow/api/v1/validate.py  | 57 +++++++++++++++++++
 4 files changed, 197 insertions(+)
 create mode 100644 src/backend/langflow/api/v1/chat.py
 create mode 100644 src/backend/langflow/api/v1/endpoints.py
 create mode 100644 src/backend/langflow/api/v1/schemas.py
 create mode 100644 src/backend/langflow/api/v1/validate.py

diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
new file mode 100644
index 000000000..7df4c65ed
--- /dev/null
+++ b/src/backend/langflow/api/v1/chat.py
@@ -0,0 +1,26 @@
+from fastapi import (
+    APIRouter,
+    WebSocket,
+    WebSocketDisconnect,
+    WebSocketException,
+    status,
+)
+
+from langflow.chat.manager import ChatManager
+from langflow.utils.logger import logger
+
+router = APIRouter()
+chat_manager = ChatManager()
+
+
+@router.websocket("/chat/{client_id}")
+async def websocket_endpoint(client_id: str, websocket: WebSocket):
+    """Websocket endpoint for chat."""
+    try:
+        await chat_manager.handle_websocket(client_id, websocket)
+    except WebSocketException as exc:
+        logger.error(exc)
+        await websocket.close(code=status.WS_1011_INTERNAL_ERROR, reason=str(exc))
+    except WebSocketDisconnect as exc:
+        logger.error(exc)
+        await websocket.close(code=status.WS_1000_NORMAL_CLOSURE, reason=str(exc))
diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py
new file mode 100644
index 000000000..1e9b0deb1
--- /dev/null
+++ b/src/backend/langflow/api/v1/endpoints.py
@@ -0,0 +1,44 @@
+import logging
+from importlib.metadata import version
+
+from fastapi import APIRouter, HTTPException
+
+from langflow.api.v1.schemas import (
+    ExportedFlow,
+    GraphData,
+    PredictRequest,
+    PredictResponse,
+)
+
+from langflow.interface.types import build_langchain_types_dict
+
+# build router
+router = APIRouter()
+logger = logging.getLogger(__name__)
+
+
+@router.get("/all")
+def get_all():
+    return build_langchain_types_dict()
+
+
+@router.post("/predict", response_model=PredictResponse)
+async def get_load(predict_request: PredictRequest):
+    try:
+        from langflow.processing.process import process_graph_cached
+
+        exported_flow: ExportedFlow = predict_request.exported_flow
+        graph_data: GraphData = exported_flow.data
+        data = graph_data.dict()
+        response = process_graph_cached(data, predict_request.message)
+        return PredictResponse(result=response.get("result", ""))
+    except Exception as e:
+        # Log stack trace
+        logger.exception(e)
+        raise HTTPException(status_code=500, detail=str(e)) from e
+
+
+# get endpoint to return version of langflow
+@router.get("/version")
+def get_version():
+    return {"version": version("langflow")}
diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py
new file mode 100644
index 000000000..f73b0642d
--- /dev/null
+++ b/src/backend/langflow/api/v1/schemas.py
@@ -0,0 +1,70 @@
+from typing import Any, Dict, List, Union
+
+from pydantic import BaseModel, validator
+
+
+class GraphData(BaseModel):
+    """Data inside the exported flow."""
+
+    nodes: List[Dict[str, Any]]
+    edges: List[Dict[str, Any]]
+
+
+class ExportedFlow(BaseModel):
+    """Exported flow from LangFlow."""
+
+    description: str
+    name: str
+    id: str
+    data: GraphData
+
+
+class PredictRequest(BaseModel):
+    """Predict request schema."""
+
+    message: str
+    exported_flow: ExportedFlow
+
+
+class PredictResponse(BaseModel):
+    """Predict response schema."""
+
+    result: str
+
+
+class ChatMessage(BaseModel):
+    """Chat message schema."""
+
+    is_bot: bool = False
+    message: Union[str, None] = None
+    type: str = "human"
+
+
+class ChatResponse(ChatMessage):
+    """Chat response schema."""
+
+    intermediate_steps: str
+    type: str
+    is_bot: bool = True
+    files: list = []
+
+    @validator("type")
+    def validate_message_type(cls, v):
+        if v not in ["start", "stream", "end", "error", "info", "file"]:
+            raise ValueError("type must be start, stream, end, error, info, or file")
+        return v
+
+
+class FileResponse(ChatMessage):
+    """File response schema."""
+
+    data: Any
+    data_type: str
+    type: str = "file"
+    is_bot: bool = True
+
+    @validator("data_type")
+    def validate_data_type(cls, v):
+        if v not in ["image", "csv"]:
+            raise ValueError("data_type must be image or csv")
+        return v
diff --git a/src/backend/langflow/api/v1/validate.py b/src/backend/langflow/api/v1/validate.py
new file mode 100644
index 000000000..009cb9a30
--- /dev/null
+++ b/src/backend/langflow/api/v1/validate.py
@@ -0,0 +1,57 @@
+import json
+
+from fastapi import APIRouter, HTTPException
+
+from langflow.api.v1.base import (
+    Code,
+    CodeValidationResponse,
+    Prompt,
+    PromptValidationResponse,
+    validate_prompt,
+)
+from langflow.graph.vertex.types import VectorStoreVertex
+from langflow.graph import Graph
+from langflow.utils.logger import logger
+from langflow.utils.validate import validate_code
+
+# build router
+router = APIRouter(prefix="/validate", tags=["validate"])
+
+
+@router.post("/code", status_code=200, response_model=CodeValidationResponse)
+def post_validate_code(code: Code):
+    try:
+        errors = validate_code(code.code)
+        return CodeValidationResponse(
+            imports=errors.get("imports", {}),
+            function=errors.get("function", {}),
+        )
+    except Exception as e:
+        return HTTPException(status_code=500, detail=str(e))
+
+
+@router.post("/prompt", status_code=200, response_model=PromptValidationResponse)
+def post_validate_prompt(prompt: Prompt):
+    try:
+        return validate_prompt(prompt.template)
+    except Exception as e:
+        logger.exception(e)
+        raise HTTPException(status_code=500, detail=str(e)) from e
+
+
+# validate node
+@router.post("/node/{node_id}", status_code=200)
+def post_validate_node(node_id: str, data: dict):
+    try:
+        # build graph
+        graph = Graph.from_payload(data)
+        # validate node
+        node = graph.get_node(node_id)
+        if node is None:
+            raise ValueError(f"Node {node_id} not found")
+        if not isinstance(node, VectorStoreVertex):
+            node.build()
+        return json.dumps({"valid": True, "params": str(node._built_object_repr())})
+    except Exception as e:
+        logger.exception(e)
+        return json.dumps({"valid": False, "params": str(e)})

From 7f4eea1e593f29438986b7d21568b06664bb5c13 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:02:21 -0300
Subject: [PATCH 48/59] =?UTF-8?q?=F0=9F=9A=80=20feat(chat):=20add=20ChatMa?=
 =?UTF-8?q?nager=20and=20ChatHistory=20classes=20to=20manage=20chat=20hist?=
 =?UTF-8?q?ory=20and=20active=20connections=20=E2=9C=A8=20feat(utils.py):?=
 =?UTF-8?q?=20add=20process=5Fgraph=20function=20to=20process=20graph=20da?=
 =?UTF-8?q?ta=20and=20generate=20result=20and=20thought=20The=20ChatManage?=
 =?UTF-8?q?r=20class=20manages=20active=20connections=20and=20chat=20histo?=
 =?UTF-8?q?ry.=20The=20ChatHistory=20class=20manages=20the=20chat=20histor?=
 =?UTF-8?q?y=20for=20a=20client.=20The=20process=5Fgraph=20function=20proc?=
 =?UTF-8?q?esses=20graph=20data=20and=20generates=20a=20result=20and=20tho?=
 =?UTF-8?q?ught.=20This=20function=20is=20used=20in=20the=20ChatManager=20?=
 =?UTF-8?q?class=20to=20generate=20a=20response=20back=20to=20the=20fronte?=
 =?UTF-8?q?nd.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/chat/__init__.py |   0
 src/backend/langflow/chat/manager.py  | 190 ++++++++++++++++++++++++++
 src/backend/langflow/chat/utils.py    |  41 ++++++
 3 files changed, 231 insertions(+)
 create mode 100644 src/backend/langflow/chat/__init__.py
 create mode 100644 src/backend/langflow/chat/manager.py
 create mode 100644 src/backend/langflow/chat/utils.py

diff --git a/src/backend/langflow/chat/__init__.py b/src/backend/langflow/chat/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/chat/manager.py b/src/backend/langflow/chat/manager.py
new file mode 100644
index 000000000..d24057b68
--- /dev/null
+++ b/src/backend/langflow/chat/manager.py
@@ -0,0 +1,190 @@
+from collections import defaultdict
+from fastapi import WebSocket, status
+from langflow.api.v1.schemas import ChatMessage, ChatResponse, FileResponse
+from langflow.cache import cache_manager
+from langflow.cache.manager import Subject
+from langflow.chat.utils import process_graph
+from langflow.interface.utils import pil_to_base64
+from langflow.utils.logger import logger
+
+
+import asyncio
+import json
+from typing import Dict, List
+
+
+class ChatHistory(Subject):
+    def __init__(self):
+        super().__init__()
+        self.history: Dict[str, List[ChatMessage]] = defaultdict(list)
+
+    def add_message(self, client_id: str, message: ChatMessage):
+        """Add a message to the chat history."""
+
+        self.history[client_id].append(message)
+
+        if not isinstance(message, FileResponse):
+            self.notify()
+
+    def get_history(self, client_id: str, filter_messages=True) -> List[ChatMessage]:
+        """Get the chat history for a client."""
+        if history := self.history.get(client_id, []):
+            if filter_messages:
+                return [msg for msg in history if msg.type not in ["start", "stream"]]
+            return history
+        else:
+            return []
+
+    def empty_history(self, client_id: str):
+        """Empty the chat history for a client."""
+        self.history[client_id] = []
+
+
+class ChatManager:
+    def __init__(self):
+        self.active_connections: Dict[str, WebSocket] = {}
+        self.chat_history = ChatHistory()
+        self.cache_manager = cache_manager
+        self.cache_manager.attach(self.update)
+
+    def on_chat_history_update(self):
+        """Send the last chat message to the client."""
+        client_id = self.cache_manager.current_client_id
+        if client_id in self.active_connections:
+            chat_response = self.chat_history.get_history(
+                client_id, filter_messages=False
+            )[-1]
+            if chat_response.is_bot:
+                # Process FileResponse
+                if isinstance(chat_response, FileResponse):
+                    # If data_type is pandas, convert to csv
+                    if chat_response.data_type == "pandas":
+                        chat_response.data = chat_response.data.to_csv()
+                    elif chat_response.data_type == "image":
+                        # Base64 encode the image
+                        chat_response.data = pil_to_base64(chat_response.data)
+                # get event loop
+                loop = asyncio.get_event_loop()
+
+                coroutine = self.send_json(client_id, chat_response)
+                asyncio.run_coroutine_threadsafe(coroutine, loop)
+
+    def update(self):
+        if self.cache_manager.current_client_id in self.active_connections:
+            self.last_cached_object_dict = self.cache_manager.get_last()
+            # Add a new ChatResponse with the data
+            chat_response = FileResponse(
+                message=None,
+                type="file",
+                data=self.last_cached_object_dict["obj"],
+                data_type=self.last_cached_object_dict["type"],
+            )
+
+            self.chat_history.add_message(
+                self.cache_manager.current_client_id, chat_response
+            )
+
+    async def connect(self, client_id: str, websocket: WebSocket):
+        await websocket.accept()
+        self.active_connections[client_id] = websocket
+
+    def disconnect(self, client_id: str):
+        self.active_connections.pop(client_id, None)
+
+    async def send_message(self, client_id: str, message: str):
+        websocket = self.active_connections[client_id]
+        await websocket.send_text(message)
+
+    async def send_json(self, client_id: str, message: ChatMessage):
+        websocket = self.active_connections[client_id]
+        await websocket.send_json(message.dict())
+
+    async def process_message(self, client_id: str, payload: Dict):
+        # Process the graph data and chat message
+        chat_message = payload.pop("message", "")
+        chat_message = ChatMessage(message=chat_message)
+        self.chat_history.add_message(client_id, chat_message)
+
+        graph_data = payload
+        start_resp = ChatResponse(message=None, type="start", intermediate_steps="")
+        await self.send_json(client_id, start_resp)
+
+        is_first_message = len(self.chat_history.get_history(client_id=client_id)) <= 1
+        # Generate result and thought
+        try:
+            logger.debug("Generating result and thought")
+
+            result, intermediate_steps = await process_graph(
+                graph_data=graph_data,
+                is_first_message=is_first_message,
+                chat_message=chat_message,
+                websocket=self.active_connections[client_id],
+            )
+        except Exception as e:
+            # Log stack trace
+            logger.exception(e)
+            self.chat_history.empty_history(client_id)
+            raise e
+        # Send a response back to the frontend, if needed
+        intermediate_steps = intermediate_steps or ""
+        history = self.chat_history.get_history(client_id, filter_messages=False)
+        file_responses = []
+        if history:
+            # Iterate backwards through the history
+            for msg in reversed(history):
+                if isinstance(msg, FileResponse):
+                    if msg.data_type == "image":
+                        # Base64 encode the image
+                        msg.data = pil_to_base64(msg.data)
+                    file_responses.append(msg)
+                if msg.type == "start":
+                    break
+
+        response = ChatResponse(
+            message=result,
+            intermediate_steps=intermediate_steps.strip(),
+            type="end",
+            files=file_responses,
+        )
+        await self.send_json(client_id, response)
+        self.chat_history.add_message(client_id, response)
+
+    async def handle_websocket(self, client_id: str, websocket: WebSocket):
+        await self.connect(client_id, websocket)
+
+        try:
+            chat_history = self.chat_history.get_history(client_id)
+            # iterate and make BaseModel into dict
+            chat_history = [chat.dict() for chat in chat_history]
+            await websocket.send_json(chat_history)
+
+            while True:
+                json_payload = await websocket.receive_json()
+                try:
+                    payload = json.loads(json_payload)
+                except TypeError:
+                    payload = json_payload
+                if "clear_history" in payload:
+                    self.chat_history.history[client_id] = []
+                    continue
+
+                with self.cache_manager.set_client_id(client_id):
+                    await self.process_message(client_id, payload)
+
+        except Exception as e:
+            # Handle any exceptions that might occur
+            logger.exception(e)
+            # send a message to the client
+            await self.active_connections[client_id].close(
+                code=status.WS_1011_INTERNAL_ERROR, reason=str(e)[:120]
+            )
+            self.disconnect(client_id)
+        finally:
+            try:
+                connection = self.active_connections.get(client_id)
+                if connection:
+                    await connection.close(code=1000, reason="Client disconnected")
+                    self.disconnect(client_id)
+            except Exception as e:
+                logger.exception(e)
+            self.disconnect(client_id)
diff --git a/src/backend/langflow/chat/utils.py b/src/backend/langflow/chat/utils.py
new file mode 100644
index 000000000..410a442be
--- /dev/null
+++ b/src/backend/langflow/chat/utils.py
@@ -0,0 +1,41 @@
+from fastapi import WebSocket
+from langflow.api.v1.schemas import ChatMessage
+from langflow.processing.process import (
+    load_or_build_langchain_object,
+)
+from langflow.processing.base import get_result_and_steps
+from langflow.interface.utils import try_setting_streaming_options
+from langflow.utils.logger import logger
+
+
+from typing import Dict
+
+
+async def process_graph(
+    graph_data: Dict,
+    is_first_message: bool,
+    chat_message: ChatMessage,
+    websocket: WebSocket,
+):
+    langchain_object = load_or_build_langchain_object(graph_data, is_first_message)
+    langchain_object = try_setting_streaming_options(langchain_object, websocket)
+    logger.debug("Loaded langchain object")
+
+    if langchain_object is None:
+        # Raise user facing error
+        raise ValueError(
+            "There was an error loading the langchain_object. Please, check all the nodes and try again."
+        )
+
+    # Generate result and thought
+    try:
+        logger.debug("Generating result and thought")
+        result, intermediate_steps = await get_result_and_steps(
+            langchain_object, chat_message.message or "", websocket=websocket
+        )
+        logger.debug("Generated result and intermediate_steps")
+        return result, intermediate_steps
+    except Exception as e:
+        # Log stack trace
+        logger.exception(e)
+        raise e

From 3bfee4d4455af8ae18a947ca31785f7978e1384d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:05:01 -0300
Subject: [PATCH 49/59] =?UTF-8?q?=F0=9F=9A=80=20feat(graph):=20add=20from?=
 =?UTF-8?q?=5Fpayload=20class=20method=20to=20Graph=20class=20=F0=9F=9A=80?=
 =?UTF-8?q?=20feat(utils.py):=20import=20extract=5Finput=5Fvariables=5Ffro?=
 =?UTF-8?q?m=5Fprompt=20from=20langflow.interface.utils=20The=20`from=5Fpa?=
 =?UTF-8?q?yload`=20class=20method=20is=20added=20to=20the=20`Graph`=20cla?=
 =?UTF-8?q?ss=20to=20create=20a=20graph=20from=20a=20payload.=20This=20met?=
 =?UTF-8?q?hod=20takes=20a=20dictionary=20as=20input=20and=20returns=20a?=
 =?UTF-8?q?=20`Graph`=20object.=20The=20`extract=5Finput=5Fvariables=5Ffro?=
 =?UTF-8?q?m=5Fprompt`=20function=20is=20imported=20from=20`langflow.inter?=
 =?UTF-8?q?face.utils`=20to=20extract=20input=20variables=20from=20a=20pro?=
 =?UTF-8?q?mpt.=20This=20function=20is=20used=20in=20other=20parts=20of=20?=
 =?UTF-8?q?the=20codebase=20to=20extract=20input=20variables=20from=20prom?=
 =?UTF-8?q?pts.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/graph/graph/base.py  | 21 +++++++++++++++++++++
 src/backend/langflow/graph/graph/utils.py |  0
 src/backend/langflow/graph/utils.py       |  8 ++------
 3 files changed, 23 insertions(+), 6 deletions(-)
 create mode 100644 src/backend/langflow/graph/graph/utils.py

diff --git a/src/backend/langflow/graph/graph/base.py b/src/backend/langflow/graph/graph/base.py
index 020f539ec..5fd00d09b 100644
--- a/src/backend/langflow/graph/graph/base.py
+++ b/src/backend/langflow/graph/graph/base.py
@@ -24,6 +24,27 @@ class Graph:
         self._edges = edges
         self._build_graph()
 
+    @classmethod
+    @classmethod
+    def from_payload(cls, payload: Dict) -> "Graph":
+        """
+        Creates a graph from a payload.
+
+        Args:
+            payload (Dict): The payload to create the graph from.
+
+        Returns:
+            Graph: The created graph.
+        """
+        if "data" in payload:
+            payload = payload["data"]
+        try:
+            nodes = payload["nodes"]
+            edges = payload["edges"]
+            return cls(nodes, edges)
+        except KeyError as exc:
+            raise ValueError("Invalid payload") from exc
+
     def _build_graph(self) -> None:
         """Builds the graph from the nodes and edges."""
         self.nodes = self._build_vertices()
diff --git a/src/backend/langflow/graph/graph/utils.py b/src/backend/langflow/graph/graph/utils.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/graph/utils.py b/src/backend/langflow/graph/utils.py
index e22b27cf5..b78b2f961 100644
--- a/src/backend/langflow/graph/utils.py
+++ b/src/backend/langflow/graph/utils.py
@@ -1,6 +1,7 @@
-import re
 from typing import Any, Union
 
+from langflow.interface.utils import extract_input_variables_from_prompt
+
 
 def validate_prompt(prompt: str):
     """Validate prompt."""
@@ -15,11 +16,6 @@ def fix_prompt(prompt: str):
     return prompt + " {input}"
 
 
-def extract_input_variables_from_prompt(prompt: str) -> list[str]:
-    """Extract input variables from prompt."""
-    return re.findall(r"{(.*?)}", prompt)
-
-
 def flatten_list(list_of_lists: list[Union[list, Any]]) -> list:
     """Flatten list of lists."""
     new_list = []

From 228f938cd8371ab5ecd5aa2d5b68622ad40ab03b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:05:46 -0300
Subject: [PATCH 50/59] =?UTF-8?q?=F0=9F=94=A8=20refactor(types.py):=20move?=
 =?UTF-8?q?=20extract=5Finput=5Fvariables=5Ffrom=5Fprompt=20import=20to=20?=
 =?UTF-8?q?interface.utils=20module=20=F0=9F=94=A8=20refactor(custom.py,?=
 =?UTF-8?q?=20loading.py,=20prompts/custom.py,=20run.py):=20update=20impor?=
 =?UTF-8?q?t=20statements=20to=20use=20extract=5Finput=5Fvariables=5Ffrom?=
 =?UTF-8?q?=5Fprompt=20from=20interface.utils=20module=20=F0=9F=94=A8=20re?=
 =?UTF-8?q?factor(run.py):=20remove=20unused=20imports=20and=20functions?=
 =?UTF-8?q?=20=F0=9F=94=A8=20refactor(utils.py):=20add=20type=20hinting=20?=
 =?UTF-8?q?to=20extract=5Finput=5Fvariables=5Ffrom=5Fprompt=20function=20a?=
 =?UTF-8?q?nd=20remove=20unused=20imports=20The=20extract=5Finput=5Fvariab?=
 =?UTF-8?q?les=5Ffrom=5Fprompt=20function=20has=20been=20moved=20to=20the?=
 =?UTF-8?q?=20interface.utils=20module=20to=20improve=20code=20organizatio?=
 =?UTF-8?q?n.=20The=20import=20statements=20in=20the=20affected=20modules?=
 =?UTF-8?q?=20have=20been=20updated=20to=20reflect=20this=20change.=20Unus?=
 =?UTF-8?q?ed=20imports=20and=20functions=20have=20been=20removed=20from?=
 =?UTF-8?q?=20the=20run.py=20module.=20Type=20hinting=20has=20been=20added?=
 =?UTF-8?q?=20to=20the=20extract=5Finput=5Fvariables=5Ffrom=5Fprompt=20fun?=
 =?UTF-8?q?ction=20in=20the=20interface.utils=20module.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🚀 feat(processing): add processing module with get_result_and_steps and fix_memory_inputs functions
The processing module was added to the project with two functions: get_result_and_steps and fix_memory_inputs. The get_result_and_steps function extracts the result and thought from a LangChain object and returns them. The fix_memory_inputs function checks if a LangChain object has a memory attribute and if that memory key exists in the object's input variables. If not, it gets a possible new memory key using the get_memory_key function and updates the memory keys using the update_memory_keys function.
---
 src/backend/langflow/graph/vertex/types.py    |   3 +-
 .../langflow/interface/chains/custom.py       |   2 +-
 src/backend/langflow/interface/loading.py     |  33 ---
 .../langflow/interface/prompts/custom.py      |   2 +-
 src/backend/langflow/interface/run.py         | 191 +-----------------
 src/backend/langflow/interface/utils.py       |   6 +
 src/backend/langflow/processing/__init__.py   |   0
 src/backend/langflow/processing/base.py       |  55 +++++
 src/backend/langflow/processing/process.py    | 172 ++++++++++++++++
 9 files changed, 238 insertions(+), 226 deletions(-)
 create mode 100644 src/backend/langflow/processing/__init__.py
 create mode 100644 src/backend/langflow/processing/base.py
 create mode 100644 src/backend/langflow/processing/process.py

diff --git a/src/backend/langflow/graph/vertex/types.py b/src/backend/langflow/graph/vertex/types.py
index b81e72439..4eb20f416 100644
--- a/src/backend/langflow/graph/vertex/types.py
+++ b/src/backend/langflow/graph/vertex/types.py
@@ -1,7 +1,8 @@
 from typing import Any, Dict, List, Optional, Union
 
 from langflow.graph.vertex.base import Vertex
-from langflow.graph.utils import extract_input_variables_from_prompt, flatten_list
+from langflow.graph.utils import flatten_list
+from langflow.interface.utils import extract_input_variables_from_prompt
 
 
 class AgentVertex(Vertex):
diff --git a/src/backend/langflow/interface/chains/custom.py b/src/backend/langflow/interface/chains/custom.py
index cb76a53c8..ba4ba8b62 100644
--- a/src/backend/langflow/interface/chains/custom.py
+++ b/src/backend/langflow/interface/chains/custom.py
@@ -5,7 +5,7 @@ from langchain.memory.buffer import ConversationBufferMemory
 from langchain.schema import BaseMemory
 from pydantic import Field, root_validator
 
-from langflow.graph.utils import extract_input_variables_from_prompt
+from langflow.interface.utils import extract_input_variables_from_prompt
 
 DEFAULT_SUFFIX = """"
 Current conversation:
diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py
index 16a7b186c..eb4623f5a 100644
--- a/src/backend/langflow/interface/loading.py
+++ b/src/backend/langflow/interface/loading.py
@@ -12,7 +12,6 @@ from langchain.agents.load_tools import (
     _LLM_TOOLS,
 )
 from langchain.agents.loading import load_agent_from_config
-from langflow.graph import Graph
 from langchain.agents.tools import Tool
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.base import BaseCallbackManager
@@ -22,7 +21,6 @@ from pydantic import ValidationError
 
 from langflow.interface.agents.custom import CUSTOM_AGENTS
 from langflow.interface.importing.utils import get_function, import_by_type
-from langflow.interface.run import fix_memory_inputs
 from langflow.interface.toolkits.base import toolkits_creator
 from langflow.interface.types import get_type_list
 from langflow.interface.utils import load_file_into_dict
@@ -163,37 +161,6 @@ def instantiate_utility(node_type, class_object, params):
     return class_object(**params)
 
 
-def load_flow_from_json(path: str, build=True):
-    """Load flow from json file"""
-    # This is done to avoid circular imports
-
-    with open(path, "r", encoding="utf-8") as f:
-        flow_graph = json.load(f)
-    data_graph = flow_graph["data"]
-    nodes = data_graph["nodes"]
-    # Substitute ZeroShotPrompt with PromptTemplate
-    # nodes = replace_zero_shot_prompt_with_prompt_template(nodes)
-    # Add input variables
-    # nodes = payload.extract_input_variables(nodes)
-
-    # Nodes, edges and root node
-    edges = data_graph["edges"]
-    graph = Graph(nodes, edges)
-    if build:
-        langchain_object = graph.build()
-        if hasattr(langchain_object, "verbose"):
-            langchain_object.verbose = True
-
-        if hasattr(langchain_object, "return_intermediate_steps"):
-            # https://github.com/hwchase17/langchain/issues/2068
-            # Deactivating until we have a frontend solution
-            # to display intermediate steps
-            langchain_object.return_intermediate_steps = False
-        fix_memory_inputs(langchain_object)
-        return langchain_object
-    return graph
-
-
 def replace_zero_shot_prompt_with_prompt_template(nodes):
     """Replace ZeroShotPrompt with PromptTemplate"""
     for node in nodes:
diff --git a/src/backend/langflow/interface/prompts/custom.py b/src/backend/langflow/interface/prompts/custom.py
index b1dbef370..286210271 100644
--- a/src/backend/langflow/interface/prompts/custom.py
+++ b/src/backend/langflow/interface/prompts/custom.py
@@ -3,7 +3,7 @@ from typing import Dict, List, Optional, Type
 from langchain.prompts import PromptTemplate
 from pydantic import root_validator
 
-from langflow.graph.utils import extract_input_variables_from_prompt
+from langflow.interface.utils import extract_input_variables_from_prompt
 
 # Steps to create a BaseCustomPrompt:
 # 1. Create a prompt template that endes with:
diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py
index c2483416f..89f71fd8b 100644
--- a/src/backend/langflow/interface/run.py
+++ b/src/backend/langflow/interface/run.py
@@ -1,10 +1,3 @@
-import contextlib
-import io
-from typing import Any, Dict, List, Tuple
-
-from langchain.schema import AgentAction
-
-from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler  # type: ignore
 from langflow.cache.base import compute_dict_hash, load_cache, memoize_dict
 from langflow.graph import Graph
 from langflow.utils.logger import logger
@@ -24,15 +17,6 @@ def load_langchain_object(data_graph, is_first_message=False):
     return computed_hash, langchain_object
 
 
-def load_or_build_langchain_object(data_graph, is_first_message=False):
-    """
-    Load langchain object from cache if it exists, otherwise build it.
-    """
-    if is_first_message:
-        build_langchain_object_with_caching.clear_cache()
-    return build_langchain_object_with_caching(data_graph)
-
-
 @memoize_dict(maxsize=10)
 def build_langchain_object_with_caching(data_graph):
     """
@@ -40,16 +24,10 @@ def build_langchain_object_with_caching(data_graph):
     """
 
     logger.debug("Building langchain object")
-    graph = build_graph(data_graph)
+    graph = Graph.from_payload(data_graph)
     return graph.build()
 
 
-def build_graph(data_graph):
-    nodes = data_graph["nodes"]
-    edges = data_graph["edges"]
-    return Graph(nodes, edges)
-
-
 def build_langchain_object(data_graph):
     """
     Build langchain object from data_graph.
@@ -66,29 +44,6 @@ def build_langchain_object(data_graph):
     return graph.build()
 
 
-def process_graph_cached(data_graph: Dict[str, Any], message: str):
-    """
-    Process graph by extracting input variables and replacing ZeroShotPrompt
-    with PromptTemplate,then run the graph and return the result and thought.
-    """
-    # Load langchain object
-    is_first_message = len(data_graph.get("chatHistory", [])) == 0
-    langchain_object = load_or_build_langchain_object(data_graph, is_first_message)
-    logger.debug("Loaded langchain object")
-
-    if langchain_object is None:
-        # Raise user facing error
-        raise ValueError(
-            "There was an error loading the langchain_object. Please, check all the nodes and try again."
-        )
-
-    # Generate result and thought
-    logger.debug("Generating result and thought")
-    result, thought = get_result_and_thought(langchain_object, message)
-    logger.debug("Generated result and thought")
-    return {"result": str(result), "thought": thought.strip()}
-
-
 def get_memory_key(langchain_object):
     """
     Given a LangChain object, this function retrieves the current memory key from the object's memory attribute.
@@ -124,147 +79,3 @@ def update_memory_keys(langchain_object, possible_new_mem_key):
     langchain_object.memory.input_key = input_key
     langchain_object.memory.output_key = output_key
     langchain_object.memory.memory_key = possible_new_mem_key
-
-
-def fix_memory_inputs(langchain_object):
-    """
-    Given a LangChain object, this function checks if it has a memory attribute and if that memory key exists in the
-    object's input variables. If so, it does nothing. Otherwise, it gets a possible new memory key using the
-    get_memory_key function and updates the memory keys using the update_memory_keys function.
-    """
-    if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
-        try:
-            if langchain_object.memory.memory_key in langchain_object.input_variables:
-                return
-        except AttributeError:
-            input_variables = (
-                langchain_object.prompt.input_variables
-                if hasattr(langchain_object, "prompt")
-                else langchain_object.input_keys
-            )
-            if langchain_object.memory.memory_key in input_variables:
-                return
-
-        possible_new_mem_key = get_memory_key(langchain_object)
-        if possible_new_mem_key is not None:
-            update_memory_keys(langchain_object, possible_new_mem_key)
-
-
-async def get_result_and_steps(langchain_object, message: str, **kwargs):
-    """Get result and thought from extracted json"""
-
-    try:
-        if hasattr(langchain_object, "verbose"):
-            langchain_object.verbose = True
-        chat_input = None
-        memory_key = ""
-        if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
-            memory_key = langchain_object.memory.memory_key
-
-        if hasattr(langchain_object, "input_keys"):
-            for key in langchain_object.input_keys:
-                if key not in [memory_key, "chat_history"]:
-                    chat_input = {key: message}
-        else:
-            chat_input = message  # type: ignore
-
-        if hasattr(langchain_object, "return_intermediate_steps"):
-            # https://github.com/hwchase17/langchain/issues/2068
-            # Deactivating until we have a frontend solution
-            # to display intermediate steps
-            langchain_object.return_intermediate_steps = True
-
-        fix_memory_inputs(langchain_object)
-        try:
-            async_callbacks = [AsyncStreamingLLMCallbackHandler(**kwargs)]
-            output = await langchain_object.acall(chat_input, callbacks=async_callbacks)
-        except Exception as exc:
-            # make the error message more informative
-            logger.debug(f"Error: {str(exc)}")
-            sync_callbacks = [StreamingLLMCallbackHandler(**kwargs)]
-            output = langchain_object(chat_input, callbacks=sync_callbacks)
-
-        intermediate_steps = (
-            output.get("intermediate_steps", []) if isinstance(output, dict) else []
-        )
-
-        result = (
-            output.get(langchain_object.output_keys[0])
-            if isinstance(output, dict)
-            else output
-        )
-        thought = format_actions(intermediate_steps) if intermediate_steps else ""
-    except Exception as exc:
-        raise ValueError(f"Error: {str(exc)}") from exc
-    return result, thought
-
-
-def get_result_and_thought(langchain_object, message: str):
-    """Get result and thought from extracted json"""
-    try:
-        if hasattr(langchain_object, "verbose"):
-            langchain_object.verbose = True
-        chat_input = None
-        memory_key = ""
-        if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
-            memory_key = langchain_object.memory.memory_key
-
-        if hasattr(langchain_object, "input_keys"):
-            for key in langchain_object.input_keys:
-                if key not in [memory_key, "chat_history"]:
-                    chat_input = {key: message}
-        else:
-            chat_input = message  # type: ignore
-
-        if hasattr(langchain_object, "return_intermediate_steps"):
-            # https://github.com/hwchase17/langchain/issues/2068
-            # Deactivating until we have a frontend solution
-            # to display intermediate steps
-            langchain_object.return_intermediate_steps = False
-
-        fix_memory_inputs(langchain_object)
-
-        with io.StringIO() as output_buffer, contextlib.redirect_stdout(output_buffer):
-            try:
-                # if hasattr(langchain_object, "acall"):
-                #     output = await langchain_object.acall(chat_input)
-                # else:
-                output = langchain_object(chat_input)
-            except ValueError as exc:
-                # make the error message more informative
-                logger.debug(f"Error: {str(exc)}")
-                output = langchain_object.run(chat_input)
-
-            intermediate_steps = (
-                output.get("intermediate_steps", []) if isinstance(output, dict) else []
-            )
-
-            result = (
-                output.get(langchain_object.output_keys[0])
-                if isinstance(output, dict)
-                else output
-            )
-            if intermediate_steps:
-                thought = format_actions(intermediate_steps)
-            else:
-                thought = output_buffer.getvalue()
-
-    except Exception as exc:
-        raise ValueError(f"Error: {str(exc)}") from exc
-    return result, thought
-
-
-def format_actions(actions: List[Tuple[AgentAction, str]]) -> str:
-    """Format a list of (AgentAction, answer) tuples into a string."""
-    output = []
-    for action, answer in actions:
-        log = action.log
-        tool = action.tool
-        tool_input = action.tool_input
-        output.append(f"Log: {log}")
-        if "Action" not in log and "Action Input" not in log:
-            output.append(f"Tool: {tool}")
-            output.append(f"Tool Input: {tool_input}")
-        output.append(f"Answer: {answer}")
-        output.append("")  # Add a blank line
-    return "\n".join(output)
diff --git a/src/backend/langflow/interface/utils.py b/src/backend/langflow/interface/utils.py
index 2b7c5acd1..32c605654 100644
--- a/src/backend/langflow/interface/utils.py
+++ b/src/backend/langflow/interface/utils.py
@@ -2,6 +2,7 @@ import base64
 import json
 import os
 from io import BytesIO
+import re
 
 import yaml
 from langchain.base_language import BaseLanguageModel
@@ -48,3 +49,8 @@ def try_setting_streaming_options(langchain_object, websocket):
         llm.streaming = True
 
     return langchain_object
+
+
+def extract_input_variables_from_prompt(prompt: str) -> list[str]:
+    """Extract input variables from prompt."""
+    return re.findall(r"{(.*?)}", prompt)
diff --git a/src/backend/langflow/processing/__init__.py b/src/backend/langflow/processing/__init__.py
new file mode 100644
index 000000000..e69de29bb
diff --git a/src/backend/langflow/processing/base.py b/src/backend/langflow/processing/base.py
new file mode 100644
index 000000000..97b0d5be0
--- /dev/null
+++ b/src/backend/langflow/processing/base.py
@@ -0,0 +1,55 @@
+from langflow.api.v1.callback import (
+    AsyncStreamingLLMCallbackHandler,
+    StreamingLLMCallbackHandler,
+)
+from langflow.processing.process import fix_memory_inputs, format_actions
+from langflow.utils.logger import logger
+
+
+async def get_result_and_steps(langchain_object, message: str, **kwargs):
+    """Get result and thought from extracted json"""
+
+    try:
+        if hasattr(langchain_object, "verbose"):
+            langchain_object.verbose = True
+        chat_input = None
+        memory_key = ""
+        if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
+            memory_key = langchain_object.memory.memory_key
+
+        if hasattr(langchain_object, "input_keys"):
+            for key in langchain_object.input_keys:
+                if key not in [memory_key, "chat_history"]:
+                    chat_input = {key: message}
+        else:
+            chat_input = message  # type: ignore
+
+        if hasattr(langchain_object, "return_intermediate_steps"):
+            # https://github.com/hwchase17/langchain/issues/2068
+            # Deactivating until we have a frontend solution
+            # to display intermediate steps
+            langchain_object.return_intermediate_steps = True
+
+        fix_memory_inputs(langchain_object)
+        try:
+            async_callbacks = [AsyncStreamingLLMCallbackHandler(**kwargs)]
+            output = await langchain_object.acall(chat_input, callbacks=async_callbacks)
+        except Exception as exc:
+            # make the error message more informative
+            logger.debug(f"Error: {str(exc)}")
+            sync_callbacks = [StreamingLLMCallbackHandler(**kwargs)]
+            output = langchain_object(chat_input, callbacks=sync_callbacks)
+
+        intermediate_steps = (
+            output.get("intermediate_steps", []) if isinstance(output, dict) else []
+        )
+
+        result = (
+            output.get(langchain_object.output_keys[0])
+            if isinstance(output, dict)
+            else output
+        )
+        thought = format_actions(intermediate_steps) if intermediate_steps else ""
+    except Exception as exc:
+        raise ValueError(f"Error: {str(exc)}") from exc
+    return result, thought
diff --git a/src/backend/langflow/processing/process.py b/src/backend/langflow/processing/process.py
new file mode 100644
index 000000000..3b8852e00
--- /dev/null
+++ b/src/backend/langflow/processing/process.py
@@ -0,0 +1,172 @@
+import contextlib
+import io
+from langchain.schema import AgentAction
+import json
+from langflow.interface.run import (
+    build_langchain_object_with_caching,
+    get_memory_key,
+    update_memory_keys,
+)
+from langflow.utils.logger import logger
+from langflow.graph import Graph
+
+
+from typing import Any, Dict, List, Tuple
+
+
+def fix_memory_inputs(langchain_object):
+    """
+    Given a LangChain object, this function checks if it has a memory attribute and if that memory key exists in the
+    object's input variables. If so, it does nothing. Otherwise, it gets a possible new memory key using the
+    get_memory_key function and updates the memory keys using the update_memory_keys function.
+    """
+    if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
+        try:
+            if langchain_object.memory.memory_key in langchain_object.input_variables:
+                return
+        except AttributeError:
+            input_variables = (
+                langchain_object.prompt.input_variables
+                if hasattr(langchain_object, "prompt")
+                else langchain_object.input_keys
+            )
+            if langchain_object.memory.memory_key in input_variables:
+                return
+
+        possible_new_mem_key = get_memory_key(langchain_object)
+        if possible_new_mem_key is not None:
+            update_memory_keys(langchain_object, possible_new_mem_key)
+
+
+def format_actions(actions: List[Tuple[AgentAction, str]]) -> str:
+    """Format a list of (AgentAction, answer) tuples into a string."""
+    output = []
+    for action, answer in actions:
+        log = action.log
+        tool = action.tool
+        tool_input = action.tool_input
+        output.append(f"Log: {log}")
+        if "Action" not in log and "Action Input" not in log:
+            output.append(f"Tool: {tool}")
+            output.append(f"Tool Input: {tool_input}")
+        output.append(f"Answer: {answer}")
+        output.append("")  # Add a blank line
+    return "\n".join(output)
+
+
+def get_result_and_thought(langchain_object, message: str):
+    """Get result and thought from extracted json"""
+    try:
+        if hasattr(langchain_object, "verbose"):
+            langchain_object.verbose = True
+        chat_input = None
+        memory_key = ""
+        if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
+            memory_key = langchain_object.memory.memory_key
+
+        if hasattr(langchain_object, "input_keys"):
+            for key in langchain_object.input_keys:
+                if key not in [memory_key, "chat_history"]:
+                    chat_input = {key: message}
+        else:
+            chat_input = message  # type: ignore
+
+        if hasattr(langchain_object, "return_intermediate_steps"):
+            # https://github.com/hwchase17/langchain/issues/2068
+            # Deactivating until we have a frontend solution
+            # to display intermediate steps
+            langchain_object.return_intermediate_steps = False
+
+        fix_memory_inputs(langchain_object)
+
+        with io.StringIO() as output_buffer, contextlib.redirect_stdout(output_buffer):
+            try:
+                # if hasattr(langchain_object, "acall"):
+                #     output = await langchain_object.acall(chat_input)
+                # else:
+                output = langchain_object(chat_input)
+            except ValueError as exc:
+                # make the error message more informative
+                logger.debug(f"Error: {str(exc)}")
+                output = langchain_object.run(chat_input)
+
+            intermediate_steps = (
+                output.get("intermediate_steps", []) if isinstance(output, dict) else []
+            )
+
+            result = (
+                output.get(langchain_object.output_keys[0])
+                if isinstance(output, dict)
+                else output
+            )
+            if intermediate_steps:
+                thought = format_actions(intermediate_steps)
+            else:
+                thought = output_buffer.getvalue()
+
+    except Exception as exc:
+        raise ValueError(f"Error: {str(exc)}") from exc
+    return result, thought
+
+
+def load_or_build_langchain_object(data_graph, is_first_message=False):
+    """
+    Load langchain object from cache if it exists, otherwise build it.
+    """
+    if is_first_message:
+        build_langchain_object_with_caching.clear_cache()
+    return build_langchain_object_with_caching(data_graph)
+
+
+def process_graph_cached(data_graph: Dict[str, Any], message: str):
+    """
+    Process graph by extracting input variables and replacing ZeroShotPrompt
+    with PromptTemplate,then run the graph and return the result and thought.
+    """
+    # Load langchain object
+    is_first_message = len(data_graph.get("chatHistory", [])) == 0
+    langchain_object = load_or_build_langchain_object(data_graph, is_first_message)
+    logger.debug("Loaded langchain object")
+
+    if langchain_object is None:
+        # Raise user facing error
+        raise ValueError(
+            "There was an error loading the langchain_object. Please, check all the nodes and try again."
+        )
+
+    # Generate result and thought
+    logger.debug("Generating result and thought")
+    result, thought = get_result_and_thought(langchain_object, message)
+    logger.debug("Generated result and thought")
+    return {"result": str(result), "thought": thought.strip()}
+
+
+def load_flow_from_json(path: str, build=True):
+    """Load flow from json file"""
+    # This is done to avoid circular imports
+
+    with open(path, "r", encoding="utf-8") as f:
+        flow_graph = json.load(f)
+    data_graph = flow_graph["data"]
+    nodes = data_graph["nodes"]
+    # Substitute ZeroShotPrompt with PromptTemplate
+    # nodes = replace_zero_shot_prompt_with_prompt_template(nodes)
+    # Add input variables
+    # nodes = payload.extract_input_variables(nodes)
+
+    # Nodes, edges and root node
+    edges = data_graph["edges"]
+    graph = Graph(nodes, edges)
+    if build:
+        langchain_object = graph.build()
+        if hasattr(langchain_object, "verbose"):
+            langchain_object.verbose = True
+
+        if hasattr(langchain_object, "return_intermediate_steps"):
+            # https://github.com/hwchase17/langchain/issues/2068
+            # Deactivating until we have a frontend solution
+            # to display intermediate steps
+            langchain_object.return_intermediate_steps = False
+        fix_memory_inputs(langchain_object)
+        return langchain_object
+    return graph

From 478bb446c3fd9ee6077c7abad6698a583e075e5b Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:06:14 -0300
Subject: [PATCH 51/59] =?UTF-8?q?=F0=9F=90=9B=20fix(frontend):=20add=20mis?=
 =?UTF-8?q?sing=20api/v1=20prefix=20to=20API=20routes=20=F0=9F=90=9B=20fix?=
 =?UTF-8?q?(frontend):=20add=20missing=20api/v1=20prefix=20to=20WebSocket?=
 =?UTF-8?q?=20URL=20=F0=9F=90=9B=20fix(frontend):=20add=20missing=20api/v1?=
 =?UTF-8?q?=20prefix=20to=20Vite=20proxy=20target=20The=20API=20routes,=20?=
 =?UTF-8?q?WebSocket=20URL,=20and=20Vite=20proxy=20target=20were=20missing?=
 =?UTF-8?q?=20the=20"api/v1"=20prefix,=20causing=20the=20frontend=20to=20n?=
 =?UTF-8?q?ot=20be=20able=20to=20communicate=20with=20the=20backend.=20Thi?=
 =?UTF-8?q?s=20commit=20adds=20the=20missing=20prefix=20to=20all=20three?=
 =?UTF-8?q?=20locations=20to=20fix=20the=20issue.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/frontend/src/controllers/API/index.ts   | 4 ++--
 src/frontend/src/modals/chatModal/index.tsx | 4 ++--
 src/frontend/vite.config.ts                 | 2 +-
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/frontend/src/controllers/API/index.ts b/src/frontend/src/controllers/API/index.ts
index f6f46404b..0cffd04bf 100644
--- a/src/frontend/src/controllers/API/index.ts
+++ b/src/frontend/src/controllers/API/index.ts
@@ -14,13 +14,13 @@ export async function sendAll(data: sendAllProps) {
 export async function checkCode(
   code: string
 ): Promise<AxiosResponse<errorsTypeAPI>> {
-  return await axios.post("/validate/code", { code });
+  return await axios.post("api/v1/validate/code", { code });
 }
 
 export async function checkPrompt(
   template: string
 ): Promise<AxiosResponse<PromptTypeAPI>> {
-  return await axios.post("/validate/prompt", { template });
+  return await axios.post("api/v1/validate/prompt", { template });
 }
 
 export async function getExamples(): Promise<FlowType[]> {
diff --git a/src/frontend/src/modals/chatModal/index.tsx b/src/frontend/src/modals/chatModal/index.tsx
index cf2b52aac..39bb72994 100644
--- a/src/frontend/src/modals/chatModal/index.tsx
+++ b/src/frontend/src/modals/chatModal/index.tsx
@@ -182,10 +182,10 @@ export default function ChatModal({
     try {
       const urlWs =
         process.env.NODE_ENV === "development"
-          ? `ws://localhost:7860/chat/${id.current}`
+          ? `ws://localhost:7860/api/v1/chat/${id.current}`
           : `${window.location.protocol === "https:" ? "wss" : "ws"}://${
               window.location.host
-            }/chat/${id.current}`;
+            }api/v1/chat/${id.current}`;
       const newWs = new WebSocket(urlWs);
       newWs.onopen = () => {
         console.log("WebSocket connection established!");
diff --git a/src/frontend/vite.config.ts b/src/frontend/vite.config.ts
index 172b37733..d4fa2248b 100644
--- a/src/frontend/vite.config.ts
+++ b/src/frontend/vite.config.ts
@@ -11,7 +11,7 @@ const apiRoutes = [
 ];
 
 // Use environment variable to determine the target.
-const target = process.env.VITE_PROXY_TARGET || "http://127.0.0.1:7860";
+const target = process.env.VITE_PROXY_TARGET || "http://127.0.0.1:7860/api/v1";
 
 const proxyTargets = apiRoutes.reduce((proxyObj, route) => {
   proxyObj[route] = {

From 6b5539232fa98fcb8323aa66289d6414b4fa1dd2 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:06:40 -0300
Subject: [PATCH 52/59] =?UTF-8?q?=F0=9F=9A=80=20chore(server,=20tests):=20?=
 =?UTF-8?q?update=20API=20endpoint=20URLs=20to=20include=20version=20numbe?=
 =?UTF-8?q?r=20The=20API=20endpoint=20URLs=20have=20been=20updated=20to=20?=
 =?UTF-8?q?include=20the=20version=20number=20to=20improve=20the=20API's?=
 =?UTF-8?q?=20versioning=20and=20maintainability.=20The=20changes=20were?=
 =?UTF-8?q?=20made=20to=20the=20server.ts=20file=20and=20the=20tests=20tha?=
 =?UTF-8?q?t=20use=20the=20API=20endpoints.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

🐛 fix(tests): update API endpoint paths in test files
The API endpoint paths in the test files were outdated and have been updated to reflect the current API version. This ensures that the tests are running against the correct endpoints and that the tests are up-to-date with the current API version.
---
 tests/test_agents_template.py  | 10 +++++-----
 tests/test_cache.py            |  6 +++---
 tests/test_chains_template.py  | 16 ++++++++--------
 tests/test_endpoints.py        | 20 ++++++++++----------
 tests/test_graph.py            |  2 +-
 tests/test_llms_template.py    |  8 ++++----
 tests/test_loading.py          |  2 +-
 tests/test_prompts_template.py |  8 ++++----
 tests/test_websocket.py        |  8 ++++----
 9 files changed, 40 insertions(+), 40 deletions(-)

diff --git a/tests/test_agents_template.py b/tests/test_agents_template.py
index 7aa8de176..8e181711f 100644
--- a/tests/test_agents_template.py
+++ b/tests/test_agents_template.py
@@ -5,7 +5,7 @@ from langflow.settings import settings
 # check that all agents are in settings.agents
 # are in json_response["agents"]
 def test_agents_settings(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     agents = json_response["agents"]
@@ -13,7 +13,7 @@ def test_agents_settings(client: TestClient):
 
 
 def test_zero_shot_agent(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     agents = json_response["agents"]
@@ -52,7 +52,7 @@ def test_zero_shot_agent(client: TestClient):
 
 
 def test_json_agent(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     agents = json_response["agents"]
@@ -87,7 +87,7 @@ def test_json_agent(client: TestClient):
 
 
 def test_csv_agent(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     agents = json_response["agents"]
@@ -126,7 +126,7 @@ def test_csv_agent(client: TestClient):
 
 
 def test_initialize_agent(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     agents = json_response["agents"]
diff --git a/tests/test_cache.py b/tests/test_cache.py
index 3d3e951fc..3214e7d15 100644
--- a/tests/test_cache.py
+++ b/tests/test_cache.py
@@ -1,10 +1,10 @@
 import json
+from langflow.graph import Graph
+from langflow.processing.process import load_or_build_langchain_object
 
 import pytest
 from langflow.interface.run import (
-    build_graph,
     build_langchain_object_with_caching,
-    load_or_build_langchain_object,
 )
 
 
@@ -62,7 +62,7 @@ def test_build_langchain_object_with_caching(basic_data_graph):
 
 # Test build_graph
 def test_build_graph(basic_data_graph):
-    graph = build_graph(basic_data_graph)
+    graph = Graph.from_payload(basic_data_graph)
     assert graph is not None
     assert len(graph.nodes) == len(basic_data_graph["nodes"])
     assert len(graph.edges) == len(basic_data_graph["edges"])
diff --git a/tests/test_chains_template.py b/tests/test_chains_template.py
index c958cf64d..0c7af56ad 100644
--- a/tests/test_chains_template.py
+++ b/tests/test_chains_template.py
@@ -3,7 +3,7 @@ from langflow.settings import settings
 
 
 def test_chains_settings(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -12,7 +12,7 @@ def test_chains_settings(client: TestClient):
 
 # Test the ConversationChain object
 def test_conversation_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -94,7 +94,7 @@ def test_conversation_chain(client: TestClient):
 
 
 def test_llm_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -152,7 +152,7 @@ def test_llm_chain(client: TestClient):
 
 
 def test_llm_checker_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -228,7 +228,7 @@ def test_llm_checker_chain(client: TestClient):
 
 
 def test_llm_math_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -306,7 +306,7 @@ def test_llm_math_chain(client: TestClient):
 
 
 def test_series_character_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -368,7 +368,7 @@ def test_series_character_chain(client: TestClient):
 
 
 def test_mid_journey_prompt_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
@@ -407,7 +407,7 @@ def test_mid_journey_prompt_chain(client: TestClient):
 
 
 def test_time_travel_guide_chain(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     chains = json_response["chains"]
diff --git a/tests/test_endpoints.py b/tests/test_endpoints.py
index 83f6c62b1..9e07dfb24 100644
--- a/tests/test_endpoints.py
+++ b/tests/test_endpoints.py
@@ -4,7 +4,7 @@ from langflow.interface.tools.constants import CUSTOM_TOOLS
 
 
 def test_get_all(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     # We need to test the custom nodes
@@ -21,7 +21,7 @@ import math
 def square(x):
     return x ** 2
 """
-    response1 = client.post("/validate/code", json={"code": code1})
+    response1 = client.post("api/v1/validate/code", json={"code": code1})
     assert response1.status_code == 200
     assert response1.json() == {"imports": {"errors": []}, "function": {"errors": []}}
 
@@ -32,7 +32,7 @@ import non_existent_module
 def square(x):
     return x ** 2
 """
-    response2 = client.post("/validate/code", json={"code": code2})
+    response2 = client.post("api/v1/validate/code", json={"code": code2})
     assert response2.status_code == 200
     assert response2.json() == {
         "imports": {"errors": ["No module named 'non_existent_module'"]},
@@ -46,7 +46,7 @@ import math
 def square(x)
     return x ** 2
 """
-    response3 = client.post("/validate/code", json={"code": code3})
+    response3 = client.post("api/v1/validate/code", json={"code": code3})
     assert response3.status_code == 200
     assert response3.json() == {
         "imports": {"errors": []},
@@ -54,11 +54,11 @@ def square(x)
     }
 
     # Test case with invalid JSON payload
-    response4 = client.post("/validate/code", json={"invalid_key": code1})
+    response4 = client.post("api/v1/validate/code", json={"invalid_key": code1})
     assert response4.status_code == 422
 
     # Test case with an empty code string
-    response5 = client.post("/validate/code", json={"code": ""})
+    response5 = client.post("api/v1/validate/code", json={"code": ""})
     assert response5.status_code == 200
     assert response5.json() == {"imports": {"errors": []}, "function": {"errors": []}}
 
@@ -69,7 +69,7 @@ import math
 def square(x)
     return x ** 2
 """
-    response6 = client.post("/validate/code", json={"code": code6})
+    response6 = client.post("api/v1/validate/code", json={"code": code6})
     assert response6.status_code == 200
     assert response6.json() == {
         "imports": {"errors": []},
@@ -95,13 +95,13 @@ INVALID_PROMPT = "This is an invalid prompt without any input variable."
 
 
 def test_valid_prompt(client: TestClient):
-    response = client.post("/validate/prompt", json={"template": VALID_PROMPT})
+    response = client.post("api/v1/validate/prompt", json={"template": VALID_PROMPT})
     assert response.status_code == 200
     assert response.json() == {"input_variables": ["product"]}
 
 
 def test_invalid_prompt(client: TestClient):
-    response = client.post("/validate/prompt", json={"template": INVALID_PROMPT})
+    response = client.post("api/v1/validate/prompt", json={"template": INVALID_PROMPT})
     assert response.status_code == 200
     assert response.json() == {"input_variables": []}
 
@@ -116,7 +116,7 @@ def test_invalid_prompt(client: TestClient):
     ],
 )
 def test_various_prompts(client, prompt, expected_input_variables):
-    response = client.post("/validate/prompt", json={"template": prompt})
+    response = client.post("api/v1/validate/prompt", json={"template": prompt})
     assert response.status_code == 200
     assert response.json() == {
         "input_variables": expected_input_variables,
diff --git a/tests/test_graph.py b/tests/test_graph.py
index 8c6560d54..69a926cc3 100644
--- a/tests/test_graph.py
+++ b/tests/test_graph.py
@@ -16,7 +16,7 @@ from langflow.graph.vertex.types import (
     ToolVertex,
     WrapperVertex,
 )
-from langflow.interface.run import get_result_and_thought
+from langflow.processing.process import get_result_and_thought
 from langflow.utils.payload import get_root_node
 
 # Test cases for the graph module
diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index ccf2f6388..da0b94318 100644
--- a/tests/test_llms_template.py
+++ b/tests/test_llms_template.py
@@ -3,7 +3,7 @@ from langflow.settings import settings
 
 
 def test_llms_settings(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     llms = json_response["llms"]
@@ -11,7 +11,7 @@ def test_llms_settings(client: TestClient):
 
 
 # def test_hugging_face_hub(client: TestClient):
-#     response = client.get("/all")
+#     response = client.get("api/v1/all")
 #     assert response.status_code == 200
 #     json_response = response.json()
 #     language_models = json_response["llms"]
@@ -103,7 +103,7 @@ def test_llms_settings(client: TestClient):
 
 
 def test_openai(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     language_models = json_response["llms"]
@@ -333,7 +333,7 @@ def test_openai(client: TestClient):
 
 
 def test_chat_open_ai(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     language_models = json_response["llms"]
diff --git a/tests/test_loading.py b/tests/test_loading.py
index 872314699..885eb7a82 100644
--- a/tests/test_loading.py
+++ b/tests/test_loading.py
@@ -2,7 +2,7 @@ import json
 
 import pytest
 from langchain.chains.base import Chain
-from langflow import load_flow_from_json
+from langflow.processing.process import load_flow_from_json
 from langflow.graph import Graph
 from langflow.utils.payload import get_root_node
 
diff --git a/tests/test_prompts_template.py b/tests/test_prompts_template.py
index 83da2f14d..a8562898c 100644
--- a/tests/test_prompts_template.py
+++ b/tests/test_prompts_template.py
@@ -3,7 +3,7 @@ from langflow.settings import settings
 
 
 def test_prompts_settings(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     prompts = json_response["prompts"]
@@ -11,7 +11,7 @@ def test_prompts_settings(client: TestClient):
 
 
 def test_prompt_template(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     prompts = json_response["prompts"]
@@ -89,7 +89,7 @@ def test_prompt_template(client: TestClient):
 
 
 def test_few_shot_prompt_template(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     prompts = json_response["prompts"]
@@ -168,7 +168,7 @@ def test_few_shot_prompt_template(client: TestClient):
 
 
 def test_zero_shot_prompt(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     prompts = json_response["prompts"]
diff --git a/tests/test_websocket.py b/tests/test_websocket.py
index 5b60d0fed..611faff79 100644
--- a/tests/test_websocket.py
+++ b/tests/test_websocket.py
@@ -5,17 +5,17 @@ from fastapi.testclient import TestClient
 
 
 def test_websocket_connection(client: TestClient):
-    with client.websocket_connect("/chat/test_client") as websocket:
+    with client.websocket_connect("api/v1/chat/test_client") as websocket:
         assert websocket.scope["client"] == ["testclient", 50000]
-        assert websocket.scope["path"] == "/chat/test_client"
+        assert websocket.scope["path"] == "/api/v1/chat/test_client"
 
 
 def test_chat_history(client: TestClient):
     # Mock the process_graph function to return a specific value
-    with patch("langflow.api.chat_manager.process_graph") as mock_process_graph:
+    with patch("langflow.chat.manager.process_graph") as mock_process_graph:
         mock_process_graph.return_value = ("Hello, I'm a mock response!", "")
 
-        with client.websocket_connect("/chat/test_client") as websocket:
+        with client.websocket_connect("api/v1/chat/test_client") as websocket:
             # First message should be the history
             history = websocket.receive_json()
             assert history == []  # Empty history

From 1e854fc4694b6404b6f09050be9566f8fd1d6784 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 10:06:51 -0300
Subject: [PATCH 53/59] update endpoint

---
 tests/test_vectorstore_template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_vectorstore_template.py b/tests/test_vectorstore_template.py
index 5b1d7e5bc..0aa823786 100644
--- a/tests/test_vectorstore_template.py
+++ b/tests/test_vectorstore_template.py
@@ -5,7 +5,7 @@ from langflow.settings import settings
 # check that all agents are in settings.agents
 # are in json_response["agents"]
 def test_vectorstores_settings(client: TestClient):
-    response = client.get("/all")
+    response = client.get("api/v1/all")
     assert response.status_code == 200
     json_response = response.json()
     vectorstores = json_response["vectorstores"]

From 3de23e345f4d02f7fb48a03c20891508df11af71 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 11:40:39 -0300
Subject: [PATCH 54/59] =?UTF-8?q?=F0=9F=9A=80=20feat(customs.py):=20add=20?=
 =?UTF-8?q?PythonFunction=20to=20CUSTOM=5FNODES=20=F0=9F=9A=80=20feat(load?=
 =?UTF-8?q?ing.py):=20add=20support=20for=20PythonFunction=20node=20type?=
 =?UTF-8?q?=20=F0=9F=9A=80=20feat(constants.py):=20add=20PythonFunction=20?=
 =?UTF-8?q?to=20CUSTOM=5FTOOLS=20=F0=9F=9A=80=20feat(custom.py):=20add=20P?=
 =?UTF-8?q?ythonFunction=20class=20=F0=9F=9A=80=20feat(frontend=5Fnode/too?=
 =?UTF-8?q?ls.py):=20add=20PythonFunctionNode=20class=20=F0=9F=A7=AA=20tes?=
 =?UTF-8?q?t(test=5Fcustom=5Ftypes.py):=20add=20test=20for=20PythonFunctio?=
 =?UTF-8?q?n=20class=20=F0=9F=A7=AA=20test(test=5Fllms=5Ftemplate.py):=20c?=
 =?UTF-8?q?omment=20out=20tests=20for=20AzureOpenAI=20and=20AzureChatOpenA?=
 =?UTF-8?q?I=20The=20changes=20add=20support=20for=20a=20new=20node=20type?=
 =?UTF-8?q?,=20PythonFunction,=20which=20allows=20users=20to=20define=20a?=
 =?UTF-8?q?=20Python=20function=20to=20be=20executed.=20The=20node=20type?=
 =?UTF-8?q?=20is=20added=20to=20CUSTOM=5FNODES=20in=20customs.py,=20and=20?=
 =?UTF-8?q?support=20for=20the=20node=20type=20is=20added=20to=20loading.p?=
 =?UTF-8?q?y.=20The=20node=20type=20is=20also=20added=20to=20CUSTOM=5FTOOL?=
 =?UTF-8?q?S=20in=20constants.py,=20and=20the=20PythonFunction=20class=20i?=
 =?UTF-8?q?s=20added=20to=20custom.py.=20The=20PythonFunctionNode=20class?=
 =?UTF-8?q?=20is=20added=20to=20frontend=5Fnode/tools.py.=20Tests=20for=20?=
 =?UTF-8?q?the=20new=20PythonFunction=20class=20are=20added=20to=20test=5F?=
 =?UTF-8?q?custom=5Ftypes.py.=20Tests=20for=20AzureOpenAI=20and=20AzureCha?=
 =?UTF-8?q?tOpenAI=20are=20commented=20out=20in=20test=5Fllms=5Ftemplate.p?=
 =?UTF-8?q?y.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/custom/customs.py        |   1 +
 src/backend/langflow/interface/loading.py     |   8 +-
 .../langflow/interface/tools/constants.py     |   8 +-
 .../langflow/interface/tools/custom.py        |  15 +-
 .../langflow/template/frontend_node/tools.py  |  24 ++++
 tests/test_custom_types.py                    |  16 ++-
 tests/test_llms_template.py                   | 133 +++++++++---------
 7 files changed, 133 insertions(+), 72 deletions(-)

diff --git a/src/backend/langflow/custom/customs.py b/src/backend/langflow/custom/customs.py
index f7a82e4a3..92e1fc2d8 100644
--- a/src/backend/langflow/custom/customs.py
+++ b/src/backend/langflow/custom/customs.py
@@ -5,6 +5,7 @@ CUSTOM_NODES = {
     "prompts": {"ZeroShotPrompt": frontend_node.prompts.ZeroShotPromptNode()},
     "tools": {
         "PythonFunctionTool": frontend_node.tools.PythonFunctionToolNode(),
+        "PythonFunction": frontend_node.tools.PythonFunctionNode(),
         "Tool": frontend_node.tools.ToolNode(),
     },
     "agents": {
diff --git a/src/backend/langflow/interface/loading.py b/src/backend/langflow/interface/loading.py
index 16a7b186c..95f5041ba 100644
--- a/src/backend/langflow/interface/loading.py
+++ b/src/backend/langflow/interface/loading.py
@@ -26,7 +26,7 @@ from langflow.interface.run import fix_memory_inputs
 from langflow.interface.toolkits.base import toolkits_creator
 from langflow.interface.types import get_type_list
 from langflow.interface.utils import load_file_into_dict
-from langflow.utils import util
+from langflow.utils import util, validate
 
 
 def instantiate_class(node_type: str, base_type: str, params: Dict) -> Any:
@@ -103,6 +103,12 @@ def instantiate_tool(node_type, class_object, params):
     elif node_type == "PythonFunctionTool":
         params["func"] = get_function(params.get("code"))
         return class_object(**params)
+    # For backward compatibility
+    elif node_type == "PythonFunction":
+        function_string = params["code"]
+        if isinstance(function_string, str):
+            return validate.eval_function(function_string)
+        raise ValueError("Function should be a string")
     elif node_type.lower() == "tool":
         return class_object(**params)
     return class_object(**params)
diff --git a/src/backend/langflow/interface/tools/constants.py b/src/backend/langflow/interface/tools/constants.py
index 31c75ec08..fea3c5237 100644
--- a/src/backend/langflow/interface/tools/constants.py
+++ b/src/backend/langflow/interface/tools/constants.py
@@ -9,10 +9,14 @@ from langchain.agents.load_tools import (
 from langchain.tools.json.tool import JsonSpec
 
 from langflow.interface.importing.utils import import_class
-from langflow.interface.tools.custom import PythonFunctionTool
+from langflow.interface.tools.custom import PythonFunctionTool, PythonFunction
 
 FILE_TOOLS = {"JsonSpec": JsonSpec}
-CUSTOM_TOOLS = {"Tool": Tool, "PythonFunctionTool": PythonFunctionTool}
+CUSTOM_TOOLS = {
+    "Tool": Tool,
+    "PythonFunctionTool": PythonFunctionTool,
+    "PythonFunction": PythonFunction,
+}
 
 OTHER_TOOLS = {tool: import_class(f"langchain.tools.{tool}") for tool in tools.__all__}
 
diff --git a/src/backend/langflow/interface/tools/custom.py b/src/backend/langflow/interface/tools/custom.py
index b2d43565d..0e2e5ff57 100644
--- a/src/backend/langflow/interface/tools/custom.py
+++ b/src/backend/langflow/interface/tools/custom.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Callable, Optional
 from langflow.interface.importing.utils import get_function
 
 from pydantic import BaseModel, validator
@@ -9,6 +9,7 @@ from langchain.agents.tools import Tool
 
 class Function(BaseModel):
     code: str
+    function: Optional[Callable] = None
     imports: Optional[str] = None
 
     # Eval code and store the function
@@ -25,6 +26,12 @@ class Function(BaseModel):
 
         return v
 
+    def get_function(self):
+        """Get the function"""
+        function_name = validate.extract_function_name(self.code)
+
+        return validate.create_function(self.code, function_name)
+
 
 class PythonFunctionTool(Function, Tool):
     """Python function"""
@@ -39,3 +46,9 @@ class PythonFunctionTool(Function, Tool):
         self.code = code
         self.func = get_function(self.code)
         super().__init__(name=name, description=description, func=self.func)
+
+
+class PythonFunction(Function):
+    """Python function"""
+
+    code: str
diff --git a/src/backend/langflow/template/frontend_node/tools.py b/src/backend/langflow/template/frontend_node/tools.py
index 4e97fec8c..3094f3568 100644
--- a/src/backend/langflow/template/frontend_node/tools.py
+++ b/src/backend/langflow/template/frontend_node/tools.py
@@ -103,3 +103,27 @@ class PythonFunctionToolNode(FrontendNode):
 
     def to_dict(self):
         return super().to_dict()
+
+
+class PythonFunctionNode(FrontendNode):
+    name: str = "PythonFunction"
+    template: Template = Template(
+        type_name="python_function",
+        fields=[
+            TemplateField(
+                field_type="code",
+                required=True,
+                placeholder="",
+                is_list=False,
+                show=True,
+                value=DEFAULT_PYTHON_FUNCTION,
+                name="code",
+                advanced=False,
+            )
+        ],
+    )
+    description: str = "Python function to be executed."
+    base_classes: list[str] = ["function"]
+
+    def to_dict(self):
+        return super().to_dict()
diff --git a/tests/test_custom_types.py b/tests/test_custom_types.py
index 7503426ab..b65f58d0a 100644
--- a/tests/test_custom_types.py
+++ b/tests/test_custom_types.py
@@ -1,11 +1,11 @@
 # Test this:
 from langflow.interface.importing.utils import get_function
 import pytest
-from langflow.interface.tools.custom import PythonFunctionTool
+from langflow.interface.tools.custom import PythonFunctionTool, PythonFunction
 from langflow.utils import constants
 
 
-def test_python_function():
+def test_python_function_tool():
     """Test Python function"""
     code = constants.DEFAULT_PYTHON_FUNCTION
     func = get_function(code)
@@ -21,3 +21,15 @@ def test_python_function():
         func = PythonFunctionTool(
             name="Test", description="Testing", code=code, func=func
         )
+
+
+def test_python_function():
+    """Test Python function"""
+    func = PythonFunction(code=constants.DEFAULT_PYTHON_FUNCTION)
+    assert get_function(func.code)("text") == "text"
+    # the tool decorator should raise an error if
+    # the function is not str -> str
+
+    # This raises ValidationError
+    with pytest.raises(SyntaxError):
+        func = PythonFunction(code=pytest.CODE_WITH_SYNTAX_ERROR)
diff --git a/tests/test_llms_template.py b/tests/test_llms_template.py
index f54b452f1..6c117deb1 100644
--- a/tests/test_llms_template.py
+++ b/tests/test_llms_template.py
@@ -484,75 +484,76 @@ def test_chat_open_ai(client: TestClient):
     }
 
 
-def test_azure_open_ai(client: TestClient):
-    response = client.get("/all")
-    assert response.status_code == 200
-    json_response = response.json()
-    language_models = json_response["llms"]
+# Commenting this out for now, as it requires to activate the nodes
+# def test_azure_open_ai(client: TestClient):
+#     response = client.get("/all")
+#     assert response.status_code == 200
+#     json_response = response.json()
+#     language_models = json_response["llms"]
 
-    model = language_models["AzureOpenAI"]
-    template = model["template"]
+#     model = language_models["AzureOpenAI"]
+#     template = model["template"]
 
-    assert template["model_name"].show is False
-    assert template["deployment_name"] == {
-        "required": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "",
-        "password": False,
-        "name": "deployment_name",
-        "advanced": False,
-        "type": "str",
-        "list": False,
-    }
+#     assert template["model_name"]["show"] is False
+#     assert template["deployment_name"] == {
+#         "required": False,
+#         "placeholder": "",
+#         "show": True,
+#         "multiline": False,
+#         "value": "",
+#         "password": False,
+#         "name": "deployment_name",
+#         "advanced": False,
+#         "type": "str",
+#         "list": False,
+#     }
 
 
-def test_azure_chat_open_ai(client: TestClient):
-    response = client.get("/all")
-    assert response.status_code == 200
-    json_response = response.json()
-    language_models = json_response["llms"]
+# def test_azure_chat_open_ai(client: TestClient):
+#     response = client.get("/all")
+#     assert response.status_code == 200
+#     json_response = response.json()
+#     language_models = json_response["llms"]
 
-    model = language_models["AzureChatOpenAI"]
-    template = model["template"]
+#     model = language_models["AzureChatOpenAI"]
+#     template = model["template"]
 
-    assert template["model_name"].show is False
-    assert template["deployment_name"] == {
-        "required": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "",
-        "password": False,
-        "name": "deployment_name",
-        "advanced": False,
-        "type": "str",
-        "list": False,
-    }
-    assert template["openai_api_type"] == {
-        "required": False,
-        "placeholder": "",
-        "show": False,
-        "multiline": False,
-        "value": "azure",
-        "password": False,
-        "name": "openai_api_type",
-        "display_name": "OpenAI API Type",
-        "advanced": False,
-        "type": "str",
-        "list": False,
-    }
-    assert template["openai_api_version"] == {
-        "required": False,
-        "placeholder": "",
-        "show": True,
-        "multiline": False,
-        "value": "2023-03-15-preview",
-        "password": False,
-        "name": "openai_api_version",
-        "display_name": "OpenAI API Version",
-        "advanced": False,
-        "type": "str",
-        "list": False,
-    }
+#     assert template["model_name"]["show"] is False
+#     assert template["deployment_name"] == {
+#         "required": False,
+#         "placeholder": "",
+#         "show": True,
+#         "multiline": False,
+#         "value": "",
+#         "password": False,
+#         "name": "deployment_name",
+#         "advanced": False,
+#         "type": "str",
+#         "list": False,
+#     }
+#     assert template["openai_api_type"] == {
+#         "required": False,
+#         "placeholder": "",
+#         "show": False,
+#         "multiline": False,
+#         "value": "azure",
+#         "password": False,
+#         "name": "openai_api_type",
+#         "display_name": "OpenAI API Type",
+#         "advanced": False,
+#         "type": "str",
+#         "list": False,
+#     }
+#     assert template["openai_api_version"] == {
+#         "required": False,
+#         "placeholder": "",
+#         "show": True,
+#         "multiline": False,
+#         "value": "2023-03-15-preview",
+#         "password": False,
+#         "name": "openai_api_version",
+#         "display_name": "OpenAI API Version",
+#         "advanced": False,
+#         "type": "str",
+#         "list": False,
+#     }

From 73bad871f25dc6c7f50e09d2551eff64986005e1 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 11:44:57 -0300
Subject: [PATCH 55/59] =?UTF-8?q?=F0=9F=90=9B=20fix(llms.py):=20add=20null?=
 =?UTF-8?q?=20check=20for=20name=20variable=20in=20azure=20field=20formatt?=
 =?UTF-8?q?ing=20The=20code=20was=20updated=20to=20add=20a=20null=20check?=
 =?UTF-8?q?=20for=20the=20name=20variable=20before=20checking=20if=20it=20?=
 =?UTF-8?q?contains=20the=20string=20"azure".=20This=20prevents=20a=20pote?=
 =?UTF-8?q?ntial=20runtime=20error=20if=20the=20name=20variable=20is=20nul?=
 =?UTF-8?q?l.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/template/frontend_node/llms.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/backend/langflow/template/frontend_node/llms.py b/src/backend/langflow/template/frontend_node/llms.py
index 39e82422f..c2aedb45e 100644
--- a/src/backend/langflow/template/frontend_node/llms.py
+++ b/src/backend/langflow/template/frontend_node/llms.py
@@ -66,5 +66,5 @@ class LLMFrontendNode(FrontendNode):
             field.show = True
 
         LLMFrontendNode.format_openai_field(field)
-        if "azure" in name.lower():
+        if name and "azure" in name.lower():
             LLMFrontendNode.format_azure_field(field)

From 998d1bf5ed071b4152c0472a73ebdc0d9e80a8f2 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 12:55:12 -0300
Subject: [PATCH 56/59] =?UTF-8?q?=F0=9F=9A=80=20chore(Makefile):=20add=20i?=
 =?UTF-8?q?nstall=5Fbackend=20command=20to=20backend=20target=20The=20Make?=
 =?UTF-8?q?file=20has=20been=20updated=20to=20include=20the=20`install=5Fb?=
 =?UTF-8?q?ackend`=20command=20as=20a=20dependency=20of=20the=20`backend`?=
 =?UTF-8?q?=20target.=20This=20ensures=20that=20the=20backend=20dependenci?=
 =?UTF-8?q?es=20are=20installed=20before=20running=20the=20backend=20serve?=
 =?UTF-8?q?r.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Makefile | 1 +
 1 file changed, 1 insertion(+)

diff --git a/Makefile b/Makefile
index 15337f65b..baf4220ff 100644
--- a/Makefile
+++ b/Makefile
@@ -43,6 +43,7 @@ install_backend:
 	poetry install
 
 backend:
+	make install_backend
 	poetry run uvicorn langflow.main:app --port 7860 --reload --log-level debug
 
 build_frontend:

From 683b7ab4d0ae8e9d0c439e8545df5b6b0a476960 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 6 Jun 2023 17:41:17 -0300
Subject: [PATCH 57/59] =?UTF-8?q?=F0=9F=9A=80=20feat(config.yaml):=20add?=
 =?UTF-8?q?=20PythonFunction=20to=20the=20list=20of=20available=20tools=20?=
 =?UTF-8?q?The=20PythonFunction=20tool=20has=20been=20added=20to=20the=20l?=
 =?UTF-8?q?ist=20of=20available=20tools=20in=20the=20config.yaml=20file.?=
 =?UTF-8?q?=20This=20allows=20the=20backend=20to=20use=20Python=20function?=
 =?UTF-8?q?s=20as=20part=20of=20the=20language=20processing=20pipeline.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/backend/langflow/config.yaml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/backend/langflow/config.yaml b/src/backend/langflow/config.yaml
index cff623c09..2d052eb67 100644
--- a/src/backend/langflow/config.yaml
+++ b/src/backend/langflow/config.yaml
@@ -84,6 +84,7 @@ tools:
   - Serper Search
   - Tool
   - PythonFunctionTool
+  - PythonFunction
   - JsonSpec
   - News API
   - TMDB API

From b21bd3b8e0b0ffbe5ab75e3ad473e89431939d3d Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 7 Jun 2023 16:28:48 -0300
Subject: [PATCH 58/59] fix: now the version badge text aligns when there's no
 version

---
 src/frontend/src/App.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/frontend/src/App.tsx b/src/frontend/src/App.tsx
index e00019822..9e84a6430 100644
--- a/src/frontend/src/App.tsx
+++ b/src/frontend/src/App.tsx
@@ -170,7 +170,7 @@ export default function App() {
         className="absolute left-7 bottom-2 flex h-6 cursor-pointer flex-col items-center justify-start overflow-hidden rounded-lg bg-gray-800 px-2 text-center font-sans text-xs tracking-wide text-gray-300 transition-all duration-500 ease-in-out hover:h-12 dark:bg-gray-100 dark:text-gray-800"
       >
         {version && <div className="mt-1">⛓️ LangFlow v{version}</div>}
-        <div className="mt-2">Created by Logspace</div>
+        <div className={version ? "mt-2" : "mt-1"}>Created by Logspace</div>
       </a>
     </div>
   );

From 23e760b05296521d709e9621bc15f5d5d6ab9451 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Thu, 8 Jun 2023 08:45:20 -0300
Subject: [PATCH 59/59] =?UTF-8?q?=F0=9F=94=BA=20chore(pyproject.toml):=20u?=
 =?UTF-8?q?pdate=20langchain=20dependency=20from=200.0.186=20to=200.0.194?=
 =?UTF-8?q?=20This=20commit=20updates=20the=20langchain=20dependency=20fro?=
 =?UTF-8?q?m=20version=200.0.186=20to=20version=200.0.194.=20This=20is=20a?=
 =?UTF-8?q?=20chore=20as=20it=20updates=20a=20dependency=20to=20a=20newer?=
 =?UTF-8?q?=20version.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 poetry.lock    | 30 ++++++++++++++++++++++++------
 pyproject.toml |  2 +-
 2 files changed, 25 insertions(+), 7 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index d4d30eb67..43a305b1a 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -2377,20 +2377,21 @@ test = ["ipykernel", "pre-commit", "pytest", "pytest-cov", "pytest-timeout"]
 
 [[package]]
 name = "langchain"
-version = "0.0.186"
+version = "0.0.194"
 description = "Building applications with LLMs through composability"
 category = "main"
 optional = false
 python-versions = ">=3.8.1,<4.0"
 files = [
-    {file = "langchain-0.0.186-py3-none-any.whl", hash = "sha256:c53ac8943351602dbde84759d32d57fe2e6599279576935a004688e43ee8ffbf"},
-    {file = "langchain-0.0.186.tar.gz", hash = "sha256:36d6d3872727a6f7d6db1b05b13caac35fed19a0d395d2264ed82aae53cfddfd"},
+    {file = "langchain-0.0.194-py3-none-any.whl", hash = "sha256:b1d47f96c3556eebb5b330492e64fed1f5585c943be7e1fe675ff31a84b010e3"},
+    {file = "langchain-0.0.194.tar.gz", hash = "sha256:480c9cbce12161b3aece3b6fdf03f533c157539ae6243712b61b0d2558f9a96c"},
 ]
 
 [package.dependencies]
 aiohttp = ">=3.8.3,<4.0.0"
 async-timeout = {version = ">=4.0.0,<5.0.0", markers = "python_version < \"3.11\""}
 dataclasses-json = ">=0.5.7,<0.6.0"
+langchainplus-sdk = ">=0.0.6"
 numexpr = ">=2.8.4,<3.0.0"
 numpy = ">=1,<2"
 openapi-schema-pydantic = ">=1.2,<2.0"
@@ -2401,12 +2402,12 @@ SQLAlchemy = ">=1.4,<3"
 tenacity = ">=8.1.0,<9.0.0"
 
 [package.extras]
-all = ["O365 (>=2.0.26,<3.0.0)", "aleph-alpha-client (>=2.15.0,<3.0.0)", "anthropic (>=0.2.6,<0.3.0)", "arxiv (>=1.4,<2.0)", "atlassian-python-api (>=3.36.0,<4.0.0)", "azure-ai-formrecognizer (>=3.2.1,<4.0.0)", "azure-ai-vision (>=0.11.1b1,<0.12.0)", "azure-cognitiveservices-speech (>=1.28.0,<2.0.0)", "azure-cosmos (>=4.4.0b1,<5.0.0)", "azure-identity (>=1.12.0,<2.0.0)", "beautifulsoup4 (>=4,<5)", "clickhouse-connect (>=0.5.14,<0.6.0)", "cohere (>=3,<4)", "deeplake (>=3.3.0,<4.0.0)", "docarray[hnswlib] (>=0.32.0,<0.33.0)", "duckduckgo-search (>=2.8.6,<3.0.0)", "elasticsearch (>=8,<9)", "faiss-cpu (>=1,<2)", "google-api-python-client (==2.70.0)", "google-search-results (>=2,<3)", "gptcache (>=0.1.7)", "html2text (>=2020.1.16,<2021.0.0)", "huggingface_hub (>=0,<1)", "jina (>=3.14,<4.0)", "jinja2 (>=3,<4)", "jq (>=1.4.1,<2.0.0)", "lancedb (>=0.1,<0.2)", "langkit (>=0.0.1.dev3,<0.1.0)", "lark (>=1.1.5,<2.0.0)", "lxml (>=4.9.2,<5.0.0)", "manifest-ml (>=0.0.1,<0.0.2)", "momento (>=1.5.0,<2.0.0)", "neo4j (>=5.8.1,<6.0.0)", "networkx (>=2.6.3,<3.0.0)", "nlpcloud (>=1,<2)", "nltk (>=3,<4)", "nomic (>=1.0.43,<2.0.0)", "openai (>=0,<1)", "openlm (>=0.0.5,<0.0.6)", "opensearch-py (>=2.0.0,<3.0.0)", "pdfminer-six (>=20221105,<20221106)", "pexpect (>=4.8.0,<5.0.0)", "pgvector (>=0.1.6,<0.2.0)", "pinecone-client (>=2,<3)", "pinecone-text (>=0.4.2,<0.5.0)", "psycopg2-binary (>=2.9.5,<3.0.0)", "pymongo (>=4.3.3,<5.0.0)", "pyowm (>=3.3.0,<4.0.0)", "pypdf (>=3.4.0,<4.0.0)", "pytesseract (>=0.3.10,<0.4.0)", "pyvespa (>=0.33.0,<0.34.0)", "qdrant-client (>=1.1.2,<2.0.0)", "redis (>=4,<5)", "requests-toolbelt (>=1.0.0,<2.0.0)", "sentence-transformers (>=2,<3)", "spacy (>=3,<4)", "steamship (>=2.16.9,<3.0.0)", "tensorflow-text (>=2.11.0,<3.0.0)", "tiktoken (>=0.3.2,<0.4.0)", "torch (>=1,<3)", "transformers (>=4,<5)", "weaviate-client (>=3,<4)", "wikipedia (>=1,<2)", "wolframalpha (==5.0.0)"]
+all = ["O365 (>=2.0.26,<3.0.0)", "aleph-alpha-client (>=2.15.0,<3.0.0)", "anthropic (>=0.2.6,<0.3.0)", "arxiv (>=1.4,<2.0)", "atlassian-python-api (>=3.36.0,<4.0.0)", "azure-ai-formrecognizer (>=3.2.1,<4.0.0)", "azure-ai-vision (>=0.11.1b1,<0.12.0)", "azure-cognitiveservices-speech (>=1.28.0,<2.0.0)", "azure-cosmos (>=4.4.0b1,<5.0.0)", "azure-identity (>=1.12.0,<2.0.0)", "beautifulsoup4 (>=4,<5)", "clickhouse-connect (>=0.5.14,<0.6.0)", "cohere (>=3,<4)", "deeplake (>=3.3.0,<4.0.0)", "docarray[hnswlib] (>=0.32.0,<0.33.0)", "duckduckgo-search (>=2.8.6,<3.0.0)", "elasticsearch (>=8,<9)", "faiss-cpu (>=1,<2)", "google-api-python-client (==2.70.0)", "google-auth (>=2.18.1,<3.0.0)", "google-search-results (>=2,<3)", "gptcache (>=0.1.7)", "html2text (>=2020.1.16,<2021.0.0)", "huggingface_hub (>=0,<1)", "jina (>=3.14,<4.0)", "jinja2 (>=3,<4)", "jq (>=1.4.1,<2.0.0)", "lancedb (>=0.1,<0.2)", "langkit (>=0.0.1.dev3,<0.1.0)", "lark (>=1.1.5,<2.0.0)", "lxml (>=4.9.2,<5.0.0)", "manifest-ml (>=0.0.1,<0.0.2)", "momento (>=1.5.0,<2.0.0)", "nebula3-python (>=3.4.0,<4.0.0)", "neo4j (>=5.8.1,<6.0.0)", "networkx (>=2.6.3,<3.0.0)", "nlpcloud (>=1,<2)", "nltk (>=3,<4)", "nomic (>=1.0.43,<2.0.0)", "openai (>=0,<1)", "openlm (>=0.0.5,<0.0.6)", "opensearch-py (>=2.0.0,<3.0.0)", "pdfminer-six (>=20221105,<20221106)", "pexpect (>=4.8.0,<5.0.0)", "pgvector (>=0.1.6,<0.2.0)", "pinecone-client (>=2,<3)", "pinecone-text (>=0.4.2,<0.5.0)", "psycopg2-binary (>=2.9.5,<3.0.0)", "pymongo (>=4.3.3,<5.0.0)", "pyowm (>=3.3.0,<4.0.0)", "pypdf (>=3.4.0,<4.0.0)", "pytesseract (>=0.3.10,<0.4.0)", "pyvespa (>=0.33.0,<0.34.0)", "qdrant-client (>=1.1.2,<2.0.0)", "redis (>=4,<5)", "requests-toolbelt (>=1.0.0,<2.0.0)", "sentence-transformers (>=2,<3)", "singlestoredb (>=0.6.1,<0.7.0)", "spacy (>=3,<4)", "steamship (>=2.16.9,<3.0.0)", "tensorflow-text (>=2.11.0,<3.0.0)", "tigrisdb (>=1.0.0b6,<2.0.0)", "tiktoken (>=0.3.2,<0.4.0)", "torch (>=1,<3)", "transformers (>=4,<5)", "weaviate-client (>=3,<4)", "wikipedia (>=1,<2)", "wolframalpha (==5.0.0)"]
 azure = ["azure-ai-formrecognizer (>=3.2.1,<4.0.0)", "azure-ai-vision (>=0.11.1b1,<0.12.0)", "azure-cognitiveservices-speech (>=1.28.0,<2.0.0)", "azure-core (>=1.26.4,<2.0.0)", "azure-cosmos (>=4.4.0b1,<5.0.0)", "azure-identity (>=1.12.0,<2.0.0)", "openai (>=0,<1)"]
 cohere = ["cohere (>=3,<4)"]
 docarray = ["docarray[hnswlib] (>=0.32.0,<0.33.0)"]
 embeddings = ["sentence-transformers (>=2,<3)"]
-extended-testing = ["atlassian-python-api (>=3.36.0,<4.0.0)", "beautifulsoup4 (>=4,<5)", "bibtexparser (>=1.4.0,<2.0.0)", "chardet (>=5.1.0,<6.0.0)", "gql (>=3.4.1,<4.0.0)", "html2text (>=2020.1.16,<2021.0.0)", "jq (>=1.4.1,<2.0.0)", "lxml (>=4.9.2,<5.0.0)", "pandas (>=2.0.1,<3.0.0)", "pdfminer-six (>=20221105,<20221106)", "psychicapi (>=0.2,<0.3)", "py-trello (>=0.19.0,<0.20.0)", "pymupdf (>=1.22.3,<2.0.0)", "pypdf (>=3.4.0,<4.0.0)", "pypdfium2 (>=4.10.0,<5.0.0)", "pyspark (>=3.4.0,<4.0.0)", "requests-toolbelt (>=1.0.0,<2.0.0)", "scikit-learn (>=1.2.2,<2.0.0)", "telethon (>=1.28.5,<2.0.0)", "tqdm (>=4.48.0)", "zep-python (>=0.30,<0.31)"]
+extended-testing = ["atlassian-python-api (>=3.36.0,<4.0.0)", "beautifulsoup4 (>=4,<5)", "bibtexparser (>=1.4.0,<2.0.0)", "chardet (>=5.1.0,<6.0.0)", "gql (>=3.4.1,<4.0.0)", "html2text (>=2020.1.16,<2021.0.0)", "jq (>=1.4.1,<2.0.0)", "lxml (>=4.9.2,<5.0.0)", "pandas (>=2.0.1,<3.0.0)", "pdfminer-six (>=20221105,<20221106)", "psychicapi (>=0.5,<0.6)", "py-trello (>=0.19.0,<0.20.0)", "pymupdf (>=1.22.3,<2.0.0)", "pypdf (>=3.4.0,<4.0.0)", "pypdfium2 (>=4.10.0,<5.0.0)", "pyspark (>=3.4.0,<4.0.0)", "requests-toolbelt (>=1.0.0,<2.0.0)", "scikit-learn (>=1.2.2,<2.0.0)", "telethon (>=1.28.5,<2.0.0)", "tqdm (>=4.48.0)", "zep-python (>=0.31)"]
 llms = ["anthropic (>=0.2.6,<0.3.0)", "cohere (>=3,<4)", "huggingface_hub (>=0,<1)", "manifest-ml (>=0.0.1,<0.0.2)", "nlpcloud (>=1,<2)", "openai (>=0,<1)", "openlm (>=0.0.5,<0.0.6)", "torch (>=1,<3)", "transformers (>=4,<5)"]
 openai = ["openai (>=0,<1)", "tiktoken (>=0.3.2,<0.4.0)"]
 qdrant = ["qdrant-client (>=1.1.2,<2.0.0)"]
@@ -2439,6 +2440,23 @@ typing-inspect = "0.8.0"
 [package.extras]
 test = ["psutil", "pytest", "pytest-asyncio"]
 
+[[package]]
+name = "langchainplus-sdk"
+version = "0.0.6"
+description = "Client library to connect to the LangChainPlus LLM Tracing and Evaluation Platform."
+category = "main"
+optional = false
+python-versions = ">=3.8.1,<4.0"
+files = [
+    {file = "langchainplus_sdk-0.0.6-py3-none-any.whl", hash = "sha256:43fe01c66442b88403c969b8812f6be81e023c0d2a6d5d3572a8d87961438658"},
+    {file = "langchainplus_sdk-0.0.6.tar.gz", hash = "sha256:c911a98fd2d02baa48f742b7d700fd6a55f11c9a545ee5d66b08825940c9a32e"},
+]
+
+[package.dependencies]
+pydantic = ">=1,<2"
+requests = ">=2,<3"
+tenacity = ">=8.1.0,<9.0.0"
+
 [[package]]
 name = "linkify-it-py"
 version = "2.0.2"
@@ -6278,4 +6296,4 @@ deploy = ["langchain-serve"]
 [metadata]
 lock-version = "2.0"
 python-versions = ">=3.9,<3.12"
-content-hash = "0b72cff85b2228a6f41d81ac2207cecf1d94c6adb914a3ef4fb19774d757f9f6"
+content-hash = "94e4a4ca5ec150ef6f673d4a3ab4964a876ce795ae398b78c76719695c63758e"
diff --git a/pyproject.toml b/pyproject.toml
index 47a4d02cc..f66350975 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -29,7 +29,7 @@ google-search-results = "^2.4.1"
 google-api-python-client = "^2.79.0"
 typer = "^0.7.0"
 gunicorn = "^20.1.0"
-langchain = "^0.0.186"
+langchain = "^0.0.194"
 openai = "^0.27.7"
 types-pyyaml = "^6.0.12.8"
 dill = "^0.3.6"