rust-kgdb 0.5.12 → 0.6.0

package/examples/underwriting-agent.js

@@ -35,7 +35,19 @@ const {
   DatalogProgram,
   evaluateDatalog,
   GraphFrame,
-  getVersion
+  getVersion,
+  // Memory Layer (v0.5.13+) - GraphDB-Powered Agent Memory
+  AgentState,
+  AgentRuntime,
+  WorkingMemory,
+  EpisodicMemory,
+  LongTermMemory,
+  MemoryManager,
+  // Governance Layer
+  GovernancePolicy,
+  GovernanceEngine,
+  // Scope Layer
+  AgentScope
 } = require('../index.js')
 
 // ═══════════════════════════════════════════════════════════════════════════
@@ -283,6 +295,67 @@ async function main() {
   console.log(`  Agent "${agent.getName()}" spawned with model: ${agent.getModel()}`)
   console.log()
 
+  // ─────────────────────────────────────────────────────────────────────────
+  // PHASE 2.5: Memory Layer Initialization (NEW in v0.5.13)
+  // ─────────────────────────────────────────────────────────────────────────
+
+  console.log('┌─ MEMORY: AgentRuntime + MemoryManager + Governance ────────────────────┐')
+  console.log('│  Memory Layer: Working + Episodic + LongTerm (GraphDB-backed)          │')
+  console.log('└─────────────────────────────────────────────────────────────────────────┘')
+
+  // Create agent runtime with identity
+  const runtime = new AgentRuntime({
+    name: 'underwriter',
+    model: MODEL,
+    tools: ['kg.sparql.query', 'kg.datalog.apply', 'kg.embeddings.search'],
+    memoryCapacity: 100,
+    episodeLimit: 1000
+  })
+
+  // Initialize memory manager with GraphDB
+  const memoryManager = new MemoryManager(runtime, db, {
+    recencyWeight: 0.3,
+    relevanceWeight: 0.5,
+    importanceWeight: 0.2
+  })
+
+  // Create governance policy for underwriting
+  const policy = new GovernancePolicy({
+    capabilities: ['ReadKG', 'ExecuteTool', 'UseEmbeddings', 'AccessMemory'],
+    maxMemoryMB: 256,
+    maxExecutionTimeMs: 60000,
+    maxToolCalls: 100
+  })
+
+  // Initialize governance engine
+  const governance = new GovernanceEngine(policy)
+
+  // Create agent scope with namespace isolation
+  const scope = new AgentScope({
+    name: 'underwriting-scope',
+    graphUri: 'http://underwriting.org/data',
+    allowedGraphs: ['http://underwriting.org/data', 'http://underwriting.org/memory'],
+    maxToolCalls: 50
+  })
+
+  console.log(`  Runtime ID: ${runtime.id}`)
+  console.log(`  Memory: recency=${memoryManager.weights.recency}, relevance=${memoryManager.weights.relevance}, importance=${memoryManager.weights.importance}`)
+  console.log(`  Governance: ${policy.capabilities.size} capabilities, ${policy.limits.maxToolCalls} max tool calls`)
+  console.log()
+
+  // Store initial context in working memory
+  memoryManager.addToWorking({
+    type: 'session_start',
+    task: 'underwriting_analysis',
+    triple_count: tripleCount
+  })
+
+  // Transition to READY state first
+  runtime.transitionTo(AgentState.READY)
+
+  // Start execution for episodic tracking
+  const executionId = runtime.startExecution('Commercial underwriting risk analysis')
+
   // ─────────────────────────────────────────────────────────────────────────
   // PHASE 3: Execute Underwriting Tools
   // ─────────────────────────────────────────────────────────────────────────
@@ -475,6 +548,54 @@ async function main() {
     console.log()
   }
 
+  // ─────────────────────────────────────────────────────────────────────────
+  // Memory Layer: Store Tool Executions as Episodes (NEW in v0.5.13)
+  // ─────────────────────────────────────────────────────────────────────────
+
+  await memoryManager.storeExecution({
+    id: `${executionId}-sparql`,
+    prompt: 'Query all business accounts',
+    tool: 'kg.sparql.query',
+    output: { accountCount: accounts.length },
+    success: true,
+    durationMs: 50
+  })
+
+  await memoryManager.storeExecution({
+    id: `${executionId}-graphframe`,
+    prompt: 'Risk network analysis',
+    tool: 'kg.graphframe.analyze',
+    output: { vertices: riskVertices.length, pageRank: 'complete' },
+    success: true,
+    durationMs: 30
+  })
+
+  await memoryManager.storeExecution({
+    id: `${executionId}-datalog`,
+    prompt: 'Apply underwriting rules',
+    tool: 'kg.datalog.infer',
+    output: { decisions: decisions.length, autoApprove: autoApprove.length, referToUW: referToUW.length },
+    success: true,
+    durationMs: 20
+  })
+
+  // Track resource usage in scope
+  scope.trackUsage('toolCalls', 4)
+  scope.trackUsage('graphQueries', 2)
+
+  // Add findings to working memory
+  memoryManager.addToWorking({
+    type: 'findings',
+    accountsProcessed: accounts.length,
+    autoApproveCount: autoApprove.length,
+    referToUWCount: referToUW.length
+  })
+
+  // Complete execution
+  runtime.completeExecution({ accounts: accounts.length, decisions: decisions.length }, true)
+  const memStats = memoryManager.getStats()
+  const remaining = scope.getRemainingResources()
+
   // ─────────────────────────────────────────────────────────────────────────
   // PHASE 5: Generate Execution Witness (Proof Theory)
   // ─────────────────────────────────────────────────────────────────────────
@@ -498,6 +619,16 @@ async function main() {
   console.log('    - BUS004 (Downtown): MODERATE - flood exposure, established')
   console.log()
 
+  // Memory Layer Statistics (NEW in v0.5.13)
+  console.log('  MEMORY LAYER STATISTICS (v0.5.13+):')
+  console.log(`    Runtime ID:             ${(runtime.id || 'unknown').substring(0, 36)}`)
+  console.log(`    Final State:            ${runtime.state}`)
+  console.log(`    Working Memory Items:   ${memStats.working.contextSize}`)
+  console.log(`    Episodic Episodes:      ${memStats.episodic.episodeCount}`)
+  console.log(`    Tool Calls Remaining:   ${remaining.toolCalls}`)
+  console.log(`    Graph Queries Used:     ${scope.usage.graphQueries}`)
+  console.log()
+
   // Execution Witness
   const witness = {
     agent: agent.getName(),
@@ -515,7 +646,16 @@ async function main() {
       referToUW: referToUW.map(d => d.args[0])
     },
     premiums: premiums.map(p => ({ id: p.id, premium: p.premium, decision: p.decision })),
-    trace: agent.getTrace()
+    trace: agent.getTrace(),
+    // Memory Layer (v0.5.13+)
+    memory: {
+      runtimeId: runtime.id,
+      finalState: runtime.state,
+      workingMemoryItems: memStats.working.contextSize,
+      episodicMemoryEpisodes: memStats.episodic.episodeCount,
+      toolCallsUsed: scope.usage.toolCalls,
+      governanceAuditEntries: governance.auditLog.length
+    }
   }
 
   // Generate deterministic hash