serve

type ApprovalItem struct {
	Approved bool   `json:"approved"`
	Reason   string `json:"reason,omitempty"`
}

type ApprovalRequestPayload struct {
	ApprovalID string                  `json:"approval_id"`
	Targets    []ApprovalTargetPayload `json:"targets"`
}

type ApprovalResponsePayload struct {
	ApprovalID string                  `json:"approval_id"`
	Results    map[string]ApprovalItem `json:"results"`
}

type ApprovalTargetPayload struct {
	ID      string `json:"id"`
	Tool    string `json:"tool"`
	Details string `json:"details"`
}

type ChatRequest struct {
	ChatName string        `json:"chat_name,omitempty"`
	Message  string        `json:"message,omitempty"`
	Files    []FilePayload `json:"files,omitempty"`
}

type ChatSelectedPayload struct {
	SessionID    string `json:"session_id"`
	ChatName     string `json:"chat_name"`
	Description  string `json:"description"`
	Message      string `json:"message"`
	MessageCount int    `json:"message_count"`
}

type ChunkPayload struct {
	Content     string `json:"content"`
	First       bool   `json:"first"`
	Last        bool   `json:"last"`
	ContentType string `json:"content_type"` // "response" or "thinking"
}

type ClearedPayload struct {
	ChatName     string `json:"chat_name,omitempty"`
	Message      string `json:"message"`
	MessageCount int    `json:"message_count"`
}

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

type ClientOption func(*Client)

type CompletePayload struct {
	Message string `json:"message"`
}

type ErrorPayload struct {
	Error string `json:"error"`
}

type EventHandler interface {
	// OnSessionInit is called when the server sends the session_init message.
	OnSessionInit(payload *SessionInitPayload)

	// OnChatSelected is called when a chat is successfully selected.
	OnChatSelected(payload *ChatSelectedPayload)

	// OnChunk is called for each streaming content chunk received.
	OnChunk(payload *ChunkPayload)

	// OnToolCall is called when the model invokes a tool.
	OnToolCall(payload *ToolCallPayload)

	// OnThinking is called when the thinking/reasoning state changes.
	OnThinking(payload *ThinkingPayload)

	// OnComplete is called when the model response is complete.
	OnComplete(payload *CompletePayload)

	// OnError is called when an error occurs.
	OnError(payload *ErrorPayload)

	// OnApprovalRequest is called when tool execution requires user approval.
	// The handler should call SendApprovalResponse to provide the decision.
	OnApprovalRequest(payload *ApprovalRequestPayload)

	// OnMessageCount is called when the message count changes.
	OnMessageCount(payload *MessageCountPayload)

	// OnStopped is called when a response is stopped by the user.
	OnStopped(payload *StoppedPayload)

	// OnKept is called after a keep hook executes.
	OnKept(payload *KeptPayload)

	// OnCleared is called after the conversation context is cleared.
	OnCleared(payload *ClearedPayload)

	// OnDisconnected is called when the WebSocket connection is lost.
	// err is nil for intentional disconnection.
	OnDisconnected(err error)

	// OnReconnected is called when the client successfully reconnects.
	OnReconnected()
}

type FilePayload struct {
	URL      string `json:"url"`
	Type     string `json:"type"`
	Name     string `json:"name"`
	FileSize int64  `json:"file_size,omitempty"`
}

type KeptPayload struct {
	ChatName string `json:"chat_name,omitempty"`
	Message  string `json:"message"`
}

type MessageCountPayload struct {
	Count int `json:"count"`
}

type SessionInitPayload struct {
	SessionID string `json:"session_id"`
}

type StoppedPayload struct {
	Message string `json:"message"`
}

type ThinkingPayload struct {
	Status bool `json:"status"`
}

type ToolCallPayload struct {
	Name      string `json:"name"`
	Arguments string `json:"arguments"`
	Index     string `json:"index"`
	Streaming bool   `json:"streaming"`
}

type WSMessage struct {
	Type    string          `json:"type"`
	Payload json.RawMessage `json:"payload"`
}