Updated the code

Author: djthorpe

README.md

@@ -405,15 +405,59 @@ Commands:
   agents       Return a list of agents
   models       Return a list of models
   tools        Return a list of tools
-  download     Download a model
+  download     Download a model (for Ollama)
   chat         Start a chat session
-  complete     Complete a prompt
+  complete     Complete a prompt, generate image or speech from text
   embedding    Generate an embedding
   version      Print the version of this tool
 
 Run "llm <command> --help" for more information on a command.
 ```
 
+### Prompt Completion
+
+To have the model respond to a prompt, you can use the `complete` command. For example, to
+have the model respond to the prompt "What is the capital of France?" using the `claude-3-5-haiku-20241022`
+model, you can use the following command:
+
+```bash
+llm complete "What is the capital of France?"
+```
+
+The first time you use the command use the ``--model`` flag to specify the model you want to use. Your
+choice of model will be remembered for subsequent completions.
+
+### Explain computer code
+
+To have the model explain a piece of computer code, you can pipe the code into the `complete` command.
+For example, to have the model explain the code in the file `example.go`, you can use the following command:
+
+```bash
+cat example.go | llm complete
+```
+
+### Caption an image
+
+To have the model generate a caption for an image, you can use the `complete` command with the `--file`
+flag. For example, to have the model generate a caption for the image in the file `example.jpg`, you can use
+the following command:
+
+```bash
+llm complete --model gpt-4o --file picture.png "Explain this image"
+```
+
+### Generate an image
+
+To have the model generate an image from a prompt, you can use the `complete` command with the `--format image`
+option. For example, to have the model generate an image from the prompt "A picture of a cat", you can use
+the following command:
+
+```bash
+llm complete --model dall-e-3 --format image "A picture of a cat"
+```
+
+It will write the file in the current working directory.
+
 ## Contributing & Distribution
 
 _This module is currently in development and subject to change_. Please do file

cmd/llm/main.go

@@ -77,10 +77,10 @@ type CLI struct {
 	Tools  ListToolsCmd  `cmd:"" help:"Return a list of tools"`
 
 	// Commands
-	Download  DownloadModelCmd `cmd:"" help:"Download a model"`
+	Download  DownloadModelCmd `cmd:"" help:"Download a model (for Ollama)"`
 	Chat      ChatCmd          `cmd:"" help:"Start a chat session"`
 	Chat2     Chat2Cmd         `cmd:"" help:"Start a chat session (2)"`
-	Complete  CompleteCmd      `cmd:"" help:"Complete a prompt"`
+	Complete  CompleteCmd      `cmd:"" help:"Complete a prompt, generate image or speech from text"`
 	Embedding EmbeddingCmd     `cmd:"" help:"Generate an embedding"`
 	Version   VersionCmd       `cmd:"" help:"Print the version of this tool"`
 }

examples/image_generate/main.go

@@ -0,0 +1,70 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+
+	"github.com/mutablelogic/go-llm"
+	"github.com/mutablelogic/go-llm/pkg/openai"
+)
+
+const (
+	model   = "dall-e-3"
+	size    = "1024x1024"
+	quality = "standard"
+	style   = "vivid"
+)
+
+func main() {
+	// Create a new OpenAI agent
+	agent, err := openai.New(os.Getenv("OPENAI_API_KEY"))
+	if err != nil {
+		panic(err)
+	}
+
+	// Check args
+	if len(os.Args) != 2 {
+		fmt.Println("Usage: image_generate <caption>")
+		os.Exit(-1)
+	}
+
+	// Get a model
+	model := agent.Model(context.TODO(), model)
+
+	// Get image
+	images, err := model.Completion(
+		context.TODO(),
+		os.Args[1],
+		llm.WithFormat("image"),
+		llm.WithQuality(quality),
+		llm.WithSize(size),
+		llm.WithStyle(style),
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	// Write out image(s)
+	for i := 0; i < images.Num(); i++ {
+		attachment := images.Attachment(i)
+		if attachment == nil || attachment.Filename() == "" {
+			continue
+		}
+
+		// Create a file
+		f, err := os.Create(attachment.Filename())
+		if err != nil {
+			panic(err)
+		}
+		defer f.Close()
+
+		// Write the image
+		if _, err := f.Write(attachment.Data()); err != nil {
+			panic(err)
+		} else {
+			fmt.Printf("%q written to %s\n", attachment.Caption(), attachment.Filename())
+		}
+	}
+
+}

pkg/openai/audio.go

@@ -0,0 +1 @@
+package openai