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.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",
|
||||||
]
|
]
|
||||||
|
|
|
||||||
|
|
@ -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,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue