import type {
  AxChatRequest,
  AxChatResponseResult,
  AxFunctionResult,
} from '../ai/types.js';

export type AxMemoryData = {
  tags?: string[];
  role: AxChatRequest['chatPrompt'][number]['role'];
  updatable?: boolean;
  chat: {
    index: number;
    value: Omit<AxChatRequest['chatPrompt'][number], 'role'>;
  }[];
}[];

export interface AxAIMemory {
  addRequest(result: AxChatRequest['chatPrompt'], sessionId?: string): void;

  addResponse(
    results: Readonly<AxChatResponseResult[]>,
    sessionId?: string
  ): void;

  updateResult(
    results: Readonly<AxChatResponseResult> & { delta?: string },
    sessionId?: string
  ): void;

  addFunctionResults(
    results: Readonly<AxFunctionResult[]>,
    sessionId?: string
  ): void;

  history(index: number, sessionId?: string): AxChatRequest['chatPrompt'];
  reset(sessionId?: string): void;

  getLast(sessionId?: string): AxMemoryData[number] | undefined;

  addTag(name: string, sessionId?: string): void;
  rewindToTag(name: string, sessionId?: string): AxMemoryData;
  removeByTag(name: string, sessionId?: string): AxMemoryData;
}