neptune-ai · kgodlewski · Jan 26, 2025 · Jan 30, 2025 · Jan 30, 2025 · Feb 11, 2025
@@ -1,9 +1,9 @@
 project_name_override: neptune-api
 package_version_override: 0.1.0
 post_hooks:
-   - "ruff check . --fix"
+   - "ruff check . --fix --target-version=py38"
    - "ruff format "
    - "isort ."
-   - "echo >> py.typed"
+   - "echo -n > py.typed"
 content_type_overrides:
   application/x-protobuf: application/octet-stream
@@ -85,7 +85,7 @@ exclude = ["src/neptune_api/proto", "src/neptune_retrieval_api/proto"]
 select = ["F", "UP"]
 
 [tool.mypy]
-files = ['src/neptune_api', 'src/neptune_retrieval_api']
+files = ['src/neptune_api', 'src/neptune_retrieval_api', 'src/neptune_storage_api']
 mypy_path = "stubs"
 install_types = "True"
 non_interactive = "True"

diff --git a/scripts/update-storagebridge.sh b/scripts/update-storagebridge.sh
@@ -0,0 +1,63 @@
+#!/usr/bin/env bash
+
+if [ ! -f "storagebridge_swagger.json" ]; then
+  echo "storagebridge_swagger.json not found. Make sure it's in CWD when running the script"
+  exit 1
+fi
+
+# Show every command and exit on error
+set -ex
+
+## Preserve specific files
+mkdir -p tmp
+
+# Modify the Swagger JSON to support application/x-protobuf
+jq '
+  .paths |= with_entries(
+    .value |= with_entries(
+      if .value.produces? and (.value.produces | index("application/json")) and (.value.produces | index("application/x-protobuf"))
+      then
+        .value.produces = ["application/x-protobuf"]
+      else
+        .
+      end
+    )
+  )
+' storagebridge_swagger.json > tmp/storagebridge_swagger.json && mv tmp/storagebridge_swagger.json storagebridge_swagger.json
+
+jq '
+  .paths |= with_entries(
+    .value |= with_entries(
+      if .value.produces? and (.value.produces | index("application/x-protobuf"))
+      then
+        .value.responses."200".schema = {
+          "type": "string",
+          "format": "binary"
+        }
+      else
+        .
+      end
+    )
+  )
+' storagebridge_swagger.json > tmp/storagebridge_swagger.json && mv tmp/storagebridge_swagger.json storagebridge_swagger.json
+
+# Add license information using jq
+jq '.info.license = {"name": "", "url": ""}' storagebridge_openapi.json > tmp_storagebridge_openapi.json && mv tmp_storagebridge_openapi.json storagebridge_openapi.json
+
+
+openapi-python-client generate \
+    --overwrite \
+    --meta none \
+    --path "storagebridge_openapi.json" \
+    --custom-template-path=templates/ \
+    --config openapi-generator-config.yaml \
+    --output-path src/neptune_storage_api
+
+
+# Clean tmp directories
+rm -rf tmp
+
+
+cat scripts/preserve_files.txt | while read entry; do
+    git checkout HEAD -- $entry
+done
diff --git a/src/neptune_api/proto/google_rpc/code_pb2.py b/src/neptune_api/proto/google_rpc/code_pb2.py
@@ -1,15 +1,17 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(_runtime_version.Domain.PUBLIC, 5, 28, 1, '', 'google_rpc/code.proto')
 _sym_db = _symbol_database.Default()
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15google_rpc/code.proto\x12\ngoogle_rpc*\xb7\x02\n\x04Code\x12\x06\n\x02OK\x10\x00\x12\r\n\tCANCELLED\x10\x01\x12\x0b\n\x07UNKNOWN\x10\x02\x12\x14\n\x10INVALID_ARGUMENT\x10\x03\x12\x15\n\x11DEADLINE_EXCEEDED\x10\x04\x12\r\n\tNOT_FOUND\x10\x05\x12\x12\n\x0eALREADY_EXISTS\x10\x06\x12\x15\n\x11PERMISSION_DENIED\x10\x07\x12\x13\n\x0fUNAUTHENTICATED\x10\x10\x12\x16\n\x12RESOURCE_EXHAUSTED\x10\x08\x12\x17\n\x13FAILED_PRECONDITION\x10\t\x12\x0b\n\x07ABORTED\x10\n\x12\x10\n\x0cOUT_OF_RANGE\x10\x0b\x12\x11\n\rUNIMPLEMENTED\x10\x0c\x12\x0c\n\x08INTERNAL\x10\r\x12\x0f\n\x0bUNAVAILABLE\x10\x0e\x12\r\n\tDATA_LOSS\x10\x0fBX\n\x0ecom.google.rpcB\tCodeProtoP\x01Z3google.golang.org/genproto/googleapis/rpc/code;code\xa2\x02\x03RPCb\x06proto3')
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google_rpc.code_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-    _globals['DESCRIPTOR']._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals['DESCRIPTOR']._loaded_options = None
     _globals['DESCRIPTOR']._serialized_options = b'\n\x0ecom.google.rpcB\tCodeProtoP\x01Z3google.golang.org/genproto/googleapis/rpc/code;code\xa2\x02\x03RPC'
     _globals['_CODE']._serialized_start = 38
     _globals['_CODE']._serialized_end = 349
