utils/helpers.py

from typing import List, Dict
def to_safe_items(res) -> List[Dict]:
    items = []
    if res and res.get("ids"):
        for i in range(len(res["ids"][0])):
            items.append({
                "text": res["documents"][0][i],
                "title": res["metadatas"][0][i].get("title","(untitled)"),
                "source": res["metadatas"][0][i].get("source","")
            })
    return items