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

LangChain

import { ChatOpenAI } from '@langchain/openai';
import { ConversationChain } from 'langchain/chains';
import { TDAIChatHistory } from '@ag-kit/adapter-langchain';

// Create TDAI-based chat history
const chatHistory = new TDAIChatHistory({
  sessionId: 'user-123',
  apiKey: process.env.TDAI_API_KEY!,
  baseURL: process.env.TDAI_BASE_URL
});

// Use with LangChain
const model = new ChatOpenAI();
const chain = new ConversationChain({
  llm: model,
  memory: chatHistory
});

const response = await chain.call({
  input: 'Hello, I am building a chatbot'
});

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