Skip to content

Commit 61e6fa3

Browse files
poissoncorppoissoncorp
committed
RDBC-1013 Code format
1 parent 98829b7 commit 61e6fa3

3 files changed

Lines changed: 55 additions & 27 deletions

File tree

‎ravendb/documents/session/query.py‎

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2038,14 +2038,13 @@ def vector_search_with_base64(
20382038
)
20392039
return self
20402040

2041-
20422041
def vector_search_with_base64_i8(
2043-
self,
2044-
embedding_field: str,
2045-
vector: list[float] | str,
2046-
minimum_similarity: float = None,
2047-
number_of_candidates: int = None,
2048-
is_exact: bool = VectorSearch.DEFAULT_IS_EXACT,
2042+
self,
2043+
embedding_field: str,
2044+
vector: list[float] | str,
2045+
minimum_similarity: float = None,
2046+
number_of_candidates: int = None,
2047+
is_exact: bool = VectorSearch.DEFAULT_IS_EXACT,
20492048
):
20502049
"""Perform vector search over a base64-encoded int8 vector field.
20512050
The vector parameter can be either a list of int or a base64 string."""
@@ -2060,14 +2059,13 @@ def vector_search_with_base64_i8(
20602059
)
20612060
return self
20622061

2063-
20642062
def vector_search_with_base64_i1(
2065-
self,
2066-
embedding_field: str,
2067-
vector: list[int] | str,
2068-
minimum_similarity: float = None,
2069-
number_of_candidates: int = None,
2070-
is_exact: bool = VectorSearch.DEFAULT_IS_EXACT,
2063+
self,
2064+
embedding_field: str,
2065+
vector: list[int] | str,
2066+
minimum_similarity: float = None,
2067+
number_of_candidates: int = None,
2068+
is_exact: bool = VectorSearch.DEFAULT_IS_EXACT,
20712069
):
20722070
"""Perform vector search over a base64-encoded int1 vector field.
20732071
The vector parameter can be either a list of int or a base64 string."""

‎ravendb/tests/documents_tests/query_tests/test_vector_search.py‎

Lines changed: 21 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,9 @@ def test_to_int1_padding_trimmed(self):
128128
class TestVectorSearch(TestBase):
129129
def test_should_generate_rql_with_text_field_using_named_ai_task(self):
130130
with self.store.open_session() as session:
131-
q = session.query(object_type=Dto).vector_search_text("EmbeddingField", "fishing", embedding_generation_task_identifier="my-ai-task")
131+
q = session.query(object_type=Dto).vector_search_text(
132+
"EmbeddingField", "fishing", embedding_generation_task_identifier="my-ai-task"
133+
)
132134
self.assertEqual(
133135
"from 'Dtoes' where vector.search(embedding.text(EmbeddingField, ai.task('my-ai-task')), $p0)",
134136
q._to_string(),
@@ -143,31 +145,45 @@ def test_should_generate_rql_with_text_field_using_named_ai_task(self):
143145
)
144146

145147
q2 = session.query(object_type=Dto).vector_search_text(
146-
"EmbeddingField", "fishing", target_quantization=VectorEmbeddingType.BINARY, embedding_generation_task_identifier="my-ai-task"
148+
"EmbeddingField",
149+
"fishing",
150+
target_quantization=VectorEmbeddingType.BINARY,
151+
embedding_generation_task_identifier="my-ai-task",
147152
)
148153
self.assertEqual(
149154
"from 'Dtoes' where vector.search(embedding.text_i1(EmbeddingField, ai.task('my-ai-task')), $p0)",
150155
q2._to_string(),
151156
)
152157

153158
q2_exact = session.query(object_type=Dto).vector_search_text(
154-
"EmbeddingField", "fishing", target_quantization=VectorEmbeddingType.BINARY, embedding_generation_task_identifier="my-ai-task", is_exact=True
159+
"EmbeddingField",
160+
"fishing",
161+
target_quantization=VectorEmbeddingType.BINARY,
162+
embedding_generation_task_identifier="my-ai-task",
163+
is_exact=True,
155164
)
156165
self.assertEqual(
157166
"from 'Dtoes' where exact(vector.search(embedding.text_i1(EmbeddingField, ai.task('my-ai-task')), $p0))",
158167
q2_exact._to_string(),
159168
)
160169

161170
q3 = session.query(object_type=Dto).vector_search_text(
162-
"EmbeddingField", "fishing", target_quantization=VectorEmbeddingType.INT8, embedding_generation_task_identifier="my-ai-task",
171+
"EmbeddingField",
172+
"fishing",
173+
target_quantization=VectorEmbeddingType.INT8,
174+
embedding_generation_task_identifier="my-ai-task",
163175
)
164176
self.assertEqual(
165177
"from 'Dtoes' where vector.search(embedding.text_i8(EmbeddingField, ai.task('my-ai-task')), $p0)",
166178
q3._to_string(),
167179
)
168180

