Update langflow custom component endpoints

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 16:55:35 -03:00
commit a4b2cfef87

View file

@ -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