assistant.php
1 year ago
assistfeedback.php
1 year ago
base.php
1 year ago
droppedfile.php
1 year ago
embed.php
1 year ago
feedback.php
1 year ago
function.php
1 year ago
image.php
1 year ago
parameter.php
2 years ago
text.php
1 year ago
transcribe.php
1 year ago
function.php
149 lines
| 1 | <?php |
| 2 | |
| 3 | class Meow_MWAI_Query_Function { |
| 4 | public string $name; |
| 5 | public string $description; |
| 6 | public array $parameters; |
| 7 | public string $type; // 'code-engine', etc... |
| 8 | public string $target; // 'server' or 'client' |
| 9 | public ?string $id; |
| 10 | |
| 11 | public function __construct( string $name, string $description, |
| 12 | array $parameters = [], string $type = null, string $id = null, string $target = null ) { |
| 13 | // $name: The name of the function to be called. |
| 14 | // Must be a-z, A-Z, 0-9, or contain underscores and dashes, with a maximum length of 64. |
| 15 | if ( !preg_match( '/^[a-zA-Z0-9_-]{1,64}$/', $name ) ) { |
| 16 | throw new InvalidArgumentException( "AI Engine: Invalid function name ($name) for Meow_MWAI_Query_Function." ); |
| 17 | } |
| 18 | |
| 19 | foreach ( $parameters as $parameter ) { |
| 20 | if ( !( $parameter instanceof Meow_MWAI_Query_Parameter ) ) { |
| 21 | throw new InvalidArgumentException( "AI Engine: Invalid parameter for Meow_MWAI_Query_Function." ); |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | $this->name = $name; |
| 26 | $this->description = $description; |
| 27 | $this->parameters = $parameters; |
| 28 | $this->type = $type ?? 'manual'; |
| 29 | $this->id = $id; |
| 30 | $this->target = $target ?? 'server'; |
| 31 | } |
| 32 | |
| 33 | public function serializeForOpenAI() { |
| 34 | // Initialize the base structure with name and description |
| 35 | $json = [ 'name' => $this->name, 'description' => $this->description ]; |
| 36 | |
| 37 | // Check if parameters are set and not empty |
| 38 | if ( !empty( $this->parameters ) ) { |
| 39 | $properties = []; |
| 40 | $required = []; |
| 41 | |
| 42 | // Loop through each parameter to construct the properties object |
| 43 | foreach ( $this->parameters as $parameter ) { |
| 44 | $properties[$parameter->name] = [ |
| 45 | 'type' => $parameter->type, // Assuming each parameter has a 'type' attribute |
| 46 | 'description' => $parameter->description, // Assuming each parameter has a 'description' attribute |
| 47 | ]; |
| 48 | |
| 49 | // If an enum is set for the parameter, include it |
| 50 | if ( isset($parameter->enum) ) { |
| 51 | $properties[$parameter->name]['enum'] = $parameter->enum; |
| 52 | } |
| 53 | |
| 54 | // If the parameter is required, add its name to the required array |
| 55 | if ( $parameter->required ) { |
| 56 | $required[] = $parameter->name; |
| 57 | } |
| 58 | } |
| 59 | |
| 60 | // Assemble the parameters part of the JSON |
| 61 | $json['parameters'] = [ |
| 62 | 'type' => 'object', |
| 63 | 'properties' => $properties, |
| 64 | 'required' => $required, |
| 65 | ]; |
| 66 | } |
| 67 | |
| 68 | return $json; |
| 69 | } |
| 70 | |
| 71 | public function serializeForAnthropic() { |
| 72 | $json = [ |
| 73 | 'name' => $this->name, |
| 74 | 'description' => $this->description, |
| 75 | 'input_schema' => [ |
| 76 | 'type' => 'object', |
| 77 | 'properties' => new stdClass() |
| 78 | ], |
| 79 | ]; |
| 80 | |
| 81 | if ( !empty( $this->parameters ) ) { |
| 82 | $properties = []; |
| 83 | $required = []; |
| 84 | foreach ( $this->parameters as $parameter ) |
| 85 | { |
| 86 | $properties[$parameter->name] = [ |
| 87 | 'type' => $parameter->type, |
| 88 | 'description' => $parameter->description, |
| 89 | ]; |
| 90 | if ( isset( $parameter->enum ) ) { |
| 91 | $properties[$parameter->name]['enum'] = $parameter->enum; |
| 92 | } |
| 93 | if ( $parameter->required ) { |
| 94 | $required[] = $parameter->name; |
| 95 | } |
| 96 | } |
| 97 | $json['input_schema']['properties'] = empty( $properties ) ? new stdClass() : $properties; |
| 98 | if ( !empty( $required ) ) { |
| 99 | $json['input_schema']['required'] = $required; |
| 100 | } |
| 101 | } |
| 102 | |
| 103 | return $json; |
| 104 | } |
| 105 | |
| 106 | public static function fromJson( array $json ): Meow_MWAI_Query_Function { |
| 107 | $funcName = $json['name']; |
| 108 | $funcDesc = $json['desc']; |
| 109 | $funcType = $json['type'] ?? null; |
| 110 | $funcId = $json['id'] ?? null; |
| 111 | $funcTarget = $json['target'] ?? null; |
| 112 | if ( $funcId === null && !empty( $json['snippetId'] ) ) { |
| 113 | $funcId = $json['snippetId']; |
| 114 | } |
| 115 | $args = []; |
| 116 | if ( !empty( $json['args'] ) ) { |
| 117 | foreach ( $json['args'] as $arg ) { |
| 118 | $name = ltrim( $arg['name'], '$' ); |
| 119 | $desc = $arg['desc'] ?? null; |
| 120 | $type = $arg['type'] ?? 'string'; |
| 121 | $required = $arg['required'] ?? false; |
| 122 | $args[] = new Meow_MWAI_Query_Parameter( $name, $desc, $type, $required ); |
| 123 | } |
| 124 | } |
| 125 | return new self( $funcName, $funcDesc, $args, $funcType, $funcId, $funcTarget ); |
| 126 | } |
| 127 | |
| 128 | public static function toJson( Meow_MWAI_Query_Function $function ): array { |
| 129 | $json = [ |
| 130 | 'name' => $function->name, |
| 131 | 'desc' => $function->description, |
| 132 | 'type' => $function->type, |
| 133 | 'id' => $function->id, |
| 134 | 'target' => $function->target, |
| 135 | 'args' => [], |
| 136 | ]; |
| 137 | |
| 138 | foreach ( $function->parameters as $param ) { |
| 139 | $json['args'][] = [ |
| 140 | 'name' => $param->name, |
| 141 | 'desc' => $param->description, |
| 142 | 'type' => $param->type, |
| 143 | 'required' => $param->required, |
| 144 | ]; |
| 145 | } |
| 146 | |
| 147 | return $json; |
| 148 | } |
| 149 | } |