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