Fix JSON

neo4j · Nov 28, 2024 · 4d8eddb · 4d8eddb
1 parent 4b80926
commit 4d8eddb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/examples/customize/build_graph/pipeline/simple_kg_pipeline_config.json b/examples/customize/build_graph/pipeline/simple_kg_pipeline_config.json
@@ -46,15 +46,12 @@
         ["Person", "WORKS_FOR", "Organization"],
         ["Organization", "DIRECTED_BY", "Person"]
     ],
-    "components": [
-        {
-            "_name": "text_splitter",
-            "_class": "text_splitters.fixed_size_splitter.FixedSizeSplitter",
-            "_params": {
-                "chunk_size":  100,
-                "chunk_overlap": 10
-            }
+    "text_splitter": {
+        "class_": "fixed_size_splitter.FixedSizeSplitter",
+        "params_": {
+            "chunk_size":  100,
+            "chunk_overlap": 10
         }
-    ],
+    },
     "perform_entity_resolution": false
 }
diff --git a/src/neo4j_graphrag/experimental/pipeline/config/types.py b/src/neo4j_graphrag/experimental/pipeline/config/types.py
@@ -41,6 +41,7 @@ class ParamFromEnvConfig(ParamToResolveConfig):
 
 
 ParamConfig = Union[
+    float,
     str,
     ParamFromEnvConfig,
     dict[str, Any],
-Original file line number
+Diff line change
@@ Expand Up / @@ -41,6 +41,7 @@ class ParamFromEnvConfig(ParamToResolveConfig): @@
     ParamConfig = Union[
+        float,
         str,
         ParamFromEnvConfig,
         dict[str, Any],
@@ Expand Down @@