diff --git a/graphiti_core/driver/kuzu_driver.py b/graphiti_core/driver/kuzu_driver.py index a99a0147..27bfa5ba 100644 --- a/graphiti_core/driver/kuzu_driver.py +++ b/graphiti_core/driver/kuzu_driver.py @@ -15,7 +15,7 @@ limitations under the License. """ import logging -from typing import Any, LiteralString +from typing import Any import kuzu @@ -151,7 +151,7 @@ class KuzuDriver(GraphDriver): if delete_existing: 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 # 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( *[ diff --git a/graphiti_core/driver/neptune_driver.py b/graphiti_core/driver/neptune_driver.py index e8bd7772..aefb85a7 100644 --- a/graphiti_core/driver/neptune_driver.py +++ b/graphiti_core/driver/neptune_driver.py @@ -18,7 +18,7 @@ import asyncio import datetime import logging from collections.abc import Coroutine -from typing import Any, LiteralString +from typing import Any import boto3 from langchain_aws.graphs import NeptuneAnalyticsGraph, NeptuneGraph @@ -273,9 +273,9 @@ class NeptuneDriver(GraphDriver): if delete_existing: 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( *[