diff --git a/projects/fal/src/fal/toolkit/file/providers/__init__.py b/projects/fal/src/fal/toolkit/file/providers/__init__.py index e69de29b..ccce1864 100644 --- a/projects/fal/src/fal/toolkit/file/providers/__init__.py +++ b/projects/fal/src/fal/toolkit/file/providers/__init__.py @@ -0,0 +1,10 @@ +from .fal import FalFileRepository, InMemoryRepository +from .gcp import GoogleStorageRepository +from .r2 import R2Repository + +__all__ = [ + "FalFileRepository", + "InMemoryRepository", + "GoogleStorageRepository", + "R2Repository", +] diff --git a/projects/fal/tests/stability_test.py b/projects/fal/tests/stability_test.py index c023e530..af968315 100644 --- a/projects/fal/tests/stability_test.py +++ b/projects/fal/tests/stability_test.py @@ -419,7 +419,7 @@ class MathQuery(BaseModel): class MathResult(BaseModel): result: int = Field(description="The result of the operation") - @isolated_client("virtualenv", requirements=["pydantic<2"]) + @isolated_client("virtualenv", requirements=["pydantic>=2"]) def add(query: MathQuery) -> MathResult: return MathResult(result=query.x + query.y) diff --git a/projects/fal/tests/toolkit/file_test.py b/projects/fal/tests/toolkit/file_test.py index 4059906f..b24f479d 100644 --- a/projects/fal/tests/toolkit/file_test.py +++ b/projects/fal/tests/toolkit/file_test.py @@ -4,7 +4,8 @@ from base64 import b64encode import pytest -from fal.toolkit.file.file import File, GoogleStorageRepository +from fal.toolkit.file.file import File +from fal.toolkit.file.providers import GoogleStorageRepository def test_binary_content_matches():