WIP

griptape-ai · Jul 30, 2024 · a2f7f6e · a2f7f6e
1 parent b28971e
commit a2f7f6e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/griptape/mixins/activity_mixin.py b/griptape/mixins/activity_mixin.py
@@ -20,7 +20,7 @@ class ActivityMixin:
 
     allowlist: Optional[list[str]] = field(default=None, kw_only=True)
     denylist: Optional[list[str]] = field(default=None, kw_only=True)
-    extra_schema_properties: dict[str, dict] = field(default=None, kw_only=True)
+    extra_schema_properties: Optional[dict[str, dict]] = field(default=None, kw_only=True)
 
     @allowlist.validator  # pyright: ignore[reportAttributeAccessIssue]
     def validate_allowlist(self, _: Attribute, allowlist: Optional[list[str]]) -> None:

diff --git a/tests/unit/mixins/test_activity_mixin.py b/tests/unit/mixins/test_activity_mixin.py
@@ -74,12 +74,12 @@ def test_enable_activities(self, tool):
         assert len(tool.activities()) > 0
 
     def test_extra_schema_properties(self):
-        tool = MockTool(
+        tool_with_extra_properties = MockTool(
             test_field="hello",
             test_int=5,
             extra_schema_properties={"test": {Literal("new_property"): str, Optional("optional_property"): int}},
         )
-        schema = tool.activity_schema(tool.test).json_schema("InputSchema")
+        schema = tool_with_extra_properties.activity_schema(tool_with_extra_properties.test).json_schema("InputSchema")
 
         assert schema == {
             "$id": "InputSchema",