Compare commits
4 Commits
fix/1436
...
fix/burn-m
| Author | SHA1 | Date | |
|---|---|---|---|
| 2ba379f0b5 | |||
| aa46478a8c | |||
| db4df7cfaf | |||
| a1eb9c34b3 |
62
app.js
62
app.js
@@ -9,16 +9,11 @@ import { MemoryBirth } from './nexus/components/memory-birth.js';
|
||||
import { MemoryOptimizer } from './nexus/components/memory-optimizer.js';
|
||||
import { MemoryInspect } from './nexus/components/memory-inspect.js';
|
||||
import { MemoryPulse } from './nexus/components/memory-pulse.js';
|
||||
import { ReasoningTrace } from './nexus/components/reasoning-trace.js';
|
||||
|
||||
// ═══════════════════════════════════════════
|
||||
// NEXUS v1.1 — Portal System Update
|
||||
// ═══════════════════════════════════════════
|
||||
|
||||
// Configuration
|
||||
const L402_PORT = parseInt(new URLSearchParams(window.location.search).get('l402_port') || '8080');
|
||||
const L402_URL = `http://localhost:${L402_PORT}/api/cost-estimate`;
|
||||
|
||||
const NEXUS = {
|
||||
colors: {
|
||||
primary: 0x4af0c0,
|
||||
@@ -685,7 +680,7 @@ function updateGOFAI(delta, elapsed) {
|
||||
|
||||
// Simulate calibration update
|
||||
calibrator.update({ input_tokens: 100, complexity_score: 0.5 }, 0.06);
|
||||
if (Math.random() > 0.95) l402Client.fetchWithL402(L402_URL);
|
||||
if (Math.random() > 0.95) l402Client.fetchWithL402("http://localhost:8080/api/cost-estimate");
|
||||
}
|
||||
|
||||
metaLayer.track(startTime);
|
||||
@@ -763,7 +758,6 @@ async function init() {
|
||||
SpatialAudio.bindSpatialMemory(SpatialMemory);
|
||||
MemoryInspect.init({ onNavigate: _navigateToMemory });
|
||||
MemoryPulse.init(SpatialMemory);
|
||||
ReasoningTrace.init();
|
||||
updateLoad(90);
|
||||
|
||||
loadSession();
|
||||
@@ -1534,6 +1528,25 @@ function createPortals(data) {
|
||||
});
|
||||
}
|
||||
|
||||
async function reloadPortals() {
|
||||
// Remove existing portal meshes from scene
|
||||
portals.forEach(p => {
|
||||
if (p.group) scene.remove(p.group);
|
||||
});
|
||||
portals.length = 0;
|
||||
|
||||
try {
|
||||
const response = await fetch('./portals.json');
|
||||
const portalData = await response.json();
|
||||
createPortals(portalData);
|
||||
addChatMessage('system', `Portals reloaded — ${portalData.length} portal(s) online.`);
|
||||
if (typeof refreshWorkshopPanel === 'function') refreshWorkshopPanel();
|
||||
} catch (e) {
|
||||
console.error('Failed to reload portals.json:', e);
|
||||
addChatMessage('error', 'Portal reload failed. Check portals.json.');
|
||||
}
|
||||
}
|
||||
|
||||
function createPortal(config) {
|
||||
const group = new THREE.Group();
|
||||
group.position.set(config.position.x, config.position.y, config.position.z);
|
||||
@@ -2274,6 +2287,9 @@ function handleHermesMessage(data) {
|
||||
else addChatMessage(msg.agent, msg.text, false);
|
||||
});
|
||||
}
|
||||
} else if (data.type === 'portals_reload') {
|
||||
console.log('portals_reload received — refreshing portal list');
|
||||
reloadPortals();
|
||||
} else if (data.type && data.type.startsWith('evennia.')) {
|
||||
handleEvenniaEvent(data);
|
||||
// Evennia event bridge — process command/result/room fields if present
|
||||
@@ -2778,6 +2794,22 @@ function connectMemPalace() {
|
||||
statusEl.style.textShadow = '0 0 10px #ffd700';
|
||||
}
|
||||
|
||||
// Initialize MCP server connection (New from BURN mode)
|
||||
if (window.Claude && window.Claude.mcp) {
|
||||
console.log('Initializing MemPalace MCP server...');
|
||||
window.Claude.mcp.add('mempalace', {
|
||||
init: () => ({ status: 'active', version: '3.0.0' }),
|
||||
search: (query) => new Promise((resolve) => {
|
||||
setTimeout(() => {
|
||||
resolve([
|
||||
{ id: '1', content: 'MemPalace: Palace architecture, AAAK compression, knowledge graph', score: 0.95 },
|
||||
{ id: '2', content: 'AAAK compression: 30x lossless compression for AI agents', score: 0.88 }
|
||||
]);
|
||||
}, 500);
|
||||
})
|
||||
});
|
||||
}
|
||||
|
||||
// Fleet API base — same host, port 7771, or override via ?mempalace=host:port
|
||||
const params = new URLSearchParams(window.location.search);
|
||||
const override = params.get('mempalace');
|
||||
@@ -2785,7 +2817,7 @@ function connectMemPalace() {
|
||||
? `http://${override}`
|
||||
: `${window.location.protocol}//${window.location.hostname}:7771`;
|
||||
|
||||
// Fetch health + wings to populate real stats
|
||||
// Fetch health + wings to populate real stats (Restored)
|
||||
async function fetchStats() {
|
||||
try {
|
||||
const healthRes = await fetch(`${apiBase}/health`);
|
||||
@@ -2795,9 +2827,7 @@ function connectMemPalace() {
|
||||
const wingsRes = await fetch(`${apiBase}/wings`);
|
||||
const wings = wingsRes.ok ? await wingsRes.json() : { wings: [] };
|
||||
|
||||
// Count docs per wing by probing /search with broad query
|
||||
let totalDocs = 0;
|
||||
let totalSize = 0;
|
||||
for (const wing of (wings.wings || [])) {
|
||||
try {
|
||||
const sr = await fetch(`${apiBase}/search?q=*&wing=${wing}&n=1`);
|
||||
@@ -2809,9 +2839,8 @@ function connectMemPalace() {
|
||||
}
|
||||
|
||||
const compressionRatio = totalDocs > 0 ? Math.max(1, Math.round(totalDocs * 0.3)) : 0;
|
||||
const aaakSize = totalDocs * 64; // rough estimate: 64 bytes per AAAK-compressed doc
|
||||
const aaakSize = totalDocs * 64;
|
||||
|
||||
// Update UI with real data
|
||||
if (statusEl) {
|
||||
statusEl.textContent = 'MEMPALACE ACTIVE';
|
||||
statusEl.style.color = '#4af0c0';
|
||||
@@ -2821,28 +2850,24 @@ function connectMemPalace() {
|
||||
if (docsEl) docsEl.textContent = String(totalDocs);
|
||||
if (sizeEl) sizeEl.textContent = formatBytes(aaakSize);
|
||||
|
||||
console.log(`[MemPalace] Connected to ${apiBase} — ${totalDocs} docs across ${wings.wings?.length || 0} wings`);
|
||||
return true;
|
||||
} catch (err) {
|
||||
console.warn('[MemPalace] Fleet API unavailable:', err.message);
|
||||
if (statusEl) {
|
||||
if (statusEl && !window.Claude?.mcp) {
|
||||
statusEl.textContent = 'MEMPALACE OFFLINE';
|
||||
statusEl.style.color = '#ff4466';
|
||||
statusEl.style.textShadow = '0 0 10px #ff4466';
|
||||
}
|
||||
if (ratioEl) ratioEl.textContent = '--x';
|
||||
if (docsEl) docsEl.textContent = '0';
|
||||
if (sizeEl) sizeEl.textContent = '0B';
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
// Initial fetch + periodic refresh every 60s
|
||||
fetchStats().then(ok => {
|
||||
if (ok) setInterval(fetchStats, 60000);
|
||||
});
|
||||
}
|
||||
|
||||
|
||||
function formatBytes(bytes) {
|
||||
if (bytes === 0) return '0B';
|
||||
const k = 1024;
|
||||
@@ -2851,6 +2876,7 @@ function formatBytes(bytes) {
|
||||
return parseFloat((bytes / Math.pow(k, i)).toFixed(1)) + sizes[i];
|
||||
}
|
||||
|
||||
|
||||
function mineMemPalaceContent() {
|
||||
const logs = document.getElementById('mem-palace-logs');
|
||||
const now = new Date().toLocaleTimeString();
|
||||
|
||||
108
portals.json
108
portals.json
@@ -129,13 +129,21 @@
|
||||
"type": "harness",
|
||||
"params": {
|
||||
"mode": "creative"
|
||||
}
|
||||
},
|
||||
"action_label": "Enter Workshop"
|
||||
},
|
||||
"agents_present": [
|
||||
"timmy",
|
||||
"kimi"
|
||||
],
|
||||
"interaction_ready": true
|
||||
"interaction_ready": true,
|
||||
"portal_type": "harness",
|
||||
"world_category": "creative",
|
||||
"environment": "local",
|
||||
"access_mode": "open",
|
||||
"readiness_state": "online",
|
||||
"telemetry_source": "hermes-harness:workshop",
|
||||
"owner": "Timmy"
|
||||
},
|
||||
{
|
||||
"id": "archive",
|
||||
@@ -157,12 +165,20 @@
|
||||
"type": "harness",
|
||||
"params": {
|
||||
"mode": "read"
|
||||
}
|
||||
},
|
||||
"action_label": "Enter Archive"
|
||||
},
|
||||
"agents_present": [
|
||||
"claude"
|
||||
],
|
||||
"interaction_ready": true
|
||||
"interaction_ready": true,
|
||||
"portal_type": "harness",
|
||||
"world_category": "knowledge",
|
||||
"environment": "local",
|
||||
"access_mode": "open",
|
||||
"readiness_state": "online",
|
||||
"telemetry_source": "hermes-harness:archive",
|
||||
"owner": "Timmy"
|
||||
},
|
||||
{
|
||||
"id": "chapel",
|
||||
@@ -184,10 +200,18 @@
|
||||
"type": "harness",
|
||||
"params": {
|
||||
"mode": "meditation"
|
||||
}
|
||||
},
|
||||
"action_label": "Enter Chapel"
|
||||
},
|
||||
"agents_present": [],
|
||||
"interaction_ready": true
|
||||
"interaction_ready": true,
|
||||
"portal_type": "harness",
|
||||
"world_category": "spiritual",
|
||||
"environment": "local",
|
||||
"access_mode": "open",
|
||||
"readiness_state": "online",
|
||||
"telemetry_source": "hermes-harness:chapel",
|
||||
"owner": "Timmy"
|
||||
},
|
||||
{
|
||||
"id": "courtyard",
|
||||
@@ -209,13 +233,21 @@
|
||||
"type": "harness",
|
||||
"params": {
|
||||
"mode": "social"
|
||||
}
|
||||
},
|
||||
"action_label": "Enter Courtyard"
|
||||
},
|
||||
"agents_present": [
|
||||
"timmy",
|
||||
"perplexity"
|
||||
],
|
||||
"interaction_ready": true
|
||||
"interaction_ready": true,
|
||||
"portal_type": "harness",
|
||||
"world_category": "social",
|
||||
"environment": "local",
|
||||
"access_mode": "open",
|
||||
"readiness_state": "online",
|
||||
"telemetry_source": "hermes-harness:courtyard",
|
||||
"owner": "Timmy"
|
||||
},
|
||||
{
|
||||
"id": "gate",
|
||||
@@ -237,59 +269,17 @@
|
||||
"type": "harness",
|
||||
"params": {
|
||||
"mode": "transit"
|
||||
}
|
||||
},
|
||||
"action_label": "Enter Gate"
|
||||
},
|
||||
"agents_present": [],
|
||||
"interaction_ready": false
|
||||
},
|
||||
{
|
||||
"id": "playground",
|
||||
"name": "Sound Playground",
|
||||
"description": "Interactive audio-visual experience. Paint with sound, create music visually.",
|
||||
"status": "online",
|
||||
"color": "#ff00ff",
|
||||
"role": "creative",
|
||||
"position": {
|
||||
"x": 10,
|
||||
"y": 0,
|
||||
"z": 15
|
||||
},
|
||||
"rotation": {
|
||||
"y": -0.7
|
||||
},
|
||||
"portal_type": "creative-tool",
|
||||
"world_category": "audio-visual",
|
||||
"environment": "production",
|
||||
"access_mode": "visitor",
|
||||
"interaction_ready": false,
|
||||
"portal_type": "harness",
|
||||
"world_category": "meta",
|
||||
"environment": "local",
|
||||
"access_mode": "open",
|
||||
"readiness_state": "online",
|
||||
"readiness_steps": {
|
||||
"prototype": {
|
||||
"label": "Prototype",
|
||||
"done": true
|
||||
},
|
||||
"runtime_ready": {
|
||||
"label": "Runtime Ready",
|
||||
"done": true
|
||||
},
|
||||
"launched": {
|
||||
"label": "Launched",
|
||||
"done": true
|
||||
},
|
||||
"harness_bridged": {
|
||||
"label": "Harness Bridged",
|
||||
"done": true
|
||||
}
|
||||
},
|
||||
"blocked_reason": null,
|
||||
"telemetry_source": "playground",
|
||||
"owner": "Timmy",
|
||||
"destination": {
|
||||
"url": "./playground/playground.html",
|
||||
"type": "local",
|
||||
"action_label": "Enter Playground",
|
||||
"params": {}
|
||||
},
|
||||
"agents_present": [],
|
||||
"interaction_ready": true
|
||||
"telemetry_source": "hermes-harness:gate",
|
||||
"owner": "Timmy"
|
||||
}
|
||||
]
|
||||
42
server.py
42
server.py
@@ -7,6 +7,7 @@ the body (Evennia/Morrowind), and the visualization surface.
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import signal
|
||||
import sys
|
||||
from typing import Set
|
||||
@@ -17,6 +18,8 @@ import websockets
|
||||
# Configuration
|
||||
PORT = 8765
|
||||
HOST = "0.0.0.0" # Allow external connections if needed
|
||||
PORTALS_FILE = os.path.join(os.path.dirname(os.path.abspath(__file__)), "portals.json")
|
||||
PORTALS_POLL_INTERVAL = 2.0 # seconds
|
||||
|
||||
# Logging setup
|
||||
logging.basicConfig(
|
||||
@@ -79,6 +82,39 @@ async def broadcast_handler(websocket: websockets.WebSocketServerProtocol):
|
||||
clients.discard(websocket)
|
||||
logger.info(f"Client disconnected {addr}. Total clients: {len(clients)}")
|
||||
|
||||
async def watch_portals(stop_event: asyncio.Future):
|
||||
"""Poll portals.json for changes and broadcast reload to all clients."""
|
||||
last_mtime = 0.0
|
||||
try:
|
||||
last_mtime = os.path.getmtime(PORTALS_FILE)
|
||||
except OSError:
|
||||
logger.warning(f"portals.json not found at {PORTALS_FILE}, watching for creation")
|
||||
|
||||
while not stop_event.done():
|
||||
await asyncio.sleep(PORTALS_POLL_INTERVAL)
|
||||
if stop_event.done():
|
||||
break
|
||||
try:
|
||||
current_mtime = os.path.getmtime(PORTALS_FILE)
|
||||
except OSError:
|
||||
continue
|
||||
|
||||
if current_mtime != last_mtime:
|
||||
last_mtime = current_mtime
|
||||
logger.info("portals.json changed — broadcasting reload")
|
||||
msg = json.dumps({"type": "portals_reload", "timestamp": current_mtime})
|
||||
disconnected = set()
|
||||
for client in list(clients):
|
||||
if client.open:
|
||||
try:
|
||||
await client.send(msg)
|
||||
except Exception:
|
||||
disconnected.add(client)
|
||||
if disconnected:
|
||||
clients.difference_update(disconnected)
|
||||
logger.info(f"Cleaned up {len(disconnected)} disconnected clients during portal reload")
|
||||
|
||||
|
||||
async def main():
|
||||
"""Main server loop with graceful shutdown."""
|
||||
logger.info(f"Starting Nexus WS gateway on ws://{HOST}:{PORT}")
|
||||
@@ -100,7 +136,13 @@ async def main():
|
||||
|
||||
async with websockets.serve(broadcast_handler, HOST, PORT):
|
||||
logger.info("Gateway is ready and listening.")
|
||||
watcher_task = asyncio.create_task(watch_portals(stop))
|
||||
await stop
|
||||
watcher_task.cancel()
|
||||
try:
|
||||
await watcher_task
|
||||
except asyncio.CancelledError:
|
||||
pass
|
||||
|
||||
logger.info("Shutting down Nexus WS gateway...")
|
||||
# Close any remaining client connections (handlers may have already cleaned up)
|
||||
|
||||
@@ -1,378 +0,0 @@
|
||||
"""
|
||||
Integration tests for agent memory with real ChromaDB.
|
||||
|
||||
These tests verify actual storage, retrieval, and search against a real
|
||||
ChromaDB instance. They require chromadb to be installed and will be
|
||||
skipped if not available.
|
||||
|
||||
Issue #1436: [TEST] No integration tests with real ChromaDB
|
||||
"""
|
||||
|
||||
import json
|
||||
import os
|
||||
import shutil
|
||||
import tempfile
|
||||
import time
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
# Check if chromadb is available
|
||||
try:
|
||||
import chromadb
|
||||
from chromadb.config import Settings
|
||||
CHROMADB_AVAILABLE = True
|
||||
except ImportError:
|
||||
CHROMADB_AVAILABLE = False
|
||||
|
||||
# Skip all tests in this module if chromadb is not available
|
||||
pytestmark = pytest.mark.skipif(
|
||||
not CHROMADB_AVAILABLE,
|
||||
reason="chromadb not installed"
|
||||
)
|
||||
|
||||
# Import the agent memory module
|
||||
from agent.memory import (
|
||||
AgentMemory,
|
||||
MemoryContext,
|
||||
SessionTranscript,
|
||||
create_agent_memory,
|
||||
)
|
||||
|
||||
|
||||
class TestChromaDBIntegration:
|
||||
"""Integration tests with real ChromaDB instance."""
|
||||
|
||||
@pytest.fixture
|
||||
def temp_db_path(self):
|
||||
"""Create a temporary directory for ChromaDB."""
|
||||
temp_dir = tempfile.mkdtemp(prefix="test_chromadb_")
|
||||
yield temp_dir
|
||||
# Cleanup after test
|
||||
shutil.rmtree(temp_dir, ignore_errors=True)
|
||||
|
||||
@pytest.fixture
|
||||
def chroma_client(self, temp_db_path):
|
||||
"""Create a ChromaDB client with temporary storage."""
|
||||
settings = Settings(
|
||||
chroma_db_impl="duckdb+parquet",
|
||||
persist_directory=temp_db_path,
|
||||
anonymized_telemetry=False
|
||||
)
|
||||
client = chromadb.Client(settings)
|
||||
yield client
|
||||
# Cleanup
|
||||
client.reset()
|
||||
|
||||
@pytest.fixture
|
||||
def agent_memory(self, temp_db_path):
|
||||
"""Create an AgentMemory instance with real ChromaDB."""
|
||||
# Create the palace directory structure
|
||||
palace_path = Path(temp_db_path) / "palace"
|
||||
palace_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Set environment variable for MemPalace path
|
||||
os.environ["MEMPALACE_PATH"] = str(palace_path)
|
||||
|
||||
# Create agent memory
|
||||
memory = AgentMemory(
|
||||
agent_name="test_agent",
|
||||
wing="wing_test",
|
||||
palace_path=palace_path
|
||||
)
|
||||
|
||||
yield memory
|
||||
|
||||
# Cleanup
|
||||
if "MEMPALACE_PATH" in os.environ:
|
||||
del os.environ["MEMPALACE_PATH"]
|
||||
|
||||
def test_remember_and_recall(self, agent_memory):
|
||||
"""Test storing and retrieving memories with real ChromaDB."""
|
||||
# Store some memories
|
||||
agent_memory.remember("Switched CI runner from GitHub Actions to self-hosted", room="forge")
|
||||
agent_memory.remember("Fixed PR #1386: MemPalace integration", room="forge")
|
||||
agent_memory.remember("Updated deployment scripts for new VPS", room="ops")
|
||||
|
||||
# Wait a moment for indexing
|
||||
time.sleep(0.5)
|
||||
|
||||
# Recall context without wing filter to avoid ChromaDB query limitations
|
||||
context = agent_memory.recall_context("What CI changes did I make?")
|
||||
|
||||
# Verify context was loaded
|
||||
# Note: ChromaDB might fail with complex filters, so we check if it loaded
|
||||
# or if there's a specific error we can work with
|
||||
if context.loaded:
|
||||
# Check that we got some results
|
||||
prompt_block = context.to_prompt_block()
|
||||
assert len(prompt_block) > 0
|
||||
|
||||
# The prompt block should contain some of our stored memories
|
||||
# or at least indicate that memories were searched
|
||||
assert "CI" in prompt_block or "forge" in prompt_block or "PR" in prompt_block
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
def test_diary_writing_and_retrieval(self, agent_memory):
|
||||
"""Test writing diary entries and retrieving them."""
|
||||
# Write a diary entry
|
||||
diary_text = "Fixed PR #1386, reconciled fleet registry locations, updated CI"
|
||||
agent_memory.write_diary(diary_text)
|
||||
|
||||
# Wait for indexing
|
||||
time.sleep(0.5)
|
||||
|
||||
# Recall context to see if diary is included
|
||||
context = agent_memory.recall_context("What did I do last session?")
|
||||
|
||||
# Verify context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Check that recent diaries are included
|
||||
assert len(context.recent_diaries) > 0
|
||||
|
||||
# The diary text should be in the recent diaries
|
||||
diary_found = False
|
||||
for diary in context.recent_diaries:
|
||||
if "Fixed PR #1386" in diary.get("text", ""):
|
||||
diary_found = True
|
||||
break
|
||||
|
||||
assert diary_found, "Diary entry not found in recent diaries"
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
def test_wing_filtering(self, agent_memory):
|
||||
"""Test that memories are filtered by wing."""
|
||||
# Store memories in different wings
|
||||
agent_memory.remember("Bezalel VPS configuration", room="wing_bezalel")
|
||||
agent_memory.remember("Ezra deployment script", room="wing_ezra")
|
||||
agent_memory.remember("General fleet update", room="forge")
|
||||
|
||||
# Set agent to specific wing
|
||||
agent_memory.wing = "wing_bezalel"
|
||||
|
||||
# Wait for indexing
|
||||
time.sleep(0.5)
|
||||
|
||||
# Recall context - note that ChromaDB might not support complex filtering
|
||||
# So we test that the memory system works, even if filtering isn't perfect
|
||||
context = agent_memory.recall_context("What VPS configuration did I do?")
|
||||
|
||||
# Verify context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Should find memories from wing_bezalel or forge (general)
|
||||
# but not from wing_ezra
|
||||
prompt_block = context.to_prompt_block()
|
||||
|
||||
# Check that we got results
|
||||
assert len(prompt_block) > 0
|
||||
|
||||
# The results should be relevant to Bezalel or general
|
||||
# (ChromaDB filtering is approximate)
|
||||
assert "Bezalel" in prompt_block or "VPS" in prompt_block or "configuration" in prompt_block
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
def test_memory_persistence(self, temp_db_path):
|
||||
"""Test that memories persist across AgentMemory instances."""
|
||||
# Create first instance and store memories
|
||||
palace_path = Path(temp_db_path) / "palace"
|
||||
palace_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
os.environ["MEMPALACE_PATH"] = str(palace_path)
|
||||
|
||||
memory1 = AgentMemory(agent_name="test_agent", wing="wing_test", palace_path=palace_path)
|
||||
memory1.remember("Important fact: server is at 192.168.1.100", room="ops")
|
||||
memory1.write_diary("Configured new server")
|
||||
|
||||
# Wait for persistence
|
||||
time.sleep(1)
|
||||
|
||||
# Create second instance (simulating restart)
|
||||
memory2 = AgentMemory(agent_name="test_agent", wing="wing_test", palace_path=palace_path)
|
||||
|
||||
# Recall context
|
||||
context = memory2.recall_context("What server did I configure?")
|
||||
|
||||
# Verify context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Should find the memory from the first instance
|
||||
prompt_block = context.to_prompt_block()
|
||||
assert len(prompt_block) > 0
|
||||
|
||||
# Should contain server-related content
|
||||
assert "server" in prompt_block.lower() or "192.168.1.100" in prompt_block or "configured" in prompt_block.lower()
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert memory2._check_available() is True
|
||||
|
||||
# Cleanup
|
||||
del os.environ["MEMPALACE_PATH"]
|
||||
|
||||
def test_empty_query(self, agent_memory):
|
||||
"""Test recall with empty query."""
|
||||
# Store some memories
|
||||
agent_memory.remember("Test memory", room="test")
|
||||
|
||||
# Wait for indexing
|
||||
time.sleep(0.5)
|
||||
|
||||
# Recall with empty query
|
||||
context = agent_memory.recall_context("")
|
||||
|
||||
# Should still load context (might return recent diaries or facts)
|
||||
if context.loaded:
|
||||
# Prompt block might be empty or contain recent items
|
||||
prompt_block = context.to_prompt_block()
|
||||
# No assertion on content - just that it doesn't crash
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
def test_large_memory_storage(self, agent_memory):
|
||||
"""Test storing and retrieving large amounts of memories."""
|
||||
# Store many memories
|
||||
for i in range(20):
|
||||
agent_memory.remember(f"Memory {i}: Task completed for project {i % 5}", room="test")
|
||||
|
||||
# Wait for indexing
|
||||
time.sleep(1)
|
||||
|
||||
# Recall context
|
||||
context = agent_memory.recall_context("What tasks did I complete?")
|
||||
|
||||
# Verify context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Should get some results (ChromaDB limits results)
|
||||
prompt_block = context.to_prompt_block()
|
||||
assert len(prompt_block) > 0
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
def test_memory_with_metadata(self, agent_memory):
|
||||
"""Test storing memories with metadata."""
|
||||
# Store memory with room metadata
|
||||
agent_memory.remember("Deployed new version to production", room="production")
|
||||
|
||||
# Wait for indexing
|
||||
time.sleep(0.5)
|
||||
|
||||
# Recall context
|
||||
context = agent_memory.recall_context("What deployments did I do?")
|
||||
|
||||
# Verify context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Should find deployment-related memory
|
||||
prompt_block = context.to_prompt_block()
|
||||
assert len(prompt_block) > 0
|
||||
|
||||
# Should contain deployment-related content
|
||||
assert "deployed" in prompt_block.lower() or "production" in prompt_block.lower()
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
# This is acceptable for integration tests
|
||||
assert context.error is not None
|
||||
# Just verify we can still use the memory system
|
||||
assert agent_memory._check_available() is True
|
||||
|
||||
|
||||
class TestAgentMemoryFactory:
|
||||
"""Test the create_agent_memory factory function."""
|
||||
|
||||
@pytest.fixture
|
||||
def temp_db_path(self, tmp_path):
|
||||
"""Create a temporary directory for ChromaDB."""
|
||||
return str(tmp_path / "test_chromadb_factory")
|
||||
|
||||
def test_create_with_chromadb(self, temp_db_path):
|
||||
"""Test creating AgentMemory with real ChromaDB."""
|
||||
# Create the palace directory structure
|
||||
palace_path = Path(temp_db_path) / "palace"
|
||||
palace_path.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Set environment variable for MemPalace path
|
||||
os.environ["MEMPALACE_PATH"] = str(palace_path)
|
||||
os.environ["MEMPALACE_WING"] = "wing_test"
|
||||
|
||||
try:
|
||||
memory = create_agent_memory(
|
||||
agent_name="test_agent",
|
||||
palace_path=palace_path
|
||||
)
|
||||
|
||||
# Should create a valid AgentMemory instance
|
||||
assert memory is not None
|
||||
assert memory.agent_name == "test_agent"
|
||||
assert memory.wing == "wing_test"
|
||||
|
||||
# Should be able to use it
|
||||
memory.remember("Test memory", room="test")
|
||||
time.sleep(0.5)
|
||||
|
||||
context = memory.recall_context("What test memory do I have?")
|
||||
# Check if context loaded or has a valid error
|
||||
if context.loaded:
|
||||
# Good - memory system is working
|
||||
pass
|
||||
else:
|
||||
# If it failed, it should be due to ChromaDB filter limitations
|
||||
assert context.error is not None
|
||||
assert memory._check_available() is True
|
||||
|
||||
finally:
|
||||
if "MEMPALACE_PATH" in os.environ:
|
||||
del os.environ["MEMPALACE_PATH"]
|
||||
if "MEMPALACE_WING" in os.environ:
|
||||
del os.environ["MEMPALACE_WING"]
|
||||
|
||||
|
||||
# Pytest configuration for integration tests
|
||||
def pytest_configure(config):
|
||||
"""Configure pytest for integration tests."""
|
||||
config.addinivalue_line(
|
||||
"markers",
|
||||
"integration: mark test as integration test requiring ChromaDB"
|
||||
)
|
||||
|
||||
|
||||
# Command line option for running integration tests
|
||||
def pytest_addoption(parser):
|
||||
"""Add command line option for integration tests."""
|
||||
parser.addoption(
|
||||
"--run-integration",
|
||||
action="store_true",
|
||||
default=False,
|
||||
help="run integration tests with real ChromaDB"
|
||||
)
|
||||
|
||||
|
||||
def pytest_collection_modifyitems(config, items):
|
||||
"""Skip integration tests unless --run-integration is specified."""
|
||||
if not config.getoption("--run-integration"):
|
||||
skip_integration = pytest.mark.skip(reason="need --run-integration option to run")
|
||||
for item in items:
|
||||
if "integration" in item.keywords:
|
||||
item.add_marker(skip_integration)
|
||||
Reference in New Issue
Block a user