INNER CODE UNIT · Python

embed

scub-france/docling-Studio · embedding-service/main.py:62

async def embed(request: EmbedRequest) -> EmbedResponse:
    """Generate embeddings for a batch of texts."""
    if model is None:
        raise HTTPException(status_code=503, detail="Model not loaded yet")

    vectors = model.encode(
        request.texts,
        batch_size=BATCH_SIZE,
        show_progress_bar=False,
        normalize_embeddings=True,
    )
    return EmbedResponse(
        embeddings=vectors.tolist(),
        model=MODEL_NAME,
        dimension=model.get_sentence_embedding_dimension(),
    )

View source record →