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

Update langflow custom component endpoints

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-04 16:55:35 -03:00
commit a4b2cfef87
1 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/backend/langflow/api/v1/endpoints.py
View file

@ -3,6 +3,8 @@ from typing import Annotated, Optional, Union
import sqlalchemy as sa import sqlalchemy as sa
from fastapi import APIRouter, Body, Depends, HTTPException, UploadFile, status 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.utils import update_frontend_node_with_template_values
from langflow.api.v1.schemas import ( from langflow.api.v1.schemas import (
CustomComponentCode, 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.flow import Flow
from langflow.services.database.models.user.model import User 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 langflow.services.deps import get_session, get_session_service, get_settings_service, get_task_service
from loguru import logger
try: try:
from langflow.worker import process_graph_cached_task from langflow.worker import process_graph_cached_task
@ -30,9 +31,10 @@ except ImportError:
raise NotImplementedError("Celery is not installed") raise NotImplementedError("Celery is not installed")
from langflow.services.task.service import TaskService
from sqlmodel import Session from sqlmodel import Session
from langflow.services.task.service import TaskService
# build router # build router
router = APIRouter(tags=["Base"]) 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) component_node = build_langchain_template_custom_component(component, user_id=user.id, update_field=raw_code.field)
# Update the field # Update the field
return component_node return component_node
# Update the field
return component_node
# Update the field
return component_node
# Update the field
return component_node