Add python bindings for new types and getValue for float

kuzudb · Apr 2, 2023 · b736723 · b736723
1 parent afed1c7
commit b736723
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/include/common/types/value.h b/src/include/common/types/value.h
@@ -383,6 +383,15 @@ inline int64_t Value::getValue() const {
     return val.int64Val;
 }
 
+/**
+ * @return float value.
+ */
+KUZU_API template<>
+inline float Value::getValue() const {
+    assert(dataType.getTypeID() == FLOAT);
+    return val.floatVal;
+}
+
 /**
  * @return double value.
  */

diff --git a/tools/python_api/src_cpp/py_query_result.cpp b/tools/python_api/src_cpp/py_query_result.cpp
@@ -70,9 +70,18 @@ py::object PyQueryResult::convertValueToPyObject(const Value& value) {
     case BOOL: {
         return py::cast(value.getValue<bool>());
     }
+    case INT16: {
+        return py::cast(value.getValue<int16_t>());
+    }
+    case INT32: {
+        return py::cast(value.getValue<int32_t>());
+    }
     case INT64: {
         return py::cast(value.getValue<int64_t>());
     }
+    case FLOAT: {
+        return py::cast(value.getValue<float>());
+    }
     case DOUBLE: {
         return py::cast(value.getValue<double>());
     }