node_service_pb2.py 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. # -*- coding: utf-8 -*-
  2. # Generated by the protocol buffer compiler. DO NOT EDIT!
  3. # source: node_service.proto
  4. # Protobuf Python Version: 5.26.1
  5. """Generated protocol buffer code."""
  6. from google.protobuf import descriptor as _descriptor
  7. from google.protobuf import descriptor_pool as _descriptor_pool
  8. from google.protobuf import symbol_database as _symbol_database
  9. from google.protobuf.internal import builder as _builder
  10. # @@protoc_insertion_point(imports)
  11. _sym_db = _symbol_database.Default()
  12. DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x12node_service.proto\x12\x0cnode_service\"S\n\x05Shard\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x13\n\x0bstart_layer\x18\x02 \x01(\x05\x12\x11\n\tend_layer\x18\x03 \x01(\x05\x12\x10\n\x08n_layers\x18\x04 \x01(\x05\"\xc3\x01\n\rPromptRequest\x12\"\n\x05shard\x18\x01 \x01(\x0b\x32\x13.node_service.Shard\x12\x0e\n\x06prompt\x18\x02 \x01(\t\x12\x16\n\timage_str\x18\x03 \x01(\tH\x00\x88\x01\x01\x12\x17\n\nrequest_id\x18\x04 \x01(\tH\x01\x88\x01\x01\x12\x1c\n\x0finference_state\x18\x05 \x01(\tH\x02\x88\x01\x01\x42\x0c\n\n_image_strB\r\n\x0b_request_idB\x12\n\x10_inference_state\"\xb3\x01\n\rTensorRequest\x12\"\n\x05shard\x18\x01 \x01(\x0b\x32\x13.node_service.Shard\x12$\n\x06tensor\x18\x02 \x01(\x0b\x32\x14.node_service.Tensor\x12\x17\n\nrequest_id\x18\x03 \x01(\tH\x00\x88\x01\x01\x12\x1c\n\x0finference_state\x18\x04 \x01(\tH\x01\x88\x01\x01\x42\r\n\x0b_request_idB\x12\n\x10_inference_state\"/\n\x19GetInferenceResultRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\t\"\\\n\x0fInferenceResult\x12)\n\x06tensor\x18\x01 \x01(\x0b\x32\x14.node_service.TensorH\x00\x88\x01\x01\x12\x13\n\x0bis_finished\x18\x02 \x01(\x08\x42\t\n\x07_tensor\";\n\x06Tensor\x12\x13\n\x0btensor_data\x18\x01 \x01(\x0c\x12\r\n\x05shape\x18\x02 \x03(\x05\x12\r\n\x05\x64type\x18\x03 \x01(\t\"<\n\x16\x43ollectTopologyRequest\x12\x0f\n\x07visited\x18\x01 \x03(\t\x12\x11\n\tmax_depth\x18\x02 \x01(\x05\"\x8e\x02\n\x08Topology\x12\x30\n\x05nodes\x18\x01 \x03(\x0b\x32!.node_service.Topology.NodesEntry\x12\x39\n\npeer_graph\x18\x02 \x03(\x0b\x32%.node_service.Topology.PeerGraphEntry\x1aN\n\nNodesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12/\n\x05value\x18\x02 \x01(\x0b\x32 .node_service.DeviceCapabilities:\x02\x38\x01\x1a\x45\n\x0ePeerGraphEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\"\n\x05value\x18\x02 \x01(\x0b\x32\x13.node_service.Peers:\x02\x38\x01\"\x19\n\x05Peers\x12\x10\n\x08peer_ids\x18\x01 \x03(\t\"7\n\x0b\x44\x65viceFlops\x12\x0c\n\x04\x66p32\x18\x01 \x01(\x02\x12\x0c\n\x04\x66p16\x18\x02 \x01(\x02\x12\x0c\n\x04int8\x18\x03 \x01(\x02\"k\n\x12\x44\x65viceCapabilities\x12\r\n\x05model\x18\x01 \x01(\t\x12\x0c\n\x04\x63hip\x18\x02 \x01(\t\x12\x0e\n\x06memory\x18\x03 \x01(\x05\x12(\n\x05\x66lops\x18\x04 \x01(\x0b\x32\x19.node_service.DeviceFlops\"L\n\x11SendResultRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\t\x12\x0e\n\x06result\x18\x02 \x03(\x05\x12\x13\n\x0bis_finished\x18\x03 \x01(\x08\"=\n\x17SendOpaqueStatusRequest\x12\x12\n\nrequest_id\x18\x01 \x01(\t\x12\x0e\n\x06status\x18\x02 \x01(\t\"\x14\n\x12HealthCheckRequest\")\n\x13HealthCheckResponse\x12\x12\n\nis_healthy\x18\x01 \x01(\x08\"\x07\n\x05\x45mpty2\xb4\x04\n\x0bNodeService\x12\x41\n\nSendPrompt\x12\x1b.node_service.PromptRequest\x1a\x14.node_service.Tensor\"\x00\x12\x41\n\nSendTensor\x12\x1b.node_service.TensorRequest\x1a\x14.node_service.Tensor\"\x00\x12^\n\x12GetInferenceResult\x12\'.node_service.GetInferenceResultRequest\x1a\x1d.node_service.InferenceResult\"\x00\x12Q\n\x0f\x43ollectTopology\x12$.node_service.CollectTopologyRequest\x1a\x16.node_service.Topology\"\x00\x12\x44\n\nSendResult\x12\x1f.node_service.SendResultRequest\x1a\x13.node_service.Empty\"\x00\x12P\n\x10SendOpaqueStatus\x12%.node_service.SendOpaqueStatusRequest\x1a\x13.node_service.Empty\"\x00\x12T\n\x0bHealthCheck\x12 .node_service.HealthCheckRequest\x1a!.node_service.HealthCheckResponse\"\x00\x62\x06proto3')
  13. _globals = globals()
  14. _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
  15. _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'node_service_pb2', _globals)
  16. if not _descriptor._USE_C_DESCRIPTORS:
  17. DESCRIPTOR._loaded_options = None
  18. _globals['_TOPOLOGY_NODESENTRY']._loaded_options = None
  19. _globals['_TOPOLOGY_NODESENTRY']._serialized_options = b'8\001'
  20. _globals['_TOPOLOGY_PEERGRAPHENTRY']._loaded_options = None
  21. _globals['_TOPOLOGY_PEERGRAPHENTRY']._serialized_options = b'8\001'
  22. _globals['_SHARD']._serialized_start=36
  23. _globals['_SHARD']._serialized_end=119
  24. _globals['_PROMPTREQUEST']._serialized_start=122
  25. _globals['_PROMPTREQUEST']._serialized_end=317
  26. _globals['_TENSORREQUEST']._serialized_start=320
  27. _globals['_TENSORREQUEST']._serialized_end=499
  28. _globals['_GETINFERENCERESULTREQUEST']._serialized_start=501
  29. _globals['_GETINFERENCERESULTREQUEST']._serialized_end=548
  30. _globals['_INFERENCERESULT']._serialized_start=550
  31. _globals['_INFERENCERESULT']._serialized_end=642
  32. _globals['_TENSOR']._serialized_start=644
  33. _globals['_TENSOR']._serialized_end=703
  34. _globals['_COLLECTTOPOLOGYREQUEST']._serialized_start=705
  35. _globals['_COLLECTTOPOLOGYREQUEST']._serialized_end=765
  36. _globals['_TOPOLOGY']._serialized_start=768
  37. _globals['_TOPOLOGY']._serialized_end=1038
  38. _globals['_TOPOLOGY_NODESENTRY']._serialized_start=889
  39. _globals['_TOPOLOGY_NODESENTRY']._serialized_end=967
  40. _globals['_TOPOLOGY_PEERGRAPHENTRY']._serialized_start=969
  41. _globals['_TOPOLOGY_PEERGRAPHENTRY']._serialized_end=1038
  42. _globals['_PEERS']._serialized_start=1040
  43. _globals['_PEERS']._serialized_end=1065
  44. _globals['_DEVICEFLOPS']._serialized_start=1067
  45. _globals['_DEVICEFLOPS']._serialized_end=1122
  46. _globals['_DEVICECAPABILITIES']._serialized_start=1124
  47. _globals['_DEVICECAPABILITIES']._serialized_end=1231
  48. _globals['_SENDRESULTREQUEST']._serialized_start=1233
  49. _globals['_SENDRESULTREQUEST']._serialized_end=1309
  50. _globals['_SENDOPAQUESTATUSREQUEST']._serialized_start=1311
  51. _globals['_SENDOPAQUESTATUSREQUEST']._serialized_end=1372
  52. _globals['_HEALTHCHECKREQUEST']._serialized_start=1374
  53. _globals['_HEALTHCHECKREQUEST']._serialized_end=1394
  54. _globals['_HEALTHCHECKRESPONSE']._serialized_start=1396
  55. _globals['_HEALTHCHECKRESPONSE']._serialized_end=1437
  56. _globals['_EMPTY']._serialized_start=1439
  57. _globals['_EMPTY']._serialized_end=1446
  58. _globals['_NODESERVICE']._serialized_start=1449
  59. _globals['_NODESERVICE']._serialized_end=2013
  60. # @@protoc_insertion_point(module_scope)