From ba86bd0b2a586034abfb04e372fc0144c51e07ff Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 30 Jan 2024 23:07:55 -0300
Subject: [PATCH] Add monitor router and log vertex build

---
 src/backend/langflow/api/v1/__init__.py |  2 ++
 src/backend/langflow/api/v1/chat.py     | 15 ++++++++++++---
 2 files changed, 14 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/api/v1/__init__.py b/src/backend/langflow/api/v1/__init__.py
index f42d3164a..39659748e 100644
--- a/src/backend/langflow/api/v1/__init__.py
+++ b/src/backend/langflow/api/v1/__init__.py
@@ -5,6 +5,7 @@ from langflow.api.v1.endpoints import router as endpoints_router
 from langflow.api.v1.files import router as files_router
 from langflow.api.v1.flows import router as flows_router
 from langflow.api.v1.login import router as login_router
+from langflow.api.v1.monitor import router as monitor_router
 from langflow.api.v1.store import router as store_router
 from langflow.api.v1.users import router as users_router
 from langflow.api.v1.validate import router as validate_router
@@ -19,5 +20,6 @@ __all__ = [
     "api_key_router",
     "login_router",
     "credentials_router",
+    "monitor_router",
     "files_router",
 ]
diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py
index c9f2897df..97162991e 100644
--- a/src/backend/langflow/api/v1/chat.py
+++ b/src/backend/langflow/api/v1/chat.py
@@ -2,6 +2,9 @@ import time
 
 from fastapi import APIRouter, Body, Depends, HTTPException, Query, WebSocket, WebSocketException, status
 from fastapi.responses import StreamingResponse
+from loguru import logger
+from sqlmodel import Session
+
 from langflow.api.utils import build_input_keys_response, format_elapsed_time
 from langflow.api.v1.schemas import (
     BuildStatus,
@@ -21,8 +24,7 @@ from langflow.services.cache.utils import update_build_status
 from langflow.services.chat.service import ChatService
 from langflow.services.database.models.flow import Flow
 from langflow.services.deps import get_cache_service, get_chat_service, get_session
-from loguru import logger
-from sqlmodel import Session
+from langflow.services.monitor.utils import log_vertex_build
 
 router = APIRouter(tags=["Chat"])
 
@@ -308,7 +310,14 @@ async def build_vertex(
             result_dict = ResultDict(results={})
             artifacts = {}
         chat_service.set_cache(flow_id, graph)
-
+        await log_vertex_build(
+            flow_id=flow_id,
+            vertex_id=vertex_id,
+            valid=valid,
+            params=params,
+            data=result_dict,
+            artifacts=artifacts,
+        )
         return VertexBuildResponse(
             valid=valid,
             params=params,