Copybara import of the project:

gcf-owl-bot[bot] · copybara-github · commit d64c414549e9 · 2025-05-07T13:29:53.000-07:00
-- a9fed09 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md COPYBARA_INTEGRATE_REVIEW=#5194 from googleapis:owl-bot-copy cd97261 PiperOrigin-RevId: 755982628
diff --git a/google/cloud/aiplatform_v1/services/migration_service/client.py b/google/cloud/aiplatform_v1/services/migration_service/client.py
@@ -281,23 +281,18 @@ def parse_dataset_path(path: str) -> Dict[str, str]:
     @staticmethod
     def dataset_path(
         project: str,
-        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
+        return "projects/{project}/datasets/{dataset}".format(
             project=project,
-            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(
-            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
-            path,
-        )
+        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
         return m.groupdict() if m else {}
 
     @staticmethod
diff --git a/google/cloud/aiplatform_v1/types/vertex_rag_data.py b/google/cloud/aiplatform_v1/types/vertex_rag_data.py
@@ -615,12 +615,21 @@ class RagFileTransformationConfig(proto.Message):
 class RagFileParsingConfig(proto.Message):
     r"""Specifies the parsing config for RagFiles.
 
+    This message has `oneof`_ fields (mutually exclusive fields).
+    For each oneof, at most one member field can be set at the same time.
+    Setting any member of the oneof automatically clears all other
+    members.
+
     .. _oneof: https://proto-plus-python.readthedocs.io/en/stable/fields.html#oneofs-mutually-exclusive-fields
 
     Attributes:
         layout_parser (google.cloud.aiplatform_v1.types.RagFileParsingConfig.LayoutParser):
             The Layout Parser to use for RagFiles.
 
+            This field is a member of `oneof`_ ``parser``.
+        llm_parser (google.cloud.aiplatform_v1.types.RagFileParsingConfig.LlmParser):
+            The LLM Parser to use for RagFiles.
+
             This field is a member of `oneof`_ ``parser``.
     """
 
@@ -656,12 +665,52 @@ class LayoutParser(proto.Message):
             number=2,
         )
 
+    class LlmParser(proto.Message):
+        r"""Specifies the advanced parsing for RagFiles.
+
+        Attributes:
+            model_name (str):
+                The name of a LLM model used for parsing. Format:
+
+                -  ``projects/{project_id}/locations/{location}/publishers/{publisher}/models/{model}``
+            max_parsing_requests_per_min (int):
+                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.
+            custom_parsing_prompt (str):
+                The prompt to use for parsing. If not
+                specified, a default prompt will be used.
+        """
+
+        model_name: str = proto.Field(
+            proto.STRING,
+            number=1,
+        )
+        max_parsing_requests_per_min: int = proto.Field(
+            proto.INT32,
+            number=2,
+        )
+        custom_parsing_prompt: str = proto.Field(
+            proto.STRING,
+            number=3,
+        )
+
     layout_parser: LayoutParser = proto.Field(
         proto.MESSAGE,
         number=4,
         oneof="parser",
         message=LayoutParser,
     )
+    llm_parser: LlmParser = proto.Field(
+        proto.MESSAGE,
+        number=5,
+        oneof="parser",
+        message=LlmParser,
+    )
 
 
 class UploadRagFileConfig(proto.Message):
diff --git a/tests/unit/gapic/aiplatform_v1/test_migration_service.py b/tests/unit/gapic/aiplatform_v1/test_migration_service.py
@@ -5420,9 +5420,9 @@ def test_parse_dataset_path():
 
 
 def test_dataset_path():
-    project = "scallop"
-    location = "abalone"
-    dataset = "squid"
+    project = "squid"
+    location = "clam"
+    dataset = "whelk"
     expected = "projects/{project}/locations/{location}/datasets/{dataset}".format(
         project=project,
         location=location,
@@ -5446,22 +5446,19 @@ def test_parse_dataset_path():
 
 
 def test_dataset_path():
-    project = "oyster"
-    location = "nudibranch"
-    dataset = "cuttlefish"
-    expected = "projects/{project}/locations/{location}/datasets/{dataset}".format(
+    project = "cuttlefish"
+    dataset = "mussel"
+    expected = "projects/{project}/datasets/{dataset}".format(
         project=project,
-        location=location,
         dataset=dataset,
     )
-    actual = MigrationServiceClient.dataset_path(project, location, dataset)
+    actual = MigrationServiceClient.dataset_path(project, dataset)
     assert expected == actual
 
 
 def test_parse_dataset_path():
     expected = {
-        "project": "mussel",
-        "location": "winkle",
+        "project": "winkle",
         "dataset": "nautilus",
     }
     path = MigrationServiceClient.dataset_path(**expected)
