chore: remove temp fix for optional params support

src/toolbox_llamaindex/utils.py

@@ -97,9 +97,7 @@ def _schema_to_model(model_name: str, schema: list[ParameterSchema]) -> Type[Bas
         field_definitions[field.name] = cast(
             Any,
             (
-                # TODO: Remove the hardcoded optional types once optional fields
-                # are supported by Toolbox.
-                Optional[_parse_type(field)],
+                _parse_type(field),
                 Field(description=field.description),
             ),
         )

tests/test_utils.py

@@ -154,9 +154,9 @@ def test_schema_to_model(self):
         model = _schema_to_model("TestModel", schema)
         assert issubclass(model, BaseModel)
 
-        assert model.model_fields["param1"].annotation == Union[str, None]
+        assert model.model_fields["param1"].annotation == str
         assert model.model_fields["param1"].description == "Parameter 1"
-        assert model.model_fields["param2"].annotation == Union[int, None]
+        assert model.model_fields["param2"].annotation == int
         assert model.model_fields["param2"].description == "Parameter 2"
 
     def test_schema_to_model_empty(self):