Merge pull request #21 from SheepFromHeaven/main

README.md

@@ -80,6 +80,19 @@ $complete = $open_ai->complete([
 ]);
 ```
 
+## Images (Dall-E)
+
+Given a prompt, the model will return one or more generated images as urls or base64 encoded.
+
+ ```php
+$complete = $open_ai->image([
+    'prompt' => 'A cat drinking milk',
+    'n' => 1,
+    'size' => '256x256',
+    'response_format' => 'url',
+]);
+```
+
 ## Searches
 
 Given a query and a set of documents or labels, the model ranks each document based on its semantic similarity to the

src/OpenAi.php

@@ -34,6 +34,17 @@ public function complete($opts)
         return $this->sendRequest($url, 'POST', $opts);
     }
 
+    /**
+     * @param $opts
+     * @return bool|string
+     */
+    public function image($opts)
+    {
+        $url = Url::imageUrl();
+
+        return $this->sendRequest($url, 'POST', $opts);
+    }
+
     /**
      * @param $opts
      * @return bool|string

src/Url.php

@@ -97,4 +97,13 @@ public static function answersUrl(): string
     {
         return self::OPEN_AI_URL . "/answers";
     }
+
+    /**
+     * @param
+     * @return string
+     */
+    public static function imageUrl(): string
+    {
+        return self::OPEN_AI_URL . "/images/generations";
+    }
 }

tests/OpenAiTest.php

@@ -145,3 +145,12 @@
     $result = $open_ai->engine($engine_id);
     $this->assertStringContainsString($engine_id, $result);
 });
+
+it('should handle image', function () use ($open_ai) {
+  $result = $open_ai->image([
+    'prompt' => "a picture of a cat",
+    'n' => 1,
+    'size' => "256x256",
+  ]);
+  $this->assertStringContainsString('data', $result);
+});