From e8f2e018e4ae227bbb19d9e4190b55ce62b80f86 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Fri, 2 Aug 2024 15:10:07 -0300
Subject: [PATCH] fix: add chat_service to build_vertex method call so freeze
 works (#3175)

---
 src/backend/base/langflow/api/v1/chat.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/backend/base/langflow/api/v1/chat.py b/src/backend/base/langflow/api/v1/chat.py
index 9cbc6bdd5..03c3c96b6 100644
--- a/src/backend/base/langflow/api/v1/chat.py
+++ b/src/backend/base/langflow/api/v1/chat.py
@@ -15,13 +15,13 @@ from starlette.types import Receive
 
 from langflow.api.utils import (
     build_and_cache_graph_from_data,
+    build_graph_from_data,
     build_graph_from_db,
+    build_graph_from_db_no_cache,
     format_elapsed_time,
     format_exception_message,
     get_top_level_vertices,
     parse_exception,
-    build_graph_from_db_no_cache,
-    build_graph_from_data,
 )
 from langflow.api.v1.schemas import (
     FlowDataRequest,
@@ -232,7 +232,7 @@ async def build_flow(
                     artifacts,
                     vertex,
                 ) = await graph.build_vertex(
-                    chat_service=None,
+                    chat_service=chat_service,
                     vertex_id=vertex_id,
                     user_id=current_user.id,
                     inputs_dict=inputs.model_dump() if inputs else {},