File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -175,7 +175,7 @@ def similarity_generate_query(
175175 for ref_obj , ref_fields in references .items ():
176176 fields_clause = ", " .join (ref_fields )
177177 type_clauses .append (f"WHEN { ref_obj } THEN { fields_clause } " )
178- type_clause = f"TYPEOF { relationship } { ' ' .join (type_clauses )} END"
178+ type_clause = f"TYPEOF { relationship } { ' ' .join (type_clauses )} ELSE Id END"
179179 query_fields .append (type_clause )
180180
181181 # Add regular fields to the query
@@ -206,7 +206,7 @@ def similarity_generate_query(
206206 if "Id" not in fields :
207207 fields .insert (0 , "Id" )
208208
209- return query , fields # Return the original input fields with "Id"
209+ return query , fields
210210
211211
212212def similarity_post_process (
@@ -226,9 +226,9 @@ def similarity_post_process(
226226 closest_records = []
227227
228228 if complexity_constant < 1000 :
229- closest_records = annoy_post_process (load_records , query_records )
230- else :
231229 closest_records = levenshtein_post_process (load_records , query_records )
230+ else :
231+ closest_records = annoy_post_process (load_records , query_records )
232232
233233 return closest_records
234234
You can’t perform that action at this time.
0 commit comments