pagination.py 985 B

123456789101112131415161718192021222324252627
  1. """Artifacts-specific data models for handling paginated results from GraphQL queries."""
  2. from wandb._pydantic import Connection, ConnectionWithTotal
  3. from .._generated.fragments import (
  4. ArtifactCollectionFragment,
  5. ArtifactFragment,
  6. ArtifactMembershipFragment,
  7. ArtifactTypeFragment,
  8. FileFragment,
  9. FileWithUrlFragment,
  10. RegistryCollectionFragment,
  11. RegistryFragment,
  12. )
  13. ArtifactTypeConnection = Connection[ArtifactTypeFragment]
  14. ArtifactCollectionConnection = ConnectionWithTotal[ArtifactCollectionFragment]
  15. ProjectArtifactCollectionConnection = Connection[ArtifactCollectionFragment]
  16. ArtifactMembershipConnection = Connection[ArtifactMembershipFragment]
  17. FileWithUrlConnection = Connection[FileWithUrlFragment]
  18. ArtifactFileConnection = Connection[FileFragment]
  19. RunArtifactConnection = ConnectionWithTotal[ArtifactFragment]
  20. RegistryConnection = Connection[RegistryFragment]
  21. RegistryCollectionConnection = ConnectionWithTotal[RegistryCollectionFragment]