feat: add BaseLogitsProcessor core interface

Author: bhuvan002

lib/bindings/python/src/dynamo/logits_processing/__init__.py

@@ -0,0 +1,10 @@
+"""
+Dynamo Logits Processing - Backend-agnostic logits processors.
+
+This module provides the BaseLogitsProcessor protocol that can be used
+across different backend adapters (TRT-LLM, vLLM, SGLang).
+"""
+
+from .base import BaseLogitsProcessor
+
+__all__ = ["BaseLogitsProcessor"]

lib/bindings/python/src/dynamo/logits_processing/base.py

@@ -0,0 +1,36 @@
+"""
+Base logits processor protocol for Dynamo.
+
+This module defines the core BaseLogitsProcessor interface that all
+logits processors must implement.
+"""
+
+from typing import List, Protocol
+
+import torch
+
+
+class BaseLogitsProcessor(Protocol):
+    """
+    Protocol for logits processors in Dynamo.
+
+    All logits processors must implement this interface to be compatible
+    with backend adapters (TRT-LLM, vLLM, SGLang).
+    """
+
+    def __call__(
+        self,
+        input_ids: List[int],
+        logits: torch.Tensor,
+    ) -> torch.Tensor:
+        """
+        Process the logits for the next token prediction.
+
+        Args:
+            input_ids: The input token IDs generated so far.
+            logits: The raw logits for the next token. Shape: (vocab_size,)
+
+        Returns:
+            The modified logits tensor with same shape as input.
+        """
+        raise NotImplementedError