Use float to string conversion

Mozilla-Ocho · Jul 1, 2024 · 263d39b · 263d39b
1 parent 3fc00b1
commit 263d39b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/llamafile/server/json.cpp b/llamafile/server/json.cpp
@@ -76,10 +76,19 @@ encode_json(char* p, unsigned long x) noexcept
     return FormatUint64(p, x);
 }
 
+char*
+encode_json(char* p, float x) noexcept
+{
+    double_conversion::StringBuilder b(p, 256);
+    kDoubleToJson.ToShortestSingle(x, &b);
+    b.Finalize();
+    return p + strlen(p);
+}
+
 char*
 encode_json(char* p, double x) noexcept
 {
-    double_conversion::StringBuilder b(p, 128);
+    double_conversion::StringBuilder b(p, 256);
     kDoubleToJson.ToShortest(x, &b);
     b.Finalize();
     return p + strlen(p);

diff --git a/llamafile/server/json.h b/llamafile/server/json.h
@@ -27,6 +27,9 @@ encode_json(char*, int) noexcept;
 char*
 encode_json(char*, long) noexcept;
 
+char*
+encode_json(char*, float) noexcept;
+
 char*
 encode_json(char*, double) noexcept;