Skip to content

Commit 02a2240

Browse files
authored
chore: Update model and client - s3 support update (#95)
* regenerate the client * regenerate protobuf model * Update the model --------- Co-authored-by: Michał Sośnicki <michal.sosnicki@neptune.ai>
1 parent 1848394 commit 02a2240

29 files changed

+2279
-1456
lines changed

codegen/proto/neptune_pb/ingest/v1/common.proto

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,16 @@ message Owner {
144144
}
145145
}
146146

147+
message OffsetMetadata {
148+
// System-set. Version of configuration used by kafka for consumed topic (number of partitions, etc.).
149+
// Used with `message_offset` field to create robust ordering of operations
150+
optional uint32 broker_version = 1;
151+
152+
// System-set. Message offset from kafka. Set on consumption from Kafka.
153+
// Used with `message_offset` field to create robust ordering of operations
154+
optional uint64 message_offset = 2;
155+
}
156+
147157
// CreateRun can be used to create a new run. This can be done in two ways:
148158
// 1. Create a new run with no inherited state. You may specify a new run family that will be
149159
// inherited by future runs forking from this one, otherwise the new family will be selected by the server.
@@ -164,7 +174,7 @@ message Run {
164174

165175
// Optional. ForkPoint is used to identify the exact point in the parent history from which the new run continues.
166176
// If not specified, the new run will start with no inherited state.
167-
ForkPoint fork_point = 2;
177+
optional ForkPoint fork_point = 2;
168178

169179
// Specifies Family for the new run. Run Family is used to group forking runs that share common ancestry.
170180
// By default, the new forking run will be in the same family as the parent run.
@@ -192,6 +202,9 @@ message Run {
192202
// - offline mode data generation,
193203
// - data migrations of historical runs
194204
optional google.protobuf.Timestamp start_processing_time = 11;
205+
206+
// System-set. Object contains infromation to provide robust ordering of operations.
207+
optional OffsetMetadata offset_metadata = 12;
195208
}
196209

197210
// Preview allows to control whether values are preview or committed.
@@ -218,10 +231,10 @@ message Preview {
218231
message UpdateRunSnapshot {
219232
// Optional. Step value within the run. If not set, it will default to next full step of the run
220233
// (highest step across step values).
221-
Step step = 1;
234+
optional Step step = 1;
222235

223236
// Timestamp field common to all included operations
224-
google.protobuf.Timestamp timestamp = 2;
237+
optional google.protobuf.Timestamp timestamp = 2;
225238

226239
// Preview allows to control whether values are preview or committed.
227240
// When not specified, the values are considered committed.
@@ -269,6 +282,9 @@ message UpdateRunSnapshot {
269282
// - data migrations of historical runs,
270283
// - slow metric computation
271284
optional google.protobuf.Timestamp start_processing_time = 11;
285+
286+
// System-set. Object contains infromation to provide robust ordering of operations.
287+
optional OffsetMetadata offset_metadata = 12;
272288
}
273289

274290
// UpdateSnapshots updates fields for a given step. Used to update current state of the run in a single step.

codegen/proto/neptune_pb/ingest/v1/pub/client.proto

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,3 +28,13 @@ message SubmitResponse {
2828
string request_id = 1; // wire-compatibility with RequestId. Will be deprecated
2929
repeated string request_ids = 2;
3030
}
31+
32+
message StatusCheck {
33+
string project = 1;
34+
RequestId request_id = 2;
35+
}
36+
37+
message BulkStatusCheck {
38+
string project = 2;
39+
repeated RequestId ids = 1; // wire-compatibility with RequestIdList
40+
}

codegen/proto/neptune_pb/ingest/v1/pub/ingest.proto

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ message RunOperation {
1919
string run_id = 2;
2020

2121
// Optional. Will create project if it doesn't yet exist. This operation is idempotent.
22-
bool create_missing_project = 3;
22+
optional bool create_missing_project = 3;
2323

2424
// Required. Operation to be performed on the run.
2525
oneof operation {
@@ -35,5 +35,5 @@ message RunOperation {
3535
}
3636
// API Key used for authorization of operations.
3737
// See https://docs.neptune.ai/setup/setting_api_token/ for more information on how to obtain an API Key.
38-
bytes api_key = 12;
38+
optional bytes api_key = 12;
3939
}

0 commit comments

Comments
 (0)