From 5be733e68cf4c322909555e9f6f84bc35d4cc4da Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 20 Jun 2024 18:27:13 -0300 Subject: [PATCH] refactor: Update SelfQueryRetriever to handle non-string query types --- .../base/langflow/components/retrievers/SelfQueryRetriever.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/retrievers/SelfQueryRetriever.py b/src/backend/base/langflow/components/retrievers/SelfQueryRetriever.py index 441e08415..fd61a32de 100644 --- a/src/backend/base/langflow/components/retrievers/SelfQueryRetriever.py +++ b/src/backend/base/langflow/components/retrievers/SelfQueryRetriever.py @@ -60,7 +60,8 @@ class SelfQueryRetrieverComponent(CustomComponent): input_text = query.text elif isinstance(query, str): input_text = query - else: + + if not isinstance(query, str): raise ValueError(f"Query type {type(query)} not supported.") documents = self_query_retriever.invoke(input=input_text) data = [Data.from_document(document) for document in documents]