catalogue

Documentation

Index

• func FilterFromChannelName(name string) map[string]string
• func FilterFromChannelTopic(topic string) map[string]string
• type Catalog
  • func NewCatalogue(elastic *elasticsearch.TypedClient, elasticIndex string, badgerDB *badger.DB, ...) *Catalog
  • func (cat *Catalog) CommandMagic() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandMore() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandResultSize() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandSearch() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandSearchKNN() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandSimilar() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandSimilarKNN() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) CommandText() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)
  • func (cat *Catalog) GetDocuments(identifier ...string) (map[string]*schema.UBSchema, error)
  • func (cat *Catalog) GetEmbedding(queryString string) (embedding []float32, resultErr error)
  • func (cat *Catalog) InitCommands(session *discord.Session) error
  • func (cat *Catalog) Query2Embedding(queryString string) (string, error)
  • func (cat *Catalog) Result2MessageEmbed(result *index.Result, stat *channelStatus) ([]*discordgo.MessageEmbed, error)
  • func (cat *Catalog) Search(queryString string, filter map[string]string, embedding []float32, ...) (*index.Result, error)
  • func (cat *Catalog) SearchKNN(filter map[string]string, embedding []float32, searchType SearchType, k int64, ...) (*index.Result, error)
• type SearchType

Functions

func FilterFromChannelName

func FilterFromChannelName(name string) map[string]string

func FilterFromChannelTopic

func FilterFromChannelTopic(topic string) map[string]string

Types

type Catalog

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

func NewCatalogue

func NewCatalogue(elastic *elasticsearch.TypedClient, elasticIndex string, badgerDB *badger.DB, openaiApiKey string, prefix string, logger zLogger.ZLogger) *Catalog

func (*Catalog) CommandMagic

func (cat *Catalog) CommandMagic() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandMore

func (cat *Catalog) CommandMore() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandResultSize

func (cat *Catalog) CommandResultSize() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandSearch

func (cat *Catalog) CommandSearch() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandSearchKNN

func (cat *Catalog) CommandSearchKNN() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandSimilar

func (cat *Catalog) CommandSimilar() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandSimilarKNN

func (cat *Catalog) CommandSimilarKNN() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) CommandText

func (cat *Catalog) CommandText() (cmdFunc discord.CommandCreate, appCmd *discordgo.ApplicationCommand)

func (*Catalog) GetDocuments

func (cat *Catalog) GetDocuments(identifier ...string) (map[string]*schema.UBSchema, error)

func (*Catalog) GetEmbedding

func (cat *Catalog) GetEmbedding(queryString string) (embedding []float32, resultErr error)

func (*Catalog) InitCommands

func (cat *Catalog) InitCommands(session *discord.Session) error

func (*Catalog) Query2Embedding

func (cat *Catalog) Query2Embedding(queryString string) (string, error)

func (*Catalog) Result2MessageEmbed

func (cat *Catalog) Result2MessageEmbed(result *index.Result, stat *channelStatus) ([]*discordgo.MessageEmbed, error)

func (*Catalog) Search

func (cat *Catalog) Search(queryString string, filter map[string]string, embedding []float32, searchType SearchType, from, num int64) (*index.Result, error)

func (*Catalog) SearchKNN

func (cat *Catalog) SearchKNN(filter map[string]string, embedding []float32, searchType SearchType, k int64, numCandidates int64) (*index.Result, error)

type SearchType

type SearchType int

const (
	SearchTypeSimple SearchType = iota
	SearchTypeEmbeddingMARC
	SearchTypeEmbeddingProse
	SearchTypeEmbeddingJSON
)