wait for 2 seconds if its watsonx
This commit is contained in:
parent
2ca741ed32
commit
8279cec761
1 changed files with 11 additions and 0 deletions
|
|
@ -1,5 +1,6 @@
|
||||||
"""Provider health check endpoint."""
|
"""Provider health check endpoint."""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
import httpx
|
import httpx
|
||||||
from starlette.responses import JSONResponse
|
from starlette.responses import JSONResponse
|
||||||
from utils.logging_config import get_logger
|
from utils.logging_config import get_logger
|
||||||
|
|
@ -149,6 +150,16 @@ async def check_provider_health(request):
|
||||||
logger.error(f"LLM provider ({provider}) validation failed: {llm_error}")
|
logger.error(f"LLM provider ({provider}) validation failed: {llm_error}")
|
||||||
|
|
||||||
# Validate embedding provider
|
# Validate embedding provider
|
||||||
|
# For WatsonX with test_completion=True, wait 2 seconds between completion and embedding tests
|
||||||
|
if (
|
||||||
|
test_completion
|
||||||
|
and provider == "watsonx"
|
||||||
|
and embedding_provider == "watsonx"
|
||||||
|
and llm_error is None
|
||||||
|
):
|
||||||
|
logger.info("Waiting 2 seconds before WatsonX embedding test (after completion test)")
|
||||||
|
await asyncio.sleep(2)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
await validate_provider_setup(
|
await validate_provider_setup(
|
||||||
provider=embedding_provider,
|
provider=embedding_provider,
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue