File size: 381 Bytes
1668c89
3cded50
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from typing import List
from infinity_emb import AsyncEmbeddingEngine

async def embed_text(engine: AsyncEmbeddingEngine, sentences: List[str]): 
    async with engine: 
        embeddings, usage = await engine.embed(sentences=sentences)
    await engine.astart()
    embeddings, usage = await engine.embed(sentences=sentences)
    await engine.astop()
    return embeddings, usage