Skip to content

Commit 1a7421a

Browse files
style: ruff format (auto)
1 parent 63a1a1a commit 1a7421a

5 files changed

Lines changed: 156 additions & 135 deletions

File tree

src/api/provider_health.py

Lines changed: 23 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,16 @@ async def check_provider_health(
6767
project_id = getattr(provider_config, "project_id", None)
6868

6969
# Check if this provider is used for LLM or embedding
70-
llm_model = current_config.agent.llm_model if provider == current_config.agent.llm_provider else None
71-
embedding_model = current_config.knowledge.embedding_model if provider == current_config.knowledge.embedding_provider else None
70+
llm_model = (
71+
current_config.agent.llm_model
72+
if provider == current_config.agent.llm_provider
73+
else None
74+
)
75+
embedding_model = (
76+
current_config.knowledge.embedding_model
77+
if provider == current_config.knowledge.embedding_provider
78+
else None
79+
)
7280
except ValueError:
7381
# Provider not found in configuration
7482
return JSONResponse(
@@ -119,7 +127,7 @@ async def check_provider_health(
119127
return JSONResponse(cached_payload, status_code=200)
120128

121129
logger.info(f"Checking health for provider: {provider}")
122-
130+
123131
# Validate provider setup
124132
if check_provider:
125133
# Validate specific provider
@@ -183,9 +191,11 @@ async def check_provider_health(
183191
and embedding_provider == "watsonx"
184192
and llm_error is None
185193
):
186-
logger.info("Waiting 2 seconds before WatsonX embedding test (after completion test)")
194+
logger.info(
195+
"Waiting 2 seconds before WatsonX embedding test (after completion test)"
196+
)
187197
await asyncio.sleep(2)
188-
198+
189199
try:
190200
await validate_provider_setup(
191201
provider=embedding_provider,
@@ -202,10 +212,14 @@ async def check_provider_health(
202212
logger.info(f"Embedding provider ({embedding_provider}) appears busy: {str(e)}")
203213
else:
204214
embedding_error = str(e)
205-
logger.error(f"Embedding provider ({embedding_provider}) validation timed out: {embedding_error}")
215+
logger.error(
216+
f"Embedding provider ({embedding_provider}) validation timed out: {embedding_error}"
217+
)
206218
except Exception as e:
207219
embedding_error = str(e)
208-
logger.error(f"Embedding provider ({embedding_provider}) validation failed: {embedding_error}")
220+
logger.error(
221+
f"Embedding provider ({embedding_provider}) validation failed: {embedding_error}"
222+
)
209223

210224
# Return combined status
211225
if llm_error or embedding_error:
@@ -239,11 +253,11 @@ async def check_provider_health(
239253
}
240254
provider_health_cache.set_(health_cache_key, healthy_payload)
241255
return JSONResponse(healthy_payload, status_code=200)
242-
256+
243257
except Exception as e:
244258
error_message = str(e)
245259
logger.error(f"Provider health check failed for {provider}: {error_message}")
246-
260+
247261
return JSONResponse(
248262
{
249263
"status": "unhealthy",
@@ -252,4 +266,3 @@ async def check_provider_health(
252266
},
253267
status_code=503,
254268
)
255-

0 commit comments

Comments
 (0)