transformer

Documentation

Index

• type CohereRerank
  • func NewCohereRerank() *CohereRerank
  • func (c *CohereRerank) Rerank(ctx context.Context, query string, documents []document.Document) ([]document.Document, error)
  • func (c *CohereRerank) WithAPIKey(apiKey string) *CohereRerank
  • func (c *CohereRerank) WithMaxChunksPerDoc(maxChunksPerDoc int) *CohereRerank
  • func (c *CohereRerank) WithModel(model CohereRerankModel) *CohereRerank
  • func (c *CohereRerank) WithTopN(topN int) *CohereRerank
• type CohereRerankModel
• type DallE
  • func NewDallE() *DallE
  • func (d *DallE) AsFile(path string) *DallE
  • func (d *DallE) AsImage() *DallE
  • func (d *DallE) AsURL() *DallE
  • func (d *DallE) Transform(ctx context.Context, input string) (any, error)
  • func (d *DallE) WithClient(client *openai.Client) *DallE
  • func (d *DallE) WithImageSize(imageSize DallEImageSize) *DallE
• type DallEImageFormat
• type DallEImageOutput
• type DallEImageSize
• type VisualQuestionAnswering
  • func NewHFVisualQuestionAnswering(mediaFile string) *VisualQuestionAnswering
  • func (v *VisualQuestionAnswering) Transform(ctx context.Context, input string, all bool) (any, error)
  • func (v *VisualQuestionAnswering) WithImage(mediaFile string) *VisualQuestionAnswering
  • func (v *VisualQuestionAnswering) WithModel(model string) *VisualQuestionAnswering
  • func (v *VisualQuestionAnswering) WithToken(token string) *VisualQuestionAnswering
• type VisualQuestionAnsweringRequest
• type VisualQuestionAnsweringRequestInputs
• type VisualQuestionAnsweringResponse

Types

type CohereRerank

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

func NewCohereRerank

func NewCohereRerank() *CohereRerank

func (*CohereRerank) Rerank

func (c *CohereRerank) Rerank(ctx context.Context, query string, documents []document.Document) ([]document.Document, error)

func (*CohereRerank) WithAPIKey

func (c *CohereRerank) WithAPIKey(apiKey string) *CohereRerank

func (*CohereRerank) WithMaxChunksPerDoc

func (c *CohereRerank) WithMaxChunksPerDoc(maxChunksPerDoc int) *CohereRerank

func (*CohereRerank) WithModel

func (c *CohereRerank) WithModel(model CohereRerankModel) *CohereRerank

func (*CohereRerank) WithTopN

func (c *CohereRerank) WithTopN(topN int) *CohereRerank

type CohereRerankModel

type CohereRerankModel = model.RerankModel

const (
	CohereRerankScoreMetdataKey = "cohere-rerank-score"

	CohereRerankModelEnglishV20      CohereRerankModel = model.RerankModelEnglishV20
	CohereRerankModelMultilingualV20 CohereRerankModel = model.RerankModelMultilingualV20
)

type DallE

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

func NewDallE

func NewDallE() *DallE

func (*DallE) AsFile

func (d *DallE) AsFile(path string) *DallE

func (*DallE) AsImage

func (d *DallE) AsImage() *DallE

func (*DallE) AsURL

func (d *DallE) AsURL() *DallE

func (*DallE) Transform

func (d *DallE) Transform(ctx context.Context, input string) (any, error)

func (*DallE) WithClient

func (d *DallE) WithClient(client *openai.Client) *DallE

func (*DallE) WithImageSize

func (d *DallE) WithImageSize(imageSize DallEImageSize) *DallE

type DallEImageFormat

type DallEImageFormat string

const (
	DallEImageFormatURL   DallEImageFormat = "url"
	DallEImageFormatFile  DallEImageFormat = "file"
	DallEImageFormatImage DallEImageFormat = "image"
)

type DallEImageOutput

type DallEImageOutput any

type DallEImageSize

type DallEImageSize string

const (
	DallEImageSize256  DallEImageSize = openai.CreateImageSize256x256
	DallEImageSize512  DallEImageSize = openai.CreateImageSize512x512
	DallEImageSize1024 DallEImageSize = openai.CreateImageSize1024x1024
)

type VisualQuestionAnswering

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

func NewHFVisualQuestionAnswering

func NewHFVisualQuestionAnswering(mediaFile string) *VisualQuestionAnswering

func (*VisualQuestionAnswering) Transform

func (v *VisualQuestionAnswering) Transform(ctx context.Context, input string, all bool) (any, error)

func (*VisualQuestionAnswering) WithImage

func (v *VisualQuestionAnswering) WithImage(mediaFile string) *VisualQuestionAnswering

func (*VisualQuestionAnswering) WithModel

func (v *VisualQuestionAnswering) WithModel(model string) *VisualQuestionAnswering

func (*VisualQuestionAnswering) WithToken

func (v *VisualQuestionAnswering) WithToken(token string) *VisualQuestionAnswering

type VisualQuestionAnsweringRequest

type VisualQuestionAnsweringRequest struct {
	Inputs VisualQuestionAnsweringRequestInputs `json:"inputs"`
}

type VisualQuestionAnsweringRequestInputs

type VisualQuestionAnsweringRequestInputs struct {
	Image    string `json:"image"`
	Question string `json:"question"`
}

type VisualQuestionAnsweringResponse

type VisualQuestionAnsweringResponse struct {
	Score  float64 `json:"score"`
	Answer string  `json:"answer"`
}