fabric/vendors/dryrun/dryrun.go

package dryrun

import (
	"bytes"
	"context"
	"fmt"

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

type Client struct{}

func NewClient() *Client {
	return &Client{}
}

func (c *Client) GetName() string {
	return "DryRun"
}

func (c *Client) IsConfigured() bool {
	return true
}

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

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

func (c *Client) SendStream(msgs []*common.Message, 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 "system":
			output += fmt.Sprintf("System:\n%s\n\n", msg.Content)
		case "user":
			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)

	channel <- output
	close(channel)
	return nil
}

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

	for _, msg := range msgs {
		switch msg.Role {
		case "system":
			fmt.Printf("System:\n%s\n\n", msg.Content)
		case "user":
			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)

	return "", nil
}

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

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