Agent Integration

Complete guide to integrating AG-Kit’s memory system with AI agents and frameworks

AG-Kit’s memory system is designed to seamlessly integrate with AI agents, providing automatic context management, persistent knowledge storage, and intelligent memory operations. This guide covers integration patterns, framework-specific implementations, and best practices.

Integration Overview

Memory-Agent Workflow

Core Integration Patterns

1. Automatic Memory Management

• Agent automatically stores conversation events
• Context retrieval happens transparently
• Token limits managed automatically

2. Dual-Layer Memory

• Short-term: Recent conversation context
• Long-term: Persistent user knowledge and preferences

3. Intelligent Context Engineering

• Automatic summarization for long conversations
• Token-aware context trimming
• Conversation branching for experimentation

Framework Integration

llamaindex

import { LlamaIndexAgent } from '@ag-kit/adapter-llamaindex';
import { ConversationMemory } from '@ag-kit/adapter-llamaindex';
import { createMemory } from 'llamaindex';

// Create conversation memory manager
const conversationMemory = new ConversationMemory({
  memoryFactory: () => createMemory()
});

// Create LlamaIndex agent with AG-Kit integration
const agent = new LlamaIndexAgent({
  name: 'llamaindex-assistant',
  model: provider,
  conversationMemory: conversationMemory,
  tools: []
});

// Use with conversation ID for memory isolation
const response = await agent.run({
  input: 'Hello, remember my preferences',
  conversationId: 'user-123'
});

Configuration & Best Practices

Memory Configuration

TypeScript

// Production configuration
const memory = MySQLMemory.create({
  host: process.env.DB_HOST,
  database: process.env.DB_NAME,
  sessionId: userId,

  // Context engineering
  maxTokens: 8000,
  enableContextManagement: true
});

const agent = new Agent({
  name: 'production-assistant',
  model: provider,
  memory: memory,

  // Memory-aware configuration
  maxContextTokens: 8000,
  memoryRetrievalLimit: 10
});

Troubleshooting

Common Integration Issues

Memory Not Persisting

TypeScript

// ❌ Wrong: Memory instance not shared
const agent1 = new Agent({ memory: new InMemoryMemory() });
const agent2 = new Agent({ memory: new InMemoryMemory() });

// ✅ Correct: Shared memory instance
const sharedMemory = new InMemoryMemory();
const agent1 = new Agent({ memory: sharedMemory });
const agent2 = new Agent({ memory: sharedMemory });

Session Isolation Issues

TypeScript

// ❌ Wrong: Same session for different users
const agent = new Agent({ memory, sessionId: 'default' });

// ✅ Correct: Unique session per user
const agent = new Agent({
  memory,
  sessionId: `user-${userId}`
});

Next Steps

Short-term Memory

Learn about conversation context management

Long-term Memory

Implement persistent knowledge storage

Session Management

Handle multi-user scenarios

API Reference

Complete API documentation