diff --git a/src/neptune_api/proto/neptune_pb/ingest/v1/common_pb2.py b/src/neptune_api/proto/neptune_pb/ingest/v1/common_pb2.py
@@ -1,52 +1,56 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(_runtime_version.Domain.PUBLIC, 5, 28, 1, '', 'neptune_pb/ingest/v1/common.proto')
 _sym_db = _symbol_database.Default()
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!neptune_pb/ingest/v1/common.proto\x12\x11neptune.ingest.v1\x1a\x1fgoogle/protobuf/timestamp.proto"$\n\x04Step\x12\r\n\x05whole\x18\x01 \x01(\x04\x12\r\n\x05micro\x18\x02 \x01(\x04"\x9b\x01\n\tForkPoint\x12\x16\n\x0eparent_project\x18\x01 \x01(\t\x12\x15\n\rparent_run_id\x18\x02 \x01(\t\x12%\n\x04step\x18\x04 \x01(\x0b2\x17.neptune.ingest.v1.Step\x12 \n\x13requested_parent_id\x18\x0f \x01(\tH\x00\x88\x01\x01B\x16\n\x14_requested_parent_id"\x1b\n\tStringSet\x12\x0e\n\x06values\x18\x01 \x03(\t"\xbb\x01\n\x05Value\x12\x11\n\x07float64\x18\x01 \x01(\x01H\x00\x12\x0f\n\x05int64\x18\x03 \x01(\x03H\x00\x12\x0e\n\x04bool\x18\x05 \x01(\x08H\x00\x12\x10\n\x06string\x18\x06 \x01(\tH\x00\x12/\n\ttimestamp\x18\x08 \x01(\x0b2\x1a.google.protobuf.TimestampH\x00\x122\n\nstring_set\x18\x0c \x01(\x0b2\x1c.neptune.ingest.v1.StringSetH\x00B\x07\n\x05value"\xa2\x01\n\x0fModifyStringSet\x12>\n\x06values\x18\x01 \x03(\x0b2..neptune.ingest.v1.ModifyStringSet.ValuesEntry\x1aO\n\x0bValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12/\n\x05value\x18\x02 \x01(\x0e2 .neptune.ingest.v1.SET_OPERATION:\x028\x01"I\n\tModifySet\x124\n\x06string\x18\x01 \x01(\x0b2".neptune.ingest.v1.ModifyStringSetH\x00B\x06\n\x04type"F\n\x05Owner\x12\x11\n\x07user_id\x18\x01 \x01(\tH\x00\x12\x1c\n\x12service_account_id\x18\x02 \x01(\tH\x00B\x0c\n\nowner_type"\xcf\x02\n\x03Run\x12\x13\n\x06run_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1a\n\rexperiment_id\x18\x05 \x01(\tH\x01\x88\x01\x01\x120\n\nfork_point\x18\x02 \x01(\x0b2\x1c.neptune.ingest.v1.ForkPoint\x12\x13\n\x06family\x18\x04 \x01(\tH\x02\x88\x01\x01\x126\n\rcreation_time\x18\x03 \x01(\x0b2\x1a.google.protobuf.TimestampH\x03\x88\x01\x01\x12,\n\x05owner\x18\x06 \x01(\x0b2\x18.neptune.ingest.v1.OwnerH\x04\x88\x01\x01\x12\x17\n\nrequest_id\x18\x07 \x01(\tH\x05\x88\x01\x01B\t\n\x07_run_idB\x10\n\x0e_experiment_idB\t\n\x07_familyB\x10\n\x0e_creation_timeB\x08\n\x06_ownerB\r\n\x0b_request_id"\xc3\x04\n\x11UpdateRunSnapshot\x12%\n\x04step\x18\x01 \x01(\x0b2\x17.neptune.ingest.v1.Step\x12-\n\ttimestamp\x18\x02 \x01(\x0b2\x1a.google.protobuf.Timestamp\x12@\n\x06assign\x18\x04 \x03(\x0b20.neptune.ingest.v1.UpdateRunSnapshot.AssignEntry\x12I\n\x0bmodify_sets\x18\x05 \x03(\x0b24.neptune.ingest.v1.UpdateRunSnapshot.ModifySetsEntry\x12@\n\x06append\x18\x08 \x03(\x0b20.neptune.ingest.v1.UpdateRunSnapshot.AppendEntry\x12\x17\n\nrequest_id\x18\t \x01(\tH\x00\x88\x01\x01\x1aG\n\x0bAssignEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b2\x18.neptune.ingest.v1.Value:\x028\x01\x1aO\n\x0fModifySetsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b2\x1c.neptune.ingest.v1.ModifySet:\x028\x01\x1aG\n\x0bAppendEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b2\x18.neptune.ingest.v1.Value:\x028\x01B\r\n\x0b_request_id"M\n\x12UpdateRunSnapshots\x127\n\tsnapshots\x18\x01 \x03(\x0b2$.neptune.ingest.v1.UpdateRunSnapshot*.\n\rSET_OPERATION\x12\x08\n\x04NOOP\x10\x00\x12\x07\n\x03ADD\x10\x01\x12\n\n\x06REMOVE\x10\x02B5\n$ml.neptune.leaderboard.api.ingest.v1B\x0bCommonProtoP\x01b\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n!neptune_pb/ingest/v1/common.proto\x12\x11neptune.ingest.v1\x1a\x1fgoogle/protobuf/timestamp.proto"$\n\x04Step\x12\r\n\x05whole\x18\x01 \x01(\x04\x12\r\n\x05micro\x18\x02 \x01(\x04"\x9b\x01\n\tForkPoint\x12\x16\n\x0eparent_project\x18\x01 \x01(\t\x12\x15\n\rparent_run_id\x18\x02 \x01(\t\x12%\n\x04step\x18\x04 \x01(\x0b2\x17.neptune.ingest.v1.Step\x12 \n\x13requested_parent_id\x18\x0f \x01(\tH\x00\x88\x01\x01B\x16\n\x14_requested_parent_id"\x1b\n\tStringSet\x12\x0e\n\x06values\x18\x01 \x03(\t">\n\x07FileRef\x12\x0c\n\x04path\x18\x01 \x01(\t\x12\x12\n\nsize_bytes\x18\x02 \x01(\x03\x12\x11\n\tmime_type\x18\x03 \x01(\t"\xeb\x01\n\x05Value\x12\x11\n\x07float64\x18\x01 \x01(\x01H\x00\x12\x0f\n\x05int64\x18\x03 \x01(\x03H\x00\x12\x0e\n\x04bool\x18\x05 \x01(\x08H\x00\x12\x10\n\x06string\x18\x06 \x01(\tH\x00\x12/\n\ttimestamp\x18\x08 \x01(\x0b2\x1a.google.protobuf.TimestampH\x00\x122\n\nstring_set\x18\x0c \x01(\x0b2\x1c.neptune.ingest.v1.StringSetH\x00\x12.\n\x08file_ref\x18\r \x01(\x0b2\x1a.neptune.ingest.v1.FileRefH\x00B\x07\n\x05value"\xa2\x01\n\x0fModifyStringSet\x12>\n\x06values\x18\x01 \x03(\x0b2..neptune.ingest.v1.ModifyStringSet.ValuesEntry\x1aO\n\x0bValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12/\n\x05value\x18\x02 \x01(\x0e2 .neptune.ingest.v1.SET_OPERATION:\x028\x01"I\n\tModifySet\x124\n\x06string\x18\x01 \x01(\x0b2".neptune.ingest.v1.ModifyStringSetH\x00B\x06\n\x04type"F\n\x05Owner\x12\x11\n\x07user_id\x18\x01 \x01(\tH\x00\x12\x1c\n\x12service_account_id\x18\x02 \x01(\tH\x00B\x0c\n\nowner_type"\xcf\x02\n\x03Run\x12\x13\n\x06run_id\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x1a\n\rexperiment_id\x18\x05 \x01(\tH\x01\x88\x01\x01\x120\n\nfork_point\x18\x02 \x01(\x0b2\x1c.neptune.ingest.v1.ForkPoint\x12\x13\n\x06family\x18\x04 \x01(\tH\x02\x88\x01\x01\x126\n\rcreation_time\x18\x03 \x01(\x0b2\x1a.google.protobuf.TimestampH\x03\x88\x01\x01\x12,\n\x05owner\x18\x06 \x01(\x0b2\x18.neptune.ingest.v1.OwnerH\x04\x88\x01\x01\x12\x17\n\nrequest_id\x18\x07 \x01(\tH\x05\x88\x01\x01B\t\n\x07_run_idB\x10\n\x0e_experiment_idB\t\n\x07_familyB\x10\n\x0e_creation_timeB\x08\n\x06_ownerB\r\n\x0b_request_id"\xc3\x04\n\x11UpdateRunSnapshot\x12%\n\x04step\x18\x01 \x01(\x0b2\x17.neptune.ingest.v1.Step\x12-\n\ttimestamp\x18\x02 \x01(\x0b2\x1a.google.protobuf.Timestamp\x12@\n\x06assign\x18\x04 \x03(\x0b20.neptune.ingest.v1.UpdateRunSnapshot.AssignEntry\x12I\n\x0bmodify_sets\x18\x05 \x03(\x0b24.neptune.ingest.v1.UpdateRunSnapshot.ModifySetsEntry\x12@\n\x06append\x18\x08 \x03(\x0b20.neptune.ingest.v1.UpdateRunSnapshot.AppendEntry\x12\x17\n\nrequest_id\x18\t \x01(\tH\x00\x88\x01\x01\x1aG\n\x0bAssignEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b2\x18.neptune.ingest.v1.Value:\x028\x01\x1aO\n\x0fModifySetsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12+\n\x05value\x18\x02 \x01(\x0b2\x1c.neptune.ingest.v1.ModifySet:\x028\x01\x1aG\n\x0bAppendEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\'\n\x05value\x18\x02 \x01(\x0b2\x18.neptune.ingest.v1.Value:\x028\x01B\r\n\x0b_request_id"M\n\x12UpdateRunSnapshots\x127\n\tsnapshots\x18\x01 \x03(\x0b2$.neptune.ingest.v1.UpdateRunSnapshot*.\n\rSET_OPERATION\x12\x08\n\x04NOOP\x10\x00\x12\x07\n\x03ADD\x10\x01\x12\n\n\x06REMOVE\x10\x02B5\n$ml.neptune.leaderboard.api.ingest.v1B\x0bCommonProtoP\x01b\x06proto3')
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'neptune_pb.ingest.v1.common_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-    _globals['DESCRIPTOR']._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+    _globals['DESCRIPTOR']._loaded_options = None
     _globals['DESCRIPTOR']._serialized_options = b'\n$ml.neptune.leaderboard.api.ingest.v1B\x0bCommonProtoP\x01'
-    _globals['_MODIFYSTRINGSET_VALUESENTRY']._options = None
+    _globals['_MODIFYSTRINGSET_VALUESENTRY']._loaded_options = None
     _globals['_MODIFYSTRINGSET_VALUESENTRY']._serialized_options = b'8\x01'
-    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._options = None
+    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._loaded_options = None
     _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._serialized_options = b'8\x01'
-    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._options = None
+    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._loaded_options = None
     _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._serialized_options = b'8\x01'
-    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._options = None
+    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._loaded_options = None
     _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._serialized_options = b'8\x01'
-    _globals['_SET_OPERATION']._serialized_start = 1815
-    _globals['_SET_OPERATION']._serialized_end = 1861
+    _globals['_SET_OPERATION']._serialized_start = 1927
+    _globals['_SET_OPERATION']._serialized_end = 1973
     _globals['_STEP']._serialized_start = 89
     _globals['_STEP']._serialized_end = 125
     _globals['_FORKPOINT']._serialized_start = 128
     _globals['_FORKPOINT']._serialized_end = 283
     _globals['_STRINGSET']._serialized_start = 285
     _globals['_STRINGSET']._serialized_end = 312
-    _globals['_VALUE']._serialized_start = 315
-    _globals['_VALUE']._serialized_end = 502
-    _globals['_MODIFYSTRINGSET']._serialized_start = 505
-    _globals['_MODIFYSTRINGSET']._serialized_end = 667
-    _globals['_MODIFYSTRINGSET_VALUESENTRY']._serialized_start = 588
-    _globals['_MODIFYSTRINGSET_VALUESENTRY']._serialized_end = 667
-    _globals['_MODIFYSET']._serialized_start = 669
-    _globals['_MODIFYSET']._serialized_end = 742
-    _globals['_OWNER']._serialized_start = 744
-    _globals['_OWNER']._serialized_end = 814
-    _globals['_RUN']._serialized_start = 817
-    _globals['_RUN']._serialized_end = 1152
-    _globals['_UPDATERUNSNAPSHOT']._serialized_start = 1155
-    _globals['_UPDATERUNSNAPSHOT']._serialized_end = 1734
-    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._serialized_start = 1494
-    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._serialized_end = 1565
-    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._serialized_start = 1567
-    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._serialized_end = 1646
-    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._serialized_start = 1648
-    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._serialized_end = 1719
-    _globals['_UPDATERUNSNAPSHOTS']._serialized_start = 1736
-    _globals['_UPDATERUNSNAPSHOTS']._serialized_end = 1813
+    _globals['_FILEREF']._serialized_start = 314
+    _globals['_FILEREF']._serialized_end = 376
+    _globals['_VALUE']._serialized_start = 379
+    _globals['_VALUE']._serialized_end = 614
+    _globals['_MODIFYSTRINGSET']._serialized_start = 617
+    _globals['_MODIFYSTRINGSET']._serialized_end = 779
+    _globals['_MODIFYSTRINGSET_VALUESENTRY']._serialized_start = 700
+    _globals['_MODIFYSTRINGSET_VALUESENTRY']._serialized_end = 779
+    _globals['_MODIFYSET']._serialized_start = 781
+    _globals['_MODIFYSET']._serialized_end = 854
+    _globals['_OWNER']._serialized_start = 856
+    _globals['_OWNER']._serialized_end = 926
+    _globals['_RUN']._serialized_start = 929
+    _globals['_RUN']._serialized_end = 1264
+    _globals['_UPDATERUNSNAPSHOT']._serialized_start = 1267
+    _globals['_UPDATERUNSNAPSHOT']._serialized_end = 1846
+    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._serialized_start = 1606
+    _globals['_UPDATERUNSNAPSHOT_ASSIGNENTRY']._serialized_end = 1677
+    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._serialized_start = 1679
+    _globals['_UPDATERUNSNAPSHOT_MODIFYSETSENTRY']._serialized_end = 1758
+    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._serialized_start = 1760
+    _globals['_UPDATERUNSNAPSHOT_APPENDENTRY']._serialized_end = 1831
+    _globals['_UPDATERUNSNAPSHOTS']._serialized_start = 1848
+    _globals['_UPDATERUNSNAPSHOTS']._serialized_end = 1925
diff --git a/src/neptune_api/proto/neptune_pb/ingest/v1/common_pb2.pyi b/src/neptune_api/proto/neptune_pb/ingest/v1/common_pb2.pyi
@@ -106,6 +106,23 @@ class StringSet(google.protobuf.message.Message):
         ...
 global___StringSet = StringSet
 
+@typing.final
+class FileRef(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    PATH_FIELD_NUMBER: builtins.int
+    SIZE_BYTES_FIELD_NUMBER: builtins.int
+    MIME_TYPE_FIELD_NUMBER: builtins.int
+    path: builtins.str
+    size_bytes: builtins.int
+    mime_type: builtins.str
+
+    def __init__(self, *, path: builtins.str=..., size_bytes: builtins.int=..., mime_type: builtins.str=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['mime_type', b'mime_type', 'path', b'path', 'size_bytes', b'size_bytes']) -> None:
+        ...
+global___FileRef = FileRef
+
 @typing.final
 class Value(google.protobuf.message.Message):
     """Value is a union of all supported types that can be used to update a field.
@@ -118,6 +135,7 @@ class Value(google.protobuf.message.Message):
     STRING_FIELD_NUMBER: builtins.int
     TIMESTAMP_FIELD_NUMBER: builtins.int
     STRING_SET_FIELD_NUMBER: builtins.int
+    FILE_REF_FIELD_NUMBER: builtins.int
     float64: builtins.float
     int64: builtins.int
     bool: builtins.bool
@@ -131,16 +149,20 @@ class Value(google.protobuf.message.Message):
     def string_set(self) -> global___StringSet:
         ...
 
-    def __init__(self, *, float64: builtins.float=..., int64: builtins.int=..., bool: builtins.bool=..., string: builtins.str=..., timestamp: google.protobuf.timestamp_pb2.Timestamp | None=..., string_set: global___StringSet | None=...) -> None:
+    @property
+    def file_ref(self) -> global___FileRef:
+        ...
+
+    def __init__(self, *, float64: builtins.float=..., int64: builtins.int=..., bool: builtins.bool=..., string: builtins.str=..., timestamp: google.protobuf.timestamp_pb2.Timestamp | None=..., string_set: global___StringSet | None=..., file_ref: global___FileRef | None=...) -> None:
         ...
 
-    def HasField(self, field_name: typing.Literal['bool', b'bool', 'float64', b'float64', 'int64', b'int64', 'string', b'string', 'string_set', b'string_set', 'timestamp', b'timestamp', 'value', b'value']) -> builtins.bool:
+    def HasField(self, field_name: typing.Literal['bool', b'bool', 'file_ref', b'file_ref', 'float64', b'float64', 'int64', b'int64', 'string', b'string', 'string_set', b'string_set', 'timestamp', b'timestamp', 'value', b'value']) -> builtins.bool:
         ...
 
-    def ClearField(self, field_name: typing.Literal['bool', b'bool', 'float64', b'float64', 'int64', b'int64', 'string', b'string', 'string_set', b'string_set', 'timestamp', b'timestamp', 'value', b'value']) -> None:
+    def ClearField(self, field_name: typing.Literal['bool', b'bool', 'file_ref', b'file_ref', 'float64', b'float64', 'int64', b'int64', 'string', b'string', 'string_set', b'string_set', 'timestamp', b'timestamp', 'value', b'value']) -> None:
         ...
 
-    def WhichOneof(self, oneof_group: typing.Literal['value', b'value']) -> typing.Literal['float64', 'int64', 'bool', 'string', 'timestamp', 'string_set'] | None:
+    def WhichOneof(self, oneof_group: typing.Literal['value', b'value']) -> typing.Literal['float64', 'int64', 'bool', 'string', 'timestamp', 'string_set', 'file_ref'] | None:
         ...
 global___Value = Value