fix: use Mapping for dict params, Any for ValidatedParameters

py/src/braintrust/framework2.py

@@ -1,6 +1,6 @@
 import dataclasses
 import json
-from collections.abc import Callable, Sequence
+from collections.abc import Callable, Mapping, Sequence
 from typing import Any, overload
 
 import slugify
@@ -127,7 +127,7 @@ class CodeParameters:
     description: str | None
     schema: EvalParameters
     if_exists: IfExists | None
-    metadata: dict[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
 
     def to_function_definition(self, if_exists: IfExists | None, project_ids: ProjectIdCache) -> dict[str, Any]:
         schema = parameters_to_json_schema(self.schema)
@@ -352,7 +352,7 @@ def create(
         slug: str | None = None,
         description: str | None = None,
         if_exists: IfExists | None = None,
-        metadata: dict[str, Any] | None = None,
+        metadata: Mapping[str, Any] | None = None,
     ) -> EvalParameters:
         if slug is None or len(slug) == 0:
             slug = slugify.slugify(name)
@@ -408,7 +408,7 @@ def create(
         model: str,
         params: ModelParams | None = None,
         use_cot: bool,
-        choice_scores: dict[str, float],
+        choice_scores: Mapping[str, float],
     ) -> CodePrompt: ...
 
     # LLM scorer with messages.
@@ -426,7 +426,7 @@ def create(
         model: str,
         params: ModelParams | None = None,
         use_cot: bool,
-        choice_scores: dict[str, float],
+        choice_scores: Mapping[str, float],
     ) -> CodePrompt: ...
 
     def create(
@@ -448,7 +448,7 @@ def create(
         model: str | None = None,
         params: ModelParams | None = None,
         use_cot: bool | None = None,
-        choice_scores: dict[str, float] | None = None,
+        choice_scores: Mapping[str, float] | None = None,
     ) -> CodeFunction | CodePrompt:
         """Creates a scorer.
 

py/src/braintrust/parameters.py

@@ -35,7 +35,7 @@ class ModelParameter(TypedDict):
 
 
 JSONValue = None | bool | int | float | str | list["JSONValue"] | dict[str, "JSONValue"]
-ValidatedParameters = dict[str, object]
+ValidatedParameters = dict[str, Any]
 ParameterSchema = PromptParameter | ModelParameter | type[object] | None
 EvalParameters = Mapping[str, ParameterSchema]
 ParametersSchema = dict[str, Any]