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 (
	OpenAIModels = []string{"gpt-3.5", "gpt-3.5-turbo"}
	GeminiModels = []string{"gemini-pro"}
)

var (
	ZhiPuAILite  string = "chatglm_lite"
	ZhiPuAIStd   string = "chatglm_std"
	ZhiPuAIPro   string = "chatglm_pro"
	ZhiPuAITurbo string = "chatglm_turbo"
	// ChatGLM3
	ZhiPuAIGLM3Turbo string = "glm-3-turbo"
	// ChatGLM4
	ZhiPuAIGLM4 string = "glm-4"
	// Character LLM
	ZhiPuAICharGLM3 string = "charglm-3"
)

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

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"
	Gemini    LLMType = "gemini"
	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)