feat(guardrails): straiker guardrail speaks the v3 platform API (/api/v3/detect)

litellm/proxy/guardrails/guardrail_hooks/straiker/__init__.py

@@ -1,4 +1,6 @@
-from typing import TYPE_CHECKING, Final
+from typing import TYPE_CHECKING, Final, Literal
+
+from pydantic import BaseModel
 
 import litellm
 from litellm.types.guardrails import SupportedGuardrailIntegrations
@@ -8,6 +10,14 @@
 if TYPE_CHECKING:
     from litellm.types.guardrails import Guardrail, LitellmParams
 
+
+class _V3Routing(BaseModel):
+    api_version: Literal["v1", "v3"] | None = None
+    agent_ref: str | None = None
+    client: str | None = None
+    format_hint: Literal["anthropic.messages", "openai.chat"] | None = None
+
+
 _OPTIONAL_INIT_FIELDS: Final = (
     "timeout",
     "max_retries",
@@ -48,13 +58,23 @@ def initialize_guardrail(litellm_params: "LitellmParams", guardrail: "Guardrail"
         for value in [_get_config_value(litellm_params, optional_params, field)]
         if value is not None
     }
+    routing: Final = _V3Routing.model_validate(
+        {
+            field: _get_config_value(litellm_params, optional_params, field)
+            for field in ("api_version", "agent_ref", "client", "format_hint")
+        }
+    )
     _callback: Final = StraikerGuardrail(
         api_key=api_key,
         api_base=api_base if isinstance(api_base, str) else "https://api.prod.straiker.ai",
         source=source,
         guardrail_name=guardrail.get("guardrail_name", "straiker"),
         event_hook=litellm_params.mode,
         default_on=litellm_params.default_on,
+        api_version=routing.api_version,
+        agent_ref=routing.agent_ref,
+        client=routing.client,
+        format_hint=routing.format_hint,
         **kwargs,
     )