Function process_user

evaluation/scripts/personamem/pm_search.py:174–265

(row_data, conv_idx, frame, version, top_k=20)

def process_user(row_data, conv_idx, frame, version, top_k=20):
    persona_id = row_data["persona_id"]
    question_id = row_data["question_id"]
    question_type = row_data["question_type"]
    topic = row_data["topic"]
    question = row_data["user_question_or_message"]
    correct_answer = row_data["correct_answer"]
    all_options = row_data["all_options"]
    user_id = f"pm_exper_user_{conv_idx}_{version}"
    print(f"\n🔍 Processing conversation {conv_idx} for user {user_id}...")

    search_results = defaultdict(list)
    print("\n" + "-" * 80)
    print(f"🔎 [{conv_idx + 1}/589] Processing conversation {conv_idx}")
    print(f"❓ Question: {question}")
    print(f"🗂️  Options: {all_options}")
    print(f"🏷️  Type: {question_type}")
    print("-" * 80)

    existing_results, exists = load_existing_results(frame, version, conv_idx)
    if exists:
        print(f"♻️  Using existing results for conversation {conv_idx}")
        return existing_results

    if frame == "zep":
        from utils.client import ZepClient

        client = ZepClient()
        print("🔌 Using Zep client for search...")
        context, duration_ms = zep_search(client, user_id, question)

    elif frame == "mem0" or frame == "mem0-graph":
        from utils.client import Mem0Client

        client = Mem0Client(enable_graph="graph" in frame)
        print("🔌 Using Mem0 API client for search...")
        context, duration_ms = mem0_search(client, question, user_id, top_k)
    elif frame == "memos-api":
        from utils.client import MemosApiClient

        client = MemosApiClient()
        print("🔌 Using Memos API client for search...")
        context, duration_ms = memos_search(client, user_id, question, top_k=top_k)
    elif frame == "supermemory":
        from utils.client import SupermemoryClient

        client = SupermemoryClient()
        print("🔌 Using supermemory client for search...")
        context, duration_ms = supermemory_search(client, question, user_id, top_k)
    elif frame == "memu":
        from utils.client import MemuClient

        client = MemuClient()
        print("🔌 Using memu client for search...")
        context, duration_ms = memu_search(client, question, user_id, top_k)
    elif frame == "memobase":
        from utils.client import MemobaseClient