4bbd760204
* change my PR to match Caren's * add path parameter validation for agent id first * remove old import * remove old agent_id_pattern pattern * add example and fix max/min calculation to include hyphen * fix regex string interpolation * example deprecated in favour of examples * openapi autogen * change template test to expect 422 * fix 422 swallow * expect 422 or 400 * rewrite error codes * fix hallucinated uuid * tweaked error message test * print docker logs on failure
28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
import re
|
|
|
|
from fastapi import Path
|
|
|
|
# TODO: extract this list from routers/v1/__init__.py and ROUTERS
|
|
primitives = ["agent", "message", "run", "job", "group", "block", "file", "folder", "source", "tool", "mcp_server"]
|
|
|
|
PRIMITIVE_ID_PATTERNS = {
|
|
# f-string interpolation gets confused because of the regex's required curly braces {}
|
|
primitive: re.compile("^" + primitive + "-[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$")
|
|
for primitive in primitives
|
|
}
|
|
|
|
PATH_VALIDATORS = {}
|
|
for primitive in primitives:
|
|
PATH_VALIDATORS[primitive] = Path(
|
|
description=f"The ID of the {primitive} in the format '{primitive}-<uuid4>'",
|
|
pattern=PRIMITIVE_ID_PATTERNS[primitive].pattern,
|
|
examples=[f"{primitive}-123e4567-e89b-42d3-8456-426614174000"],
|
|
# len(agent) + len("-") + len(uuid4)
|
|
min_length=len(primitive) + 1 + 36,
|
|
max_length=len(primitive) + 1 + 36,
|
|
)
|
|
|
|
|
|
def is_valid_id(primitive: str, id: str) -> bool:
|
|
return PRIMITIVE_ID_PATTERNS[primitive].match(id) is not None
|