fix: Imports

Author: AlejandroFernandezLuces

tests/test_endpoints_splitter.py

@@ -23,17 +23,17 @@
 import base64
 from pathlib import Path
 
-from app.app import app
-from app.endpoints.splitter import validate_request
-from app.models.splitter import SplitterRequest
+from allie.flowkit import flowkit_service
+from allie.flowkit.endpoints.splitter import validate_request
+from allie.flowkit.models.splitter import SplitterRequest
 from fastapi import HTTPException
 from fastapi.testclient import TestClient
 import pytest
 
 from tests.conftest import MOCK_API_KEY
 
 # Create a test client
-client = TestClient(app)
+client = TestClient(flowkit_service)
 
 
 def encode_file_to_base64(file_path):

tests/test_list_functions.py

@@ -20,12 +20,12 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 """Test module for list functions."""
-from app.app import app
+from allie.flowkit import flowkit_service
 from fastapi.testclient import TestClient
 import pytest
 
 # Initialize the test client
-client = TestClient(app)
+client = TestClient(flowkit_service)
 
 
 @pytest.mark.asyncio