Changeset 0.17.0 (#75)

2026-02-27 16:25:39 +00:00
parent 8bb77710b9
commit c9141a6896
37 changed files with 2778 additions and 968 deletions
--- a/server/store/postgres/persona.go
+++ b/server/store/postgres/persona.go
@@ -320,3 +320,84 @@ func scanPersonas(rows *sql.Rows) ([]models.Persona, error) {
 	}
 	return result, rows.Err()
 }
+
+// ── Persona-KB Bindings (v0.17.0) ───────────
+
+func (s *PersonaStore) SetKBs(ctx context.Context, personaID string, kbIDs []string, autoSearch map[string]bool) error {
+	tx, err := DB.BeginTx(ctx, nil)
+	if err != nil {
+		return err
+	}
+	defer tx.Rollback()
+
+	_, err = tx.ExecContext(ctx, "DELETE FROM persona_knowledge_bases WHERE persona_id = $1", personaID)
+	if err != nil {
+		return err
+	}
+
+	for _, kbID := range kbIDs {
+		auto := false
+		if autoSearch != nil {
+			auto = autoSearch[kbID]
+		}
+		_, err = tx.ExecContext(ctx, `
+			INSERT INTO persona_knowledge_bases (persona_id, kb_id, auto_search)
+			VALUES ($1, $2, $3)`,
+			personaID, kbID, auto)
+		if err != nil {
+			return fmt.Errorf("persona KB %s: %w", kbID, err)
+		}
+	}
+
+	return tx.Commit()
+}
+
+func (s *PersonaStore) GetKBs(ctx context.Context, personaID string) ([]models.PersonaKB, error) {
+	rows, err := DB.QueryContext(ctx, `
+		SELECT pkb.persona_id, pkb.kb_id, pkb.auto_search, pkb.added_at,
+		       kb.name AS kb_name, kb.document_count, kb.chunk_count
+		FROM persona_knowledge_bases pkb
+		JOIN knowledge_bases kb ON kb.id = pkb.kb_id
+		WHERE pkb.persona_id = $1
+		ORDER BY kb.name`, personaID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var result []models.PersonaKB
+	for rows.Next() {
+		var pkb models.PersonaKB
+		if err := rows.Scan(&pkb.PersonaID, &pkb.KBID, &pkb.AutoSearch, &pkb.AddedAt,
+			&pkb.KBName, &pkb.DocumentCount, &pkb.ChunkCount); err != nil {
+			return nil, err
+		}
+		result = append(result, pkb)
+	}
+	if result == nil {
+		result = make([]models.PersonaKB, 0)
+	}
+	return result, rows.Err()
+}
+
+func (s *PersonaStore) GetKBIDs(ctx context.Context, personaID string) ([]string, error) {
+	rows, err := DB.QueryContext(ctx,
+		"SELECT kb_id FROM persona_knowledge_bases WHERE persona_id = $1", personaID)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+
+	var ids []string
+	for rows.Next() {
+		var id string
+		if err := rows.Scan(&id); err != nil {
+			return nil, err
+		}
+		ids = append(ids, id)
+	}
+	if ids == nil {
+		ids = make([]string, 0)
+	}
+	return ids, rows.Err()
+}