✨ Use field notation

bananalyzer/data/schemas.py

@@ -96,5 +96,5 @@ def set_goal_if_fetch_id_provided(cls, values: Dict[str, Any]) -> Dict[str, Any]
         if goal is not None:
             raise ValueError("goal must not be provided if fetch_id is provided")
 
-        values["goal"] = get_fetch_schema(fetch_id).model_json_schema()
+        values["goal"] = get_fetch_schema(fetch_id).model_fields
         return values

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bananalyzer"
-version = "0.6.1"
+version = "0.6.2"
 description = "Open source AI Agent evaluation framework for web tasks 🐒🍌"
 authors = ["asim-shrestha <[email protected]>"]
 readme = "README.md"

tests/test_example_eval.py

@@ -149,5 +149,5 @@ def test_fetch_with_fetch_id_and_goal_should_raise_validation_error() -> None:
 def test_fetch_with_fetch_id_and_no_goal_sets_default_goal() -> None:
     example_data = create_default_example({"fetch_id": "contact", "goal": None})
     example = Example(**example_data)
-    print(get_fetch_schema("contact").model_json_schema())
-    assert example.goal == get_fetch_schema("contact").model_json_schema()
+    print(get_fetch_schema("contact").model_fields)
+    assert example.goal == get_fetch_schema("contact").model_fields