chatglm

Documentation

Index

• Constants
• type GLMClient
  • func (g *GLMClient) BuildHTTPOptions() ([]poc.PocConfigOption, error)
  • func (g *GLMClient) Chat(s string, f ...aispec.Function) (string, error)
  • func (g *GLMClient) ChatEx(details []aispec.ChatDetail, function ...aispec.Function) ([]aispec.ChatChoice, error)
  • func (g *GLMClient) ExtractData(msg string, desc string, fields map[string]string) (map[string]any, error)
  • func (g *GLMClient) LoadOption(opt ...aispec.AIConfigOption)
• type ModelAPI
  • func NewGLMMessage(msg string) *ModelAPI
  • func (m ModelAPI) Invoke(apiKey string) (map[string]interface{}, error)

Constants

const (
	BaseURL    = "https://open.bigmodel.cn/api/paas/v4/chat/completions"
	APITimeout = 300 * time.Second
)

const (
	APITokenTTLSeconds = 3 * 60
	APIKeyPartCount    = 2
)

Types

type GLMClient

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

func (*GLMClient) BuildHTTPOptions

func (g *GLMClient) BuildHTTPOptions() ([]poc.PocConfigOption, error)

func (*GLMClient) Chat

func (g *GLMClient) Chat(s string, f ...aispec.Function) (string, error)

func (*GLMClient) ChatEx

func (g *GLMClient) ChatEx(details []aispec.ChatDetail, function ...aispec.Function) ([]aispec.ChatChoice, error)

func (*GLMClient) ExtractData

func (g *GLMClient) ExtractData(msg string, desc string, fields map[string]string) (map[string]any, error)

func (*GLMClient) LoadOption

func (g *GLMClient) LoadOption(opt ...aispec.AIConfigOption)

type ModelAPI

type ModelAPI struct {
	Model       string
	Prompt      []map[string]interface{}
	TopP        float64
	Temperature float64
}

func NewGLMMessage

func NewGLMMessage(msg string) *ModelAPI

func (ModelAPI) Invoke

func (m ModelAPI) Invoke(apiKey string) (map[string]interface{}, error)