fix: fix broken check_llm_provider healthcheck method
This commit is contained in:
parent
c9c5406aca
commit
ada3753534
1 changed files with 11 additions and 2 deletions
|
|
@ -193,8 +193,17 @@ class HealthChecker:
|
||||||
|
|
||||||
config = get_llm_config()
|
config = get_llm_config()
|
||||||
|
|
||||||
# Test actual API connection with minimal request
|
from pydantic import BaseModel
|
||||||
LLMGateway.show_prompt("test", "test.txt")
|
|
||||||
|
class TestResponse(BaseModel):
|
||||||
|
status: str
|
||||||
|
|
||||||
|
# Use a simple test to verify LLM connectivity
|
||||||
|
await LLMGateway.acreate_structured_output(
|
||||||
|
text_input="Respond with status: ok",
|
||||||
|
system_prompt="You are a test assistant. Respond with a simple status.",
|
||||||
|
response_model=TestResponse,
|
||||||
|
)
|
||||||
|
|
||||||
response_time = int((time.time() - start_time) * 1000)
|
response_time = int((time.time() - start_time) * 1000)
|
||||||
return ComponentHealth(
|
return ComponentHealth(
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue