node_service_pb2.py 2.6 KB

1234567891011121314151617181920212223242526272829303132333435363738
  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\"c\n\rPromptRequest\x12\"\n\x05shard\x18\x01 \x01(\x0b\x32\x13.node_service.Shard\x12\x0e\n\x06prompt\x18\x02 \x01(\t\x12\x13\n\x06target\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\t\n\x07_target\"y\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\x13\n\x06target\x18\x03 \x01(\tH\x00\x88\x01\x01\x42\t\n\x07_target\";\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\"7\n\x11ResetShardRequest\x12\"\n\x05shard\x18\x01 \x01(\x0b\x32\x13.node_service.Shard\"\x07\n\x05\x45mpty2\xd9\x01\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\x44\n\nResetShard\x12\x1f.node_service.ResetShardRequest\x1a\x13.node_service.Empty\"\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['_SHARD']._serialized_start=36
  19. _globals['_SHARD']._serialized_end=119
  20. _globals['_PROMPTREQUEST']._serialized_start=121
  21. _globals['_PROMPTREQUEST']._serialized_end=220
  22. _globals['_TENSORREQUEST']._serialized_start=222
  23. _globals['_TENSORREQUEST']._serialized_end=343
  24. _globals['_TENSOR']._serialized_start=345
  25. _globals['_TENSOR']._serialized_end=404
  26. _globals['_RESETSHARDREQUEST']._serialized_start=406
  27. _globals['_RESETSHARDREQUEST']._serialized_end=461
  28. _globals['_EMPTY']._serialized_start=463
  29. _globals['_EMPTY']._serialized_end=470
  30. _globals['_NODESERVICE']._serialized_start=473
  31. _globals['_NODESERVICE']._serialized_end=690
  32. # @@protoc_insertion_point(module_scope)