|
| 1 | +package autonomous |
| 2 | + |
| 3 | +import ( |
| 4 | + "encoding/json" |
| 5 | + "log" |
| 6 | + "time" |
| 7 | +) |
| 8 | + |
| 9 | +type AgentMessage struct { |
| 10 | + Role string `json:"role"` |
| 11 | + Content string `json:"content"` |
| 12 | + Timestamp time.Time `json:"timestamp"` |
| 13 | + Tokens int `json:"tokens"` |
| 14 | +} |
| 15 | + |
| 16 | +type DebateSession struct { |
| 17 | + ID string `json:"id"` |
| 18 | + Topic string `json:"topic"` |
| 19 | + Status string `json:"status"` // "ongoing", "resolved", "escalated" |
| 20 | + Messages []AgentMessage `json:"messages"` |
| 21 | + StartTime time.Time `json:"startTime"` |
| 22 | + EndTime time.Time `json:"endTime,omitempty"` |
| 23 | +} |
| 24 | + |
| 25 | +var ActiveDebates = make(map[string]*DebateSession) |
| 26 | + |
| 27 | +func InitiateDebate(topic string) *DebateSession { |
| 28 | + session := &DebateSession{ |
| 29 | + ID: "debate_" + time.Now().Format("20060102150405"), |
| 30 | + Topic: topic, |
| 31 | + Status: "ongoing", |
| 32 | + StartTime: time.Now(), |
| 33 | + Messages: []AgentMessage{}, |
| 34 | + } |
| 35 | + ActiveDebates[session.ID] = session |
| 36 | + log.Printf("[COUNCIL SUPERVISOR] Initiated debate session: %s on topic: %s", session.ID, topic) |
| 37 | + return session |
| 38 | +} |
| 39 | + |
| 40 | +func AddDebateMessage(sessionID, role, content string, tokens int) { |
| 41 | + if session, exists := ActiveDebates[sessionID]; exists { |
| 42 | + msg := AgentMessage{ |
| 43 | + Role: role, |
| 44 | + Content: content, |
| 45 | + Timestamp: time.Now(), |
| 46 | + Tokens: tokens, |
| 47 | + } |
| 48 | + session.Messages = append(session.Messages, msg) |
| 49 | + log.Printf("[DEBATE %s] %s: %s (%d tokens)", sessionID, role, content, tokens) |
| 50 | + } |
| 51 | +} |
| 52 | + |
| 53 | +func ResolveDebate(sessionID string) { |
| 54 | + if session, exists := ActiveDebates[sessionID]; exists { |
| 55 | + session.Status = "resolved" |
| 56 | + session.EndTime = time.Now() |
| 57 | + log.Printf("[COUNCIL SUPERVISOR] Resolved debate session: %s", sessionID) |
| 58 | + } |
| 59 | +} |
| 60 | + |
| 61 | +func GetFleetDebateSummary() []byte { |
| 62 | + summary := []map[string]interface{}{} |
| 63 | + for _, session := range ActiveDebates { |
| 64 | + summary = append(summary, map[string]interface{}{ |
| 65 | + "id": session.ID, |
| 66 | + "topic": session.Topic, |
| 67 | + "status": session.Status, |
| 68 | + "messages": len(session.Messages), |
| 69 | + }) |
| 70 | + } |
| 71 | + bytes, _ := json.Marshal(summary) |
| 72 | + return bytes |
| 73 | +} |
0 commit comments