jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor: Update SelfQueryRetriever to handle non-string query types

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-20 18:27:13 -03:00
commit 5be733e68c
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/base/langflow/components/retrievers/SelfQueryRetriever.py
View file

@ -60,7 +60,8 @@ class SelfQueryRetrieverComponent(CustomComponent):
input_text = query.text input_text = query.text
elif isinstance(query, str): elif isinstance(query, str):
input_text = query input_text = query
else:
if not isinstance(query, str):
raise ValueError(f"Query type {type(query)} not supported.") raise ValueError(f"Query type {type(query)} not supported.")
documents = self_query_retriever.invoke(input=input_text) documents = self_query_retriever.invoke(input=input_text)
data = [Data.from_document(document) for document in documents] data = [Data.from_document(document) for document in documents]