feat: release advanced parsing options for rag files

Google APIs · copybara-github · commit a353d25bae4b · 2024-08-08T13:44:49.000-07:00
PiperOrigin-RevId: 660965686
diff --git a/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto b/google/cloud/aiplatform/v1beta1/vertex_rag_data.proto
@@ -187,6 +187,12 @@ message RagFileChunkingConfig {
   int32 chunk_overlap = 2;
 }
 
+// Specifies the parsing config for RagFiles.
+message RagFileParsingConfig {
+  // Whether to use advanced PDF parsing.
+  bool use_advanced_pdf_parsing = 2;
+}
+
 // Config for uploading RagFile.
 message UploadRagFileConfig {
   // Specifies the size and overlap of chunks after uploading RagFile.
@@ -217,6 +223,9 @@ message ImportRagFilesConfig {
   // Specifies the size and overlap of chunks after importing RagFiles.
   RagFileChunkingConfig rag_file_chunking_config = 4;
 
+  // Specifies the parsing config for RagFiles.
+  RagFileParsingConfig rag_file_parsing_config = 8;
+
   // Optional. The max number of queries per minute that this job is allowed to
   // make to the embedding model specified on the corpus. This value is specific
   // to this job and not shared across other import jobs. Consult the Quotas
