mirror of
https://github.com/msoedov/agentic_security.git
synced 2026-08-09 03:06:05 +02:00
feat(add spec endpoint):
This commit is contained in:
@@ -6,6 +6,7 @@ from fastapi.responses import JSONResponse
|
|||||||
from ..primitives import FileProbeResponse, Probe
|
from ..primitives import FileProbeResponse, Probe
|
||||||
from ..probe_actor.refusal import REFUSAL_MARKS
|
from ..probe_actor.refusal import REFUSAL_MARKS
|
||||||
from ..probe_data import REGISTRY
|
from ..probe_data import REGISTRY
|
||||||
|
from ._specs import LLM_SPECS
|
||||||
|
|
||||||
router = APIRouter()
|
router = APIRouter()
|
||||||
|
|
||||||
@@ -73,6 +74,12 @@ async def data_config():
|
|||||||
return [m for m in REGISTRY]
|
return [m for m in REGISTRY]
|
||||||
|
|
||||||
|
|
||||||
|
@router.get("/v1/llm-specs", response_model=list)
|
||||||
|
def get_llm_specs():
|
||||||
|
"""Returns the LLM API specifications."""
|
||||||
|
return LLM_SPECS
|
||||||
|
|
||||||
|
|
||||||
@router.get("/health")
|
@router.get("/health")
|
||||||
async def health_check():
|
async def health_check():
|
||||||
"""Health check endpoint."""
|
"""Health check endpoint."""
|
||||||
|
|||||||
Reference in New Issue
Block a user