Extend BaseModelOptions with anonymize_algorithm and model_name_anonymize (add typing imports) and update endpoint_i to pop these new fields from the request payload.

Author: Paweł Kędzia

llm_router_api/endpoints/endpoint_i.py

@@ -865,6 +865,8 @@ def _prepare_payload_at_beginning(
         """
         # Remember general options before clear payload
         _anon_payload = FORCE_ANONYMISATION or payload.pop("anonymize", False)
+        _anonymize_algorithm = payload.pop("anonymize_algorithm", None)
+        _model_name_anonymize = payload.pop("model_name_anonymize", None)
 
         payload = self._clear_payload(payload=payload)
         if not _anon_payload:
@@ -898,7 +900,7 @@ def _clear_payload(payload: Dict[str, Any]):
             payload.pop(k, "")
 
         # If stream param is not given, then set as False
-        payload["stream"] =  payload.get("stream", False)
+        payload["stream"] = payload.get("stream", False)
         return payload
 
     def _anonymize_payload(self, payload: Dict | str | List | Any) -> Dict[str, Any]:

llm_router_lib/data_models/base_model.py

@@ -8,6 +8,7 @@
 """
 
 from pydantic import BaseModel
+from typing import Literal, Optional
 
 
 class BaseModelOptions(BaseModel):
@@ -21,6 +22,21 @@ class BaseModelOptions(BaseModel):
         library's :class:`~llm_router_lib.anonymizer.core.Anonymizer` before any
         downstream logic is executed.  The default value ``False`` disables
         anonymisation.
+
+    anonymize_algorithm : Literal["fast_masker", "genai", "priv_masker"], default ``"fast_masker"``
+        Specifies which anonymisation algorithm to use when ``anonymize`` is ``True``.
+        * ``"fast_masker"`` – a lightweight, high‑performance masker.
+        * ``"genai"`` – a generative‑AI based approach.
+        * ``"priv_masker"`` – a privacy‑focused masking technique.
+
+    model_name_anonymize : Optional[str], default ``None``
+        Name or identifier of the model to be used for anonymisation when
+        ``anonymize_algorithm`` requires a specific model (e.g., for the ``"genai"``
+        algorithm).  If ``None``, the library will fall back to its default model.
     """
 
     anonymize: bool = False
+    anonymize_algorithm: Literal["fast_masker", "genai", "priv_masker"] = (
+        "fast_masker"
+    )
+    model_name_anonymize: Optional[str] = None