gmakstutis/openrag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

fix tests

Browse source
This commit is contained in:
phact 2025-10-07 04:02:39 -04:00
parent 3efcbfd364
commit bde95a5870
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/integration/test_api_endpoints.py
View file

@ -64,15 +64,16 @@ async def test_upload_and_search_endpoint(tmp_path: Path, disable_langflow_inges
"src.main", "src.main",
"api", # Clear the api package itself "api", # Clear the api package itself
"src.api", "src.api",
"services", # Clear services that import clients
"src.services",
"services.search_service",
"src.services.search_service",
]: ]:
sys.modules.pop(mod, None) sys.modules.pop(mod, None)
from src.main import create_app, startup_tasks from src.main import create_app, startup_tasks
import src.api.router as upload_router import src.api.router as upload_router
from src.config.settings import clients, INDEX_NAME, DISABLE_INGEST_WITH_LANGFLOW from src.config.settings import clients, INDEX_NAME, DISABLE_INGEST_WITH_LANGFLOW
# Verify settings loaded correctly
print(f"Settings DISABLE_INGEST_WITH_LANGFLOW: {DISABLE_INGEST_WITH_LANGFLOW}")
# Ensure a clean index before startup # Ensure a clean index before startup
await clients.initialize() await clients.initialize()
try: try:
@ -188,15 +189,16 @@ async def test_router_upload_ingest_traditional(tmp_path: Path, disable_langflow
"src.main", "src.main",
"api", # Clear the api package itself "api", # Clear the api package itself
"src.api", "src.api",
"services", # Clear services that import clients
"src.services",
"services.search_service",
"src.services.search_service",
]: ]:
sys.modules.pop(mod, None) sys.modules.pop(mod, None)
from src.main import create_app, startup_tasks from src.main import create_app, startup_tasks
import src.api.router as upload_router import src.api.router as upload_router
from src.config.settings import clients, INDEX_NAME, DISABLE_INGEST_WITH_LANGFLOW from src.config.settings import clients, INDEX_NAME, DISABLE_INGEST_WITH_LANGFLOW
# Verify settings loaded correctly
print(f"Settings DISABLE_INGEST_WITH_LANGFLOW: {DISABLE_INGEST_WITH_LANGFLOW}")
# Ensure a clean index before startup # Ensure a clean index before startup
await clients.initialize() await clients.initialize()
try: try: