core/agent
8
0
Fork 0
Code Issues 5 Pull requests Projects Releases Packages Wiki Activity Actions

feat(brain): add workspace-aware message fields

Browse source 
Co-Authored-By: Virgil <virgil@lethean.io>
This commit is contained in:
Virgil 2026-04-01 20:04:04 +00:00
parent db7b42dc4d
commit c52ed05c83
2 changed files with 68 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

57
pkg/brain/messaging.go
View file

@ -48,18 +48,22 @@ type InboxInput struct {
Agent string `json:"agent,omitempty"`
}
// brain.MessageItem{ID: 7, From: "cladius", To: "charon", Content: "all green"}
// brain.MessageItem{ID: 7, WorkspaceID: 1, FromAgent: "cladius", ToAgent: "charon", Content: "all green"}
type MessageItem struct {
ID int `json:"id"`
From string `json:"from"`
To string `json:"to"`
Subject string `json:"subject,omitempty"`
Content string `json:"content"`
Read bool `json:"read"`
CreatedAt string `json:"created_at"`
ID int `json:"id"`
WorkspaceID int `json:"workspace_id,omitempty"`
From string `json:"from,omitempty"`
FromAgent string `json:"from_agent,omitempty"`
To string `json:"to,omitempty"`
ToAgent string `json:"to_agent,omitempty"`
Subject string `json:"subject,omitempty"`
Content string `json:"content"`
Read bool `json:"read"`
ReadAt string `json:"read_at,omitempty"`
CreatedAt string `json:"created_at"`
}
// brain.InboxOutput{Success: true, Messages: []brain.MessageItem{{ID: 1, From: "charon", To: "cladius"}}}
// brain.InboxOutput{Success: true, Messages: []brain.MessageItem{{ID: 1, FromAgent: "charon", ToAgent: "cladius"}}}
type InboxOutput struct {
Success bool `json:"success"`
Messages []MessageItem `json:"messages"`
@ -70,7 +74,7 @@ type ConversationInput struct {
Agent string `json:"agent"`
}
// brain.ConversationOutput{Success: true, Messages: []brain.MessageItem{{ID: 10, From: "cladius", To: "charon"}}}
// brain.ConversationOutput{Success: true, Messages: []brain.MessageItem{{ID: 10, FromAgent: "cladius", ToAgent: "charon"}}}
type ConversationOutput struct {
Success bool `json:"success"`
Messages []MessageItem `json:"messages"`
@ -142,19 +146,38 @@ func parseMessages(result map[string]any) []MessageItem {
data, _ := result["data"].([]any)
for _, m := range data {
mm, _ := m.(map[string]any)
from := stringFieldAny(mm, "from", "from_agent")
to := stringFieldAny(mm, "to", "to_agent")
messages = append(messages, MessageItem{
ID: toInt(mm["id"]),
From: stringField(mm, "from"),
To: stringField(mm, "to"),
Subject: stringField(mm, "subject"),
Content: stringField(mm, "content"),
Read: mm["read"] == true,
CreatedAt: stringField(mm, "created_at"),
ID: toInt(mm["id"]),
WorkspaceID: toInt(mm["workspace_id"]),
From: from,
FromAgent: from,
To: to,
ToAgent: to,
Subject: stringField(mm, "subject"),
Content: stringField(mm, "content"),
Read: mm["read"] == true,
ReadAt: stringField(mm, "read_at"),
CreatedAt: stringField(mm, "created_at"),
})
}
return messages
}
func stringFieldAny(values map[string]any, keys ...string) string {
for _, key := range keys {
value, exists := values[key]
if !exists {
continue
}
if text := core.Trim(core.Sprint(value)); text != "" {
return text
}
}
return ""
}
func toInt(v any) int {
if f, ok := v.(float64); ok {
return int(f)

45
pkg/brain/messaging_test.go
View file

@ -91,13 +91,15 @@ func TestMessaging_Inbox_Good_WithMessages(t *testing.T) {
w.Write([]byte(core.JSONMarshalString(map[string]any{
"data": []any{
map[string]any{
"id": float64(1),
"from": "charon",
"to": "cladius",
"subject": "status",
"content": "deploy done",
"read": true,
"created_at": "2026-03-10T12:00:00Z",
"workspace_id": float64(3),
"id": float64(1),
"from_agent": "charon",
"to_agent": "cladius",
"subject": "status",
"content": "deploy done",
"read": true,
"read_at": "2026-03-10T12:05:00Z",
"created_at": "2026-03-10T12:00:00Z",
},
map[string]any{
"id": float64(2),
@ -118,9 +120,13 @@ func TestMessaging_Inbox_Good_WithMessages(t *testing.T) {
assert.True(t, out.Success)
require.Len(t, out.Messages, 2)
assert.Equal(t, 1, out.Messages[0].ID)
assert.Equal(t, 3, out.Messages[0].WorkspaceID)
assert.Equal(t, "charon", out.Messages[0].From)
assert.Equal(t, "charon", out.Messages[0].FromAgent)
assert.Equal(t, "cladius", out.Messages[0].ToAgent)
assert.Equal(t, "deploy done", out.Messages[0].Content)
assert.True(t, out.Messages[0].Read)
assert.Equal(t, "2026-03-10T12:05:00Z", out.Messages[0].ReadAt)
assert.Equal(t, 2, out.Messages[1].ID)
assert.False(t, out.Messages[1].Read)
}
@ -202,24 +208,28 @@ func TestMessaging_ParseMessages_Good(t *testing.T) {
result := map[string]any{
"data": []any{
map[string]any{
"id": float64(5),
"from": "alice",
"to": "bob",
"subject": "hello",
"content": "hi there",
"read": true,
"created_at": "2026-03-10T10:00:00Z",
"id": float64(5),
"workspace_id": float64(8),
"from": "alice",
"to": "bob",
"subject": "hello",
"content": "hi there",
"read": true,
"read_at": "2026-03-10T10:01:00Z",
"created_at": "2026-03-10T10:00:00Z",
},
},
}
msgs := parseMessages(result)
require.Len(t, msgs, 1)
assert.Equal(t, 5, msgs[0].ID)
assert.Equal(t, 8, msgs[0].WorkspaceID)
assert.Equal(t, "alice", msgs[0].From)
assert.Equal(t, "bob", msgs[0].To)
assert.Equal(t, "hello", msgs[0].Subject)
assert.Equal(t, "hi there", msgs[0].Content)
assert.True(t, msgs[0].Read)
assert.Equal(t, "2026-03-10T10:01:00Z", msgs[0].ReadAt)
assert.Equal(t, "2026-03-10T10:00:00Z", msgs[0].CreatedAt)
}
@ -280,21 +290,22 @@ func TestMessaging_InboxOutput_Good_RoundTrip(t *testing.T) {
in := InboxOutput{
Success: true,
Messages: []MessageItem{
{ID: 1, From: "a", To: "b", Content: "hi", Read: false, CreatedAt: "2026-03-10T12:00:00Z"},
{ID: 1, WorkspaceID: 7, FromAgent: "a", ToAgent: "b", Content: "hi", Read: false, CreatedAt: "2026-03-10T12:00:00Z"},
},
}
var out InboxOutput
roundTrip(t, in, &out)
assert.Equal(t, in.Success, out.Success)
require.Len(t, out.Messages, 1)
assert.Equal(t, "a", out.Messages[0].From)
assert.Equal(t, 7, out.Messages[0].WorkspaceID)
assert.Equal(t, "a", out.Messages[0].FromAgent)
}
func TestMessaging_ConversationOutput_Good_RoundTrip(t *testing.T) {
in := ConversationOutput{
Success: true,
Messages: []MessageItem{
{ID: 10, From: "x", To: "y", Content: "thread", Read: true, CreatedAt: "2026-03-10T14:00:00Z"},
{ID: 10, From: "x", To: "y", Content: "thread", Read: true, ReadAt: "2026-03-10T14:01:00Z", CreatedAt: "2026-03-10T14:00:00Z"},
},
}
var out ConversationOutput