Merge branch 'main' of https://github.com/swissspidy/ai-command

Author: davidmosterd

ai-command.php

@@ -23,6 +23,13 @@
 
 	// TODO Register your tool here and add it to the collection
 
+	$image_tools = new ImageTools($client);
+
+	foreach($image_tools->get_tools() as $tool){
+		$tools->add($tool);
+	}
+
+
 	// WordPress REST calls
 	$rest_tools = new MapRESTtoMCP();
 

src/ImageTools.php

@@ -0,0 +1,45 @@
+<?php
+
+namespace WP_CLI\AiCommand;
+use WP_CLI\AiCommand\Entity\Tool;
+
+class ImageTools {
+
+	private $client;
+
+	public function __construct($client) {
+		$this->client = $client;
+	}
+
+	public function get_tools(){
+		return [
+				$this->image_generation_tool()
+		];
+	}
+
+	public function image_generation_tool() {
+		return new Tool(
+			[
+				'name' => 'generate_image',
+				'description' => 'Generates an image',
+				'inputSchema' => [
+					'type'       => 'object',
+					'properties' => [
+						'prompt' => [
+							'type'        => 'string',
+							'description' => 'The prompt for generating the image.',
+						],
+					],
+					'required'   => [ 'prompt' ],
+				],
+				'callable'    => function ( $params ) {
+
+					return $this->client->get_image_from_ai_service( $params['prompt'] );
+				},
+			]
+			);
+	}
+
+
+
+}

src/MCP/Client.php

@@ -107,7 +107,7 @@ static function () {
 					rename( $filename, $filename . '.' . $extension );
 					$filename .= '.' . $extension;
 
-					file_put_contents( $filename, $image_blob->get_binary_data() );
+					// file_put_contents( $filename, $image_blob->get_binary_data() );
 
 					$image_url = $filename;
 					$image_id = \WP_CLI\AiCommand\MediaManager::upload_to_media_library($image_url);