fix: lint and type errors

• Replace 'any' types with proper TypeScript types in test and
implementation files
• Remove non-null assertions and use type casts where appropriate
• Ensure type safety in embedding provider and benchmark code

Author: gorango

packages/ai-tool-retriever/src/core/retriever.test.ts

@@ -102,7 +102,7 @@ describe('ToolRetriever', () => {
 	})
 
 	describe('strict mode', () => {
-		let consoleWarnSpy: any
+		let consoleWarnSpy: ReturnType<typeof vi.spyOn>
 
 		beforeEach(() => {
 			consoleWarnSpy = vi.spyOn(console, 'warn').mockImplementation(() => {})

packages/ai-tool-retriever/src/providers/embedding/transformers.test.ts

@@ -17,7 +17,7 @@ vi.mock('@xenova/transformers', () => ({
 const PIPELINE_PROMISE_SYMBOL = Symbol.for('ai-tool-retriever.pipeline_promise')
 
 describe('TransformersEmbeddingProvider', () => {
-	let TransformersEmbeddingProvider: any
+	let TransformersEmbeddingProvider: typeof import('./transformers').TransformersEmbeddingProvider
 
 	beforeEach(async () => {
 		vi.resetModules()
@@ -27,7 +27,7 @@ describe('TransformersEmbeddingProvider', () => {
 
 	afterEach(() => {
 		vi.clearAllMocks()
-		delete (globalThis as any)[PIPELINE_PROMISE_SYMBOL]
+		delete (globalThis as Record<string | symbol, unknown>)[PIPELINE_PROMISE_SYMBOL]
 	})
 
 	it('should initialize the pipeline only once on multiple create calls', async () => {

packages/ai-tool-retriever/src/providers/embedding/transformers.ts

@@ -85,7 +85,7 @@ export class TransformersEmbeddingProvider implements EmbeddingProvider {
 			})()
 		}
 
-		return _global[PIPELINE_PROMISE_SYMBOL]!
+		return _global[PIPELINE_PROMISE_SYMBOL] as Promise<FeatureExtractionPipeline>
 	}
 
 	/**

packages/benchmark/src/search-scalability.ts

@@ -4,6 +4,12 @@ import type { ToolDefinition } from 'ai-tool-retriever'
 import { InMemoryStore } from 'ai-tool-retriever/providers/store/in-memory'
 import { z } from 'zod'
 
+type EmbeddingProvider = {
+	dimensions: number
+	getFloatEmbedding: (text: string) => Promise<number[]>
+	getFloatEmbeddingsBatch: (texts: string[]) => Promise<number[][]>
+}
+
 const mockEmbeddingProvider = {
 	dimensions: 4,
 	getFloatEmbedding: async () => [0, 0, 0, 0],
@@ -31,7 +37,7 @@ async function runBenchmark() {
 
 	for (const count of toolCounts) {
 		const tools = createMockTools(count)
-		await store.sync(tools, mockEmbeddingProvider as any)
+		await store.sync(tools, mockEmbeddingProvider as EmbeddingProvider)
 
 		const searchIterations = 100
 		const startTime = performance.now()

packages/e2e/src/cli.ts

@@ -23,7 +23,7 @@ async function main() {
 		}
 
 		process.exit(0)
-	} catch (e) {
+	} catch (e: unknown) {
 		if (e instanceof ToolNotFoundError) {
 			console.error(`E2E_ERROR: ${e.message}`)
 			process.exit(10) // custom exit code for specific error