11 lines
370 B
Python
11 lines
370 B
Python
from ollama import AsyncClient
|
|
|
|
from haiku.rag.config import Config
|
|
from haiku.rag.embeddings.base import EmbedderBase
|
|
|
|
|
|
class Embedder(EmbedderBase):
|
|
async def embed(self, text: str) -> list[float]:
|
|
client = AsyncClient(host=Config.OLLAMA_BASE_URL)
|
|
res = await client.embeddings(model=self._model, prompt=text)
|
|
return list(res["embedding"])
|