From e1407d059592591d630d7fa88f06498656dc4db3 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 9 Aug 2023 17:12:22 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(Metaphor.py):=20refactor?= =?UTF-8?q?=20build=20method=20to=20accept=20additional=20parameters=20for?= =?UTF-8?q?=20improved=20configurability=20=F0=9F=94=A7=20chore(Metaphor.p?= =?UTF-8?q?y):=20update=20search=20method=20to=20use=20the=20provided=20pa?= =?UTF-8?q?rameters=20for=20use=5Fautoprompt=20and=20search=5Fnum=5Fresult?= =?UTF-8?q?s=20=F0=9F=94=A7=20chore(Metaphor.py):=20update=20find=5Fsimila?= =?UTF-8?q?r=20method=20to=20use=20the=20provided=20parameter=20for=20simi?= =?UTF-8?q?lar=5Fnum=5Fresults?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/components/toolkits/Metaphor.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/toolkits/Metaphor.py b/src/backend/langflow/components/toolkits/Metaphor.py index 567071278..6f43d24b4 100644 --- a/src/backend/langflow/components/toolkits/Metaphor.py +++ b/src/backend/langflow/components/toolkits/Metaphor.py @@ -23,6 +23,9 @@ class MetaphorToolkit(CustomComponent): def build( self, metaphor_api_key: str, + use_autoprompt: bool = True, + search_num_results: int = 5, + similar_num_results: int = 5, ) -> Union[Tool, BaseToolkit]: # If documents, then we need to create a Vectara instance using .from_documents client = Metaphor(api_key=metaphor_api_key) @@ -30,7 +33,9 @@ class MetaphorToolkit(CustomComponent): @tool def search(query: str): """Call search engine with a query.""" - return client.search(query, use_autoprompt=True, num_results=5) + return client.search( + query, use_autoprompt=use_autoprompt, num_results=search_num_results + ) @tool def get_contents(ids: List[str]): @@ -46,6 +51,6 @@ class MetaphorToolkit(CustomComponent): The url passed in should be a URL returned from `search` """ - return client.find_similar(url, num_results=5) + return client.find_similar(url, num_results=similar_num_results) return [search, get_contents, find_similar] # type: ignore