fix: add retriever input_type on RetrieverToolComponent + tests (#3601)

*  (RetrieverTool.py): add "input_types" field to the configuration of the RetrieverToolComponent class

*  (generalBugs-shard-12.spec.ts): add end-to-end test for connecting RetrieverTool to other components in the application.
This commit is contained in:
Cristhian Zanforlin Lousa 2024-08-28 16:02:42 -03:00 committed by GitHub
commit 7641e81ca3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 98 additions and 0 deletions

View file

@ -15,6 +15,7 @@ class RetrieverToolComponent(CustomComponent):
"display_name": "Retriever",
"info": "Retriever to interact with",
"type": BaseRetriever,
"input_types": ["Retriever"],
},
"name": {"display_name": "Name", "info": "Name of the tool"},
"description": {"display_name": "Description", "info": "Description of the tool"},