@@ -136,74 +136,3 @@ def HealthCheck(
136136 metadata ,
137137 )
138138
139-
140- class MultiAdapterManagerStub (object ):
141- """MultiAdapterManagerStub."""
142-
143- def __init__ (self , channel ):
144- """Constructor.
145-
146- Args:
147- channel: A grpc.Channel.
148- """
149- self .ListAdapters = channel .unary_unary (
150- '/jetstream_proto.MultiAdapterManager/ListAdapters' ,
151- request_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .ListAdaptersRequest .SerializeToString ,
152- response_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .ListAdaptersResponse .FromString ,
153- _registered_method = True )
154- self .LoadAdapter = channel .unary_unary (
155- '/jetstream_proto.MultiAdapterManager/LoadAdapter' ,
156- request_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .LoadAdapterRequest .SerializeToString ,
157- response_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .LoadAdapterResponse .FromString ,
158- _registered_method = True )
159- self .UnloadAdapter = channel .unary_unary (
160- '/jetstream_proto.MultiAdapterManager/UnloadAdapter' ,
161- request_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .UnloadAdapterRequest .SerializeToString ,
162- response_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .UnloadAdapterResponse .FromString ,
163- _registered_method = True )
164-
165-
166- class MultiAdapterManagerServicer (object ):
167- """TODO: Merge this with main JetStream core once we settle on an API."""
168-
169- def ListAdapters (self , request , context ):
170- """Lists all the currently loaded LoRA adapters."""
171- context .set_code (grpc .StatusCode .UNIMPLEMENTED )
172- context .set_details ("Method not implemented!" )
173- raise NotImplementedError ("Method not implemented!" )
174-
175- def LoadAdapter (self , request , context ):
176- """Check the feasibility and load the new LoRA adapter."""
177- context .set_code (grpc .StatusCode .UNIMPLEMENTED )
178- context .set_details ("Method not implemented!" )
179- raise NotImplementedError ("Method not implemented!" )
180-
181- def UnloadAdapter (self , request , context ):
182- """Unload a LoRA adapter."""
183- context .set_code (grpc .StatusCode .UNIMPLEMENTED )
184- context .set_details ("Method not implemented!" )
185- raise NotImplementedError ("Method not implemented!" )
186-
187-
188- def add_MultiAdapterManagerServicer_to_server (servicer , server ):
189- rpc_method_handlers = {
190- "ListAdapters" : grpc .unary_unary_rpc_method_handler (
191- servicer .ListAdapters ,
192- request_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .ListAdaptersRequest .FromString ,
193- response_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .ListAdaptersResponse .SerializeToString ,
194- ),
195- "LoadAdapter" : grpc .unary_unary_rpc_method_handler (
196- servicer .LoadAdapter ,
197- request_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .LoadAdapterRequest .FromString ,
198- response_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .LoadAdapterResponse .SerializeToString ,
199- ),
200- "UnloadAdapter" : grpc .unary_unary_rpc_method_handler (
201- servicer .UnloadAdapter ,
202- request_deserializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .UnloadAdapterRequest .FromString ,
203- response_serializer = jetstream_dot_core_dot_proto_dot_jetstream__pb2 .UnloadAdapterResponse .SerializeToString ,
204- ),
205- }
206- generic_handler = grpc .method_handlers_generic_handler (
207- "jetstream_proto.MultiAdapterManager" , rpc_method_handlers
208- )
209- server .add_generic_rpc_handlers ((generic_handler ,))
0 commit comments