src/index.js

/**
 * AI Agents Framework
 *
 * A lightweight, educational implementation of LangChain/LangGraph
 * using node-llama-cpp for local inference.
 *
 * @module ai-agents-framework
 */

// Core
export {
  Runnable,
  RunnableSequence,
  RunnableParallel,
  BaseMessage,
  HumanMessage,
  AIMessage,
  SystemMessage,
  ToolMessage,
  RunnableConfig
} from './core/index.js';

// LLM
export {
  BaseLLM,
  LlamaCppLLM,
  ChatModel,
  StreamingLLM
} from './llm/index.js';

// Prompts
export {
  BasePromptTemplate,
  PromptTemplate,
  ChatPromptTemplate,
  FewShotPromptTemplate,
  PipelinePromptTemplate,
  SystemMessagePromptTemplate
} from './prompts/index.js';

// Output Parsers
export {
  BaseOutputParser,
  StringOutputParser,
  JsonOutputParser,
  StructuredOutputParser,
  ListOutputParser,
  RegexOutputParser
} from './output-parsers/index.js';

// Chains
export {
  BaseChain,
  LLMChain,
  SequentialChain,
  RouterChain,
  MapReduceChain,
  TransformChain
} from './chains/index.js';

// Tools
export {
  BaseTool,
  ToolExecutor,
  ToolRegistry,
  Calculator,
  WebSearch,
  WebScraper,
  FileReader,
  FileWriter,
  CodeExecutor
} from './tools/index.js';

// Agents
export {
  BaseAgent,
  AgentExecutor,
  ToolCallingAgent,
  ReActAgent,
  StructuredChatAgent,
  ConversationalAgent
} from './agents/index.js';

// Memory
export {
  BaseMemory,
  BufferMemory,
  WindowMemory,
  SummaryMemory,
  VectorMemory,
  EntityMemory
} from './memory/index.js';

// Graph
export {
  StateGraph,
  MessageGraph,
  CompiledGraph,
  GraphNode,
  GraphEdge,
  ConditionalEdge,
  Checkpoint,
  BaseCheckpointer,
  MemoryCheckpointer,
  FileCheckpointer,
  END
} from './graph/index.js';

// Utils
export {
  CallbackManager,
  TokenCounter,
  RetryManager,
  TimeoutManager,
  Logger,
  SchemaValidator
} from './utils/index.js';