assistants.php
3 years ago
chatbot.php
2 years ago
chatbot_legacy.php
2 years ago
discussions.php
2 years ago
security.php
3 years ago
discussions.php
298 lines
| 1 | <?php |
| 2 | |
| 3 | class Meow_MWAI_Modules_Discussions { |
| 4 | private $wpdb = null; |
| 5 | private $core = null; |
| 6 | private $table_chats = null; |
| 7 | private $db_check = false; |
| 8 | private $namespace_admin = 'mwai/v1'; |
| 9 | private $namespace_ui = 'mwai-ui/v1'; |
| 10 | |
| 11 | public function __construct() { |
| 12 | global $wpdb; |
| 13 | global $mwai_core; |
| 14 | $this->core = $mwai_core; |
| 15 | $this->wpdb = $wpdb; |
| 16 | $this->table_chats = $wpdb->prefix . 'mwai_chats'; |
| 17 | |
| 18 | if ( $this->core->get_option( 'shortcode_chat_discussions' ) ) { |
| 19 | add_filter( 'mwai_chatbot_reply', [ $this, 'chatbot_reply' ], 10, 4 ); |
| 20 | add_action( 'rest_api_init', [ $this, 'rest_api_init' ] ); |
| 21 | } |
| 22 | } |
| 23 | |
| 24 | public function rest_api_init() { |
| 25 | |
| 26 | // Admin |
| 27 | register_rest_route( $this->namespace_admin, '/discussions/list', [ |
| 28 | 'methods' => 'POST', |
| 29 | 'callback' => [ $this, 'rest_discussions_list' ], |
| 30 | 'permission_callback' => [ $this->core, 'can_access_settings' ], |
| 31 | ] ); |
| 32 | register_rest_route( $this->namespace_admin, '/discussions/delete', [ |
| 33 | 'methods' => 'POST', |
| 34 | 'callback' => [ $this, 'rest_discussions_delete' ], |
| 35 | 'permission_callback' => [ $this->core, 'can_access_settings' ], |
| 36 | ] ); |
| 37 | |
| 38 | // UI |
| 39 | register_rest_route( $this->namespace_ui, '/discussions/list', [ |
| 40 | 'methods' => 'POST', |
| 41 | 'callback' => [ $this, 'rest_discussions_ui_list' ], |
| 42 | 'permission_callback' => '__return_true' |
| 43 | ] ); |
| 44 | } |
| 45 | |
| 46 | function rest_discussions_list( $request ) { |
| 47 | try { |
| 48 | $params = $request->get_json_params(); |
| 49 | $offset = $params['offset']; |
| 50 | $limit = $params['limit']; |
| 51 | $filters = $params['filters']; |
| 52 | $sort = $params['sort']; |
| 53 | $chats = $this->chats_query( [], $offset, $limit, $filters, $sort ); |
| 54 | return new WP_REST_Response([ 'success' => true, 'total' => $chats['total'], 'chats' => $chats['rows'] ], 200 ); |
| 55 | } |
| 56 | catch ( Exception $e ) { |
| 57 | return new WP_REST_Response([ 'success' => false, 'message' => $e->getMessage() ], 500 ); |
| 58 | } |
| 59 | } |
| 60 | |
| 61 | function rest_discussions_ui_list( $request ) { |
| 62 | try { |
| 63 | $params = $request->get_json_params(); |
| 64 | $offset = isset( $params['offset'] ) ? $params['offset'] : 0; |
| 65 | $limit = isset( $params['limit'] ) ? $params['limit'] : 10; |
| 66 | $botId = isset( $params['botId'] ) ? $params['botId'] : null; |
| 67 | $customId = isset( $params['customId'] ) ? $params['customId'] : null; |
| 68 | |
| 69 | if ( !is_null( $customId ) ) { |
| 70 | $botId = $customId; |
| 71 | } |
| 72 | |
| 73 | if ( is_null( $botId ) ) { |
| 74 | return new WP_REST_Response([ 'success' => false, 'message' => "Bot ID is required." ], 200 ); |
| 75 | } |
| 76 | |
| 77 | $userId = get_current_user_id(); |
| 78 | if ( !$userId ) { |
| 79 | return new WP_REST_Response([ 'success' => false, 'message' => "You need to be connected." ], 200 ); |
| 80 | } |
| 81 | $filters = [ |
| 82 | [ 'accessor' => 'user', 'value' => $userId ], |
| 83 | [ 'accessor' => 'botId', 'value' => $botId ], |
| 84 | ]; |
| 85 | $chats = $this->chats_query( [], $offset, $limit, $filters ); |
| 86 | return new WP_REST_Response([ 'success' => true, 'total' => $chats['total'], 'chats' => $chats['rows'] ], 200 ); |
| 87 | } |
| 88 | catch ( Exception $e ) { |
| 89 | return new WP_REST_Response([ 'success' => false, 'message' => $e->getMessage() ], 500 ); |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | function rest_discussions_delete( $request ) { |
| 94 | try { |
| 95 | $params = $request->get_json_params(); |
| 96 | $chatsIds = $params['chatIds']; |
| 97 | if ( is_array( $chatsIds ) ) { |
| 98 | if ( count( $chatsIds ) === 0 ) { |
| 99 | $this->wpdb->query( "TRUNCATE TABLE $this->table_chats" ); |
| 100 | } |
| 101 | foreach( $chatsIds as $chatId ) { |
| 102 | $this->wpdb->delete( $this->table_chats, [ 'chatId' => $chatId ] ); |
| 103 | } |
| 104 | } |
| 105 | return new WP_REST_Response([ 'success' => true ], 200 ); |
| 106 | } |
| 107 | catch ( Exception $e ) { |
| 108 | return new WP_REST_Response([ 'success' => false, 'message' => $e->getMessage() ], 500 ); |
| 109 | } |
| 110 | } |
| 111 | |
| 112 | // Get latest discussion for the given parameter |
| 113 | function get_discussion( $botId, $chatId ) { |
| 114 | $this->check_db(); |
| 115 | $chat = $this->wpdb->get_row( $this->wpdb->prepare( "SELECT * |
| 116 | FROM $this->table_chats |
| 117 | WHERE chatId = %s AND botId = %s", $chatId, $botId |
| 118 | ) ); |
| 119 | if ( $chat ) { |
| 120 | $chat->messages = json_decode( $chat->messages ); |
| 121 | return $chat; |
| 122 | } |
| 123 | return null; |
| 124 | } |
| 125 | |
| 126 | function chats_query( $chats = [], $offset = 0, $limit = null, $filters = null, $sort = null ) { |
| 127 | $this->check_db(); |
| 128 | $offset = !empty( $offset ) ? intval( $offset ) : 0; |
| 129 | $limit = !empty( $limit ) ? intval( $limit ) : 5; |
| 130 | $filters = !empty( $filters ) ? $filters : []; |
| 131 | $sort = !empty( $sort ) ? $sort : [ 'accessor' => 'updated', 'by' => 'desc' ]; |
| 132 | $query = "SELECT * FROM $this->table_chats"; |
| 133 | |
| 134 | // Filters |
| 135 | if ( is_array( $filters ) ) { |
| 136 | $where = array(); |
| 137 | foreach ( $filters as $filter ) { |
| 138 | if ( $filter['accessor'] === 'user' ) { |
| 139 | $value = esc_sql( $filter['value'] ); |
| 140 | if ( is_null( $value ) || $value === '' ) { |
| 141 | continue; |
| 142 | } |
| 143 | $isIP = filter_var( $value, FILTER_VALIDATE_IP ); |
| 144 | if ( $isIP ) { |
| 145 | $where[] = "ip = '{$value}'"; |
| 146 | } |
| 147 | else { |
| 148 | $where[] = "userId = '{$value}'"; |
| 149 | } |
| 150 | } |
| 151 | if ( $filter['accessor'] === 'botId' ) { |
| 152 | $value = esc_sql( $filter['value'] ); |
| 153 | if ( is_null( $value ) || $value === '' ) { |
| 154 | continue; |
| 155 | } |
| 156 | $where[] = "botId = '{$value}'"; |
| 157 | } |
| 158 | if ( $filter['accessor'] === 'preview' ) { |
| 159 | $value = $filter['value']; |
| 160 | if ( empty( $value ) ) { |
| 161 | continue; |
| 162 | } |
| 163 | $where[] = "messages LIKE '%{$value}%'"; |
| 164 | } |
| 165 | } |
| 166 | if ( count( $where ) > 0 ) { |
| 167 | $query .= " WHERE " . implode( " AND ", $where ); |
| 168 | } |
| 169 | } |
| 170 | |
| 171 | // Count based on this query |
| 172 | $chats['total'] = $this->wpdb->get_var( "SELECT COUNT(*) FROM ($query) AS t" ); |
| 173 | |
| 174 | // Order by |
| 175 | $query .= " ORDER BY " . esc_sql( $sort['accessor'] ) . " " . esc_sql( $sort['by'] ); |
| 176 | |
| 177 | // Limits |
| 178 | if ( $limit > 0 ) { |
| 179 | $query .= " LIMIT $offset, $limit"; |
| 180 | } |
| 181 | |
| 182 | $chats['rows'] = $this->wpdb->get_results( $query, ARRAY_A ); |
| 183 | return $chats; |
| 184 | } |
| 185 | |
| 186 | function chatbot_reply( $rawText, $query, $params, $extra ) { |
| 187 | global $mwai_core; |
| 188 | $userIp = $mwai_core->get_ip_address(); |
| 189 | $userId = $mwai_core->get_user_id(); |
| 190 | $botId = isset( $params['botId'] ) ? $params['botId'] : null; |
| 191 | $chatId = isset( $params['chatId'] ) ? $params['chatId'] : $query->session; |
| 192 | $customId = isset( $params['customId'] ) ? $params['customId'] : null; |
| 193 | if ( !empty( $customId ) ) { |
| 194 | $botId = $customId; |
| 195 | } |
| 196 | $newMessage = isset( $params['newMessage'] ) ? $params['newMessage'] : $query->prompt; |
| 197 | //$chatId = hash( 'sha256', $userIp . $userId . $clientChatId ); |
| 198 | $this->check_db(); |
| 199 | $chat = $this->wpdb->get_row( $this->wpdb->prepare( "SELECT * FROM $this->table_chats WHERE chatId = %s", $chatId ) ); |
| 200 | $extra = [ |
| 201 | 'embeddings' => isset( $extra['embeddings'] ) ? $extra['embeddings'] : null |
| 202 | ]; |
| 203 | if ( $chat ) { |
| 204 | $chat->messages = json_decode( $chat->messages ); |
| 205 | $chat->messages[] = [ |
| 206 | 'role' => 'user', |
| 207 | 'content' => $newMessage |
| 208 | ]; |
| 209 | $chat->messages[] = [ |
| 210 | 'role' => 'assistant', |
| 211 | 'content' => $rawText, |
| 212 | 'extra' => $extra |
| 213 | ]; |
| 214 | $chat->messages = json_encode( $chat->messages ); |
| 215 | $this->wpdb->update( $this->table_chats, [ |
| 216 | 'userId' => $userId, |
| 217 | 'messages' => $chat->messages, |
| 218 | 'updated' => date( 'Y-m-d H:i:s' ) |
| 219 | ], [ 'id' => $chat->id ] ); |
| 220 | } |
| 221 | else { |
| 222 | $chat = [ |
| 223 | 'userId' => $userId, |
| 224 | 'ip' => $userIp, |
| 225 | 'messages' => json_encode( [ |
| 226 | [ |
| 227 | 'role' => 'user', |
| 228 | 'content' => $newMessage |
| 229 | ], |
| 230 | [ |
| 231 | 'role' => 'assistant', |
| 232 | 'content' => $rawText, |
| 233 | 'extra' => $extra |
| 234 | ] |
| 235 | ] ), |
| 236 | 'extra' => json_encode( [ |
| 237 | 'session' => $query->session, |
| 238 | 'model' => $query->model, |
| 239 | 'temperature' => $query->temperature, |
| 240 | 'context' => $query->context, |
| 241 | ] ), |
| 242 | 'botId' => $botId, |
| 243 | 'chatId' => $chatId, |
| 244 | 'created' => date( 'Y-m-d H:i:s' ), |
| 245 | 'updated' => date( 'Y-m-d H:i:s' ) |
| 246 | ]; |
| 247 | $this->wpdb->insert( $this->table_chats, $chat ); |
| 248 | } |
| 249 | return $rawText; |
| 250 | } |
| 251 | |
| 252 | function check_db() { |
| 253 | if ( $this->db_check ) { |
| 254 | return true; |
| 255 | } |
| 256 | $this->db_check = !( strtolower( |
| 257 | $this->wpdb->get_var( "SHOW TABLES LIKE '$this->table_chats'" ) ) != strtolower( $this->table_chats ) |
| 258 | ); |
| 259 | if ( !$this->db_check ) { |
| 260 | $this->create_db(); |
| 261 | $this->db_check = !( strtolower( |
| 262 | $this->wpdb->get_var( "SHOW TABLES LIKE '$this->table_chats'" ) ) != strtolower( $this->table_chats ) |
| 263 | ); |
| 264 | } |
| 265 | |
| 266 | // LATER: REMOVE THIS AFTER SEPTEMBER 2023 |
| 267 | // Make sure the column "userId" and "ip "exist in the $this->table_chats table |
| 268 | $this->db_check = $this->db_check && $this->wpdb->get_var( "SHOW COLUMNS FROM $this->table_chats LIKE 'userId'" ); |
| 269 | if ( !$this->db_check ) { |
| 270 | $this->wpdb->query( "ALTER TABLE $this->table_chats ADD COLUMN userId BIGINT(20) NULL" ); |
| 271 | $this->wpdb->query( "ALTER TABLE $this->table_chats ADD COLUMN ip VARCHAR(64) NULL" ); |
| 272 | $this->wpdb->query( "ALTER TABLE $this->table_chats ADD COLUMN botId VARCHAR(64) NULL" ); |
| 273 | $this->db_check = true; |
| 274 | } |
| 275 | |
| 276 | return $this->db_check; |
| 277 | } |
| 278 | |
| 279 | function create_db() { |
| 280 | $charset_collate = $this->wpdb->get_charset_collate(); |
| 281 | $sqlLogs = "CREATE TABLE $this->table_chats ( |
| 282 | id BIGINT(20) NOT NULL AUTO_INCREMENT, |
| 283 | userId BIGINT(20) NULL, |
| 284 | ip VARCHAR(64) NULL, |
| 285 | messages TEXT NOT NULL NULL, |
| 286 | extra TEXT NOT NULL NULL, |
| 287 | botId VARCHAR(64) NULL, |
| 288 | chatId VARCHAR(64) NOT NULL, |
| 289 | created DATETIME NOT NULL, |
| 290 | updated DATETIME NOT NULL, |
| 291 | PRIMARY KEY (id), |
| 292 | INDEX chatId (chatId) |
| 293 | ) $charset_collate;"; |
| 294 | require_once( ABSPATH . 'wp-admin/includes/upgrade.php' ); |
| 295 | dbDelta( $sqlLogs ); |
| 296 | } |
| 297 | |
| 298 | } |