WP_AI_Client_Ability_Function_Resolver::execute_ability( WordPressAiClientToolsDTOFunctionCall $call ): WordPressAiClientToolsDTOFunctionResponse

Source

public function execute_ability( FunctionCall $call ): FunctionResponse {
	$function_name = $call->getName() ?? 'unknown';
	$function_id   = $call->getId() ?? 'unknown';

	if ( ! $this->is_ability_call( $call ) ) {
		return new FunctionResponse(
			$function_id,
			$function_name,
			array(
				'error' => __( 'Not an ability function call' ),
				'code'  => 'invalid_ability_call',
			)
		);
	}

	$ability_name = self::function_name_to_ability_name( $function_name );

	if ( ! isset( $this->allowed_abilities[ $ability_name ] ) ) {
		return new FunctionResponse(
			$function_id,
			$function_name,
			array(
				/* translators: %s: ability name */
				'error' => sprintf( __( 'Ability "%s" was not specified in the allowed abilities list.' ), $ability_name ),
				'code'  => 'ability_not_allowed',
			)
		);
	}

	$ability = wp_get_ability( $ability_name );

	if ( ! $ability instanceof WP_Ability ) {
		return new FunctionResponse(
			$function_id,
			$function_name,
			array(
				/* translators: %s: ability name */
				'error' => sprintf( __( 'Ability "%s" not found' ), $ability_name ),
				'code'  => 'ability_not_found',
			)
		);
	}

	$args   = $call->getArgs();
	$result = $ability->execute( ! empty( $args ) ? $args : null );

	if ( is_wp_error( $result ) ) {
		return new FunctionResponse(
			$function_id,
			$function_name,
			array(
				'error' => $result->get_error_message(),
				'code'  => $result->get_error_code(),
				'data'  => $result->get_error_data(),
			)
		);
	}

	return new FunctionResponse(
		$function_id,
		$function_name,
		$result
	);
}

View all references View on Trac View on GitHub