diff --git a/agentic_security/routes/probe.py b/agentic_security/routes/probe.py index 8ac7196..2c8d498 100644 --- a/agentic_security/routes/probe.py +++ b/agentic_security/routes/probe.py @@ -77,3 +77,11 @@ async def data_config(): async def health_check(): """Health check endpoint.""" return JSONResponse(content={"status": "ok"}) + + +@router.post("/v1/self-probe-t5") +def self_probe_t5(probe: Probe): + import languagemodels as lm # noqa + + message = lm.do(probe.prompt) + return make_mock_response(message)