feat: support array for ToolParameter

devscast · Jan 16, 2025 · 9506324 · 9506324
1 parent 64aebe5
commit 9506324
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/Model/Completion/Tools/ToolParameter.php b/src/Model/Completion/Tools/ToolParameter.php
@@ -26,14 +26,18 @@
 
     public function __construct(
         public string $name,
-        public string $type,
+        public string|array $type,
         public ?string $description = null,
         public ?array $enum = null,
         public bool $required = false
     ) {
         Assert::notEmpty($name);
-        Assert::oneOf($type, self::SUPPORTED_TYPES);
         Assert::notEmpty($description);
+
+        match (\is_array($type)) {
+            true => Assert::allInArray($type, self::SUPPORTED_TYPES),
+            default => Assert::oneOf($type, self::SUPPORTED_TYPES)
+        };
     }
 
     public function definition(): array