chore: use explicit repository provider module APIs

fal-ai · Feb 11, 2024 · b04c928 · b04c928
1 parent 93bae14
commit b04c928
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 1 deletion.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -37,7 +37,8 @@ repos:
         exclude: |
           (?x)(
             ^projects/fal/src/fal/toolkit/image/image.py|
-            ^projects/fal/src/fal/toolkit/file/file.py
+            ^projects/fal/src/fal/toolkit/file/file/standard.py|
+            ^projects/fal/src/fal/toolkit/file/file/zipped.py
           )
         files: |
           (?x)(

diff --git a/projects/fal/src/fal/toolkit/file/providers/__init__.py b/projects/fal/src/fal/toolkit/file/providers/__init__.py
diff --git a/projects/fal/src/fal/toolkit/file/providers/fal.py b/projects/fal/src/fal/toolkit/file/providers/fal.py
@@ -11,6 +11,8 @@
 from fal.toolkit.file.types import FileData, FileRepository
 from fal.toolkit.mainify import mainify
 
+__all__ = ["FalFileRepository", "InMemoryRepository"]
+
 
 @mainify
 @dataclass

diff --git a/projects/fal/src/fal/toolkit/file/providers/gcp.py b/projects/fal/src/fal/toolkit/file/providers/gcp.py
@@ -8,6 +8,8 @@
 from fal.toolkit.file.types import FileData, FileRepository
 from fal.toolkit.mainify import mainify
 
+__all__ = ["DEFAULT_URL_TIMEOUT", "GoogleStorageRepository"]
+
 DEFAULT_URL_TIMEOUT = 60 * 15  # 15 minutes
 
 

diff --git a/projects/fal/src/fal/toolkit/file/providers/r2.py b/projects/fal/src/fal/toolkit/file/providers/r2.py
@@ -8,6 +8,8 @@
 from fal.toolkit.file.types import FileData, FileRepository
 from fal.toolkit.mainify import mainify
 
+__all__ = ["DEFAULT_URL_TIMEOUT", "R2Repository"]
+
 DEFAULT_URL_TIMEOUT = 60 * 15  # 15 minutes