Skip to content

Commit b3af69c

Browse files
fix(dspy): updating syntax for Snowflake cos similarity method which will be deprecated at the end of the month
1 parent a048a51 commit b3af69c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dspy/retrieve/snowflake_rm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
try:
77
from snowflake.snowpark import Session
88
from snowflake.snowpark import functions as snow_fn
9-
from snowflake.snowpark.functions import lit, vector_cosine_distance
9+
from snowflake.snowpark.functions import lit, vector_cosine_similarity
1010
from snowflake.snowpark.types import VectorType
1111

1212
except ImportError:
@@ -88,7 +88,7 @@ def _top_k_similar_chunks(self, query_embeddings, k):
8888
doc_embeddings.select(
8989
doc_table_value,
9090
doc_table_key,
91-
vector_cosine_distance(
91+
vector_cosine_similarity(
9292
doc_embeddings.col(doc_table_value),
9393
lit(query_embeddings).cast(VectorType(float, len(query_embeddings))),
9494
).as_("dist"),

0 commit comments

Comments
 (0)