Как установить первичный ключ, который находится внутри структуры? ⇐ Python

[Anonymous]

Я пытаюсь получить файл json в кузу, но я сталкиваюсь с некоторыми проблемами.

Код: Выделить всё

import kuzu

def put_into_kuzu():
db = kuzu.Database("rag.kuzu")
conn = kuzu.Connection(db)
cex = conn.execute
cex("LOAD json;")
cex("CREATE NODE TABLE Chunks(embedding_dim INT, data STRUCT(__id__ STRING, __created_at__ INT, content STRING, full_doc_id STRING, file_path STRING), PRIMARY KEY (__id__));")
cex('COPY Chunks FROM "./rag_storage/vdb_chunks.json"')
response = conn.execute(
"""
RETURN Chunks;
"""
)
for row in response:
print(row)

if __name__ == "__main__":
put_into_kuzu()
< /code>
json-структура файла: < /p>
{
"embedding_dim": 4096,
"data": [
{
"__id__": "chunk-ed1b1b28b5cf4df59ca4bf1031e4cc27",
"__created_at__": 1753946350,
"content": "content of the text document",
"full_doc_id": "doc-ed1b1b28b5cf4df59ca4bf1031e4cc27",
"file_path": "test.txt"
},
{
a lot more of this
},
],
"matrix" : "bunch of random characters, but they are unneeded"
}

Как можно увидеть из кода, который я хочу __id __ в качестве основного ключа, но не могу его достичь. Я даже пробовал данные.

Код: Выделить всё

RuntimeError: Binder exception: Primary key __id__ does not match any of the predefined node properties.

[*] при попытке data. `` `` :

Код: Выделить всё

RuntimeError: Parser exception: Invalid input < TABLE>: expected rule kU_CreateNodeTable (line: 1, offset: 54)
"CREATE NODE TABLE Chunks(embedding_dim INT, data NODE TABLE Chunks(__id__ STRING, __created_at__ INT, content STRING, full_doc_id STRING, file_path STRING), PRIMARY KEY (__id__);"

Подробнее здесь: https://stackoverflow.com/questions/797 ... e-a-struct