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

🔧 chore(Metaphor.py): refactor build method to accept additional parameters for improved configurability

Browse source 
🔧 chore(Metaphor.py): update search method to use the provided parameters for use_autoprompt and search_num_results
🔧 chore(Metaphor.py): update find_similar method to use the provided parameter for similar_num_results
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-09 17:12:22 -03:00
commit e1407d0595
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/langflow/components/toolkits/Metaphor.py
View file

@ -23,6 +23,9 @@ class MetaphorToolkit(CustomComponent):
def build( def build(
self, self,
metaphor_api_key: str, metaphor_api_key: str,
use_autoprompt: bool = True,
search_num_results: int = 5,
similar_num_results: int = 5,
) -> Union[Tool, BaseToolkit]: ) -> Union[Tool, BaseToolkit]:
# If documents, then we need to create a Vectara instance using .from_documents # If documents, then we need to create a Vectara instance using .from_documents
client = Metaphor(api_key=metaphor_api_key) client = Metaphor(api_key=metaphor_api_key)
@ -30,7 +33,9 @@ class MetaphorToolkit(CustomComponent):
@tool @tool
def search(query: str): def search(query: str):
"""Call search engine with a query.""" """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 @tool
def get_contents(ids: List[str]): def get_contents(ids: List[str]):
@ -46,6 +51,6 @@ class MetaphorToolkit(CustomComponent):
The url passed in should be a URL returned from `search` 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 return [search, get_contents, find_similar] # type: ignore