mirror of
https://github.com/hanzoai/agent.git
synced 2026-09-11 21:08:51 +00:00
zeekay
ff7ef7114f
Consolidate all Hanzo agent infrastructure into one repo: - control-plane/: Go control plane server (was agents-new) - sdk/: Go, Python, TypeScript SDKs - deployments/: Docker, Kubernetes, Helm configs - Dockerfile.sdk: Python SDK image (ghcr.io/hanzoai/agent:latest) - CI builds both SDK + control-plane images No more agentfield — everything is hanzoai/agent.
435 lines
12 KiB
Go
435 lines
12 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInMemoryBackend(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
|
|
t.Run("Set and Get", func(t *testing.T) {
|
|
err := backend.Set(ScopeSession, "session-1", "key1", "value1")
|
|
require.NoError(t, err)
|
|
|
|
val, found, err := backend.Get(ScopeSession, "session-1", "key1")
|
|
require.NoError(t, err)
|
|
assert.True(t, found)
|
|
assert.Equal(t, "value1", val)
|
|
})
|
|
|
|
t.Run("Get non-existent key", func(t *testing.T) {
|
|
val, found, err := backend.Get(ScopeSession, "session-1", "nonexistent")
|
|
require.NoError(t, err)
|
|
assert.False(t, found)
|
|
assert.Nil(t, val)
|
|
})
|
|
|
|
t.Run("Get from non-existent scope", func(t *testing.T) {
|
|
val, found, err := backend.Get(ScopeSession, "nonexistent-session", "key1")
|
|
require.NoError(t, err)
|
|
assert.False(t, found)
|
|
assert.Nil(t, val)
|
|
})
|
|
|
|
t.Run("Delete key", func(t *testing.T) {
|
|
err := backend.Set(ScopeSession, "session-1", "to-delete", "value")
|
|
require.NoError(t, err)
|
|
|
|
err = backend.Delete(ScopeSession, "session-1", "to-delete")
|
|
require.NoError(t, err)
|
|
|
|
val, found, err := backend.Get(ScopeSession, "session-1", "to-delete")
|
|
require.NoError(t, err)
|
|
assert.False(t, found)
|
|
assert.Nil(t, val)
|
|
})
|
|
|
|
t.Run("Delete non-existent key (no error)", func(t *testing.T) {
|
|
err := backend.Delete(ScopeSession, "session-1", "nonexistent")
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("List keys", func(t *testing.T) {
|
|
// Clear and set up fresh data
|
|
backend.ClearScope(ScopeWorkflow, "workflow-1")
|
|
err := backend.Set(ScopeWorkflow, "workflow-1", "key-a", "value-a")
|
|
require.NoError(t, err)
|
|
err = backend.Set(ScopeWorkflow, "workflow-1", "key-b", "value-b")
|
|
require.NoError(t, err)
|
|
|
|
keys, err := backend.List(ScopeWorkflow, "workflow-1")
|
|
require.NoError(t, err)
|
|
assert.Len(t, keys, 2)
|
|
assert.Contains(t, keys, "key-a")
|
|
assert.Contains(t, keys, "key-b")
|
|
})
|
|
|
|
t.Run("List empty scope", func(t *testing.T) {
|
|
keys, err := backend.List(ScopeGlobal, "nonexistent")
|
|
require.NoError(t, err)
|
|
assert.Nil(t, keys)
|
|
})
|
|
|
|
t.Run("Scope isolation", func(t *testing.T) {
|
|
// Set same key in different scopes
|
|
err := backend.Set(ScopeSession, "id-1", "shared-key", "session-value")
|
|
require.NoError(t, err)
|
|
err = backend.Set(ScopeWorkflow, "id-1", "shared-key", "workflow-value")
|
|
require.NoError(t, err)
|
|
|
|
sessionVal, _, _ := backend.Get(ScopeSession, "id-1", "shared-key")
|
|
workflowVal, _, _ := backend.Get(ScopeWorkflow, "id-1", "shared-key")
|
|
|
|
assert.Equal(t, "session-value", sessionVal)
|
|
assert.Equal(t, "workflow-value", workflowVal)
|
|
})
|
|
|
|
t.Run("ScopeID isolation", func(t *testing.T) {
|
|
// Same scope, different IDs
|
|
err := backend.Set(ScopeSession, "session-a", "key", "value-a")
|
|
require.NoError(t, err)
|
|
err = backend.Set(ScopeSession, "session-b", "key", "value-b")
|
|
require.NoError(t, err)
|
|
|
|
valA, _, _ := backend.Get(ScopeSession, "session-a", "key")
|
|
valB, _, _ := backend.Get(ScopeSession, "session-b", "key")
|
|
|
|
assert.Equal(t, "value-a", valA)
|
|
assert.Equal(t, "value-b", valB)
|
|
})
|
|
|
|
t.Run("Clear all data", func(t *testing.T) {
|
|
err := backend.Set(ScopeGlobal, "global", "test", "value")
|
|
require.NoError(t, err)
|
|
|
|
backend.Clear()
|
|
|
|
val, found, _ := backend.Get(ScopeGlobal, "global", "test")
|
|
assert.False(t, found)
|
|
assert.Nil(t, val)
|
|
})
|
|
|
|
t.Run("Store complex types", func(t *testing.T) {
|
|
complexData := map[string]any{
|
|
"name": "test",
|
|
"count": 42,
|
|
"nested": map[string]any{"key": "value"},
|
|
}
|
|
err := backend.Set(ScopeSession, "session-1", "complex", complexData)
|
|
require.NoError(t, err)
|
|
|
|
val, found, err := backend.Get(ScopeSession, "session-1", "complex")
|
|
require.NoError(t, err)
|
|
assert.True(t, found)
|
|
assert.Equal(t, complexData, val)
|
|
})
|
|
}
|
|
|
|
func TestMemory_DefaultScope(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
// Create a context with execution context
|
|
execCtx := ExecutionContext{
|
|
SessionID: "test-session",
|
|
WorkflowID: "test-workflow",
|
|
RunID: "test-run",
|
|
}
|
|
ctx := contextWithExecution(context.Background(), execCtx)
|
|
|
|
t.Run("Set and Get", func(t *testing.T) {
|
|
err := memory.Set(ctx, "key1", "value1")
|
|
require.NoError(t, err)
|
|
|
|
val, err := memory.Get(ctx, "key1")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "value1", val)
|
|
})
|
|
|
|
t.Run("GetWithDefault - key exists", func(t *testing.T) {
|
|
err := memory.Set(ctx, "existing", "real-value")
|
|
require.NoError(t, err)
|
|
|
|
val, err := memory.GetWithDefault(ctx, "existing", "default")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "real-value", val)
|
|
})
|
|
|
|
t.Run("GetWithDefault - key missing", func(t *testing.T) {
|
|
val, err := memory.GetWithDefault(ctx, "missing", "default-value")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "default-value", val)
|
|
})
|
|
|
|
t.Run("Delete", func(t *testing.T) {
|
|
err := memory.Set(ctx, "to-delete", "value")
|
|
require.NoError(t, err)
|
|
|
|
err = memory.Delete(ctx, "to-delete")
|
|
require.NoError(t, err)
|
|
|
|
val, err := memory.Get(ctx, "to-delete")
|
|
require.NoError(t, err)
|
|
assert.Nil(t, val)
|
|
})
|
|
|
|
t.Run("List", func(t *testing.T) {
|
|
// Clear the session scope
|
|
backend.ClearScope(ScopeSession, "test-session")
|
|
|
|
err := memory.Set(ctx, "list-key-1", "v1")
|
|
require.NoError(t, err)
|
|
err = memory.Set(ctx, "list-key-2", "v2")
|
|
require.NoError(t, err)
|
|
|
|
keys, err := memory.List(ctx)
|
|
require.NoError(t, err)
|
|
assert.Len(t, keys, 2)
|
|
assert.Contains(t, keys, "list-key-1")
|
|
assert.Contains(t, keys, "list-key-2")
|
|
})
|
|
}
|
|
|
|
func TestMemory_WorkflowScope(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
execCtx := ExecutionContext{
|
|
SessionID: "test-session",
|
|
WorkflowID: "test-workflow",
|
|
RunID: "test-run",
|
|
}
|
|
ctx := contextWithExecution(context.Background(), execCtx)
|
|
|
|
t.Run("Set and Get", func(t *testing.T) {
|
|
err := memory.WorkflowScope().Set(ctx, "wf-key", "wf-value")
|
|
require.NoError(t, err)
|
|
|
|
val, err := memory.WorkflowScope().Get(ctx, "wf-key")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "wf-value", val)
|
|
})
|
|
|
|
t.Run("Isolation from session scope", func(t *testing.T) {
|
|
err := memory.Set(ctx, "shared-key", "session-val")
|
|
require.NoError(t, err)
|
|
err = memory.WorkflowScope().Set(ctx, "shared-key", "workflow-val")
|
|
require.NoError(t, err)
|
|
|
|
sessionVal, _ := memory.Get(ctx, "shared-key")
|
|
workflowVal, _ := memory.WorkflowScope().Get(ctx, "shared-key")
|
|
|
|
assert.Equal(t, "session-val", sessionVal)
|
|
assert.Equal(t, "workflow-val", workflowVal)
|
|
})
|
|
}
|
|
|
|
func TestMemory_GlobalScope(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
// Two different execution contexts
|
|
ctx1 := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "session-1",
|
|
})
|
|
ctx2 := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "session-2",
|
|
})
|
|
|
|
t.Run("Global data shared across sessions", func(t *testing.T) {
|
|
err := memory.GlobalScope().Set(ctx1, "global-key", "global-value")
|
|
require.NoError(t, err)
|
|
|
|
// Access from different session
|
|
val, err := memory.GlobalScope().Get(ctx2, "global-key")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "global-value", val)
|
|
})
|
|
|
|
t.Run("Session data isolated", func(t *testing.T) {
|
|
err := memory.Set(ctx1, "session-key", "session-1-value")
|
|
require.NoError(t, err)
|
|
|
|
// Should not see session-1's data from session-2
|
|
val, err := memory.Get(ctx2, "session-key")
|
|
require.NoError(t, err)
|
|
assert.Nil(t, val) // Not found
|
|
})
|
|
}
|
|
|
|
func TestMemory_UserScope(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
// Same actor, different sessions
|
|
ctx1 := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "session-1",
|
|
ActorID: "user-123",
|
|
})
|
|
ctx2 := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "session-2",
|
|
ActorID: "user-123",
|
|
})
|
|
|
|
t.Run("User data persists across sessions", func(t *testing.T) {
|
|
err := memory.UserScope().Set(ctx1, "user-pref", "dark-mode")
|
|
require.NoError(t, err)
|
|
|
|
// Access from different session, same user
|
|
val, err := memory.UserScope().Get(ctx2, "user-pref")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "dark-mode", val)
|
|
})
|
|
}
|
|
|
|
func TestMemory_ScopedGetTyped(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
ctx := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "test-session",
|
|
})
|
|
|
|
t.Run("GetTyped with struct", func(t *testing.T) {
|
|
type TestData struct {
|
|
Name string `json:"name"`
|
|
Count int `json:"count"`
|
|
}
|
|
|
|
original := TestData{Name: "test", Count: 42}
|
|
err := memory.SessionScope().Set(ctx, "typed-data", original)
|
|
require.NoError(t, err)
|
|
|
|
var retrieved TestData
|
|
err = memory.SessionScope().GetTyped(ctx, "typed-data", &retrieved)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, original.Name, retrieved.Name)
|
|
assert.Equal(t, original.Count, retrieved.Count)
|
|
})
|
|
|
|
t.Run("GetTyped with slice", func(t *testing.T) {
|
|
original := []string{"a", "b", "c"}
|
|
err := memory.SessionScope().Set(ctx, "slice-data", original)
|
|
require.NoError(t, err)
|
|
|
|
var retrieved []string
|
|
err = memory.SessionScope().GetTyped(ctx, "slice-data", &retrieved)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, original, retrieved)
|
|
})
|
|
|
|
t.Run("GetTyped with non-existent key", func(t *testing.T) {
|
|
var retrieved string
|
|
err := memory.SessionScope().GetTyped(ctx, "nonexistent", &retrieved)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "", retrieved) // zero value
|
|
})
|
|
}
|
|
|
|
func TestMemory_FallbackToRunID(t *testing.T) {
|
|
backend := NewInMemoryBackend()
|
|
memory := NewMemory(backend)
|
|
|
|
// Context with only RunID (no SessionID, WorkflowID, ActorID)
|
|
ctx := contextWithExecution(context.Background(), ExecutionContext{
|
|
RunID: "run-123",
|
|
})
|
|
|
|
t.Run("Session scope falls back to RunID", func(t *testing.T) {
|
|
err := memory.Set(ctx, "key", "value")
|
|
require.NoError(t, err)
|
|
|
|
// Verify it was stored under RunID
|
|
val, found, _ := backend.Get(ScopeSession, "run-123", "key")
|
|
assert.True(t, found)
|
|
assert.Equal(t, "value", val)
|
|
})
|
|
|
|
t.Run("Workflow scope falls back to RunID", func(t *testing.T) {
|
|
err := memory.WorkflowScope().Set(ctx, "wf-key", "wf-value")
|
|
require.NoError(t, err)
|
|
|
|
val, found, _ := backend.Get(ScopeWorkflow, "run-123", "wf-key")
|
|
assert.True(t, found)
|
|
assert.Equal(t, "wf-value", val)
|
|
})
|
|
}
|
|
|
|
func TestMemory_NilBackend(t *testing.T) {
|
|
// NewMemory should create InMemoryBackend if nil is passed
|
|
memory := NewMemory(nil)
|
|
|
|
ctx := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "test",
|
|
})
|
|
|
|
err := memory.Set(ctx, "key", "value")
|
|
require.NoError(t, err)
|
|
|
|
val, err := memory.Get(ctx, "key")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "value", val)
|
|
}
|
|
|
|
func TestAgentMemory(t *testing.T) {
|
|
cfg := Config{
|
|
NodeID: "test-node",
|
|
Version: "1.0.0",
|
|
Logger: log.New(io.Discard, "", 0),
|
|
}
|
|
|
|
agent, err := New(cfg)
|
|
require.NoError(t, err)
|
|
|
|
ctx := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "test-session",
|
|
})
|
|
|
|
t.Run("Agent has memory", func(t *testing.T) {
|
|
assert.NotNil(t, agent.Memory())
|
|
})
|
|
|
|
t.Run("Agent memory works", func(t *testing.T) {
|
|
err := agent.Memory().Set(ctx, "agent-key", "agent-value")
|
|
require.NoError(t, err)
|
|
|
|
val, err := agent.Memory().Get(ctx, "agent-key")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "agent-value", val)
|
|
})
|
|
}
|
|
|
|
func TestAgentWithCustomMemoryBackend(t *testing.T) {
|
|
// Create a custom backend
|
|
customBackend := NewInMemoryBackend()
|
|
|
|
cfg := Config{
|
|
NodeID: "test-node",
|
|
Version: "1.0.0",
|
|
Logger: log.New(io.Discard, "", 0),
|
|
MemoryBackend: customBackend,
|
|
}
|
|
|
|
agent, err := New(cfg)
|
|
require.NoError(t, err)
|
|
|
|
ctx := contextWithExecution(context.Background(), ExecutionContext{
|
|
SessionID: "test-session",
|
|
})
|
|
|
|
// Set via agent
|
|
err = agent.Memory().Set(ctx, "custom-key", "custom-value")
|
|
require.NoError(t, err)
|
|
|
|
// Verify directly on backend
|
|
val, found, _ := customBackend.Get(ScopeSession, "test-session", "custom-key")
|
|
assert.True(t, found)
|
|
assert.Equal(t, "custom-value", val)
|
|
}
|