cocoindex 0.1.22__cp313-cp313-macosx_11_0_arm64.whl → 0.1.23__cp313-cp313-macosx_11_0_arm64.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,7 +1,7 @@
-cocoindex-0.1.22.dist-info/METADATA,sha256=fF2xNGwxJBVg-npfcaYtCZ8mJujJe6kjDSMnwJ5-cIk,8066
-cocoindex-0.1.22.dist-info/WHEEL,sha256=_czbP61TsBkf9T201RekHMHlqESnWn7yJwXBJC9P-w0,104
-cocoindex-0.1.22.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-cocoindex/functions.py,sha256=uukmig7qtOIda3lrM7tNSfGqx22e0EANfWWZjuhE_5E,1579
+cocoindex-0.1.23.dist-info/METADATA,sha256=IcyNWcJi0ycAqXknL1KbV-tsNRQqNf3a097Xrpbe4g8,8066
+cocoindex-0.1.23.dist-info/WHEEL,sha256=_czbP61TsBkf9T201RekHMHlqESnWn7yJwXBJC9P-w0,104
+cocoindex-0.1.23.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+cocoindex/functions.py,sha256=xcAeRQTy9JObfxpjyMn-dPY2y7XhVWjB7759xVyup6o,1657
 cocoindex/query.py,sha256=XsVY5cBJJ3a70qazkcCHjWZLE1zBqzMQ4HVSulicGMA,3273
 cocoindex/index.py,sha256=LssEOuZi6AqhwKtZM3QFeQpa9T-0ELi8G5DsrYKECvc,534
 cocoindex/lib.py,sha256=kyfzkaanrMHVwWjd2UWgLr8ArS_vjlK64qb4TbNTbbs,3464
@@ -17,7 +17,7 @@ cocoindex/sources.py,sha256=wZFU8lwSXjyofJR-syySH9fTyPnBlAPJ6-1hQNX8fGA,936
 cocoindex/setup.py,sha256=W1HshwYk_K2aeLOVn_e62ZOXBO9yWsoUboRiH4SjF48,496
 cocoindex/cli.py,sha256=jRwGCPuqkBz_cIp7LKARCRAiDsPm91bsduKzfYJohJ4,6931
 cocoindex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cocoindex/typing.py,sha256=EZstkCKOFTRST23ia8R8PV7rZbQiNWFblzzRIQ3vVcQ,7472
-cocoindex/storages.py,sha256=m6Y9-r1XgQClWUrI8a0VHd0X6ZMYDlchVJFwWew01_M,1718
-cocoindex/_engine.cpython-313-darwin.so,sha256=CbuA8iVnxo5lZvClKQgdp5La3kqlV8aNOQR6McV413k,58986432
-cocoindex-0.1.22.dist-info/RECORD,,
+cocoindex/typing.py,sha256=4mP9VXS75s3VMfF1LDc1LXsBng7uGdR5aD73N8iaeSM,7282
+cocoindex/storages.py,sha256=ZWpQABuUWTs6BdPcTBWDOiRXVGxS87Uh2i_C1wtkXYQ,1718
+cocoindex/_engine.cpython-313-darwin.so,sha256=M2BCjW1OKxXPytPLWXhJBZio30rROEaj1Edt-z2tFIs,59027024
+cocoindex-0.1.23.dist-info/RECORD,,