fix: Create schema for additional properties for object type params

Author: anubhav756

packages/toolbox-core/src/toolbox_core/protocol.py

@@ -18,6 +18,27 @@
 from pydantic import BaseModel
 
 
+class AdditionalPropertiesSchema(BaseModel):
+    """
+    Defines the value type for 'object' parameters.
+    """
+
+    type: str
+
+    def __get_type(self) -> Type:
+        """Converts the string type to a Python type."""
+        if self.type == "string":
+            return str
+        elif self.type == "integer":
+            return int
+        elif self.type == "float":
+            return float
+        elif self.type == "boolean":
+            return bool
+        else:
+            raise ValueError(f"Unsupported schema type: {self.type}")
+
+
 class ParameterSchema(BaseModel):
     """
     Schema for a tool parameter.
@@ -29,7 +50,7 @@ class ParameterSchema(BaseModel):
     description: str
     authSources: Optional[list[str]] = None
     items: Optional["ParameterSchema"] = None
-    additionalProperties: Optional[Union[bool, "ParameterSchema"]] = None
+    additionalProperties: Optional[Union[bool, "AdditionalPropertiesSchema"]] = None
 
     def __get_type(self) -> Type:
         base_type: Type
@@ -46,7 +67,7 @@ def __get_type(self) -> Type:
                 raise ValueError("Unexpected value: type is 'array' but items is None")
             base_type = list[self.items.__get_type()]  # type: ignore
         elif self.type == "object":
-            if isinstance(self.additionalProperties, ParameterSchema):
+            if isinstance(self.additionalProperties, AdditionalPropertiesSchema):
                 value_type = self.additionalProperties.__get_type()
                 base_type = dict[str, value_type]  # type: ignore
             else: