File size: 327 Bytes
0a40afa
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
"""Very small in‑memory retriever."""
from typing import List

class Retriever:
    def __init__(self):
        self.store = []

    def add(self, embedding: List[float], metadata: dict):
        self.store.append((embedding, metadata))

    def search(self, query_emb: List[float], k: int = 5):
        return self.store[:k]