export interface HistoryManager {
  addHistory(
    memoryId: string,
    previousValue: string | null,
    newValue: string | null,
    action: string,
    createdAt?: string,
    updatedAt?: string,
    isDeleted?: number,
    userId?: string,
    runId?: string,
    messageId?: string,
    happenedAt?: string,
  ): Promise<void>;
  getHistory(memoryId: string, options?: { userId?: string, runId?: string }): Promise<any[]>;
  getAllMessages(options?: { userId?: string, runId?: string, limit?: number }): Promise<any[]>;
  
  /**
   * Get messages surrounding a specific message to provide context
   * @param messageId The reference message ID
   * @param options Filter and window size options
   * @returns Array of context messages before and after the reference message
   */
  getMessageContext(
    messageId: string, 
    options: { 
      beforeCount?: number; 
      afterCount?: number; 
      userId?: string; 
      runId?: string;
    }
  ): Promise<any[]>;
  
  /**
   * Get detailed message information for a specific memory (for agents)
   * @param memoryId The memory ID to get details for
   * @returns Array of detailed message information
   */
  getMessageDetails(memoryId: string): Promise<any[]>;
  
  /**
   * Get conversation context for a specific memory (for agents)
   * @param memoryId The memory ID to get conversation context for
   * @returns Array of conversation context messages
   */
  getMessageConversationContext(memoryId: string): Promise<any[]>;
  
  // Entity management methods
  addEntity(
    entityId: string,
    label: string,
    type: string,
    userId?: string,
    runId?: string,
  ): Promise<void>;
  
  updateEntity(
    entityId: string,
    label: string,
    type: string,
    userId?: string,
    runId?: string,
  ): Promise<void>;
  
  deleteEntity(entityId: string): Promise<void>;
  
  getEntity(entityId: string): Promise<any | null>;
  
  getAllEntities(options?: { userId?: string, runId?: string }): Promise<any[]>;
  
  associateMemoryWithEntities(memoryId: string, entityIds: string[]): Promise<void>;
  
  getMemoryEntities(memoryId: string): Promise<string[]>;
  
  reset(): Promise<void>;
  close(): void;

  /**
   * Store raw messages independently of memory creation
   * @param messages Array of messages to store
   * @param metadata Message metadata including userId, runId, messageId
   * @returns Promise that resolves when messages are stored
   */
  storeRawMessages(
    messages: any[],
    metadata: {
      userId?: string;
      runId?: string;
      messageId: string;
      batchInfo?: any;
    }
  ): Promise<void>;

  /**
   * Search through all stored messages (not just memories)
   * @param query Search query or filters
   * @param options Search options including limit, userId, runId
   * @returns Array of matching messages
   */
  searchMessages(
    query: string,
    options?: {
      userId?: string;
      runId?: string;
      limit?: number;
      timeRange?: { start?: string; end?: string };
    }
  ): Promise<any[]>;

  /**
   * Get messages by message ID pattern or exact match
   * @param messageIdPattern Message ID or pattern to search for
   * @param options Additional search options
   * @returns Array of matching messages
   */
  getMessagesByPattern(
    messageIdPattern: string,
    options?: {
      userId?: string;
      runId?: string;
      exact?: boolean;
    }
  ): Promise<any[]>;
}