@@ -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