codehere 0.1.1 → 0.2.2

package/dist/search.js

@@ -1,12 +1,11 @@
 import { config } from 'dotenv';
 import { join, dirname } from 'path';
 import { fileURLToPath } from 'url';
+// CohereClient is lazy-loaded to avoid hard dependency
 import initSqlJs from 'sql.js';
 import { readFileSync, existsSync } from 'fs';
 import { optimizeContext } from './token-optimizer.js';
 import { optimizeQuery } from './query-optimizer.js';
-import { ProviderRegistry } from './infrastructure/providers/provider-registry.js';
-import { OperationProviderResolver } from './infrastructure/providers/operation-provider-resolver.js';
 // Load .env from agent directory
 try {
     const __filename = fileURLToPath(import.meta.url);
@@ -18,6 +17,38 @@ catch {
     // Fallback
 }
 config(); // Also try current directory
+function getSearchEmbedding(text) {
+    if (process.env.OPENAI_API_KEY) {
+        return fetch('https://api.openai.com/v1/embeddings', {
+            method: 'POST',
+            headers: {
+                'Authorization': `Bearer ${process.env.OPENAI_API_KEY}`,
+                'Content-Type': 'application/json',
+            },
+            body: JSON.stringify({ input: text, model: 'text-embedding-3-small' }),
+        }).then(async (r) => {
+            if (!r.ok)
+                throw new Error(`OpenAI embedding error: ${r.status}`);
+            const data = await r.json();
+            return data.data?.[0]?.embedding || [];
+        });
+    }
+    if (process.env.COHERE_API_KEY) {
+        const { CohereClient } = require('cohere-ai');
+        const cohere = new CohereClient({ token: process.env.COHERE_API_KEY });
+        return cohere.embed({
+            texts: [text],
+            model: 'embed-english-v3.0',
+            inputType: 'search_query',
+        }).then((response) => {
+            const embeddings = Array.isArray(response.embeddings)
+                ? response.embeddings
+                : response.embeddings.embeddings || [];
+            return embeddings[0] || [];
+        });
+    }
+    throw new Error('No embedding API key found. Set OPENAI_API_KEY or COHERE_API_KEY.');
+}
 /**
  * @deprecated Legacy global DB path - this module is legacy
  * Active CLI flows use per-repo embeddings via SQLiteEmbeddingRepository
@@ -53,73 +84,12 @@ function cosineSimilarity(a, b) {
     return dotProduct / denominator;
 }
 /**
- * Get embedding provider (supports all providers via registry)
- */
-async function getEmbeddingProvider() {
-    const registry = ProviderRegistry.getInstance();
-    // Ensure providers are registered
-    if (!registry.get('cohere') && !registry.get('openai') && !registry.get('fake')) {
-        if (process.env.CODEHERE_FAKE_AI === '1') {
-            const { FakeProvider } = await import('./infrastructure/providers/fake-provider.js');
-            registry.register(new FakeProvider());
-            registry.setDefault('fake');
-        }
-        else {
-            if (process.env.COHERE_API_KEY) {
-                const { CohereProvider } = await import('./infrastructure/providers/cohere-provider.js');
-                registry.register(new CohereProvider());
-            }
-            if (process.env.CODEHERE_OPENAI_API_KEY || process.env.OPENAI_API_KEY) {
-                const { OpenAIProvider } = await import('./infrastructure/providers/openai-provider.js');
-                registry.register(new OpenAIProvider());
-            }
-            if (process.env.OPENROUTER_API_KEY) {
-                const { OpenRouterProvider } = await import('./infrastructure/providers/openrouter-provider.js');
-                registry.register(new OpenRouterProvider());
-            }
-            // Set default based on CODEHERE_AI_PROVIDER
-            const requestedProvider = process.env.CODEHERE_AI_PROVIDER?.toLowerCase();
-            if (requestedProvider && registry.get(requestedProvider)) {
-                registry.setDefault(requestedProvider);
-            }
-            else if (registry.get('openai')) {
-                registry.setDefault('openai');
-            }
-            else if (registry.get('cohere')) {
-                registry.setDefault('cohere');
-            }
-        }
-    }
-    const resolver = new OperationProviderResolver(registry);
-    // Special handling for OpenRouter: fallback to OpenAI or Cohere for embeddings
-    const currentDefault = registry.getDefault();
-    if (currentDefault?.id === 'openrouter') {
-        if (registry.get('openai')?.capabilities.embedding) {
-            return registry.get('openai');
-        }
-        if (registry.get('cohere')?.capabilities.embedding) {
-            return registry.get('cohere');
-        }
-        throw new Error('OpenRouter doesn\'t support embeddings. Please set up OpenAI or Cohere:\n' +
-            '  export CODEHERE_OPENAI_API_KEY="sk-..."\n' +
-            '  or\n' +
-            '  export COHERE_API_KEY="..."\n');
-    }
-    return resolver.resolveProvider('embedding');
-}
-/**
- * Legacy search - now uses provider system instead of hardcoded Cohere
+ * Legacy search - kept for backward compatibility
  * Use getIntelligentContext() for new code
  */
 export async function searchSimilar(query, topK = 15) {
-    // Get embedding provider (respects CODEHERE_AI_PROVIDER)
-    const provider = await getEmbeddingProvider();
-    // Embed the query using provider
-    const queryEmbedding = await provider.embed(query);
-    // Handle both single embedding and array of embeddings
-    const queryVector = Array.isArray(queryEmbedding[0])
-        ? queryEmbedding[0]
-        : queryEmbedding;
+    // Embed the query using available provider
+    const queryEmbedding = await getSearchEmbedding(query);
     // Load database
     if (!existsSync(DB_PATH)) {
         return [];
@@ -145,7 +115,7 @@ export async function searchSimilar(query, topK = 15) {
     const results = [];
     for (const row of rows) {
         const chunkEmbedding = JSON.parse(row.embedding);
-        const similarity = cosineSimilarity(queryVector, chunkEmbedding);
+        const similarity = cosineSimilarity(queryEmbedding, chunkEmbedding);
         results.push({
             chunk: {
                 id: row.id,