169181
q3_exact = session.query(object_type=Dto).vector_search_text(
170-
"EmbeddingField", "fishing", embedding_generation_task_identifier="my-ai-task", is_exact=True, target_quantization=VectorEmbeddingType.INT8,
182+
"EmbeddingField",
183+
"fishing",
184+
embedding_generation_task_identifier="my-ai-task",
185+
is_exact=True,
186+
target_quantization=VectorEmbeddingType.INT8,
171187
)
172188
self.assertEqual(
173189
"from 'Dtoes' where exact(vector.search(embedding.text_i8(EmbeddingField, ai.task('my-ai-task')), $p0))",

‎ravendb/tests/dotnet_migrated_tests/test_ravenDB_22076.py‎

Lines changed: 22 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -86,15 +86,19 @@ def test_rql_generation(self):
8686
q4 = session.query(object_type=Dto).vector_search("VectorField", "aaaa==")
8787
self.assertEqual("from 'Dtoes' where vector.search(VectorField, $p0)", q4._to_string())
8888

89-
q5 = session.query(object_type=Dto).vector_search_text("TextField", "aaaa", target_quantization=VectorEmbeddingType.INT8)
89+
q5 = session.query(object_type=Dto).vector_search_text(
90+
"TextField", "aaaa", target_quantization=VectorEmbeddingType.INT8
91+
)
9092
self.assertEqual("from 'Dtoes' where vector.search(embedding.text_i8(TextField), $p0)", q5._to_string())
9193

9294
q6 = session.query(object_type=Dto).vector_search_i8("EmbeddingField", [2, 3], 0.65)
9395
self.assertEqual(
9496
"from 'Dtoes' where vector.search(embedding.i8(EmbeddingField), $p0, 0.65, null)", q6._to_string()
9597
)
9698

97-
q7 = session.query(object_type=Dto).vector_search_text("TextField", "aaaa", target_quantization=VectorEmbeddingType.INT8)
99+
q7 = session.query(object_type=Dto).vector_search_text(
100+
"TextField", "aaaa", target_quantization=VectorEmbeddingType.INT8
101+
)
98102
self.assertEqual("from 'Dtoes' where vector.search(embedding.text_i8(TextField), $p0)", q7._to_string())
99103

100104
# q8 = session.query(object_type=Dto).vector_search_with_field()
@@ -117,13 +121,16 @@ def test_rql_generation_2(self):
117121
"from 'Dtoes' where vector.search(embedding.i8(EmbeddingField), $p0, 0.65, null)", q1._to_string()
118122
)
119123

120-
q2 = session.query(object_type=Dto).vector_search("EmbeddingField", [2.5, 3.3], 0.65, target_quantization=VectorEmbeddingType.INT8
124+
q2 = session.query(object_type=Dto).vector_search(
125+
"EmbeddingField", [2.5, 3.3], 0.65, target_quantization=VectorEmbeddingType.INT8
121126
)
122127
self.assertEqual(
123128
"from 'Dtoes' where vector.search(embedding.f32_i8(EmbeddingField), $p0, 0.65, null)", q2._to_string()
124129
)
125130

126-
q3 = session.query(object_type=Dto).vector_search("EmbeddingField", "abcd==", 0.75, target_quantization=VectorEmbeddingType.INT8)
131+
q3 = session.query(object_type=Dto).vector_search(
132+
"EmbeddingField", "abcd==", 0.75, target_quantization=VectorEmbeddingType.INT8
133+
)
127134
self.assertEqual(
128135
"from 'Dtoes' where vector.search(embedding.f32_i8(EmbeddingField), $p0, 0.75, null)", q3._to_string()
129136
)
@@ -153,9 +160,12 @@ def test_rql_generation_3(self):
153160
q1 = session.query(object_type=Dto).vector_search_with_field_for_document("VectorField", "docs/1-A")
154161
self.assertEqual("from 'Dtoes' where vector.search(VectorField, embedding.forDoc($p0))", q1._to_string())
155162

156-
q2 = session.query(object_type=Dto).vector_search_text_for_document("VectorField", "docs/1-A", target_quantization=VectorEmbeddingType.INT8)
163+
q2 = session.query(object_type=Dto).vector_search_text_for_document(
164+
"VectorField", "docs/1-A", target_quantization=VectorEmbeddingType.INT8
165+
)
157166
self.assertEqual(
158-
"from 'Dtoes' where vector.search(embedding.text_i8(VectorField), embedding.forDoc($p0))", q2._to_string()
167+
"from 'Dtoes' where vector.search(embedding.text_i8(VectorField), embedding.forDoc($p0))",
168+
q2._to_string(),
159169
)
160170

161171
# withField
@@ -201,13 +211,17 @@ def test_rql_generation_3(self):
201211
self.assertEqual("from 'Dtoes' where vector.search(VectorField, $p0)", q15._to_string())
202212

203213
# embeddingTaskIdentifier
204-
q16 = session.query(object_type=Dto).vector_search_text("VectorField", "hello", embedding_generation_task_identifier="my-ai-task")
214+
q16 = session.query(object_type=Dto).vector_search_text(
215+
"VectorField", "hello", embedding_generation_task_identifier="my-ai-task"
216+
)
205217
self.assertEqual(
206218
"from 'Dtoes' where vector.search(embedding.text(VectorField, ai.task('my-ai-task')), $p0)",
207219
q16._to_string(),
208220
)
209221

210-
q17 = session.query(object_type=Dto).vector_search_text_for_document("VectorField", "hello", embedding_generation_task_identifier="my-ai-task")
222+
q17 = session.query(object_type=Dto).vector_search_text_for_document(
223+
"VectorField", "hello", embedding_generation_task_identifier="my-ai-task"
224+
)
211225
self.assertEqual(
212226
"from 'Dtoes' where vector.search(embedding.text(VectorField, ai.task('my-ai-task')), embedding.forDoc($p0))",
213227
q17._to_string(),

0 commit comments

Comments
 (0)