From 42940fd84ce114513e5030ea1f2587b71b1737fb Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Tue, 27 Oct 2026 19:52:54 +0000 Subject: [PATCH] Include keyword-only prediction inputs in validation and schemas --- .../automatically_decorate_parameters.py | 2 +- tests/test_argument_regressions.py | 19 +++++++++++++++++++ 2 files changed, 20 insertions(+), 1 deletion(-) create mode 100644 tests/test_argument_regressions.py diff --git a/great_ai/parameters/automatically_decorate_parameters.py b/great_ai/parameters/automatically_decorate_parameters.py index 9d4e9f2..be39aa6 100644 --- a/great_ai/parameters/automatically_decorate_parameters.py +++ b/great_ai/parameters/automatically_decorate_parameters.py @@ -12,7 +12,7 @@ def automatically_decorate_parameters(func: F) -> F: parameter_names = [ param.name for param in signature.parameters.values() - if param.kind == param.POSITIONAL_OR_KEYWORD + if param.kind in (param.POSITIONAL_OR_KEYWORD, param.KEYWORD_ONLY) ] metadata = get_function_metadata_store(func) diff --git a/tests/test_argument_regressions.py b/tests/test_argument_regressions.py new file mode 100644 index 0000000..959e5c5 --- /dev/null +++ b/tests/test_argument_regressions.py @@ -0,0 +1,19 @@ +import pytest +from fastapi.testclient import TestClient + +from great_ai import GreatAI + + +def test_keyword_only_inputs_are_validated_and_exposed(isolated_context: None) -> None: + @GreatAI.create + def predict(*, value: int) -> int: + return value + + trace = predict(value=7) + assert trace.output == 7 + assert trace.logged_values["arg:value:value"] == 7 + with pytest.raises(TypeError): + predict(value="wrong") + with TestClient(predict.app) as client: + assert client.post("/predict", json={"value": 8}).json()["output"] == 8 + assert client.post("/predict", json={"value": "wrong"}).status_code == 422