PR Custody

[codex] Add toolkit retrieval filters and search race safeguards

@dhawal1checks n/achecks…codex/plan-search-enabled-toolkit-filter → master30 files · +683 −158review: @abir-taheerupdated 4d ago

▸Description

Why: googlesuper was not coming in either search-

plan search retrieval was not finding googlesuper even if its the only toolkit enabled, we now push disabledToolkits filter to turbopuffer level to get improve retrieval quality
tool search was excluding googlesuper explicitly - remove this filter (behind ff)
add some timeouts

What:

Add a LaunchDarkly-gated retrieval-time disabled-toolkit filter for plan search using full catalog minus enabled toolkit slugs.
Gate GoogleSuper tool-search inclusion behind a LaunchDarkly flag.
Fix search race grace logging so timeout warnings are emitted only when the grace timeout actually wins, and add explicit 30s tool-search reranker timeouts for Gemini, OpenAI-compatible Groq/OpenAI, and Bedrock.

pnpm --dir apps/apollo exec oxfmt --check src/lib/composio_actions/actions/search_orchestrator.ts src/lib/composio_actions/utils/toolRouter/llm_reranker.ts src/common/lib/external/bedrock.ts
git diff --check
SELF_HOSTED=true DATABASE_URL=postgres://localhost:5432/test REDIS_URL=redis://localhost:6379 ENCRYPTION_KEY=test CONNECTED_ACCOUNT_ENCRYPTION_KEY=test AUTH_CONFIG_ENCRYPTION_KEY=test SERVICE_NAME=apollo-test THERMOS_URL=http://localhost:8080 pnpm --dir apps/apollo exec vitest run --config .tmp.codex-vitest.config.ts for bedrock.test.ts and llm_reranker.test.ts (37 tests passed; temporary config removed)
Earlier targeted unit run for filters.unit.test.ts passed.

Standard Apollo .test.ts Vitest config is blocked locally by global setup requiring Postgres at localhost:5432.
Filtered check-types still reports existing unrelated search_orchestrator.ts connection-status typing errors around lines 500/720; no new bedrock.ts or llm_reranker.ts errors surfaced.

loading diff…