diff --git a/src/backend/langflow/api/v1/chat.py b/src/backend/langflow/api/v1/chat.py index fbbbbefb1..adbf685e8 100644 --- a/src/backend/langflow/api/v1/chat.py +++ b/src/backend/langflow/api/v1/chat.py @@ -2,9 +2,6 @@ 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, @@ -25,6 +22,8 @@ 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 langflow.services.monitor.utils import log_vertex_build +from loguru import logger +from sqlmodel import Session router = APIRouter(tags=["Chat"]) @@ -305,8 +304,9 @@ async def build_vertex( # to the frontend vertex.set_artifacts() artifacts = vertex.artifacts - duration = format_elapsed_time(time.perf_counter() - start_time) - result_dict = ResultDict(results=result_dict, artifacts=artifacts, duration=duration) + timedelta = time.perf_counter() - start_time + duration = format_elapsed_time(timedelta) + result_dict = ResultDict(results=result_dict, artifacts=artifacts, duration=duration, timedelta=timedelta) except Exception as exc: params = str(exc) valid = False diff --git a/src/backend/langflow/api/v1/schemas.py b/src/backend/langflow/api/v1/schemas.py index dee62547e..ff66899d6 100644 --- a/src/backend/langflow/api/v1/schemas.py +++ b/src/backend/langflow/api/v1/schemas.py @@ -225,7 +225,8 @@ class ResultDict(BaseModel): results: Optional[Any] = Field(default_factory=dict) artifacts: Optional[Any] = Field(default_factory=dict) - duration: Optional[float] = None + timedelta: Optional[float] = None + duration: Optional[str] = None class VertexBuildResponse(BaseModel):