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

Refactor API endpoints and validate custom

Browse source 
component code
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 11:50:31 -03:00
commit 943ef39050
1 changed files with 16 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,33 +1,29 @@
from http import HTTPStatus from http import HTTPStatus
from typing import Annotated, Optional, Union from typing import Annotated, Optional, Union
from langflow.services.auth.utils import api_key_security, get_current_active_user
from langflow.services.cache.utils import save_uploaded_file
from langflow.services.database.models.flow import Flow
from langflow.processing.process import process_graph_cached, process_tweaks
from langflow.services.database.models.user.user import User
from langflow.services.deps import (
get_session_service,
get_settings_service,
get_task_service,
)
from loguru import logger
from fastapi import APIRouter, Depends, HTTPException, UploadFile, Body, status
import sqlalchemy as sa import sqlalchemy as sa
from langflow.interface.custom.custom_component import CustomComponent from fastapi import APIRouter, Body, Depends, HTTPException, UploadFile, status
from loguru import logger
from langflow.api.v1.schemas import ( from langflow.api.v1.schemas import (
CustomComponentCode,
ProcessResponse, ProcessResponse,
TaskResponse, TaskResponse,
TaskStatusResponse, TaskStatusResponse,
UploadFileResponse, UploadFileResponse,
CustomComponentCode,
) )
from langflow.interface.custom.custom_component import CustomComponent
from langflow.processing.process import process_graph_cached, process_tweaks
from langflow.services.deps import get_session from langflow.services.auth.utils import api_key_security, get_current_active_user
from langflow.services.cache.utils import save_uploaded_file
from langflow.services.database.models.flow import Flow
from langflow.services.database.models.user.user import User
from langflow.services.deps import (
get_session,
get_session_service,
get_settings_service,
get_task_service,
)
try: try:
from langflow.worker import process_graph_cached_task from langflow.worker import process_graph_cached_task
@ -39,7 +35,6 @@ except ImportError:
from sqlmodel import Session from sqlmodel import Session
from langflow.services.task.service import TaskService from langflow.services.task.service import TaskService
# build router # build router
@ -217,6 +212,6 @@ async def custom_component(
) )
extractor = CustomComponent(code=raw_code.code) extractor = CustomComponent(code=raw_code.code)
extractor.is_check_valid() extractor.validate()
return build_langchain_template_custom_component(extractor, user_id=user.id) return build_langchain_template_custom_component(extractor, user_id=user.id)