Lua math lib fixes

silverbulletmd · Jan 17, 2025 · 6a65912 · 6a65912
1 parent 0076208
commit 6a65912
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/common/space_lua/stdlib/math.ts b/common/space_lua/stdlib/math.ts
@@ -73,7 +73,15 @@ export const mathApi = new LuaTable({
 
   // Keep the cosine_similarity utility function
   cosine_similarity: new LuaBuiltinFunction(
-    (sf, vecA: number[], vecB: number[]) => {
+    (sf, vecA: LuaTable | number[], vecB: LuaTable | number[]) => {
+      // Convert LuaTable to number[]
+      if (vecA instanceof LuaTable) {
+        vecA = vecA.toJSArray();
+      }
+      if (vecB instanceof LuaTable) {
+        vecB = vecB.toJSArray();
+      }
+
       if (vecA.length !== vecB.length) {
         throw new LuaRuntimeError("Vectors must be of the same length", sf);
       }

diff --git a/common/space_lua/stdlib/string.ts b/common/space_lua/stdlib/string.ts
@@ -204,4 +204,13 @@ export const stringApi = new LuaTable({
   endswith: new LuaBuiltinFunction((_sf, s: string, suffix: string) => {
     return s.endsWith(suffix);
   }),
+  trim: new LuaBuiltinFunction((_sf, s: string) => {
+    return s.trim();
+  }),
+  trim_start: new LuaBuiltinFunction((_sf, s: string) => {
+    return s.trimStart();
+  }),
+  trim_end: new LuaBuiltinFunction((_sf, s: string) => {
+    return s.trimEnd();
+  }),
 });