From a4b2cfef8780ceff2f5c573b766989c92f2b656f Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 4 Dec 2023 16:55:35 -0300 Subject: [PATCH] Update langflow custom component endpoints --- src/backend/langflow/api/v1/endpoints.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/src/backend/langflow/api/v1/endpoints.py b/src/backend/langflow/api/v1/endpoints.py index 3d4449ac1..14a160c8f 100644 --- a/src/backend/langflow/api/v1/endpoints.py +++ b/src/backend/langflow/api/v1/endpoints.py @@ -3,6 +3,8 @@ from typing import Annotated, Optional, Union import sqlalchemy as sa from fastapi import APIRouter, Body, Depends, HTTPException, UploadFile, status +from loguru import logger + from langflow.api.utils import update_frontend_node_with_template_values from langflow.api.v1.schemas import ( CustomComponentCode, @@ -20,7 +22,6 @@ 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 try: from langflow.worker import process_graph_cached_task @@ -30,9 +31,10 @@ except ImportError: raise NotImplementedError("Celery is not installed") -from langflow.services.task.service import TaskService from sqlmodel import Session +from langflow.services.task.service import TaskService + # build router router = APIRouter(tags=["Base"]) @@ -247,9 +249,3 @@ async def custom_component_update( component_node = build_langchain_template_custom_component(component, user_id=user.id, update_field=raw_code.field) # Update the field return component_node - # Update the field - return component_node - # Update the field - return component_node - # Update the field - return component_node