bot

Documentation

Index

• func Start(c config.BotConfiguration) error
• type InferenceInputs
• type InferenceParameters
• type InferenceQuery
• type InferenceResult

Functions

func Start

func Start(c config.BotConfiguration) error

Types

type InferenceInputs

type InferenceInputs struct {
	PastUserInputs     []string `json:"past_user_inputs"`
	GeneratedResponses []string `json:"generated_responses"`
	Text               string   `json:"text"`
}

type InferenceParameters

type InferenceParameters struct {
	TopK              float32 `json:"top_k"`
	TopP              float32 `json:"top_p"`
	Temperature       float32 `json:"temperature"`
	RepetitionPenalty float32 `json:"repetition_penalty"`
}

type InferenceQuery

type InferenceQuery struct {
	Inputs     InferenceInputs     `json:"inputs"`
	Parameters InferenceParameters `json:"parameters"`
}

type InferenceResult

type InferenceResult struct {
	GeneratedText string `json:"generated_text"`
}