Changeset 0.14.0 (#67)

2026-02-26 15:59:26 +00:00
parent 1a71658b24
commit e2149e249d
38 changed files with 5171 additions and 141 deletions
--- a/server/models/models.go
+++ b/server/models/models.go
@@ -622,3 +622,69 @@ func StringPtr(s string) *string    { return &s }
 func Float64Ptr(f float64) *float64 { return &f }
 func IntPtr(i int) *int             { return &i }
 func BoolPtr(b bool) *bool          { return &b }
+
+// ── Knowledge Bases ────────────────────────────
+
+// KnowledgeBase is a named collection of documents with vector embeddings.
+type KnowledgeBase struct {
+	ID              string    `json:"id" db:"id"`
+	Name            string    `json:"name" db:"name"`
+	Description     string    `json:"description" db:"description"`
+	Scope           string    `json:"scope" db:"scope"` // global, team, personal
+	OwnerID         *string   `json:"owner_id,omitempty" db:"owner_id"`
+	TeamID          *string   `json:"team_id,omitempty" db:"team_id"`
+	EmbeddingConfig JSONMap   `json:"embedding_config" db:"embedding_config"`
+	DocumentCount   int       `json:"document_count" db:"document_count"`
+	ChunkCount      int       `json:"chunk_count" db:"chunk_count"`
+	TotalBytes      int64     `json:"total_bytes" db:"total_bytes"`
+	Status          string    `json:"status" db:"status"` // active, processing, error
+	CreatedAt       time.Time `json:"created_at" db:"created_at"`
+	UpdatedAt       time.Time `json:"updated_at" db:"updated_at"`
+}
+
+// KBDocument is a single uploaded file within a knowledge base.
+type KBDocument struct {
+	ID            string    `json:"id" db:"id"`
+	KBID          string    `json:"kb_id" db:"kb_id"`
+	Filename      string    `json:"filename" db:"filename"`
+	ContentType   string    `json:"content_type" db:"content_type"`
+	SizeBytes     int64     `json:"size_bytes" db:"size_bytes"`
+	StorageKey    string    `json:"storage_key" db:"storage_key"`
+	ExtractedText *string   `json:"-" db:"extracted_text"`
+	ChunkCount    int       `json:"chunk_count" db:"chunk_count"`
+	Status        string    `json:"status" db:"status"` // pending, chunking, embedding, ready, error
+	Error         *string   `json:"error,omitempty" db:"error"`
+	UploadedBy    string    `json:"uploaded_by" db:"uploaded_by"`
+	CreatedAt     time.Time `json:"created_at" db:"created_at"`
+	UpdatedAt     time.Time `json:"updated_at" db:"updated_at"`
+}
+
+// KBChunk is a text segment from a document with its embedding vector.
+type KBChunk struct {
+	ID          string    `json:"id" db:"id"`
+	KBID        string    `json:"kb_id" db:"kb_id"`
+	DocumentID  string    `json:"document_id" db:"document_id"`
+	ChunkIndex  int       `json:"chunk_index" db:"chunk_index"`
+	Content     string    `json:"content" db:"content"`
+	TokenCount  int       `json:"token_count" db:"token_count"`
+	Embedding   []float64 `json:"-" db:"embedding"` // never serialized to API
+	Metadata    JSONMap   `json:"metadata" db:"metadata"`
+	CreatedAt   time.Time `json:"created_at" db:"created_at"`
+}
+
+// KBSearchResult is a single result from similarity search.
+type KBSearchResult struct {
+	Content    string  `json:"content"`
+	Filename   string  `json:"source"`
+	KBName     string  `json:"kb"`
+	Similarity float64 `json:"similarity"`
+	Metadata   JSONMap  `json:"metadata,omitempty"`
+}
+
+// ChannelKB represents a knowledge base linked to a channel.
+type ChannelKB struct {
+	KBID          string `json:"kb_id" db:"kb_id"`
+	KBName        string `json:"kb_name" db:"name"`
+	Enabled       bool   `json:"enabled" db:"enabled"`
+	DocumentCount int    `json:"document_count" db:"document_count"`
+}