jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Pull start_time out of try block

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-14 17:44:25 -03:00
commit 5c290c0014
1 changed files with 11 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
src/backend/langflow/api/v1/chat.py
View file

@ -1,18 +1,21 @@
import time import time
from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSocketException, status from fastapi import (APIRouter, 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 BuildStatus, BuiltResponse, InitResponse, StreamData from langflow.api.v1.schemas import (BuildStatus, BuiltResponse, InitResponse,
StreamData)
from langflow.graph.graph.base import Graph from langflow.graph.graph.base import Graph
from langflow.services.auth.utils import get_current_active_user, get_current_user_by_jwt from langflow.services.auth.utils import (get_current_active_user,
get_current_user_by_jwt)
from langflow.services.cache.service import BaseCacheService from langflow.services.cache.service import BaseCacheService
from langflow.services.cache.utils import update_build_status 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.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 sqlmodel import Session
router = APIRouter(tags=["Chat"]) router = APIRouter(tags=["Chat"])
@ -149,13 +152,12 @@ async def stream_build(
logger.debug("No user_id found in cache_service") logger.debug("No user_id found in cache_service")
user_id = None user_id = None
for i, vertex in enumerate(graph.generator_build(), 1): for i, vertex in enumerate(graph.generator_build(), 1):
start_time = time.perf_counter()
try: try:
log_dict = { log_dict = {
"log": f"Building node {vertex.vertex_type}", "log": f"Building node {vertex.vertex_type}",
} }
yield str(StreamData(event="log", data=log_dict)) yield str(StreamData(event="log", data=log_dict))
# time this
start_time = time.perf_counter()
if vertex.is_task: if vertex.is_task:
vertex = await try_running_celery_task(vertex, user_id) vertex = await try_running_celery_task(vertex, user_id)
else: else: