feat(add t5 model for testing):

This commit is contained in:
Alexander Myasoedov
2025-03-02 19:13:48 +02:00
parent f8102d1ee9
commit a8dd608f06
+8
View File
@@ -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)