fix: correctly support types like integer

Author: badmonster0

examples/postgres_embedding/main.py

@@ -14,11 +14,13 @@ def postgres_message_embedding_flow(
     data_scope["messages"] = flow_builder.add_source(
         cocoindex.sources.Postgres(
             table_name="source_messages",
+            # Optional. Use the default CocoIndex database if not specified.
             database=cocoindex.add_transient_auth_entry(
                 cocoindex.sources.DatabaseConnectionSpec(
                     url=os.getenv("SOURCE_DATABASE_URL"),
                 )
             ),
+            # Optional.
             ordinal_column="created_at",
         )
     )
@@ -80,11 +82,14 @@ def postgres_product_embedding_flow(
     data_scope["products"] = flow_builder.add_source(
         cocoindex.sources.Postgres(
             table_name="source_products",
+            # Optional. Use the default CocoIndex database if not specified.
             database=cocoindex.add_transient_auth_entry(
                 cocoindex.sources.DatabaseConnectionSpec(
                     url=os.getenv("SOURCE_DATABASE_URL"),
                 )
             ),
+            # Optional.
+            ordinal_column="modified_time",
         )
     )
 
@@ -96,6 +101,11 @@ def postgres_product_embedding_flow(
             product["_key"]["product_name"],
             product["description"],
         )
+        product["total_value"] = flow_builder.transform(
+            calculate_total_value,
+            product["price"],
+            product["amount"],
+        )
         product["embedding"] = product["full_description"].transform(
             cocoindex.functions.SentenceTransformerEmbed(
                 model="sentence-transformers/all-MiniLM-L6-v2"
@@ -107,6 +117,7 @@ def postgres_product_embedding_flow(
             description=product["description"],
             price=product["price"],
             amount=product["amount"],
+            total_value=product["total_value"],
             embedding=product["embedding"],
         )
 

examples/postgres_embedding/prepare_source_data.sql

@@ -88,7 +88,7 @@ VALUES (
         'Sports',
         'Running Shoes',
         'Lightweight running shoes for daily training',
-        129.99,
+        129.5,
         60,
         NOW() - INTERVAL '1 day'
     ) ON CONFLICT (product_category, product_name) DO NOTHING;

src/ops/shared/postgres.rs

@@ -23,10 +23,10 @@ pub async fn get_db_pool(
 }
 
 pub fn key_value_fields_iter<'a>(
-    key_fields_schema: &[FieldSchema],
+    key_fields_schema: impl ExactSizeIterator<Item = &'a FieldSchema>,
     key_value: &'a KeyValue,
 ) -> Result<&'a [KeyValue]> {
-    let slice = if key_fields_schema.len() == 1 {
+    let slice = if key_fields_schema.into_iter().count() == 1 {
         std::slice::from_ref(key_value)
     } else {
         match key_value {