Update control-plane-server.py

itsbalamurali · Nov 13, 2023 · 75d4700 · 75d4700
1 parent e966e74
commit 75d4700
Showing 1 changed file with 9 additions and 9 deletions.
diff --git a/control-plane-server.py b/control-plane-server.py
@@ -186,29 +186,29 @@ class ControlPlaneSpecResponse(BaseModel):
 
 
 class ReAttachRequest(BaseModel):
-    node_id: int
+    node_id: int | str
 
 
 class ReAttachResponseTenant(BaseModel):
-    id: str
-    gen: int
+    id: str | int
+    gen: int | str
 
 
 class ReAttachResponse(BaseModel):
     tenants: List[ReAttachResponseTenant]
 
 
 class ValidateRequestTenant(BaseModel):
-    id: str
-    gen: int
+    id: str | int
+    gen: int | str
 
 
 class ValidateRequest(BaseModel):
     tenants: List[ValidateRequestTenant]
 
 
 class ValidateResponseTenant(BaseModel):
-    id: str
+    id: str | int
     valid: bool
 
 
@@ -230,8 +230,8 @@ class AttachHookRequest(BaseModel):
         tenant_id (str): The ID of the tenant.
         node_id (str): The ID of the node to attach the hook to.
     """
-    tenant_id: str
-    node_id: str
+    tenant_id: str | int
+    node_id: str | int
 
 
 class AttachHookResponse(BaseModel):
@@ -241,7 +241,7 @@ class AttachHookResponse(BaseModel):
     Attributes:
         gen (int): The generation number of the hook.
     """
-    gen: int
+    gen: int | str
 
 
 class WelcomeMessage(BaseModel):