Skip to content

Commit 298f2e1

Browse files
committed
Update KNN test query
1 parent e0f3b12 commit 298f2e1

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/test_knn_expression.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
# type: ignore
22
import abc
33
import time
4+
import random
45

56
import pytest_asyncio
67

@@ -41,7 +42,9 @@ async def embedding_bytes():
4142
@py_test_mark_asyncio
4243
async def test_vector_field(m: type[JsonModel], embedding_bytes):
4344
# Create a new instance of the Member model
44-
member = m(name="seth", embeddings=[[0.1, 0.2, 0.3]])
45+
dimensions = m.embeddings.field.vector_options.dimension
46+
embeddings = [random.uniform(-1, 1) for _ in range(dimensions)]
47+
member = m(name="seth", embeddings=[embeddings])
4548

4649
# Save the member to Redis
4750
mt = await member.save()
@@ -57,7 +60,7 @@ async def test_vector_field(m: type[JsonModel], embedding_bytes):
5760
reference_vector=embedding_bytes,
5861
)
5962

60-
query = m.find()
63+
query = m.find(knn=knn)
6164

6265
members = await query.all()
6366

0 commit comments

Comments
 (0)