llms

Documentation

Index

• Variables
• type LLM
• type LLMType
• type ModelParams
• type Response
• type UnknowLLM
  • func NewUnknowLLM() UnknowLLM
  • func (unknown UnknowLLM) Call(data []byte) (Response, error)
  • func (unknown UnknowLLM) Type() LLMType
  • func (unknown UnknowLLM) Validate(ctx context.Context, options ...langchainllms.CallOption) (Response, error)

Variables

var (
	ZhiPuAILite  string = "chatglm_lite"
	ZhiPuAIStd   string = "chatglm_std"
	ZhiPuAIPro   string = "chatglm_pro"
	ZhiPuAITurbo string = "chatglm_turbo"
)

var OpenAIModels = []string{"gpt-3.5", "gpt-3.5-turbo"}

var ZhiPuAIModels = []string{ZhiPuAILite, ZhiPuAIStd, ZhiPuAIPro, ZhiPuAITurbo}

Types

type LLM

type LLM interface {
	Type() LLMType
	Call([]byte) (Response, error)
	Validate(context.Context, ...langchainllms.CallOption) (Response, error)
}

type LLMType

type LLMType string

const (
	OpenAI    LLMType = "openai"
	ZhiPuAI   LLMType = "zhipuai"
	DashScope LLMType = "dashscope"
	Unknown   LLMType = "unknown"
)

type ModelParams

type ModelParams interface {
	Marshal() []byte
	Unmarshal([]byte) error
}

type Response

type Response interface {
	Type() LLMType
	String() string
	Bytes() []byte
	Unmarshal([]byte) error
}

type UnknowLLM

type UnknowLLM struct{}

func NewUnknowLLM

func NewUnknowLLM() UnknowLLM

func (UnknowLLM) Call

func (unknown UnknowLLM) Call(data []byte) (Response, error)

func (UnknowLLM) Type

func (unknown UnknowLLM) Type() LLMType

func (UnknowLLM) Validate

func (unknown UnknowLLM) Validate(ctx context.Context, options ...langchainllms.CallOption) (Response, error)