Adding the inference grant

Author: adrian-cowham

livekit-api/livekit/api/access_token.py

@@ -75,6 +75,10 @@ class SIPGrants:
     # make outbound calls
     call: bool = False
 
+@dataclasses.dataclass
+class InferenceGrants:
+    # perform inference
+    perform: bool = False
 
 @dataclasses.dataclass
 class Claims:
@@ -84,6 +88,7 @@ class Claims:
     metadata: str = ""
     video: Optional[VideoGrants] = None
     sip: Optional[SIPGrants] = None
+    inference: Optional[InferenceGrants] = None
     attributes: Optional[dict[str, str]] = None
     sha256: Optional[str] = None
     room_preset: Optional[str] = None
@@ -135,6 +140,10 @@ def with_grants(self, grants: VideoGrants) -> "AccessToken":
     def with_sip_grants(self, grants: SIPGrants) -> "AccessToken":
         self.claims.sip = grants
         return self
+    
+    def with_inference_grants(self, grants: InferenceGrants) -> "AccessToken":
+        self.claims.inference = grants
+        return self
 
     def with_identity(self, identity: str) -> "AccessToken":
         self.identity = identity
@@ -225,11 +234,17 @@ def verify(self, token: str, *, verify_signature: bool = True) -> Claims:
         sip_dict = {k: v for k, v in sip_dict.items() if k in SIPGrants.__dataclass_fields__}
         sip = SIPGrants(**sip_dict)
 
+        inference_dict = claims.get("inference", dict())
+        inference_dict = {camel_to_snake(k): v for k, v in inference_dict.items()}
+        inference_dict = {k: v for k, v in inference_dict.items() if k in InferenceGrants.__dataclass_fields__}
+        inference = InferenceGrants(**inference_dict)
+
         grant_claims = Claims(
             identity=claims.get("sub", ""),
             name=claims.get("name", ""),
             video=video,
             sip=sip,
+            inference=inference,
             attributes=claims.get("attributes", {}),
             metadata=claims.get("metadata", ""),
             sha256=claims.get("sha256", ""),