feat: updated some bugs

Author: abhishekmadan30

docs/concepts/transforms.rst

@@ -105,15 +105,17 @@ about the state of the tasks at given points. Here is an example:
 
 .. code-block:: python
 
+   import msgspec
+   from typing import Optional
    from taskgraph.transforms.base import TransformSequence
-   from taskgraph.util.schema import Optional, Required, Schema
+   from taskgraph.util.schema import Schema
 
-   my_schema = Schema({
-       Required("foo"): str,
-       Optional("bar"): bool,
-   })
+   class MySchema(msgspec.Struct, kw_only=True):
+       foo: str                   # Required field
+       bar: Optional[bool] = None # Optional field
 
-   transforms.add_validate(my_schema)
+   transforms = TransformSequence()
+   transforms.add_validate(Schema(MySchema))
 
 In the above example, we can be sure that every task dict has a string field
 called ``foo``, and may or may not have a boolean field called ``bar``.

docs/tutorials/creating-a-task-graph.rst

@@ -136,14 +136,17 @@ comments for explanations):
 
 .. code-block:: python
 
-   from taskgraph.util.schema import Optional, Required, Schema
+   import msgspec
+   from typing import Optional
+   from taskgraph.util.schema import Schema
    from taskgraph.transforms.base import TransformSequence
 
-   # Define the schema. We use our schema validation to ensure correctness.
-   hello_description_schema = Schema({
-       Required("text"): str,
-       Optional("description"): str,
-   })
+   # Define the schema using msgspec for better type checking and performance.
+   class HelloDescriptionSchema(msgspec.Struct, kw_only=True):
+       text: str                          # Required field
+       description: Optional[str] = None  # Optional field
+
+   hello_description_schema = Schema(HelloDescriptionSchema)
 
    # Create a 'TransformSequence' instance. This class collects transform
    # functions to run later.

src/taskgraph/transforms/run/__init__.py

@@ -34,8 +34,8 @@ class FetchesSchema(msgspec.Struct, kw_only=True, rename="kebab", omit_defaults=
 
     artifact: str
     dest: TOptional[str] = None
-    extract: bool = True
-    verify_hash: bool = True
+    extract: bool = False
+    verify_hash: bool = False
 
 
 # When configuration using msgspec

src/taskgraph/transforms/task.py

@@ -26,8 +26,6 @@
 from taskgraph.util.hash import hash_path
 from taskgraph.util.keyed_by import evaluate_keyed_by
 from taskgraph.util.schema import (
-    Optional,
-    Required,
     Schema,
     resolve_keyed_by,
     validate_schema,
@@ -200,18 +198,23 @@ def payload_builder(name, schema):
     """
     Decorator for registering payload builders.
 
-    Supports both dict schemas and msgspec.Struct types.
+    Requires msgspec.Struct schema types for type safety and performance.
     """
-    # Handle msgspec schemas
-    if isinstance(schema, type) and issubclass(schema, msgspec.Struct):
-        # Wrap msgspec schema in our compatibility Schema class
-        schema = Schema(schema)
-    else:
-        # Traditional dict schema - extend it with required fields
-        schema = Schema({Required("implementation"): name, Optional("os"): str}).extend(
-            schema
+    # Ensure we're using msgspec schemas
+    if not (isinstance(schema, type) and issubclass(schema, msgspec.Struct)):
+        raise TypeError(
+            f"payload_builder requires msgspec.Struct schema, got {type(schema).__name__}. "
+            f"Please migrate to msgspec: class {name.title()}Schema(msgspec.Struct): ..."
         )
 
+    # Verify the schema has required fields
+    fields = {f.name for f in msgspec.structs.fields(schema)}
+    if "implementation" not in fields:
+        raise ValueError(f"Schema for {name} must include 'implementation' field")
+
+    # Wrap msgspec schema in our compatibility Schema class
+    schema = Schema(schema)
+
     def wrap(func):
         assert name not in payload_builders, f"duplicate payload builder name {name}"
         payload_builders[name] = PayloadBuilder(schema, func)  # type: ignore

template/{{cookiecutter.project_name}}/taskcluster/{{cookiecutter.project_slug}}_taskgraph/transforms/hello.py

@@ -1,11 +1,14 @@
+import msgspec
+
 from taskgraph.transforms.base import TransformSequence
-from taskgraph.util.schema import Required, Schema
+from taskgraph.util.schema import Schema
+
+
+class HelloSchema(msgspec.Struct, kw_only=True):
+    noun: str  # Required field
+
 
-HELLO_SCHEMA = Schema(
-    {
-        Required("noun"): str,
-    }
-)
+HELLO_SCHEMA = Schema(HelloSchema)
 
 transforms = TransformSequence()
 transforms.add_validate(HELLO_SCHEMA)