diff --git a/sdks/python/apache_beam/yaml/pipeline.schema.yaml b/sdks/python/apache_beam/yaml/pipeline.schema.yaml
index e784531d9be1..e6b10f06e9a6 100644
--- a/sdks/python/apache_beam/yaml/pipeline.schema.yaml
+++ b/sdks/python/apache_beam/yaml/pipeline.schema.yaml
@@ -159,7 +159,10 @@ properties:
       # These are the only top-level properties defined in pipeline.
       - type: object
         properties:
-          type: { const: chain }
+          type:
+            oneOf:
+              - { const: composite }
+              - { const: chain }
           windowing:
             $ref: '#/$defs/windowing'
           transforms: {}
diff --git a/sdks/python/apache_beam/yaml/yaml_provider.py b/sdks/python/apache_beam/yaml/yaml_provider.py
index bf0231ee89d4..01e39b770c9b 100644
--- a/sdks/python/apache_beam/yaml/yaml_provider.py
+++ b/sdks/python/apache_beam/yaml/yaml_provider.py
@@ -765,6 +765,9 @@ def _affinity(self, other):
   def underlying_provider(self):
     return self._underlying_provider.underlying_provider()
 
+  def cache_artifacts(self):
+    self._underlying_provider.cache_artifacts()
+
 
 def parse_providers(provider_specs):
   providers = collections.defaultdict(list)