gmakstutis/graphiti
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

update

Browse source
This commit is contained in:
prestonrasmussen 2025-11-05 11:47:34 -05:00
parent 14c120341b
commit 38090b74c7
2 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
graphiti_core/driver/kuzu_driver.py
View file

@ -15,7 +15,7 @@ limitations under the License.
""" """
import logging import logging
from typing import Any, LiteralString from typing import Any
import kuzu import kuzu
@ -151,7 +151,7 @@ class KuzuDriver(GraphDriver):
if delete_existing: if delete_existing:
await self.delete_all_indexes(self._database) await self.delete_all_indexes(self._database)
range_indices: list[LiteralString] = get_range_indices(self.provider) range_indices: list = get_range_indices(self.provider)
# Skip creating fulltext indices if they already exist. Need to do this manually # Skip creating fulltext indices if they already exist. Need to do this manually
# until Kuzu supports `IF NOT EXISTS` for indices. # until Kuzu supports `IF NOT EXISTS` for indices.
@ -169,7 +169,7 @@ class KuzuDriver(GraphDriver):
""", """,
) )
index_queries: list[LiteralString] = range_indices + fulltext_indices index_queries: list = range_indices + fulltext_indices
await semaphore_gather( await semaphore_gather(
*[ *[

6
graphiti_core/driver/neptune_driver.py
View file

@ -18,7 +18,7 @@ import asyncio
import datetime import datetime
import logging import logging
from collections.abc import Coroutine from collections.abc import Coroutine
from typing import Any, LiteralString from typing import Any
import boto3 import boto3
from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph
@ -273,9 +273,9 @@ class NeptuneDriver(GraphDriver):
if delete_existing: if delete_existing:
await self.delete_all_indexes() await self.delete_all_indexes()
range_indices: list[LiteralString] = get_range_indices(self.provider) range_indices: list = get_range_indices(self.provider)
index_queries: list[LiteralString] = range_indices index_queries: list = range_indices
await semaphore_gather( await semaphore_gather(
*[ *[