# LLM Judge Prompts
# Used in: src/evaluators/llm_judge.py
llm_judge:
  system_prompt: |
    You are an expert grader that determines if answers to questions match a gold standard answer
  
  user_prompt: |
    Your task is to label an answer to a question as 'CORRECT' or 'WRONG'. You will be given the following data:
        (1) a question (posed by one user to another user),
        (2) a 'gold' (ground truth) answer,
        (3) a generated answer
    which you will score as CORRECT/WRONG.

    The point of the question is to ask about something one user should know about the other user based on their prior conversations.
    The gold answer will usually be a concise and short answer that includes the referenced topic, for example:
    Question: Do you remember what I got the last time I went to Hawaii?
    Gold answer: A shell necklace
    The generated answer might be much longer, but you should be generous with your grading - as long as it touches on the same topic as the gold answer, it should be counted as CORRECT.

    For time related questions, the gold answer will be a specific date, month, year, etc. The generated answer might be much longer or use relative time references (like "last Tuesday" or "next month"), but you should be generous with your grading - as long as it refers to the same date or time period as the gold answer, it should be counted as CORRECT. Even if the format differs (e.g., "May 7th" vs "7 May"), consider it CORRECT if it's the same date.

    Now it's time for the real question:
    Question: {question}
    Gold answer: {golden_answer}
    Generated answer: {generated_answer}

    First, provide a short (one sentence) explanation of your reasoning, then finish with CORRECT or WRONG.
    Do NOT include both CORRECT and WRONG in your response, or it will break the evaluation script.

    Just return the label CORRECT or WRONG in a json format with the key as "label".
  

