fake

Documentation

Index

• type LLM
  • func NewFakeLLM(responses []string) *LLM
  • func (f *LLM) AddResponse(response string)
  • func (f *LLM) Call(ctx context.Context, prompt string, options ...llms.CallOption) (string, error)
  • func (f *LLM) GenerateContent(_ context.Context, _ []llms.MessageContent, _ ...llms.CallOption) (*llms.ContentResponse, error)
  • func (f *LLM) Reset()

Types

type LLM

type LLM struct {
	// contains filtered or unexported fields
}

func NewFakeLLM

func NewFakeLLM(responses []string) *LLM

func (*LLM) AddResponse

func (f *LLM) AddResponse(response string)

AddResponse adds a response to the list of responses.

func (*LLM) Call

func (f *LLM) Call(ctx context.Context, prompt string, options ...llms.CallOption) (string, error)

Call chama o modelo com uma única string de prompt.

func (*LLM) GenerateContent

func (f *LLM) GenerateContent(_ context.Context, _ []llms.MessageContent, _ ...llms.CallOption) (*llms.ContentResponse, error)

GenerateContent generate fake content.

func (*LLM) Reset

func (f *LLM) Reset()

Reset the index to 0.