From a8dd608f0685fe61505d6fa7a3c9d2793c74411f Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Sun, 2 Mar 2025 19:13:48 +0200 Subject: [PATCH] feat(add t5 model for testing): --- agentic_security/routes/probe.py | 8 ++++++++ 1 file changed, 8 insertions(+) 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)