Skip to content

Commit 4eddc74

Browse files
Fix CI
1 parent 4109108 commit 4eddc74

1 file changed

Lines changed: 16 additions & 5 deletions

File tree

stackone_ai/toolset.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -295,13 +295,18 @@ def __call__(
295295
Returns:
296296
Tools collection with matched tools
297297
"""
298+
effective_top_k = top_k if top_k is not None else self._config.get("top_k")
299+
effective_min_sim = (
300+
min_similarity if min_similarity is not None else self._config.get("min_similarity")
301+
)
302+
effective_search = search if search is not None else self._config.get("method", "auto")
298303
return self._toolset.search_tools(
299304
query,
300305
connector=connector,
301-
top_k=top_k if top_k is not None else self._config.get("top_k"),
302-
min_similarity=min_similarity if min_similarity is not None else self._config.get("min_similarity"),
306+
top_k=effective_top_k,
307+
min_similarity=effective_min_sim,
303308
account_ids=account_ids,
304-
search=search if search is not None else self._config.get("method", "auto"),
309+
search=effective_search,
305310
)
306311

307312

@@ -494,9 +499,15 @@ def search_tools(
494499
)
495500

496501
# Merge constructor defaults with per-call overrides
497-
effective_search: SearchMode = search if search is not None else self._search_config.get("method", "auto")
502+
effective_search: SearchMode = (
503+
search if search is not None else self._search_config.get("method", "auto")
504+
)
498505
effective_top_k = top_k if top_k is not None else self._search_config.get("top_k")
499-
effective_min_sim = min_similarity if min_similarity is not None else self._search_config.get("min_similarity")
506+
effective_min_sim = (
507+
min_similarity
508+
if min_similarity is not None
509+
else self._search_config.get("min_similarity")
510+
)
500511

501512
all_tools = self.fetch_tools(account_ids=account_ids)
502513
available_connectors = all_tools.get_connectors()

0 commit comments

Comments
 (0)