diff --git a/src/backend/langflow/api/v1/validate.py b/src/backend/langflow/api/v1/validate.py index b5b886816..184657a63 100644 --- a/src/backend/langflow/api/v1/validate.py +++ b/src/backend/langflow/api/v1/validate.py @@ -42,7 +42,7 @@ def post_validate_prompt(prompt: ValidatePromptRequest): field_type="str", show=True, advanced=False, - input_types=["BaseLoader"], + input_types=["BaseLoader", "BaseOutputParser"], ) prompt.frontend_node.template[variable] = template_field.to_dict() diff --git a/src/backend/langflow/settings.py b/src/backend/langflow/settings.py index e3644e84c..8fe508210 100644 --- a/src/backend/langflow/settings.py +++ b/src/backend/langflow/settings.py @@ -19,6 +19,7 @@ class Settings(BaseSettings): toolkits: List[str] = [] textsplitters: List[str] = [] utilities: List[str] = [] + output_parsers: List[str] = [] dev: bool = False database_url: str = "sqlite:///./langflow.db" cache: str = "InMemoryCache" @@ -48,6 +49,10 @@ class Settings(BaseSettings): self.toolkits = new_settings.toolkits or [] self.textsplitters = new_settings.textsplitters or [] self.utilities = new_settings.utilities or [] + self.embeddings = new_settings.embeddings or [] + self.vectorstores = new_settings.vectorstores or [] + self.documentloaders = new_settings.documentloaders or [] + self.output_parsers = new_settings.output_parsers or [] self.dev = dev def update_settings(self, **kwargs):