apunkt/sqlite-vec
1
0
Fork 0
mirror of https://github.com/asg017/sqlite-vec.git synced 2026-04-27 01:36:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix NaN in vec_to_json

Browse source
This commit is contained in:
Alex Garcia 2024-04-27 12:05:35 -07:00
parent a6b055ea39
commit 79d6ea1aea
1 changed files with 9 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

10
sqlite-vec.c
View file

@ -1184,7 +1184,14 @@ static void vec_to_json(sqlite3_context *context, int argc,
sqlite3_str_appendall(str, ","); sqlite3_str_appendall(str, ",");
} }
if (elementType == SQLITE_VEC_ELEMENT_TYPE_FLOAT32) { if (elementType == SQLITE_VEC_ELEMENT_TYPE_FLOAT32) {
sqlite3_str_appendf(str, "%f", ((f32 *)vector)[i]); f32 value = ((f32 *)vector)[i];
if(isnan(value)) {
sqlite3_str_appendall(str, "null");
}
else {
sqlite3_str_appendf(str, "%f", value);
}
} else if (elementType == SQLITE_VEC_ELEMENT_TYPE_INT8) { } else if (elementType == SQLITE_VEC_ELEMENT_TYPE_INT8) {
sqlite3_str_appendf(str, "%d", ((i8 *)vector)[i]); sqlite3_str_appendf(str, "%d", ((i8 *)vector)[i]);
} else if (elementType == SQLITE_VEC_ELEMENT_TYPE_BIT) { } else if (elementType == SQLITE_VEC_ELEMENT_TYPE_BIT) {
@ -1197,6 +1204,7 @@ static void vec_to_json(sqlite3_context *context, int argc,
char *s = sqlite3_str_finish(str); char *s = sqlite3_str_finish(str);
if (s) { if (s) {
sqlite3_result_text(context, s, len, sqlite3_free); sqlite3_result_text(context, s, len, sqlite3_free);
sqlite3_result_subtype(context, JSON_SUBTYPE);
} else { } else {
sqlite3_result_error_nomem(context); sqlite3_result_error_nomem(context);
} }