Skip to content

Commit 500613e

Browse files
committed
fixed sample names
1 parent fb19578 commit 500613e

File tree

2 files changed

+11
-10
lines changed

2 files changed

+11
-10
lines changed

datastore/cloud-client/vector_search.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -73,8 +73,8 @@ def vector_search_prefilter(db):
7373
return vector_query
7474

7575

76-
def vector_search_distance_result_field(db):
77-
# [START datastore_vector_search_distance_result_field]
76+
def vector_search_distance_result_property(db):
77+
# [START datastore_vector_search_distance_result_property]
7878
from google.cloud.datastore.vector import DistanceMeasure
7979
from google.cloud.datastore.vector import Vector
8080
from google.cloud.datastore.vector import FindNearest
@@ -92,7 +92,7 @@ def vector_search_distance_result_field(db):
9292

9393
for entity in vector_query.fetch():
9494
print(f"{entity.id}, Distance: {entity['vector_distance']}")
95-
# [END datastore_vector_search_distance_result_field]
95+
# [END datastore_vector_search_distance_result_property]
9696
return vector_query
9797

9898

@@ -119,8 +119,8 @@ def vector_search_distance_threshold(db):
119119
return vector_query
120120

121121

122-
def vector_search_large_query(db):
123-
# [START datastore_vector_search_large_reqponse]
122+
def vector_search_large_response(db):
123+
# [START datastore_vector_search_large_response]
124124
from google.cloud.datastore.vector import DistanceMeasure
125125
from google.cloud.datastore.vector import Vector
126126
from google.cloud.datastore.vector import FindNearest

datastore/cloud-client/vector_search_test.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,10 @@
2020

2121
from vector_search import store_vectors
2222
from vector_search import vector_search_basic
23-
from vector_search import vector_search_distance_result_field
23+
from vector_search import vector_search_distance_result_property
2424
from vector_search import vector_search_distance_threshold
2525
from vector_search import vector_search_prefilter
26+
from vector_search import vector_search_large_response
2627

2728

2829
os.environ["GOOGLE_CLOUD_PROJECT"] = os.environ["FIRESTORE_PROJECT"]
@@ -85,8 +86,8 @@ def test_vector_search_prefilter(db):
8586
assert results[1].key.name == "Excelsa"
8687

8788

88-
def test_vector_search_distance_result_field(db):
89-
vector_query = vector_search_distance_result_field(db)
89+
def test_vector_search_distance_result_property(db):
90+
vector_query = vector_search_distance_result_property(db)
9091
results = list(vector_query.fetch())
9192

9293
assert len(results) == 4
@@ -108,8 +109,8 @@ def test_vector_search_distance_threshold(db):
108109
assert results[0].key.name == "Liberica"
109110
assert results[1].key.name == "Robusta"
110111

111-
def test_vector_search_large_query(db):
112-
key_list, vector_results, full_results = vector_search_large_query(db)
112+
def test_vector_search_large_response(db):
113+
key_list, vector_results, full_results = vector_search_large_response(db)
113114
assert len(key_list) == 4
114115
# each list should have same number of elements
115116
assert len(key_list) == len(vector_results)

0 commit comments

Comments
 (0)