🐛 fix(memories.py): add condition to only add output_key field if template type is not VectorStoreRetrieverMemory
The output_key field is now only added to the template if the template type is not VectorStoreRetrieverMemory. This ensures that the output_key field is not added unnecessarily for templates of this type.
This commit is contained in:
parent
2635f06a06
commit
e04eb0fd7e
1 changed files with 10 additions and 9 deletions
|
|
@ -37,6 +37,7 @@ class MemoryFrontendNode(FrontendNode):
|
||||||
value="",
|
value="",
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
if self.template.type_name not in {"VectorStoreRetrieverMemory"}:
|
||||||
self.template.add_field(
|
self.template.add_field(
|
||||||
TemplateField(
|
TemplateField(
|
||||||
field_type="str",
|
field_type="str",
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue