Fixed Template substitutions

Author: jieguangzhou

CHANGELOG.md

@@ -45,6 +45,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Fix support for keys in new queue handler.
 - Fix the bug where the query itself changes after encoding
 - Fix the dependency error in copy_vectors within vector_index. 
+- Fix Template substitutions
 
 
 ## [0.2.0](https://github.com/superduper-io/superduper/compare/0.1.3...0.2.0])    (2024-Jun-21)

superduper/base/document.py

@@ -282,7 +282,7 @@ def to_template(self, **substitutions):
 
         def substitute(x):
             if isinstance(x, str):
-                for k, v in substitutions:
+                for k, v in substitutions.items():
                     x = x.replace(k, f'<var:{v}>')
                 return x
             if isinstance(x, dict):

superduper/components/template.py

@@ -4,7 +4,7 @@
 
 from superduper.base.constant import KEY_BLOBS, KEY_FILES
 from superduper.base.datalayer import Datalayer
-from superduper.base.document import Document
+from superduper.base.document import Document, QueryUpdateDocument
 from superduper.base.leaf import Leaf
 from superduper.base.variables import _replace_variables
 from superduper.components.component import Component, _build_info_from_path
@@ -39,7 +39,9 @@ def __post_init__(self, db, artifacts, substitutions):
             self.template = self.template.encode(defaults=False, metadata=False)
         self.template = SuperDuperFlatEncode(self.template)
         if substitutions is not None:
-            self.template = self.template.to_template(**substitutions)
+            self.template = QueryUpdateDocument(self.template).to_template(
+                **substitutions
+            )
         if self.template_variables is None:
             self.template_variables = self.template.variables
         super().__post_init__(db, artifacts)