remove save from the protocol

masci · Jun 3, 2024 · 28348b7 · 28348b7
1 parent 3c3e999
commit 28348b7
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/src/banks/env.py b/src/banks/env.py
@@ -29,7 +29,6 @@ def _add_default_templates(r: TemplateRegistry):
     templates_dir = Path(os.path.dirname(__file__)) / "templates"
     for tpl_file in templates_dir.glob("*.jinja"):
         r.set(name=tpl_file.name, prompt=tpl_file.read_text())
-    r.save()
 
 
 # Init the Jinja env

diff --git a/src/banks/registries/file.py b/src/banks/registries/file.py
@@ -40,8 +40,9 @@ def set(self, *, name: str, prompt: str, version: str | None = None, overwrite:
             tpl = self.get(name, version)
             if overwrite:
                 tpl.prompt = prompt
-                return
+                self.save()
         except TemplateNotFoundError:
             tpl_id = self._make_id(name, version)
             tpl = PromptTemplate(id=tpl_id, name=name, version=version or "", prompt=prompt)
             self._index.templates.append(tpl)
+            self.save()
diff --git a/src/banks/registry.py b/src/banks/registry.py
@@ -21,8 +21,6 @@ class PromptTemplateIndex(BaseModel):
 
 
 class TemplateRegistry(Protocol):
-    def save(self) -> None: ...
-
     def get(self, *, name: str, version: str | None = None) -> "PromptTemplate": ...
 
     def set(self, *, name: str, prompt: str, version: str | None = None, overwrite: bool = False): ...