From c60fe6ba14bc74570dee381fce1e760297fd721e Mon Sep 17 00:00:00 2001
From: anovazzi1 <otavio2204@gmail.com>
Date: Fri, 19 Jan 2024 16:47:36 -0300
Subject: [PATCH] Add build_vertex endpoint to chat.py

---
 src/backend/langflow/api/v1/chat.py | 51 ++++++++++++++++++++++++++++-
 1 file changed, 50 insertions(+), 1 deletion(-)

diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
index e2454bf9d..d815930d5 100644
--- a/src/backend/langflow/api/v1/chat.py
+++ b/src/backend/langflow/api/v1/chat.py
@@ -256,4 +256,53 @@ async def get_vertices(
 
     except Exception as exc:
         logger.error(f"Error checking build status: {exc}")
-        raise HTTPException(status_code=500, detail=str(exc)) from exc
\ No newline at end of file
+        raise HTTPException(status_code=500, detail=str(exc)) from exc
+
+@router.post("/build/{flow_id}/vertices/{vertex_id}")
+async def build_vertex(
+    flow_id: str,
+    vertex_id: str,
+    chat_service: "ChatService" = Depends(get_chat_service),
+    # current_user=Depends(get_current_active_user),
+    tweaks: dict = Body(None),
+    inputs: dict = Body(None),
+):
+    """Build a vertex instead of the entire graph."""
+    try:
+        cache = chat_service.get_cache(flow_id)
+        graph = cache.get("result")
+        result_dict = {}
+        if tweaks:
+            graph = process_tweaks_on_graph(graph, tweaks)
+        if not isinstance(graph, Graph):
+            raise ValueError("Invalid graph")
+        if not (vertex := graph.get_vertex(vertex_id)):
+            raise ValueError("Invalid vertex")
+        try:
+            if isinstance(vertex, StatelessVertex) or not vertex._built:
+                await vertex.build(user_id=None)
+            params = vertex._built_object_repr()
+            valid = True
+            result_dict = vertex.get_built_result()
+            # We need to set the artifacts to pass information
+            # to the frontend
+            vertex.set_artifacts()
+            artifacts = vertex.artifacts
+            result_dict = ResultDict(results=result_dict, artifacts=artifacts)
+        except Exception as exc:
+            params = str(exc)
+            valid = False
+            result_dict = ResultDict(results={})
+            artifacts = {}
+        chat_service.set_cache(flow_id, graph)
+
+        return VertexBuildResponse(
+            valid=valid,
+            params=params,
+            id=vertex.id,
+            data=result_dict,
+        )
+    except Exception as exc:
+        logger.error(f"Error building vertex: {exc}")
+        logger.exception(exc)
+        raise HTTPException(status_code=500, detail=str(exc)) from exc