From 3213c8719363667a8d76cfe236e5cf0237fc1422 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 22 Dec 2023 10:41:49 -0300 Subject: [PATCH] Refactor API endpoints and imports --- src/backend/langflow/api/v1/endpoints.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py index 9bf081163..d6e509295 100644 --- a/src/backend/langflow/api/v1/endpoints.py +++ b/src/backend/langflow/api/v1/endpoints.py @@ -3,9 +3,6 @@ from typing import Annotated, List, Optional, Union import sqlalchemy as sa from fastapi import APIRouter, Body, Depends, HTTPException, UploadFile, status -from loguru import logger -from sqlmodel import select - from langflow.api.utils import update_frontend_node_with_template_values from langflow.api.v1.schemas import ( CustomComponentCode, @@ -23,6 +20,8 @@ from langflow.services.cache.utils import save_uploaded_file from langflow.services.database.models.flow import Flow from langflow.services.database.models.user.model import User from langflow.services.deps import get_session, get_session_service, get_settings_service, get_task_service +from loguru import logger +from sqlmodel import select try: from langflow.worker import process_graph_cached_task @@ -32,9 +31,8 @@ except ImportError: raise NotImplementedError("Celery is not installed") -from sqlmodel import Session - from langflow.services.task.service import TaskService +from sqlmodel import Session # build router router = APIRouter(tags=["Base"]) @@ -289,7 +287,6 @@ async def reload_custom_component(path: str, user: User = Depends(get_current_ac raise ValueError(content) extractor = CustomComponent(code=content) - extractor.validate() return build_custom_component_template(extractor, user_id=user.id) except Exception as exc: raise HTTPException(status_code=400, detail=str(exc))