stacklok / toolhive / 22610100940

64%

push

github

Prepare optimizer for production by refactoring and renaming (#3909)

Prepare optimizer for production by refactoring into internal sub-packages

Clean up the optimizer package to make it production-ready by extracting
concerns into focused internal packages and consolidating the implementation.

- Move token counting logic to internal/tokencounter with a Counter
  interface and JSONByteCounter implementation
- Move SQLite tool store from internal/sqlite_store to internal/toolstore,
  inlining the fake embedding helper and simplifying tests
- Replace DummyOptimizer with a single production toolOptimizer that
  delegates search to ToolStore and scopes results per session
- Add mockgen-generated mocks for ToolStore and EmbeddingClient interfaces
- Add comprehensive unit tests for toolOptimizer using mock store
- Remove unused exported aliases (EmbeddingClient, NewEmbeddingClient)

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>

119 of 167 new or added lines in 3 files covered. (71.26%)

45396 of 71890 relevant lines covered (63.15%)

76.72 hits per line