@@ -52,6 +52,52 @@ message ProcessorVersion {
5252 }];
5353 }
5454
55+ // Information about Generative AI model-based processor versions.
56+ message GenAiModelInfo {
57+ // Information for a pretrained Google-managed foundation model.
58+ message FoundationGenAiModelInfo {
59+ // Whether finetuning is allowed for this base processor version.
60+ bool finetuning_allowed = 1 ;
61+
62+ // The minimum number of labeled documents in the training dataset
63+ // required for finetuning.
64+ int32 min_train_labeled_documents = 2 ;
65+ }
66+
67+ // Information for a custom Generative AI model created by the user. These
68+ // are created with `Create New Version` in either the `Call foundation
69+ // model` or `Fine tuning` tabs.
70+ message CustomGenAiModelInfo {
71+ // The type of custom model created by the user.
72+ enum CustomModelType {
73+ // The model type is unspecified.
74+ CUSTOM_MODEL_TYPE_UNSPECIFIED = 0 ;
75+
76+ // The model is a versioned foundation model.
77+ VERSIONED_FOUNDATION = 1 ;
78+
79+ // The model is a finetuned foundation model.
80+ FINE_TUNED = 2 ;
81+ }
82+
83+ // The type of custom model created by the user.
84+ CustomModelType custom_model_type = 1 ;
85+
86+ // The base processor version ID for the custom model.
87+ string base_processor_version_id = 2 ;
88+ }
89+
90+ // The processor version is either a pretrained Google-managed foundation
91+ // model or a custom Generative AI model created by the user.
92+ oneof model_info {
93+ // Information for a pretrained Google-managed foundation model.
94+ FoundationGenAiModelInfo foundation_gen_ai_model_info = 1 ;
95+
96+ // Information for a custom Generative AI model created by the user.
97+ CustomGenAiModelInfo custom_gen_ai_model_info = 2 ;
98+ }
99+ }
100+
55101 // The possible states of the processor version.
56102 enum State {
57103 // The processor version is in an unspecified state.
@@ -134,6 +180,11 @@ message ProcessorVersion {
134180
135181 // Output only. Reserved for future use.
136182 bool satisfies_pzi = 17 [(google.api.field_behavior ) = OUTPUT_ONLY ];
183+
184+ // Output only. Information about Generative AI model-based processor
185+ // versions.
186+ GenAiModelInfo gen_ai_model_info = 18
187+ [(google.api.field_behavior ) = OUTPUT_ONLY ];
137188}
138189
139190// Contains the alias and the aliased resource name of processor version.
0 commit comments