coderd/pubsub/chatconfigevent.go

Rama main · réplica de lectura del repositorio original.

package pubsub

import (
	"context"
	"encoding/json"

	"github.com/google/uuid"
	"golang.org/x/xerrors"
)

// ChatConfigEventChannel is the pubsub channel for chat config
// changes (model configs, user prompts, advisor config).
// All replicas subscribe to this channel to invalidate their local
// caches.
const ChatConfigEventChannel = "chat:config_change"

// HandleChatConfigEvent wraps a typed callback for ChatConfigEvent
// messages, following the same pattern as HandleChatWatchEvent.
func HandleChatConfigEvent(cb func(ctx context.Context, payload ChatConfigEvent, err error)) func(ctx context.Context, message []byte, err error) {
	return func(ctx context.Context, message []byte, err error) {
		if err != nil {
			cb(ctx, ChatConfigEvent{}, xerrors.Errorf("chat config event pubsub: %w", err))
			return
		}
		var payload ChatConfigEvent
		if err := json.Unmarshal(message, &payload); err != nil {
			cb(ctx, ChatConfigEvent{}, xerrors.Errorf("unmarshal chat config event: %w", err))
			return
		}

		cb(ctx, payload, err)
	}
}

// ChatConfigEvent is published when chat configuration changes
// (model config CRUD, user prompt updates, or advisor config
// updates). Subscribers use this to invalidate their local caches.
type ChatConfigEvent struct {
	Kind ChatConfigEventKind `json:"kind"`
	// EntityID carries context for the invalidation:
	//   - For model configs: the specific config ID.
	//   - For user prompts: the user ID.
	//   - For advisor config: uuid.Nil (singleton site-config row).
	EntityID uuid.UUID `json:"entity_id"`
}

type ChatConfigEventKind string

const (
	ChatConfigEventModelConfig   ChatConfigEventKind = "model_config"
	ChatConfigEventUserPrompt    ChatConfigEventKind = "user_prompt"
	ChatConfigEventAdvisorConfig ChatConfigEventKind = "advisor_config"
)