Merge pull request #305 from code0-tech/feat/ai-adjustments

Missing fields for AI services

Author: nicosammito

proto/velorum/velorum.generate.proto

@@ -7,20 +7,25 @@ package velorum;
 import "shared.function.proto";
 import "shared.data_type.proto";
 import "shared.flow.proto";
+import "shared.flow_type.proto";
 
 message PromptRequest {
   string prompt = 1;
   int64 project_id = 2;
-  repeated shared.FunctionDefinition functions = 3;
-  repeated shared.DefinitionDataType data_types = 4;
+  string model_identifier = 3;
+  repeated shared.FunctionDefinition functions = 4;
+  repeated shared.DefinitionDataType data_types = 5;
+  repeated shared.FlowType flow_types = 6;
 }
 
 message FlowRequest {
   string prompt = 1;
   shared.GenerationFlow flow = 2;
   int64 project_id = 3;
-  repeated shared.FunctionDefinition functions = 4;
-  repeated shared.DefinitionDataType data_types = 5;
+  string model_identifier = 4;
+  repeated shared.FunctionDefinition functions = 5;
+  repeated shared.DefinitionDataType data_types = 6;
+  repeated shared.FlowType flow_types = 7;
 }
 
 message FlowResponse {

proto/velorum/velorum.info.proto

@@ -17,9 +17,10 @@ message Model {
     GENERATE = 2;
   }
 
-  string name = 1;
-  float token_cost = 2; //e.g 1.0, 4.0
-  ModelType type = 3;
+  string identifier = 1; // e.g. "gpt-3.5-turbo"
+  string name = 2;
+  float token_cost = 3; //e.g 1.0, 4.0
+  repeated ModelType type = 4;
 }
 
 message ModelsResponse {