File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -8,6 +8,13 @@ import "shared.function.proto";
88import "shared.data_type.proto" ;
99import "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+
1118message 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+ }
Original file line number Diff line number Diff line change 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+ }
Load diff This file was deleted.
You can’t perform that action at this time.
0 commit comments