Skip to content

Commit c9c0358

Browse files
Cubic suggestions
1 parent a87fa00 commit c9c0358

3 files changed

Lines changed: 5 additions & 6 deletions

File tree

stackone_ai/models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -596,7 +596,7 @@ def utility_tools(
596596
from stackone_ai.utility_tools import create_semantic_tool_search
597597

598598
search_tool = create_semantic_tool_search(
599-
semantic_client, available_connectors=self.get_connectors() or None
599+
semantic_client, available_connectors=self.get_connectors()
600600
)
601601
execute_tool = create_tool_execute(self)
602602
return Tools([search_tool, execute_tool])

stackone_ai/toolset.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -401,11 +401,10 @@ def _search_one(c: str) -> list[SemanticSearchResult]:
401401
search_tool = utility.get_tool("tool_search")
402402

403403
if search_tool:
404-
fallback_limit = top_k * 3 if top_k is not None else 100
405404
result = search_tool.execute(
406405
{
407406
"query": query,
408-
"limit": fallback_limit,
407+
"limit": top_k,
409408
"minScore": min_score,
410409
}
411410
)

stackone_ai/utility_tools.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
from stackone_ai.constants import DEFAULT_HYBRID_ALPHA
1313
from stackone_ai.models import ExecuteConfig, JsonDict, StackOneTool, ToolParameters, Tools
14-
from stackone_ai.semantic_search import SemanticSearchClient, SemanticSearchResult
14+
from stackone_ai.semantic_search import SemanticSearchClient, SemanticSearchError, SemanticSearchResult
1515
from stackone_ai.utils.normalize import _normalize_action_name
1616
from stackone_ai.utils.tfidf_index import TfidfDocument, TfidfIndex
1717

@@ -339,7 +339,7 @@ def execute_search(arguments: str | JsonDict | None = None) -> JsonDict:
339339

340340
all_results: list[SemanticSearchResult] = []
341341

342-
if available_connectors is not None:
342+
if available_connectors is not None and available_connectors:
343343
# Scoped search: query each connector in parallel
344344
if connector:
345345
connectors_to_search = {connector.lower()} & available_connectors
@@ -357,7 +357,7 @@ def execute_search(arguments: str | JsonDict | None = None) -> JsonDict:
357357
try:
358358
resp = future.result()
359359
all_results.extend(resp.results)
360-
except Exception:
360+
except SemanticSearchError:
361361
pass # Partial failures: skip failed connectors
362362
else:
363363
# No connector scoping: query full catalog (backwards compat)

0 commit comments

Comments
 (0)