modules
3 years ago
admin.php
3 years ago
ai.php
3 years ago
answer.php
3 years ago
api.php
3 years ago
core.php
3 years ago
init.php
3 years ago
openai.php
3 years ago
query.php
3 years ago
queryembed.php
3 years ago
queryimage.php
3 years ago
querytext.php
3 years ago
querytranscribe.php
3 years ago
rest.php
3 years ago
ui.php
3 years ago
core.php
287 lines
| 1 | <?php |
| 2 | |
| 3 | require_once( MWAI_PATH . '/vendor/autoload.php' ); |
| 4 | require_once( MWAI_PATH . '/constants/init.php' ); |
| 5 | |
| 6 | class Meow_MWAI_Core |
| 7 | { |
| 8 | public $admin = null; |
| 9 | public $is_rest = false; |
| 10 | public $is_cli = false; |
| 11 | public $site_url = null; |
| 12 | public $ai = null; |
| 13 | private $option_name = 'mwai_options'; |
| 14 | public $defaultChatbotParams = MWAI_CHATBOT_PARAMS; |
| 15 | |
| 16 | public function __construct() { |
| 17 | $this->site_url = get_site_url(); |
| 18 | $this->is_rest = MeowCommon_Helpers::is_rest(); |
| 19 | $this->is_cli = defined( 'WP_CLI' ) && WP_CLI; |
| 20 | $this->ai = new Meow_MWAI_AI( $this ); |
| 21 | add_action( 'plugins_loaded', array( $this, 'init' ) ); |
| 22 | } |
| 23 | |
| 24 | function init() { |
| 25 | global $mwai; |
| 26 | $mwai = new Meow_MWAI_API(); |
| 27 | if ( $this->is_rest ) { |
| 28 | new Meow_MWAI_Rest( $this ); |
| 29 | } |
| 30 | if ( is_admin() ) { |
| 31 | new Meow_MWAI_Admin( $this ); |
| 32 | new Meow_MWAI_Modules_Assistants( $this ); |
| 33 | } |
| 34 | else { |
| 35 | //new Meow_MWAI_UI( $this ); |
| 36 | if ( $this->get_option( 'shortcode_chat' ) ) { |
| 37 | new Meow_MWAI_Modules_Chatbot(); |
| 38 | } |
| 39 | } |
| 40 | |
| 41 | // Advanced core |
| 42 | if ( class_exists( 'MeowPro_MWAI_Core' ) ) { |
| 43 | new MeowPro_MWAI_Core( $this ); |
| 44 | } |
| 45 | } |
| 46 | |
| 47 | #region Roles & Capabilities |
| 48 | |
| 49 | function can_access_settings() { |
| 50 | return apply_filters( 'mwai_allow_setup', current_user_can( 'manage_options' ) ); |
| 51 | } |
| 52 | |
| 53 | function can_access_features() { |
| 54 | $editor_or_admin = current_user_can( 'editor' ) || current_user_can( 'administrator' ); |
| 55 | return apply_filters( 'mwai_allow_usage', $editor_or_admin ); |
| 56 | } |
| 57 | |
| 58 | #endregion |
| 59 | |
| 60 | #region Text-Related Helpers |
| 61 | |
| 62 | // Clean the text perfectly, resolve shortcodes, etc, etc. |
| 63 | function clean_text( $rawText = "" ) { |
| 64 | $text = strip_tags( $rawText ); |
| 65 | $text = strip_shortcodes( $text ); |
| 66 | $text = html_entity_decode( $text ); |
| 67 | $text = str_replace( array( "\r", "\n" ), "", $text ); |
| 68 | $sentences = preg_split( '/(?<=[.?!])(?=[a-zA-Z ])/', $text ); |
| 69 | foreach ( $sentences as $key => $sentence ) { |
| 70 | $sentences[$key] = trim( $sentence ); |
| 71 | } |
| 72 | $text = implode( " ", $sentences ); |
| 73 | $text = preg_replace( '/^[\pZ\pC]+|[\pZ\pC]+$/u', '', $text ); |
| 74 | return $text . " "; |
| 75 | } |
| 76 | |
| 77 | // Make sure there are no duplicate sentences, and keep the length under a maximum length. |
| 78 | function clean_sentences( $text, $maxLength = 1024 ) { |
| 79 | $sentences = preg_split( '/(?<=[.?!])(?=[a-zA-Z ])/', $text ); |
| 80 | $hashes = array(); |
| 81 | $uniqueSentences = array(); |
| 82 | $length = 0; |
| 83 | foreach ( $sentences as $sentence ) { |
| 84 | $sentence = preg_replace( '/^[\pZ\pC]+|[\pZ\pC]+$/u', '', $sentence ); |
| 85 | $hash = md5( $sentence ); |
| 86 | if ( !in_array( $hash, $hashes ) ) { |
| 87 | if ( $length + strlen( $sentence ) > $maxLength ) { |
| 88 | continue; |
| 89 | } |
| 90 | $hashes[] = $hash; |
| 91 | $uniqueSentences[] = $sentence; |
| 92 | $length += strlen( $sentence ); |
| 93 | } |
| 94 | } |
| 95 | $text = implode( " ", $uniqueSentences ); |
| 96 | $text = preg_replace( '/^[\pZ\pC]+|[\pZ\pC]+$/u', '', $text ); |
| 97 | return $text; |
| 98 | } |
| 99 | |
| 100 | function get_text_from_postId( $postId ) { |
| 101 | $post = get_post( $postId ); |
| 102 | if ( !$post ) { |
| 103 | return false; |
| 104 | } |
| 105 | $post->post_content = apply_filters( 'the_content', $post->post_content ); |
| 106 | $text = $this->clean_text( $post->post_content ); |
| 107 | $text = $this->clean_sentences( $text ); |
| 108 | return $text; |
| 109 | } |
| 110 | |
| 111 | function markdown_to_html( $content ) { |
| 112 | $Parsedown = new Parsedown(); |
| 113 | $content = $Parsedown->text( $content ); |
| 114 | return $content; |
| 115 | } |
| 116 | #endregion |
| 117 | |
| 118 | #region Users/Sessions Helpers |
| 119 | |
| 120 | function get_session_id() { |
| 121 | if ( isset( $_COOKIE['mwai_session_id'] ) ) { |
| 122 | return $_COOKIE['mwai_session_id']; |
| 123 | } |
| 124 | return "N/A"; |
| 125 | } |
| 126 | |
| 127 | // Get the UserID from the data, or from the current user |
| 128 | function get_user_id( $data = null ) { |
| 129 | if ( isset( $data ) && isset( $data['userId'] ) ) { |
| 130 | return (int)$data['userId']; |
| 131 | } |
| 132 | if ( is_user_logged_in() ) { |
| 133 | $current_user = wp_get_current_user(); |
| 134 | if ( $current_user->ID > 0 ) { |
| 135 | return $current_user->ID; |
| 136 | } |
| 137 | } |
| 138 | return null; |
| 139 | } |
| 140 | |
| 141 | function get_ip_address( $data = null ) { |
| 142 | if ( isset( $data ) && isset( $data['ip'] ) ) { |
| 143 | $data['ip'] = (string)$data['ip']; |
| 144 | } |
| 145 | else { |
| 146 | if ( isset( $_SERVER['REMOTE_ADDR'] ) ) { |
| 147 | $data['ip'] = sanitize_text_field( $_SERVER['REMOTE_ADDR'] ); |
| 148 | } |
| 149 | else if ( isset( $_SERVER['HTTP_CLIENT_IP'] ) ) { |
| 150 | $data['ip'] = sanitize_text_field( $_SERVER['HTTP_CLIENT_IP'] ); |
| 151 | } |
| 152 | else if ( isset( $_SERVER['HTTP_X_FORWARDED_FOR'] ) ) { |
| 153 | $data['ip'] = sanitize_text_field( $_SERVER['HTTP_X_FORWARDED_FOR'] ); |
| 154 | } |
| 155 | } |
| 156 | return $data['ip']; |
| 157 | } |
| 158 | |
| 159 | #endregion |
| 160 | |
| 161 | #region Other Helpers |
| 162 | |
| 163 | function isUrl( $url ) { |
| 164 | return strpos( $url, 'http' ) === 0 ? true : false; |
| 165 | } |
| 166 | |
| 167 | #endregion |
| 168 | |
| 169 | #region Usage & Costs |
| 170 | |
| 171 | public function record_tokens_usage( $model, $prompt_tokens, $completion_tokens = 0 ) { |
| 172 | if ( !is_numeric( $prompt_tokens ) ) { |
| 173 | throw new Exception( 'Record usage: prompt_tokens is not a number.' ); |
| 174 | } |
| 175 | if ( !is_numeric( $completion_tokens ) ) { |
| 176 | $completion_tokens = 0; |
| 177 | } |
| 178 | if ( !$model ) { |
| 179 | throw new Exception( 'Record usage: model is missing.' ); |
| 180 | } |
| 181 | $usage = $this->get_option( 'openai_usage' ); |
| 182 | $month = date( 'Y-m' ); |
| 183 | if ( !isset( $usage[$month] ) ) { |
| 184 | $usage[$month] = array(); |
| 185 | } |
| 186 | if ( !isset( $usage[$month][$model] ) ) { |
| 187 | $usage[$month][$model] = array( |
| 188 | 'prompt_tokens' => 0, |
| 189 | 'completion_tokens' => 0, |
| 190 | 'total_tokens' => 0 |
| 191 | ); |
| 192 | } |
| 193 | $usage[$month][$model]['prompt_tokens'] += $prompt_tokens; |
| 194 | $usage[$month][$model]['completion_tokens'] += $completion_tokens; |
| 195 | $usage[$month][$model]['total_tokens'] += $prompt_tokens + $completion_tokens; |
| 196 | $this->update_option( 'openai_usage', $usage ); |
| 197 | return [ |
| 198 | 'prompt_tokens' => $prompt_tokens, |
| 199 | 'completion_tokens' => $completion_tokens, |
| 200 | 'total_tokens' => $prompt_tokens + $completion_tokens |
| 201 | ]; |
| 202 | } |
| 203 | |
| 204 | public function record_images_usage( $model, $resolution, $images ) { |
| 205 | if ( !$model || !$resolution || !$images ) { |
| 206 | throw new Exception( 'Missing parameters for record_image_usage.' ); |
| 207 | } |
| 208 | $usage = $this->get_option( 'openai_usage' ); |
| 209 | $month = date( 'Y-m' ); |
| 210 | if ( !isset( $usage[$month] ) ) { |
| 211 | $usage[$month] = array(); |
| 212 | } |
| 213 | if ( !isset( $usage[$month][$model] ) ) { |
| 214 | $usage[$month][$model] = array( |
| 215 | 'resolution' => array(), |
| 216 | 'images' => 0 |
| 217 | ); |
| 218 | } |
| 219 | if ( !isset( $usage[$month][$model]['resolution'][$resolution] ) ) { |
| 220 | $usage[$month][$model]['resolution'][$resolution] = 0; |
| 221 | } |
| 222 | $usage[$month][$model]['resolution'][$resolution] += $images; |
| 223 | $usage[$month][$model]['images'] += $images; |
| 224 | $this->update_option( 'openai_usage', $usage ); |
| 225 | return [ |
| 226 | 'resolution' => $resolution, |
| 227 | 'images' => $images |
| 228 | ]; |
| 229 | } |
| 230 | |
| 231 | #endregion |
| 232 | |
| 233 | #region Options |
| 234 | function get_all_options() { |
| 235 | $options = get_option( $this->option_name, null ); |
| 236 | foreach ( MWAI_OPTIONS as $key => $value ) { |
| 237 | if ( !isset( $options[$key] ) ) { |
| 238 | $options[$key] = $value; |
| 239 | } |
| 240 | if ( $key === 'languages' ) { |
| 241 | // TODO: If we decide to make a set of options for languages, we can keep it in the settings |
| 242 | $options[$key] = MWAI_LANGUAGES; |
| 243 | $options[$key] = apply_filters( 'mwai_languages', $options[$key] ); |
| 244 | } |
| 245 | } |
| 246 | $options['shortcode_chat_default_params'] = MWAI_CHATBOT_PARAMS; |
| 247 | $options['default_limits'] = MWAI_LIMITS; |
| 248 | $options['openai_models'] = MWAI_OPENAI_MODELS; |
| 249 | return $options; |
| 250 | } |
| 251 | |
| 252 | // Validate and keep the options clean and logical. |
| 253 | function sanitize_options() { |
| 254 | $options = $this->get_all_options(); |
| 255 | $needs_update = false; |
| 256 | |
| 257 | // We can sanitize our future options here, let's always remember it. |
| 258 | // Now, it is empty... |
| 259 | |
| 260 | if ( $needs_update ) { |
| 261 | update_option( $this->option_name, $options, false ); |
| 262 | } |
| 263 | return $options; |
| 264 | } |
| 265 | |
| 266 | function update_options( $options ) { |
| 267 | if ( !update_option( $this->option_name, $options, false ) ) { |
| 268 | return false; |
| 269 | } |
| 270 | $options = $this->sanitize_options(); |
| 271 | return $options; |
| 272 | } |
| 273 | |
| 274 | function update_option( $option, $value ) { |
| 275 | $options = $this->get_all_options(); |
| 276 | $options[$option] = $value; |
| 277 | return $this->update_options( $options ); |
| 278 | } |
| 279 | |
| 280 | function get_option( $option, $default = null ) { |
| 281 | $options = $this->get_all_options(); |
| 282 | return $options[$option] ?? $default; |
| 283 | } |
| 284 | #endregion |
| 285 | } |
| 286 | |
| 287 | ?> |