Merge branch 'main' into feature/create_post

Author: davidmosterd

src/MCP/Client.php

@@ -110,6 +110,7 @@ static function () {
 					file_put_contents( $filename, $image_blob->get_binary_data() );
 
 					$image_url = $filename;
+					$image_id = \WP_CLI\AiCommand\MediaManager::upload_to_media_library($image_url);
 				}
 
 				break;
@@ -120,6 +121,8 @@ static function () {
 				// TODO: Save as file or so.
 				break;
 			}
+
+			return $image_id || 'no image found';
 		}
 
 		// See https://github.com/felixarntz/ai-services/blob/main/docs/Accessing-AI-Services-in-PHP.md for further processing.
@@ -188,17 +191,17 @@ static function () {
 
 			\WP_CLI::debug( 'Making request...' . print_r( $contents, true ), 'ai' );
 
-			if ( $service->get_service_slug() === 'openai' ) {
-				$model = 'gpt-4o';
-			} else {
-				$model = 'gemini-2.0-flash';
-			}
+			// if ( $service->get_service_slug() === 'openai' ) {
+			// 	$model = 'gpt-4o';
+			// } else {
+			// 	$model = 'gemini-2.0-flash';
+			// }
 
 			$candidates = $service
 				->get_model(
 					[
 						'feature'          => 'text-generation',
-						'model'            => $model,
+						// 'model'            => $model,
 						                     'tools'        => $tools,
 							'capabilities' => [
 								AI_Capability::MULTIMODAL_INPUT,
@@ -227,9 +230,7 @@ static function () {
 				} elseif ( $part instanceof Function_Call_Part ) {
 					$function_result = $this->{$part->get_name()}( $part->get_args() );
 
-					// Capture the function name here
-					$function_name = $part->get_name();
-					echo "Output generated with the '$function_name' tool:\n"; // Log the function name
+					WP_CLI::debug( 'Calling Tool: ' . $part->get_name(), 'mcp_server' );
 
 					// Odd limitation of add_function_response_part().
 					if ( ! is_array( $function_result ) ) {

src/MediaManager.php

@@ -0,0 +1,38 @@
+<?php
+
+namespace WP_CLI\AiCommand;
+
+class MediaManager {
+
+	public static function upload_to_media_library($media_path) {
+		// Get WordPress upload directory information
+		$upload_dir = wp_upload_dir();
+
+		// Get the file name from the path
+		$file_name = basename($media_path);
+
+		// Copy file to the upload directory
+		$new_file_path = $upload_dir['path'] . '/' . $file_name;
+		copy($media_path, $new_file_path);
+
+		// Prepare attachment data
+		$wp_filetype = wp_check_filetype($file_name, null);
+		$attachment = array(
+			'post_mime_type' => $wp_filetype['type'],
+			'post_title'     => sanitize_file_name($file_name),
+			'post_content'   => '',
+			'post_status'    => 'inherit'
+		);
+
+		// Insert the attachment
+		$attach_id = wp_insert_attachment($attachment, $new_file_path);
+
+		// Generate attachment metadata
+		require_once(ABSPATH . 'wp-admin/includes/image.php');
+		$attach_data = wp_generate_attachment_metadata($attach_id, $new_file_path);
+		wp_update_attachment_metadata($attach_id, $attach_data);
+
+		return $attach_id;
+
+	}
+}