Add monitor router and log vertex build
This commit is contained in:
parent
bdc4a02b8a
commit
ba86bd0b2a
2 changed files with 14 additions and 3 deletions
|
|
@ -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",
|
||||
]
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue