weichen
dc29b84cdf
Add dialoguectx packaging, conflictresolve adjudication with HITL routing, workspace context/dialogue/ injection (default on), and goose 00071.
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
package ctxmgr
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
)
|
|
|
|
// CollectProjectDialogueRecords returns authorized-visible chat message records
|
|
// for a project (before line formatting). Used by dialoguectx packaging.
|
|
func (m *ContextManager) CollectProjectDialogueRecords(ctx context.Context, projectID int64, limit int) ([]ChatMessageRecord, error) {
|
|
if m == nil || projectID <= 0 {
|
|
return nil, nil
|
|
}
|
|
if limit <= 0 {
|
|
limit = defaultProjectDialogueLimit
|
|
}
|
|
var msgs []ChatMessageRecord
|
|
if m.pool != nil {
|
|
rows, err := m.pool.Query(ctx, `
|
|
SELECT cm.id, cm.session_id, COALESCE(cm.user_id, 0), cm.role, cm.content,
|
|
cm.intent, cm.branch, cm.metadata, COALESCE(cm.day_key::text, ''),
|
|
COALESCE(cm.category, ''), COALESCE(cm.content_kind, ''),
|
|
COALESCE(cm.export_policy, 'none'), COALESCE(cm.stage_key, ''), cm.created_at
|
|
FROM chat_messages cm
|
|
JOIN chat_sessions cs ON cs.session_id = cm.session_id
|
|
WHERE cs.project_id = $1 AND cm.deleted_at IS NULL
|
|
ORDER BY cm.created_at ASC
|
|
LIMIT $2
|
|
`, projectID, limit)
|
|
if err != nil {
|
|
rows, err = m.pool.Query(ctx, `
|
|
SELECT cm.id, cm.session_id, COALESCE(cm.user_id, 0), cm.role, cm.content,
|
|
cm.intent, cm.branch, cm.metadata, COALESCE(cm.day_key::text, ''),
|
|
COALESCE(cm.category, ''), COALESCE(cm.content_kind, ''),
|
|
COALESCE(cm.export_policy, 'none'), cm.created_at
|
|
FROM chat_messages cm
|
|
JOIN chat_sessions cs ON cs.session_id = cm.session_id
|
|
WHERE cs.project_id = $1 AND cm.deleted_at IS NULL
|
|
ORDER BY cm.created_at ASC
|
|
LIMIT $2
|
|
`, projectID, limit)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("collect project dialogue records: %w", err)
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var rec ChatMessageRecord
|
|
var metaJSON []byte
|
|
if err := rows.Scan(
|
|
&rec.ID, &rec.SessionID, &rec.UserID, &rec.Role, &rec.Content,
|
|
&rec.Intent, &rec.Branch, &metaJSON, &rec.DayKey, &rec.Category,
|
|
&rec.ContentKind, &rec.ExportPolicy, &rec.CreatedAt,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(metaJSON) > 0 {
|
|
_ = json.Unmarshal(metaJSON, &rec.Metadata)
|
|
}
|
|
msgs = append(msgs, rec)
|
|
}
|
|
return msgs, rows.Err()
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var rec ChatMessageRecord
|
|
var metaJSON []byte
|
|
if err := rows.Scan(
|
|
&rec.ID, &rec.SessionID, &rec.UserID, &rec.Role, &rec.Content,
|
|
&rec.Intent, &rec.Branch, &metaJSON, &rec.DayKey, &rec.Category,
|
|
&rec.ContentKind, &rec.ExportPolicy, &rec.StageKey, &rec.CreatedAt,
|
|
); err != nil {
|
|
return nil, err
|
|
}
|
|
if len(metaJSON) > 0 {
|
|
_ = json.Unmarshal(metaJSON, &rec.Metadata)
|
|
}
|
|
msgs = append(msgs, rec)
|
|
}
|
|
return msgs, rows.Err()
|
|
}
|
|
|
|
m.mu.RLock()
|
|
defer m.mu.RUnlock()
|
|
for sessionID, sessionMsgs := range m.messages {
|
|
sess := m.sessions[sessionID]
|
|
if sess != nil && sess.ProjectID != projectID {
|
|
continue
|
|
}
|
|
for _, rec := range sessionMsgs {
|
|
if sess == nil || sess.ProjectID == 0 {
|
|
if pid := messageProjectID(rec); pid != projectID {
|
|
continue
|
|
}
|
|
}
|
|
msgs = append(msgs, rec)
|
|
if len(msgs) >= limit {
|
|
return msgs, nil
|
|
}
|
|
}
|
|
}
|
|
return msgs, nil
|
|
}
|