# Online API Answer Prompts
# Used in: src/adapters/online_base.py and its subclasses
online_api:
  default:
    answer_prompt_mem0: |
      You are an intelligent memory assistant tasked with retrieving accurate information from conversation memories.

      # CONTEXT:
      You have access to memories from two speakers in a conversation. These memories contain
      timestamped information that may be relevant to answering the question.

      # INSTRUCTIONS:
      1. Carefully analyze all provided memories from both speakers
      2. Pay special attention to the timestamps to determine the answer
      3. If the question asks about a specific event or fact, look for direct evidence in the memories
      4. If the memories contain contradictory information, prioritize the most recent memory
      5. If there is a question about time references (like "last year", "two months ago", etc.),
         calculate the actual date based on the memory timestamp. For example, if a memory from
         4 May 2022 mentions "went to India last year," then the trip occurred in 2021.
      6. Always convert relative time references to specific dates, months, or years. For example,
         convert "last year" to "2022" or "two months ago" to "March 2023" based on the memory
         timestamp. Ignore the reference while answering the question.
      7. Focus only on the content of the memories from both speakers. Do not confuse character
         names mentioned in memories with the actual users who created those memories.
      8. The answer should be less than 5-6 words.

      # APPROACH (Think step by step):
      1. First, examine all memories that contain information related to the question
      2. Examine the timestamps and content of these memories carefully
      3. Look for explicit mentions of dates, times, locations, or events that answer the question
      4. If the answer requires calculation (e.g., converting relative time references), show your work
      5. Formulate a precise, concise answer based solely on the evidence in the memories
      6. Double-check that your answer directly addresses the question asked
      7. Ensure your final answer is specific and avoids vague time references

      {context}

      Question: {question}

      Answer:
    
    answer_prompt_memos: |
      You are a knowledgeable and helpful AI assistant.

      # CONTEXT:
      You have access to memories from two speakers in a conversation. These memories contain
      timestamped information that may be relevant to answering the question.

      # INSTRUCTIONS:
      1. Carefully analyze all provided memories. Synthesize information across different entries if needed to form a complete answer.
      2. Pay close attention to the timestamps to determine the answer. If memories contain contradictory information, the **most recent memory** is the source of truth.
      3. If the question asks about a specific event or fact, look for direct evidence in the memories.
      4. Your answer must be grounded in the memories. However, you may use general world knowledge to interpret or complete information found within a memory (e.g., identifying a landmark mentioned by description).
      5. If the question involves time references (like "last year", "two months ago", etc.), you **must** calculate the actual date based on the memory's timestamp. For example, if a memory from 4 May 2022 mentions "went to India last year," then the trip occurred in 2021.
      6. Always convert relative time references to specific dates, months, or years in your final answer.
      7. Do not confuse character names mentioned in memories with the actual users who created them.
      8. The answer must be brief (under 5-6 words) and direct, with no extra description.

      # APPROACH (Think step by step):
      1. First, examine all memories that contain information related to the question.
      2. Synthesize findings from multiple memories if a single entry is insufficient.
      3. Examine timestamps and content carefully, looking for explicit dates, times, locations, or events.
      4. If the answer requires calculation (e.g., converting relative time references), perform the calculation.
      5. Formulate a precise, concise answer based on the evidence from the memories (and allowed world knowledge).
      6. Double-check that your answer directly addresses the question asked and adheres to all instructions.
      7. Ensure your final answer is specific and avoids vague time references.

      {context}

      Question: {question}

      Answer:
    
    answer_prompt_cot: |
      You are an intelligent memory assistant tasked with retrieving accurate information from episodic memories.

      # CONTEXT:
      You have access to episodic memories from conversations between two speakers. These memories contain
      timestamped information that may be relevant to answering the question.

      # INSTRUCTIONS:
      Your goal is to synthesize information from all relevant memories to provide a comprehensive and accurate answer.
      You MUST follow a structured Chain-of-Thought process to ensure no details are missed.
      Actively look for connections between people, places, and events to build a complete picture. Synthesize information from different memories to answer the user's question.
      It is CRITICAL that you move beyond simple fact extraction and perform logical inference. When the evidence strongly suggests a connection, you must state that connection. Do not dismiss reasonable inferences as "speculation." Your task is to provide the most complete answer supported by the available evidence.

      # CRITICAL REQUIREMENTS:
      1. NEVER omit specific names - use "Amy's colleague Rob" not "a colleague"
      2. ALWAYS include exact numbers, amounts, prices, percentages, dates, times
      3. PRESERVE frequencies exactly - "every Tuesday and Thursday" not "twice a week"
      4. MAINTAIN all proper nouns and entities as they appear

      # RESPONSE FORMAT (You MUST follow this structure):

      ## STEP 1: RELEVANT MEMORIES EXTRACTION
      [List each memory that relates to the question, with its timestamp]
      - Memory 1: [timestamp] - [content]
      - Memory 2: [timestamp] - [content]
      ...

      ## STEP 2: KEY INFORMATION IDENTIFICATION
      [Extract ALL specific details from the memories]
      - Names mentioned: [list all person names, place names, company names]
      - Numbers/Quantities: [list all amounts, prices, percentages]
      - Dates/Times: [list all temporal information]
      - Frequencies: [list any recurring patterns]
      - Other entities: [list brands, products, etc.]

      ## STEP 3: CROSS-MEMORY LINKING
      [Identify entities that appear in multiple memories and link related information. Make reasonable inferences when entities are strongly connected.]
      - Shared entities: [list people, places, events mentioned across different memories]
      - Connections found: [e.g., "Memory 1 mentions A moved from hometown → Memory 2 mentions A's hometown is LA → Therefore A moved from LA"]
      - Inferred facts: [list any facts that require combining information from multiple memories]

      ## STEP 4: TIME REFERENCE CALCULATION
      [If applicable, convert relative time references]
      - Original reference: [e.g., "last year" from May 2022]
      - Calculated actual time: [e.g., "2021"]

      ## STEP 5: CONTRADICTION CHECK
      [If multiple memories contain different information]
      - Conflicting information: [describe]
      - Resolution: [explain which is most recent/reliable]

      ## STEP 6: DETAIL VERIFICATION CHECKLIST
      - [ ] All person names included: [list them]
      - [ ] All locations included: [list them]
      - [ ] All numbers exact: [list them]
      - [ ] All frequencies specific: [list them]
      - [ ] All dates/times precise: [list them]
      - [ ] All proper nouns preserved: [list them]

      ## STEP 7: ANSWER FORMULATION
      [Explain how you're combining the information to answer the question]

      ## FINAL ANSWER:
      [Provide the concise answer with ALL specific details preserved]

      ---

      {context}

      Question: {question}

      Now, follow the Chain-of-Thought process above to answer the question:

    answer_prompt_zep: |
      You are a helpful expert assistant answering questions based on the provided context.

      # CONTEXT:
      You have access to facts and entities from a conversation.

      # INSTRUCTIONS:
      1. Carefully analyze all provided memories (facts and entities)
      2. Pay special attention to the timestamps (event_time) to determine when events occurred
      3. If the question asks about a specific event or fact, look for direct evidence in the memories
      4. If the memories contain contradictory information, prioritize the most recent memory
      5. Always convert relative time references to specific dates, months, or years
      6. Be as specific as possible when talking about people, places, and events
      7. Timestamps in memories represent the actual time the event occurred, not the time the event was mentioned in a message

      # IMPORTANT CLARIFICATION:
      When interpreting memories, use the event_time (timestamp) to determine when the described event happened, not when someone talked about the event.

      Example:
      Memory: FACT (event_time: 2023-03-15T16:33:00Z): I went to the vet yesterday.
      Question: What day did I go to the vet?
      Correct Answer: March 15, 2023
      Explanation: Even though the phrase says "yesterday," the timestamp shows the event was recorded as happening on March 15th. Therefore, the actual vet visit happened on that date, regardless of the word "yesterday" in the text.

      # APPROACH (Think step by step):
      1. First, examine all memories (facts and entities) that contain information related to the question
      2. Examine the timestamps and content of these memories carefully
      3. Look for explicit mentions of dates, times, locations, or events that answer the question
      4. If the answer requires calculation (e.g., converting relative time references), show your work
      5. Formulate a precise, concise answer based solely on the evidence in the memories
      6. Double-check that your answer directly addresses the question asked
      7. Ensure your final answer is specific and avoids vague time references

      {context}

      Question: {question}
      Answer:

  templates:
    # Default template (shared by Mem0, Memos, Memu)
    # Uses speaker_1/speaker_2 as unified variable names
    default: |
      Memories for {speaker_1}:

      {speaker_1_memories}

      Memories for {speaker_2}:

      {speaker_2_memories}
    
    # Zep template - uses facts and entities format
    zep: |
      FACTS and ENTITIES represent relevant context to the current conversation.

      # These are the most relevant facts for the conversation along with the datetime of the event that the fact refers to.
      # If a fact mentions something happening a week ago, then the datetime will be the date time of last week and not the datetime
      # of when the fact was stated.
      # Timestamps in memories represent the actual time the event occurred, not the time the event was mentioned in a message.
          
      <FACTS>
      {facts}
      </FACTS>

      # These are the most relevant entities
      # ENTITY_NAME: entity summary
      <ENTITIES>
      {entities}
      </ENTITIES>
    


# Memory Extraction / Add Stage Prompts
# Used during the add stage to guide memory extraction and formatting
add_stage:
  # Mem0 custom memory extraction instructions
  mem0:
    custom_instructions: |
      Generate personal memories that follow these guidelines:
      
      1. Each memory should be self-contained with complete context, including:
         - The person's name, do not use "user" while creating memories
         - Personal details (career aspirations, hobbies, life circumstances)
         - Emotional states and reactions
         - Ongoing journeys or future plans
         - Specific dates when events occurred
      
      2. Include meaningful personal narratives focusing on:
         - Identity and self-acceptance journeys
         - Family planning and parenting
         - Creative outlets and hobbies
         - Mental health and self-care activities
         - Career aspirations and education goals
         - Important life events and milestones
      
      3. Make each memory rich with specific details rather than general statements
         - Include timeframes (exact dates when possible)
         - Name specific activities (e.g., "charity race for mental health" rather than just "exercise")
         - Include emotional context and personal growth elements
      
      4. Extract memories only from user messages, not incorporating assistant responses
      
      5. Format each memory as a paragraph with a clear narrative structure that captures the person's experience, challenges, and aspirations
  
  # Future: other system memory extraction instructions can be added here
  # memos:
  #   extraction_prompt: |
  #     ...
  # 
  # memu:
  #   extraction_prompt: |
  #     ...