🔧 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

src/backend/langflow/interface/custom/custom_component.py

@@ -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.services.database.base import session_getter
-from langflow.database.models.flow import Flow
+from langflow.services.database.models.flow import Flow
 from pydantic import Extra
 import yaml
 

tests/test_cache_manager.py

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

tests/test_custom_component.py

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

tests/test_database.py

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

tests/test_websocket.py

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