Method execute

src/tools/codegraph/semantic-search.ts:44–131

(args: z.infer<typeof SemanticSearchArgs>, ctx: ToolContext)

  argsSchema = SemanticSearchArgs;

  async execute(args: z.infer<typeof SemanticSearchArgs>, ctx: ToolContext): Promise<ToolResult> {
    const root = args.path ? path.join(ctx.cwd, args.path) : ctx.cwd;
    const topK = args.top_k ?? 10;
    const model = args.embedding_model ?? 'nomic-embed-text';
    const maxFiles = args.max_files ?? 5000;
    const baseUrl = process.env.OLLAMA_BASE_URL ?? 'http://localhost:11434';

    // Verify Ollama is reachable
    try {
      const r = await fetch(`${baseUrl}/api/tags`, { signal: ctx.signal });
      if (!r.ok) throw new Error(`Ollama returned HTTP ${r.status}`);
    } catch (e: any) {
      return {
        content: `[SEMANTIC_SEARCH_ERROR] Ollama not reachable at ${baseUrl}: ${e?.message}\n\nInstall: https://ollama.com\nPull embed model: ollama pull nomic-embed-text`,
        isError: true,
      };
    }

    // Load or build index
    const idxPath = indexPath(root, model);
    let idx: Index | null = args.rebuild_index ? null : await loadIndex(idxPath);

    // Re-index if file count changed significantly (heuristic).
    if (idx && !args.rebuild_index) {
      const currentFiles = await walkSource(root, maxFiles);
      const currentChunkCount = currentFiles.reduce((a, f) => a + Math.ceil(f.content.split('\n').length / 25), 0);
      if (Math.abs(currentChunkCount - idx.chunks.length) > idx.chunks.length * 0.15) {
        logger.info('Semantic search index stale (file count drifted >15%); rebuilding');
        idx = null;
      }
    }

    const buildMessages: string[] = [];
    if (!idx) {
      try {
        idx = await buildIndex(root, model, baseUrl, maxFiles, ctx.signal, (msg) => buildMessages.push(msg));
        await persistIndex(root, model, idx);
      } catch (e: any) {
        return {
          content: `[SEMANTIC_SEARCH_ERROR] Index build failed: ${e?.message}\n\nIf the error mentions "model not found", run:\n  ollama pull ${model}`,
          isError: true,
        };
      }
    }

    // Embed query
    let queryEmbedding: number[];
    try {
      const [e] = await ollamaEmbed(baseUrl, model, [args.query], ctx.signal);
      queryEmbedding = e!;
    } catch (e: any) {
      return { content: `[SEMANTIC_SEARCH_ERROR] Query embed failed: ${e?.message}`, isError: true };
    }

    // Hybrid (BM25 + semantic) ranking over the persisted index (SQLite preferred),
    // unless the caller explicitly opts into semantic-only.
    const scored = await searchPersisted(root, model, args.query, queryEmbedding, topK, idx, !!args.semantic_only);