cocoindex 0.1.22__cp312-cp312-manylinux_2_28_aarch64.whl → 0.1.23__cp312-cp312-manylinux_2_28_aarch64.whl

cocoindex/functions.py

@@ -5,6 +5,9 @@ import sentence_transformers
 from .typing import Float32, Vector, TypeAttr
 from . import op, llm
 
+class ParseJson(op.FunctionSpec):
+    """Parse a text into a JSON object."""
+
 class SplitRecursively(op.FunctionSpec):
     """Split a document (in string) recursively."""
 

cocoindex/storages.py

@@ -29,7 +29,7 @@ class Neo4jConnection:
     db: str | None = None
 
 @dataclass
-class Neo4jFieldMapping:
+class GraphFieldMapping:
     """Mapping for a Neo4j field."""
     field_name: str
     # Field name for the node in the Knowledge Graph.
@@ -37,36 +37,36 @@ class Neo4jFieldMapping:
     node_field_name: str | None = None
 
 @dataclass
-class Neo4jRelationshipEnd:
+class GraphRelationshipEnd:
     """Spec for a Neo4j node type."""
     label: str
-    fields: list[Neo4jFieldMapping]
+    fields: list[GraphFieldMapping]
 
 @dataclass
-class Neo4jRelationshipNode:
+class GraphRelationshipNode:
     """Spec for a Neo4j node type."""
     primary_key_fields: Sequence[str]
     vector_indexes: Sequence[index.VectorIndexDef] = ()
 
 @dataclass
-class Neo4jNode:
+class GraphNode:
     """Spec for a Neo4j node type."""
     kind = "Node"
 
     label: str
 
 @dataclass
-class Neo4jRelationship:
+class GraphRelationship:
     """Spec for a Neo4j relationship."""
     kind = "Relationship"
 
     rel_type: str
-    source: Neo4jRelationshipEnd
-    target: Neo4jRelationshipEnd
-    nodes: dict[str, Neo4jRelationshipNode] | None = None
+    source: GraphRelationshipEnd
+    target: GraphRelationshipEnd
+    nodes: dict[str, GraphRelationshipNode] | None = None
 
 class Neo4j(op.StorageSpec):
     """Graph storage powered by Neo4j."""
 
     connection: AuthEntryReference
-    mapping: Neo4jNode | Neo4jRelationship
+    mapping: GraphNode | GraphRelationship

cocoindex/typing.py

@@ -183,11 +183,7 @@ def _encode_type(type_info: AnalyzedTypeInfo) -> dict[str, Any]:
         if type_info.elem_type is None:
             raise ValueError(f"{type_info.kind} type must have an element type")
         row_type_info = analyze_type_info(type_info.elem_type)
-        if row_type_info.dataclass_type is None:
-            raise ValueError(f"{type_info.kind} type must have a dataclass type")
-        encoded_type['row'] = {
-            'fields': _encode_fields_schema(row_type_info.dataclass_type),
-        }
+        encoded_type['row'] = _encode_type(row_type_info)
 
     return encoded_type
 

{cocoindex-0.1.22.dist-info → cocoindex-0.1.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cocoindex
-Version: 0.1.22
+Version: 0.1.23
 Requires-Dist: sentence-transformers>=3.3.1
 Requires-Dist: click>=8.1.8
 Requires-Dist: pytest ; extra == 'test'

{cocoindex-0.1.22.dist-info → cocoindex-0.1.23.dist-info}/RECORD

@@ -1,12 +1,12 @@
-cocoindex-0.1.22.dist-info/METADATA,sha256=fF2xNGwxJBVg-npfcaYtCZ8mJujJe6kjDSMnwJ5-cIk,8066
-cocoindex-0.1.22.dist-info/WHEEL,sha256=xOZEB-WIKOrrrITqsx8xpY2EDgJkCu53Yhd8N6qLrYE,109
-cocoindex-0.1.22.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+cocoindex-0.1.23.dist-info/METADATA,sha256=IcyNWcJi0ycAqXknL1KbV-tsNRQqNf3a097Xrpbe4g8,8066
+cocoindex-0.1.23.dist-info/WHEEL,sha256=xOZEB-WIKOrrrITqsx8xpY2EDgJkCu53Yhd8N6qLrYE,109
+cocoindex-0.1.23.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
 cocoindex/llm.py,sha256=uHXub9AWTOtxNyTaefHY-VuY_yzo6ikM1kUxHsQn-zw,298
 cocoindex/tests/test_convert.py,sha256=wSAiskmBgBM5UZussbNXmCxt47hHGlSEEQVgDZaY4TE,2058
 cocoindex/tests/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 cocoindex/sources.py,sha256=wZFU8lwSXjyofJR-syySH9fTyPnBlAPJ6-1hQNX8fGA,936
-cocoindex/typing.py,sha256=EZstkCKOFTRST23ia8R8PV7rZbQiNWFblzzRIQ3vVcQ,7472
-cocoindex/storages.py,sha256=m6Y9-r1XgQClWUrI8a0VHd0X6ZMYDlchVJFwWew01_M,1718
+cocoindex/typing.py,sha256=4mP9VXS75s3VMfF1LDc1LXsBng7uGdR5aD73N8iaeSM,7282
+cocoindex/storages.py,sha256=ZWpQABuUWTs6BdPcTBWDOiRXVGxS87Uh2i_C1wtkXYQ,1718
 cocoindex/flow.py,sha256=HwngOKZyF60pUE_LGjt02yVNWh-haoMWH1Si0w5rYqY,19856
 cocoindex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cocoindex/setup.py,sha256=W1HshwYk_K2aeLOVn_e62ZOXBO9yWsoUboRiH4SjF48,496
@@ -16,8 +16,8 @@ cocoindex/op.py,sha256=UaBgkWN8KEKqD-3UEYW_v-99OwNQZfIB4PdpAd4jd74,9984
 cocoindex/__init__.py,sha256=f4LTPg4db7Wm3QO9HirvhsT11OVykiFxGbt1JK6taFA,572
 cocoindex/cli.py,sha256=jRwGCPuqkBz_cIp7LKARCRAiDsPm91bsduKzfYJohJ4,6931
 cocoindex/lib.py,sha256=kyfzkaanrMHVwWjd2UWgLr8ArS_vjlK64qb4TbNTbbs,3464
-cocoindex/functions.py,sha256=uukmig7qtOIda3lrM7tNSfGqx22e0EANfWWZjuhE_5E,1579
+cocoindex/functions.py,sha256=xcAeRQTy9JObfxpjyMn-dPY2y7XhVWjB7759xVyup6o,1657
 cocoindex/auth_registry.py,sha256=lZ2rD5_9aC_UpGk7t4TmSYal_rjN7eHgO4_sU7FR0Zw,620
 cocoindex/index.py,sha256=LssEOuZi6AqhwKtZM3QFeQpa9T-0ELi8G5DsrYKECvc,534
-cocoindex/_engine.cpython-312-aarch64-linux-gnu.so,sha256=43j6omy6nXTFdRo-Qe_QqxGr3RsgdCqq2pk0f6pukuk,63350816
-cocoindex-0.1.22.dist-info/RECORD,,
+cocoindex/_engine.cpython-312-aarch64-linux-gnu.so,sha256=VfLg3LJ2C_ARY8XmYKW7LXN_4YDI-gGolWNFf6ZQ5Ok,63385472
+cocoindex-0.1.23.dist-info/RECORD,,