🔧 fix(custom_component.py): fix import paths for session_getter and Flow models in custom_component.py file

🔧 fix(test_cache_manager.py): fix import path for CacheManager in test_cache_manager.py file
🔧 fix(test_custom_component.py): fix import path for Flow and FlowCreate models in test_custom_component.py file
🔧 fix(test_database.py): fix import path for Flow, FlowCreate, and FlowUpdate models in test_database.py file
🔧 fix(test_websocket.py): fix import path for WebSocketDisconnect in test_websocket.py file
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-05 22:18:20 -03:00
commit 8a2358dae0
5 changed files with 7 additions and 7 deletions

View file

@ -6,8 +6,8 @@ from langflow.interface.custom.directory_reader import DirectoryReader
from langflow.utils import validate
from langflow.database.base import session_getter
from langflow.database.models.flow import Flow
from langflow.services.database.base import session_getter
from langflow.services.database.models.flow import Flow
from pydantic import Extra
import yaml

View file

@ -2,7 +2,7 @@ from io import StringIO
import pandas as pd
import pytest
from langflow.cache.manager import CacheManager
from langflow.services.cache.manager import CacheManager
from PIL import Image

View file

@ -5,7 +5,7 @@ from uuid import uuid4
from fastapi import HTTPException
from langflow.database.models.flow import Flow, FlowCreate
from langflow.services.database.models.flow import Flow, FlowCreate
from langflow.interface.custom.base import CustomComponent
from langflow.interface.custom.component import (
Component,

View file

@ -8,9 +8,9 @@ from fastapi.testclient import TestClient
from fastapi.encoders import jsonable_encoder
from langflow.api.v1.schemas import FlowListCreate
from langflow.database.models.flow import Flow, FlowCreate, FlowUpdate
from langflow.services.database.models.flow import Flow, FlowCreate, FlowUpdate
from langflow.database.models.flow_style import (
from langflow.services.database.models.flow_style import (
FlowStyleCreate,
FlowStyleRead,
FlowStyleUpdate,

View file

@ -1,6 +1,6 @@
from fastapi import WebSocketDisconnect
# from langflow.chat.manager import ChatManager
# from langflow.services.chat.manager import ChatManager
import pytest