import {type ChatMessage, type CortexVariables, getCortexResponse, type MemorySection} from "./utils";
import { logger } from '../utils/logger';

const SEARCH_MEMORY_QUERY = `
query SearchMemory($contextId: String, $chatHistory: [MultiMessage], $aiName: String, $section: String) {
  sys_search_memory(contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName, section: $section) {
    result
    tool
    warnings
    errors
  }
}
`

const MANAGE_MEMORY_QUERY = `
query ManageMemory($contextId: String, $chatHistory: [MultiMessage], $aiName: String) {
  sys_memory_manager(contextId: $contextId, chatHistory: $chatHistory, aiName: $aiName) {
    result
    tool
    warnings
    errors
  }
}
`

const READ_MEMORY = `
query ReadMemory($contextId: String, $section: String, $priority: Int, $recentHours: Int, $numResults: Int, $stripMetadata: Boolean) {
  sys_read_memory(contextId: $contextId, section: $section, priority: $priority, recentHours: $recentHours, numResults: $numResults, stripMetadata: $stripMetadata) {
    result
    tool
    warnings
    errors
  }
}
`

export async function searchMemory(contextId: string,
  aiName: string,
  chatHistory: ChatMessage[],
  section: MemorySection
) {
logger.log('Searching memory', contextId, aiName);
const variables: CortexVariables = {
chatHistory,
contextId,
aiName,
section
}

const res = await getCortexResponse(variables, SEARCH_MEMORY_QUERY);

return res.sys_search_memory;
}

export async function manageMemory(contextId: string,
                                   aiName: string,
                                   chatHistory: ChatMessage[]
) {
  logger.log('Managing memory', contextId, aiName);
  const variables: CortexVariables = {
    chatHistory,
    contextId,
    aiName,
  }

  const res = await getCortexResponse(variables, MANAGE_MEMORY_QUERY);

  return res.sys_memory_manager;
}

export async function readMemory(contextId: string,
                                 section: MemorySection,
                                 priority: number = 0,
                                 recentHours: number = 0,
                                 numResults: number = 0,
                                 stripMetadata: boolean = false
) {

  const variables: CortexVariables = {
    section,
    contextId,
    priority,
    recentHours,
    numResults,
    stripMetadata
  }

  const res = await getCortexResponse(variables, READ_MEMORY);

  return res.sys_read_memory;
}