Record Fields (#22)

* Record Fields

* 0.0.13

Author: claytongentry

lib/manifold/services/vector_service.rb

@@ -32,16 +32,35 @@ def load_vector_config(vector_name)
       private
 
       def transform_attributes_to_schema(attributes)
-        attributes.map do |name, type_str|
-          type, mode = parse_type_and_mode(type_str)
-          {
-            "name" => name,
-            "type" => type.upcase,
-            "mode" => mode
-          }
+        attributes.map { |name, type_def| transform_field(name, type_def) }
+      end
+
+      def transform_field(name, type_def)
+        if type_def.is_a?(Hash)
+          transform_record_field(name, type_def)
+        else
+          transform_scalar_field(name, type_def)
         end
       end
 
+      def transform_record_field(name, type_def)
+        {
+          "name" => name,
+          "type" => "RECORD",
+          "mode" => "NULLABLE",
+          "fields" => transform_attributes_to_schema(type_def)
+        }
+      end
+
+      def transform_scalar_field(name, type_def)
+        type, mode = parse_type_and_mode(type_def)
+        {
+          "name" => name,
+          "type" => type.upcase,
+          "mode" => mode
+        }
+      end
+
       def parse_type_and_mode(type_str)
         type, mode = type_str.split(":")
         mode = mode&.upcase || "NULLABLE"

lib/manifold/templates/vector_template.yml

@@ -6,6 +6,9 @@ attributes:
   # created_at: TIMESTAMP
   # status: STRING
   # names: STRING:REPEATED
+  # key_values:
+  #   first_key: STRING
+  #   second_key: STRING:REQUIRED
 
 # Optionally, reference a view specifying how to select vector dimensions
 # merge:

lib/manifold/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Manifold
-  VERSION = "0.0.12"
+  VERSION = "0.0.13"
 end

spec/manifold/services/vector_service_spec.rb

@@ -15,7 +15,19 @@
           "url" => "string",
           "created_at" => "timestamp",
           "tags" => "string:repeated",
-          "email" => "string:required"
+          "email" => "string:required",
+          "metadata" => {
+            "source" => "string",
+            "last_modified" => "timestamp",
+            "settings" => {
+              "enabled" => "boolean",
+              "flags" => "string:repeated"
+            }
+          },
+          "key_values" => {
+            "first_key" => "string",
+            "second_key" => "string:required"
+          }
         }
       }
     end
@@ -29,7 +41,34 @@
           { "name" => "url", "type" => "STRING", "mode" => "NULLABLE" },
           { "name" => "created_at", "type" => "TIMESTAMP", "mode" => "NULLABLE" },
           { "name" => "tags", "type" => "STRING", "mode" => "REPEATED" },
-          { "name" => "email", "type" => "STRING", "mode" => "REQUIRED" }
+          { "name" => "email", "type" => "STRING", "mode" => "REQUIRED" },
+          {
+            "name" => "metadata",
+            "type" => "RECORD",
+            "mode" => "NULLABLE",
+            "fields" => [
+              { "name" => "source", "type" => "STRING", "mode" => "NULLABLE" },
+              { "name" => "last_modified", "type" => "TIMESTAMP", "mode" => "NULLABLE" },
+              {
+                "name" => "settings",
+                "type" => "RECORD",
+                "mode" => "NULLABLE",
+                "fields" => [
+                  { "name" => "enabled", "type" => "BOOLEAN", "mode" => "NULLABLE" },
+                  { "name" => "flags", "type" => "STRING", "mode" => "REPEATED" }
+                ]
+              }
+            ]
+          },
+          {
+            "name" => "key_values",
+            "type" => "RECORD",
+            "mode" => "NULLABLE",
+            "fields" => [
+              { "name" => "first_key", "type" => "STRING", "mode" => "NULLABLE" },
+              { "name" => "second_key", "type" => "STRING", "mode" => "REQUIRED" }
+            ]
+          }
         ]
       }
     end