1818from azure .core .serialization import _datetime_as_isostr # pylint:disable=protected-access
1919
2020from ._models import (
21- JobQueue as JobQueueGenerated ,
2221 RouterWorker as RouterWorkerGenerated ,
2322 RouterJob as RouterJobGenerated ,
2423 JSON ,
@@ -48,12 +47,18 @@ class RouterJob(RouterJobGenerated):
4847 def __init__ (
4948 self ,
5049 * ,
51- notes : Dict [Union [str , datetime ], str ] = None ,
50+ channel_reference : Optional [str ] = None ,
51+ channel_id : Optional [str ] = None ,
52+ classification_policy_id : Optional [str ] = None ,
53+ queue_id : Optional [str ] = None ,
54+ priority : Optional [int ] = None ,
55+ disposition_code : Optional [str ] = None ,
56+ requested_worker_selectors : Optional [List ["_models.WorkerSelector" ]] = None ,
5257 labels : Optional [Dict [str , Union [int , float , str , bool , None ]]] = None ,
5358 tags : Optional [Dict [str , Union [int , float , str , bool , None ]]] = None ,
59+ notes : Dict [Union [str , datetime ], str ] = None ,
5460 ** kwargs
5561 ):
56-
5762 if notes :
5863 for k in [key for key in notes .keys ()]:
5964 v : str = notes [k ]
@@ -66,23 +71,44 @@ def __init__(
6671 datetime_as_str : str = _datetime_as_isostr (k ) # pylint:disable=protected-access
6772 notes .pop (k )
6873 notes [datetime_as_str ] = v
69- super ().__init__ (notes = notes , labels = labels , tags = tags , ** kwargs )
74+ super ().__init__ (
75+ channel_reference = channel_reference ,
76+ channel_id = channel_id ,
77+ classification_policy_id = classification_policy_id ,
78+ queue_id = queue_id ,
79+ priority = priority ,
80+ disposition_code = disposition_code ,
81+ requested_worker_selectors = requested_worker_selectors ,
82+ notes = notes ,
83+ labels = labels ,
84+ tags = tags ,
85+ ** kwargs )
7086
7187
7288class RouterWorker (RouterWorkerGenerated ):
7389 def __init__ (
7490 self ,
7591 * ,
7692 queue_assignments : Optional [Dict [str , Union [QueueAssignment , JSON , None ]]] = None ,
93+ total_capacity : Optional [int ] = None ,
7794 labels : Optional [Dict [str , Union [int , float , str , bool , None ]]] = None ,
7895 tags : Optional [Dict [str , Union [int , float , str , bool , None ]]] = None ,
96+ channel_configurations : Optional [Dict [str , "_models.ChannelConfiguration" ]] = None ,
97+ available_for_offers : Optional [bool ] = None ,
7998 ** kwargs
8099 ):
81100 if queue_assignments :
82101 for k , v in queue_assignments .items ():
83102 if not isinstance (v , (MutableMapping , JSON , type (None ))):
84103 raise ValueError ("tags only accept 'QueueAssignment', 'JSON' and 'NoneType' as values." )
85- super ().__init__ (queue_assignments = queue_assignments , labels = labels , tags = tags , ** kwargs )
104+ super ().__init__ (
105+ queue_assignments = queue_assignments ,
106+ total_capacity = total_capacity ,
107+ labels = labels ,
108+ tags = tags ,
109+ channel_configurations = channel_configurations ,
110+ available_for_offers = available_for_offers ,
111+ ** kwargs )
86112
87113
88114__all__ : List [str ] = [
0 commit comments