From fa5b080318cb7bf7f3b0229d587d1e2e6f851d03 Mon Sep 17 00:00:00 2001 From: Mustafa Elbehery Date: Tue, 8 Jul 2025 22:25:30 +0200 Subject: [PATCH] chore(api): add mypy coverage to meta_reference_safety Signed-off-by: Mustafa Elbehery --- llama_stack/providers/inline/agents/meta_reference/safety.py | 5 +++-- pyproject.toml | 1 - 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/safety.py b/llama_stack/providers/inline/agents/meta_reference/safety.py index 6b3573d8c..605f387b7 100644 --- a/llama_stack/providers/inline/agents/meta_reference/safety.py +++ b/llama_stack/providers/inline/agents/meta_reference/safety.py @@ -24,8 +24,8 @@ class ShieldRunnerMixin: def __init__( self, safety_api: Safety, - input_shields: list[str] = None, - output_shields: list[str] = None, + input_shields: list[str] | None = None, + output_shields: list[str] | None = None, ): self.safety_api = safety_api self.input_shields = input_shields @@ -37,6 +37,7 @@ class ShieldRunnerMixin: return await self.safety_api.run_shield( shield_id=identifier, messages=messages, + params={}, ) responses = await asyncio.gather(*[run_shield_with_span(identifier) for identifier in identifiers]) diff --git a/pyproject.toml b/pyproject.toml index f72139ed8..edf301891 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -247,7 +247,6 @@ exclude = [ "^llama_stack/providers/inline/agents/meta_reference/", "^llama_stack/providers/inline/agents/meta_reference/agent_instance\\.py$", "^llama_stack/providers/inline/agents/meta_reference/agents\\.py$", - "^llama_stack/providers/inline/agents/meta_reference/safety\\.py$", "^llama_stack/providers/inline/datasetio/localfs/", "^llama_stack/providers/inline/eval/meta_reference/eval\\.py$", "^llama_stack/providers/inline/inference/meta_reference/config\\.py$",