chatglm

README

Go ChatGLM

Synchronization with official Python SDK feature & version numbers.

This library provides unofficial Go clients for ChatGLM API. We support:

• Get Token
• ChatGLM

Installation:

go get github.com/sunny0826/go-chatglm

Select Model

• ChatGLMLite
• ChatGLMStd
• ChatGLMPro

m := chatglm.ModelAPI{
    Model:       chatglm.ChatGLMLite,
    // Model:       chatglm.ChatGLMStd,
    // Model:       chatglm.ChatGLMPro,
}

Examples

• sync
• async
• stream

license

Apache License 2.0

Documentation

Index

• Constants
• func QueryAsyncInvokeResult(apiKey, taskID string) (map[string]interface{}, error)
• type Event
• type ModelAPI
  • func (m ModelAPI) AsyncInvoke(apiKey string) (map[string]interface{}, error)
  • func (m ModelAPI) Invoke(apiKey string) (map[string]interface{}, error)
  • func (m ModelAPI) SSEInvoke(apiKey string) (*SSEClient, error)
• type SSEClient
  • func NewSSEClient(eventSource io.Reader, charEnc string) *SSEClient
  • func (c *SSEClient) ReadEvents() <-chan Event

Constants

const (
	BaseURL         = "https://open.bigmodel.cn/api/paas/v3/model-api"
	InvokeTypeSync  = "invoke"
	InvokeTypeAsync = "async-invoke"
	InvokeTypeSSE   = "sse-invoke"
	APITimeout      = 300 * time.Second
	ChatGLMLite     = "chatglm_lite"
	ChatGLMStd      = "chatglm_std"
	ChatGLMPro      = "chatglm_pro"
)

const (
	APITokenTTLSeconds = 3 * 60
	APIKeyPartCount    = 2
)

Functions

func QueryAsyncInvokeResult

func QueryAsyncInvokeResult(apiKey, taskID string) (map[string]interface{}, error)

Types

type Event

type Event struct {
	ID    string
	Event string
	Data  string
	Retry int
	Meta  map[string]string
}

Event represents an SSE event.

type ModelAPI

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

func (ModelAPI) AsyncInvoke

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

func (ModelAPI) Invoke

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

func (ModelAPI) SSEInvoke

func (m ModelAPI) SSEInvoke(apiKey string) (*SSEClient, error)

type SSEClient

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

SSEClient represents a Server-Sent Events (SSE) client.

func NewSSEClient

func NewSSEClient(eventSource io.Reader, charEnc string) *SSEClient

NewSSEClient creates a new SSEClient instance.

func (*SSEClient) ReadEvents

func (c *SSEClient) ReadEvents() <-chan Event

ReadEvents reads SSE events from the event source and returns them as a channel.