Interface for allowing providers to pull models

Part of #283
symflower · Jul 25, 2024 · 98aeb3b · 98aeb3b
1 parent c550ddf
commit 98aeb3b
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 0 deletions.
diff --git a/.mockery.yml b/.mockery.yml
@@ -20,6 +20,7 @@ packages:
     interfaces:
       Loader:
       Provider:
+      Puller:
       Query:
   github.com/symflower/eval-dev-quality/tools:
     interfaces:

diff --git a/provider/provider.go b/provider/provider.go
@@ -61,3 +61,9 @@ type Loader interface {
 	// Unload unloads the given model.
 	Unload(modelIdentifier string) (err error)
 }
+
+// Puller is a provider that is capable of pulling models.
+type Puller interface {
+	// Pull downloads the given model.
+	Pull(logger *log.Logger, modelIdentifier string) error
+}
diff --git a/provider/testing/Puller_mock_gen.go b/provider/testing/Puller_mock_gen.go