tokenizer

package

v0.0.35 Latest Latest Go to latest Published: Jul 10, 2023 License: MIT Imports: 3 Imported by: 0

Details

Valid go.mod file

The Go module system was introduced in Go 1.11 and is the official dependency management solution for Go.
Redistributable license

Redistributable licenses place minimal restrictions on how software can be used, modified, and redistributed.
Tagged version

Modules with tagged versions give importers more predictable builds.
Stable version

When a project reaches major version v1 it is considered stable.
Learn more about best practices

This section is empty.

This section is empty.

This section is empty.

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

func NewCohere(modelName string) (*Cohere, error)

func (t *Cohere) GetNumTokens(text string) (uint, error)

GetNumTokens returns the number of tokens in the provided text.

func (t *Cohere) GetNumTokensFromMessage(messages schema.ChatMessages) (uint, error)

GetNumTokensFromMessage returns the number of tokens in the provided chat messages.

func (t *Cohere) GetTokenIDs(text string) ([]uint, error)

GetTokenIDs returns the token IDs corresponding to the provided text.

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

func NewGPT2() (*GPT2, error)

func (t *GPT2) GetNumTokens(text string) (uint, error)

GetNumTokens returns the number of tokens in the provided text.

func (t *GPT2) GetNumTokensFromMessage(messages schema.ChatMessages) (uint, error)

GetNumTokensFromMessage returns the number of tokens in the provided chat messages.

func (t *GPT2) GetTokenIDs(text string) ([]uint, error)

GetTokenIDs returns the token IDs corresponding to the provided text.

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

func NewOpenAI(modelName string) *OpenAI

func (t *OpenAI) GetNumTokens(text string) (uint, error)

GetNumTokens returns the number of tokens in the provided text.

func (t *OpenAI) GetNumTokensFromMessage(messages schema.ChatMessages) (uint, error)

GetNumTokensFromMessage returns the number of tokens in the provided chat messages.

func (t *OpenAI) GetTokenIDs(text string) ([]uint, error)

GetTokenIDs returns the token IDs corresponding to the provided text.