vector.py

from sentence_transformers import SentenceTransformer
from typing import List

class VectorModel:
    def __init__(self):
        # We use the 384-dim model recommended in the design
        self.model = SentenceTransformer('all-MiniLM-L6-v2')

    def encode(self, text: str) -> List[float]:
        return self.model.encode(text).tolist()

vector_model = VectorModel()