Method embed

paperflow/providers/embedding.py:112–121

(self, text: str)

        self._timeout = timeout

    def embed(self, text: str) -> List[float]:
        response = requests.post(
            f"{self._base_url}/api/embeddings",
            json={"model": self.model, "prompt": text or " "},
            timeout=self._timeout,
        )
        response.raise_for_status()
        data = response.json()
        vector = list(data.get("embedding") or [])
        return _resize(vector, self.dimensions)

    def embed_batch(self, texts: Iterable[str]) -> List[List[float]]:
        return [self.embed(text) for text in texts]