Fix import formatting in API files

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-12-06 10:43:53 -03:00
commit d3599eb15b
4 changed files with 15 additions and 22 deletions

View file

@ -1,7 +1,6 @@
from pathlib import Path from pathlib import Path
from typing import TYPE_CHECKING, List from typing import TYPE_CHECKING, List
from langflow.services.deps import get_settings_service
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.database.models.flow.model import Flow from langflow.services.database.models.flow.model import Flow
@ -74,7 +73,6 @@ def update_frontend_node_with_template_values(frontend_node, raw_template_data):
return frontend_node return frontend_node
def is_valid_data(frontend_node, raw_template_data): def is_valid_data(frontend_node, raw_template_data):
"""Check if the data is valid for processing.""" """Check if the data is valid for processing."""
return ( return (
@ -97,7 +95,6 @@ def update_template_values(frontend_template, raw_template):
update_template_field(frontend_template, key, value_dict) update_template_field(frontend_template, key, value_dict)
def update_template_field(frontend_template, key, value_dict): def update_template_field(frontend_template, key, value_dict):
"""Updates a specific field in the frontend template.""" """Updates a specific field in the frontend template."""
"""Updates a specific field in the frontend template.""" """Updates a specific field in the frontend template."""

View file

@ -1,17 +1,13 @@
from fastapi import (APIRouter, Depends, HTTPException, Query, WebSocket, from fastapi import APIRouter, Depends, HTTPException, Query, WebSocket, WebSocketException, status
WebSocketException, status)
from fastapi.responses import StreamingResponse from fastapi.responses import StreamingResponse
from langflow.api.utils import build_input_keys_response from langflow.api.utils import build_input_keys_response
from langflow.api.v1.schemas import (BuildStatus, BuiltResponse, InitResponse, from langflow.api.v1.schemas import BuildStatus, BuiltResponse, InitResponse, StreamData
StreamData)
from langflow.graph.graph.base import Graph from langflow.graph.graph.base import Graph
from langflow.services.auth.utils import (get_current_active_user, from langflow.services.auth.utils import get_current_active_user, get_current_user_by_jwt
get_current_user_by_jwt)
from langflow.services.cache.service import BaseCacheService from langflow.services.cache.service import BaseCacheService
from langflow.services.cache.utils import update_build_status from langflow.services.cache.utils import update_build_status
from langflow.services.chat.service import ChatService from langflow.services.chat.service import ChatService
from langflow.services.deps import (get_cache_service, get_chat_service, from langflow.services.deps import get_cache_service, get_chat_service, get_session
get_session)
from loguru import logger from loguru import logger
from sqlmodel import Session from sqlmodel import Session

View file

@ -4,21 +4,22 @@ 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 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 (CustomComponentCode, ProcessResponse, from langflow.api.v1.schemas import (
TaskResponse, TaskStatusResponse, CustomComponentCode,
UploadFileResponse) ProcessResponse,
TaskResponse,
TaskStatusResponse,
UploadFileResponse,
)
from langflow.interface.custom.custom_component import CustomComponent from langflow.interface.custom.custom_component import CustomComponent
from langflow.interface.custom.directory_reader import DirectoryReader from langflow.interface.custom.directory_reader import DirectoryReader
from langflow.interface.types import (build_custom_component_template, from langflow.interface.types import build_custom_component_template, create_and_validate_component
create_and_validate_component)
from langflow.processing.process import process_graph_cached, process_tweaks from langflow.processing.process import process_graph_cached, process_tweaks
from langflow.services.auth.utils import (api_key_security, from langflow.services.auth.utils import api_key_security, get_current_active_user
get_current_active_user)
from langflow.services.cache.utils import save_uploaded_file 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, from langflow.services.deps import get_session, get_session_service, get_settings_service, get_task_service
get_settings_service, get_task_service)
from loguru import logger from loguru import logger
from sqlmodel import select from sqlmodel import select

View file

@ -2,8 +2,7 @@ from typing import TYPE_CHECKING, List, Union
from langchain.agents.agent import AgentExecutor from langchain.agents.agent import AgentExecutor
from langchain.callbacks.base import BaseCallbackHandler from langchain.callbacks.base import BaseCallbackHandler
from langflow.api.v1.callback import (AsyncStreamingLLMCallbackHandler, from langflow.api.v1.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler
StreamingLLMCallbackHandler)
from langflow.processing.process import fix_memory_inputs, format_actions from langflow.processing.process import fix_memory_inputs, format_actions
from loguru import logger from loguru import logger