diff options
Diffstat (limited to 'autogpts/forge/forge/sdk/db_test.py')
-rw-r--r-- | autogpts/forge/forge/sdk/db_test.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/autogpts/forge/forge/sdk/db_test.py b/autogpts/forge/forge/sdk/db_test.py index 5346eb3ba..14330435e 100644 --- a/autogpts/forge/forge/sdk/db_test.py +++ b/autogpts/forge/forge/sdk/db_test.py @@ -14,7 +14,13 @@ from forge.sdk.db import ( convert_to_task, ) from forge.sdk.errors import NotFoundError as DataNotFoundError -from forge.sdk.schema import * +from forge.sdk.model import ( + Artifact, + Status, + Step, + StepRequestBody, + Task, +) @pytest.mark.asyncio |