Fabric/plugins/ai/dryrun/dryrun.go

package dryrun

import (
	"bytes"
	"context"
	"fmt"

	goopenai "github.com/sashabaranov/go-openai"

	"github.com/danielmiessler/fabric/common"
	"github.com/danielmiessler/fabric/plugins"
)

type Client struct {
	*plugins.PluginBase
}

func NewClient() *Client {
	return &Client{PluginBase: &plugins.PluginBase{Name: "DryRun"}}
}

func (c *Client) ListModels() ([]string, error) {
	return []string{"dry-run-model"}, nil
}

func (c *Client) SendStream(msgs []*goopenai.ChatCompletionMessage, opts *common.ChatOptions, channel chan string) error {
	output := "Dry run: Would send the following request:\n\n"

	for _, msg := range msgs {
		switch msg.Role {
		case goopenai.ChatMessageRoleSystem:
			output += fmt.Sprintf("System:\n%s\n\n", msg.Content)
		case goopenai.ChatMessageRoleAssistant:
			output += fmt.Sprintf("Assistant:\n%s\n\n", msg.Content)
		case goopenai.ChatMessageRoleUser:
            if msg.MultiContent != nil {
                output += "User:\n"
                for _, part := range msg.MultiContent {
                    output += fmt.Sprintf("  - Type: %s\n", part.Type)
                    if part.Type == goopenai.ChatMessagePartTypeImageURL {
                        output += fmt.Sprintf("    Image URL: %s\n", part.ImageURL.URL)
                    } else {
                        output += fmt.Sprintf("    Text: %s\n", part.Text)
                    }
                }
                output += "\n"
            } else {
                output += fmt.Sprintf("User:\n%s\n\n", msg.Content)
            }
		default:
			output += fmt.Sprintf("%s:\n%s\n\n", msg.Role, msg.Content)
		}
	}

	output += "Options:\n"
	output += fmt.Sprintf("Model: %s\n", opts.Model)
	output += fmt.Sprintf("Temperature: %f\n", opts.Temperature)
	output += fmt.Sprintf("TopP: %f\n", opts.TopP)
	output += fmt.Sprintf("PresencePenalty: %f\n", opts.PresencePenalty)
	output += fmt.Sprintf("FrequencyPenalty: %f\n", opts.FrequencyPenalty)
	if opts.ModelContextLength != 0 {
		output += fmt.Sprintf("ModelContextLength: %d\n", opts.ModelContextLength)
	}

	channel <- output
	close(channel)
	return nil
}

func (c *Client) Send(_ context.Context, msgs []*goopenai.ChatCompletionMessage, opts *common.ChatOptions) (string, error) {
	fmt.Println("Dry run: Would send the following request:")

	for _, msg := range msgs {
		switch msg.Role {
		case goopenai.ChatMessageRoleSystem:
			fmt.Printf("System:\n%s\n\n", msg.Content)
		case goopenai.ChatMessageRoleAssistant:
			fmt.Printf("Assistant:\n%s\n\n", msg.Content)
		case goopenai.ChatMessageRoleUser:
			if msg.MultiContent != nil {
				fmt.Println("User:")
				for _, part := range msg.MultiContent {
					fmt.Printf("  - Type: %s\n", part.Type)
					if part.Type == goopenai.ChatMessagePartTypeImageURL {
						fmt.Printf("    Image URL: %s\n", part.ImageURL.URL)
					} else {
						fmt.Printf("    Text: %s\n", part.Text)
					}
				}
				fmt.Println()
			} else {
				fmt.Printf("User:\n%s\n\n", msg.Content)
			}
		default:
			fmt.Printf("%s:\n%s\n\n", msg.Role, msg.Content)
		}
	}

	fmt.Println("Options:")
	fmt.Printf("Model: %s\n", opts.Model)
	fmt.Printf("Temperature: %f\n", opts.Temperature)
	fmt.Printf("TopP: %f\n", opts.TopP)
	fmt.Printf("PresencePenalty: %f\n", opts.PresencePenalty)
	fmt.Printf("FrequencyPenalty: %f\n", opts.FrequencyPenalty)
	if opts.ModelContextLength != 0 {
		fmt.Printf("ModelContextLength: %d\n", opts.ModelContextLength)
	}

	return "", nil
}

func (c *Client) Setup() error {
	return nil
}

func (c *Client) SetupFillEnvFileContent(_ *bytes.Buffer) {
	// No environment variables needed for dry run
}

func (c *Client) NeedsRawMode(modelName string) bool {
	return false
}