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

View file

@ -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(
*[

View file

@ -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(
*[