Skip to content

Commit 45747a4

Browse files
committed
feat: add GenerateService and InfoService with corresponding request and response messages
1 parent b17d3a9 commit 45747a4

3 files changed

Lines changed: 43 additions & 28 deletions

File tree

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,13 @@ import "shared.function.proto";
88
import "shared.data_type.proto";
99
import "shared.flow.proto";
1010

11+
message PromptRequest {
12+
string prompt = 1;
13+
int64 project_id = 2;
14+
repeated shared.FunctionDefinition functions = 3;
15+
repeated shared.DefinitionDataType data_types = 4;
16+
}
17+
1118
message FlowRequest {
1219
string prompt = 1;
1320
shared.GenerationFlow flow = 2;
@@ -22,6 +29,8 @@ message FlowResponse {
2229
int64 usage = 3; // in tokens
2330
}
2431

25-
service FlowService {
32+
33+
service GenerateService {
34+
rpc Prompt (PromptRequest) returns (FlowResponse) {}
2635
rpc Flow (FlowRequest) returns (FlowResponse) {}
27-
}
36+
}

proto/velorum/velorum.info.proto

Lines changed: 32 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,32 @@
1+
syntax = "proto3";
2+
3+
option ruby_package = "Tucana::Velorum";
4+
5+
package velorum;
6+
7+
import "shared.function.proto";
8+
import "shared.data_type.proto";
9+
import "shared.flow.proto";
10+
11+
message ModelsRequest {}
12+
13+
message Model {
14+
enum ModelType {
15+
UNKNOWN = 0;
16+
EXPLAIN = 1;
17+
GENERATE = 2;
18+
}
19+
20+
string name = 1;
21+
float token_cost = 2; //e.g 1.0, 4.0
22+
ModelType type = 3;
23+
}
24+
25+
message ModelsResponse {
26+
repeated Model models = 1;
27+
}
28+
29+
30+
service InfoService {
31+
rpc Models (ModelsRequest) returns (ModelsResponse) {}
32+
}

proto/velorum/velorum.prompt.proto

Lines changed: 0 additions & 26 deletions
This file was deleted.

0 commit comments

Comments
 (0)