feat: add LLM parser proto to API

Google APIs · copybara-github · commit 2f81f777e4f0 · 2024-12-11T12:32:10.000-08:00
PiperOrigin-RevId: 705199740
diff --git a/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto b/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
@@ -440,13 +440,34 @@ message RagFileParsingConfig {
     int32 max_parsing_requests_per_min = 2;
   }
 
+  // Specifies the advanced parsing for RagFiles.
+  message LlmParser {
+    // The name of a LLM model used for parsing.
+    // Format: `gemini-1.5-pro-002`
+    string model_name = 1;
+
+    // The maximum number of requests the job is allowed to make to the
+    // LLM model per minute. Consult
+    // https://cloud.google.com/vertex-ai/generative-ai/docs/quotas
+    // and your document size to set an appropriate value here. If unspecified,
+    // a default value of 5000 QPM would be used.
+    int32 max_parsing_requests_per_min = 2;
+
+    // The prompt to use for parsing. If not specified, a default prompt will
+    // be used.
+    string custom_parsing_prompt = 3;
+  }
+
   // The parser to use for RagFiles.
   oneof parser {
     // The Advanced Parser to use for RagFiles.
     AdvancedParser advanced_parser = 3;
 
     // The Layout Parser to use for RagFiles.
     LayoutParser layout_parser = 4;
+
+    // The LLM Parser to use for RagFiles.
+    LlmParser llm_parser = 5;
   }
 
   // Whether to use advanced PDF parsing.
