Skip to content

Commit 40119cc

Browse files
committed
feat: update proto files to include model identifier and flow types in requests
1 parent 080e0aa commit 40119cc

2 files changed

Lines changed: 13 additions & 7 deletions

File tree

proto/velorum/velorum.generate.proto

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,20 +7,25 @@ package velorum;
77
import "shared.function.proto";
88
import "shared.data_type.proto";
99
import "shared.flow.proto";
10+
import "shared.flow_type.proto";
1011

1112
message PromptRequest {
1213
string prompt = 1;
1314
int64 project_id = 2;
14-
repeated shared.FunctionDefinition functions = 3;
15-
repeated shared.DefinitionDataType data_types = 4;
15+
string model_identifier = 3;
16+
repeated shared.FunctionDefinition functions = 4;
17+
repeated shared.DefinitionDataType data_types = 5;
18+
repeated shared.FlowType flow_types = 6;
1619
}
1720

1821
message FlowRequest {
1922
string prompt = 1;
2023
shared.GenerationFlow flow = 2;
2124
int64 project_id = 3;
22-
repeated shared.FunctionDefinition functions = 4;
23-
repeated shared.DefinitionDataType data_types = 5;
25+
string model_identifier = 4;
26+
repeated shared.FunctionDefinition functions = 5;
27+
repeated shared.DefinitionDataType data_types = 6;
28+
repeated shared.FlowType flow_types = 7;
2429
}
2530

2631
message FlowResponse {

proto/velorum/velorum.info.proto

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,10 @@ message Model {
1717
GENERATE = 2;
1818
}
1919

20-
string name = 1;
21-
float token_cost = 2; //e.g 1.0, 4.0
22-
ModelType type = 3;
20+
string identifier = 1; // e.g. "gpt-3.5-turbo"
21+
string name = 2;
22+
float token_cost = 3; //e.g 1.0, 4.0
23+
repeated ModelType type = 4;
2324
}
2425

2526
message ModelsResponse {

0 commit comments

Comments
 (0)