glm_6b

Documentation

Index

• func NewGLM(log logger.Logger, uri string) llm.LLM
• type CompletionResult
• type GLM
  • func (o *GLM) Chat(ctx context.Context, stream bool, history []map[string]string, ...) (map[string]int, error)
  • func (o *GLM) Completion(ctx context.Context, prompt prompts.PromptTemplate, ...) ([]string, map[string]int, error)
  • func (o *GLM) GetAssistantModel() string
  • func (o *GLM) GetSystemModel() string
  • func (o *GLM) GetUserModel() string

Functions

func NewGLM

func NewGLM(log logger.Logger, uri string) llm.LLM

Types

type CompletionResult

type CompletionResult struct {
	Response string     `json:"response"`
	History  [][]string `json:"history"`
	Status   int        `json:"status"`
	Time     string     `json:"time"`
}

type GLM

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

func (*GLM) Chat

func (o *GLM) Chat(ctx context.Context, stream bool, history []map[string]string, answers chan<- map[string]string) (map[string]int, error)

todo: not supported

func (*GLM) Completion

func (o *GLM) Completion(ctx context.Context, prompt prompts.PromptTemplate, parameters map[string]string) ([]string, map[string]int, error)

func (*GLM) GetAssistantModel

func (o *GLM) GetAssistantModel() string

func (*GLM) GetSystemModel

func (o *GLM) GetSystemModel() string

func (*GLM) GetUserModel

func (o *GLM) GetUserModel() string