package entity

import (
	libkg "haixun-backend/internal/library/knowledge"
)

const CollectionName = "topic_knowledge_graphs"

type Graph struct {
	ID              string              `bson:"_id"`
	TenantID        string              `bson:"tenant_id"`
	OwnerUID        string              `bson:"owner_uid"`
	BrandID         string              `bson:"brand_id"`
	TopicID         string              `bson:"topic_id,omitempty"`
	LegacyPersonaID string              `bson:"persona_id,omitempty"`
	Seed            string              `bson:"seed"`
	Nodes           []libkg.Node        `bson:"nodes"`
	Edges           []libkg.Edge        `bson:"edges"`
	BraveSources    []libkg.BraveSource `bson:"brave_sources"`
	ExpandStrategy  string              `bson:"expand_strategy,omitempty"`
	PainTagCount    int                 `bson:"pain_tag_count"`
	GeneratedAt     int64               `bson:"generated_at"`
	CreateAt        int64               `bson:"create_at"`
	UpdateAt        int64               `bson:"update_at"`
}
