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

sets starter_project fixture to Basic Prompting

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 12:26:22 -03:00
commit ac1efd99aa
1 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/conftest.py
View file

@ -10,6 +10,10 @@ import orjson
import pytest import pytest
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from httpx import AsyncClient from httpx import AsyncClient
from sqlmodel import Session, SQLModel, create_engine, select
from sqlmodel.pool import StaticPool
from typer.testing import CliRunner
from langflow.graph.graph.base import Graph from langflow.graph.graph.base import Graph
from langflow.initial_setup.setup import STARTER_FOLDER_NAME from langflow.initial_setup.setup import STARTER_FOLDER_NAME
from langflow.services.auth.utils import get_password_hash from langflow.services.auth.utils import get_password_hash
@ -18,9 +22,6 @@ from langflow.services.database.models.flow.model import Flow, FlowCreate
from langflow.services.database.models.user.model import User, UserCreate from langflow.services.database.models.user.model import User, UserCreate
from langflow.services.database.utils import session_getter from langflow.services.database.utils import session_getter
from langflow.services.deps import get_db_service from langflow.services.deps import get_db_service
from sqlmodel import Session, SQLModel, create_engine, select
from sqlmodel.pool import StaticPool
from typer.testing import CliRunner
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.database.service import DatabaseService from langflow.services.database.service import DatabaseService
@ -379,7 +380,9 @@ def created_api_key(active_user):
def get_starter_project(active_user): def get_starter_project(active_user):
# once the client is created, we can get the starter project # once the client is created, we can get the starter project
with session_getter(get_db_service()) as session: with session_getter(get_db_service()) as session:
flow = session.exec(select(Flow).where(Flow.folder == STARTER_FOLDER_NAME)).first() flow = session.exec(
select(Flow).where(Flow.folder == STARTER_FOLDER_NAME).where(Flow.name == "Basic Prompting")
).first()
if not flow: if not flow:
raise ValueError("No starter project found") raise ValueError("No starter project found")