Add monitor router and log vertex build

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-30 23:07:55 -03:00
commit ba86bd0b2a
2 changed files with 14 additions and 3 deletions

View file

@ -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.files import router as files_router
from langflow.api.v1.flows import router as flows_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.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.store import router as store_router
from langflow.api.v1.users import router as users_router from langflow.api.v1.users import router as users_router
from langflow.api.v1.validate import router as validate_router from langflow.api.v1.validate import router as validate_router
@ -19,5 +20,6 @@ __all__ = [
"api_key_router", "api_key_router",
"login_router", "login_router",
"credentials_router", "credentials_router",
"monitor_router",
"files_router", "files_router",
] ]

View file

@ -2,6 +2,9 @@ import time
from fastapi import APIRouter, Body, Depends, HTTPException, Query, WebSocket, WebSocketException, status from fastapi import APIRouter, Body, Depends, HTTPException, Query, WebSocket, WebSocketException, status
from fastapi.responses import StreamingResponse 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.utils import build_input_keys_response, format_elapsed_time
from langflow.api.v1.schemas import ( from langflow.api.v1.schemas import (
BuildStatus, BuildStatus,
@ -21,8 +24,7 @@ from langflow.services.cache.utils import update_build_status
from langflow.services.chat.service import ChatService from langflow.services.chat.service import ChatService
from langflow.services.database.models.flow import Flow from langflow.services.database.models.flow import Flow
from langflow.services.deps import get_cache_service, get_chat_service, get_session from langflow.services.deps import get_cache_service, get_chat_service, get_session
from loguru import logger from langflow.services.monitor.utils import log_vertex_build
from sqlmodel import Session
router = APIRouter(tags=["Chat"]) router = APIRouter(tags=["Chat"])
@ -308,7 +310,14 @@ async def build_vertex(
result_dict = ResultDict(results={}) result_dict = ResultDict(results={})
artifacts = {} artifacts = {}
chat_service.set_cache(flow_id, graph) 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( return VertexBuildResponse(
valid=valid, valid=valid,
params=params, params=params,