🚀 feat(chains.py): add return_source_documents field to template

The `return_source_documents` field is added to the template as a boolean field. It is not a required field and is set to `True` by default. This field is used to determine whether the source documents should be returned or not. It is displayed in the frontend and can be toggled by the user.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-06-22 13:48:44 -03:00
commit 1e61500f38

View file

@ -19,6 +19,18 @@ class ChainFrontendNode(FrontendNode):
advanced=False,
)
)
# add return_source_documents
self.template.add_field(
TemplateField(
field_type="bool",
required=False,
show=True,
name="return_source_documents",
advanced=False,
value=True,
display_name="Return source documents",
)
)
@staticmethod
def format_field(field: TemplateField, name: Optional[str] = None) -> None:
@ -59,6 +71,12 @@ class ChainFrontendNode(FrontendNode):
field.show = True
field.advanced = False
if field.name == "return_source_documents":
field.required = False
field.show = True
field.advanced = True
field.value = True
class SeriesCharacterChainNode(FrontendNode):
name: str = "SeriesCharacterChain"