Skip to content

Commit fbb7be8

Browse files
fix type annotation for _client
1 parent b26bc76 commit fbb7be8

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

integrations/supabase/src/haystack_integrations/components/downloaders/supabase/supabase_bucket_downloader.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from haystack.dataclasses import ByteStream
1111
from haystack.utils.auth import Secret, deserialize_secrets_inplace
1212

13-
from supabase import create_client
13+
from supabase import Client, create_client
1414

1515
logger = logging.getLogger(__name__)
1616

@@ -61,7 +61,7 @@ def __init__(
6161
self.supabase_key = supabase_key
6262
self.bucket_name = bucket_name
6363
self.file_extensions = [e.lower() for e in file_extensions] if file_extensions else None
64-
self._client = None
64+
self._client: Client | None = None
6565

6666
def warm_up(self) -> None:
6767
"""Initialize the Supabase client. Called once before the first run."""
@@ -85,6 +85,7 @@ def run(self, sources: list[str]) -> dict[str, list[ByteStream]]:
8585
"""
8686
if self._client is None:
8787
self.warm_up()
88+
assert self._client is not None
8889
streams = []
8990

9091
for path in sources:

0 commit comments

Comments
 (0)