Skip to content

Commit

Permalink
test: add test for additional params in FastEmbed
Browse files Browse the repository at this point in the history
  • Loading branch information
Pouyanpi committed Jan 9, 2025
1 parent db6245a commit c673f5c
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions tests/test_embeddings_fastembed.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,13 @@ def test_sync_embeddings():
assert len(result[0]) == 384


def test_aditional_params_with_fastembed():
model = FastEmbedEmbeddingModel("all-MiniLM-L6-v2", max_length=512, lazy_load=True)
result = model.encode(["test"])

assert len(result[0]) == 384


@pytest.mark.asyncio
async def test_async_embeddings():
model = FastEmbedEmbeddingModel("all-MiniLM-L6-v2")
Expand Down

0 comments on commit c673f5c

Please sign in to comment.