dsh-codebase-chat 0.25.0 → 0.25.2

package/dist/index.d.ts

@@ -300,4 +300,7 @@ declare function buildApplyPrompt(filePath: string, newContent: string, projectN
  */
 declare function buildToolPrompt(tool: string, opts: ToolPromptOptions): string;
 
-export { CONFIG_FILE, type CloneGroup, type CodeChunk, type CodeIndex, type ConfigLang, type ContextOptions, type ContextResult, type Cycle, type DiffScope, type HealthReport, type Hotspot, type ImpactDependent, type ImpactReport, type ImpactResult, type IndexedFile, type InvertedIndex, type ProjectConfig, type ToolPromptOptions, type UnusedExport, analyzeImpact, analyzeProject, brandSignature, buildApplyPrompt, buildAsciiBanner, buildAuditPrompt, buildBuildPrompt, buildCeoPrompt, buildChatPrompt, buildContext, buildCreaPrompt, buildExplainPrompt, buildGitPrompt, buildIndex, buildIntelligencePrompt, buildPlayerPrompt, buildRefactorPrompt, buildReportPrompt, buildSearchPrompt, buildTasksPrompt, buildToolPrompt, chunkByTokens, clearConfigCache, cosineSimilarity, countTokens, disposeTreeSitter, embedIndex, ensureTreeSitterForExt, extractChunks, findProjectRoot, formatHealthReport, formatHealthReportMd, formatImpactReport, formatImpactReportMd, getCacheDir, getChangedFiles, getEmbedding, getEmbeddings, getExtractor, getIndex, globToRegExp, initTreeSitter, langInstruction, loadIndex, loadProjectConfig, matchesAnyGlob, normalizeLabels, resolveProjectPath, saveIndex, scoreChunks, selectChunks, styleInstruction, treeSitterReady, truncateToTokens };
+declare function isLocalLlmEnabled(): boolean;
+declare function callLocalLlm(prompt: string, lang?: string): Promise<string>;
+
+export { CONFIG_FILE, type CloneGroup, type CodeChunk, type CodeIndex, type ConfigLang, type ContextOptions, type ContextResult, type Cycle, type DiffScope, type HealthReport, type Hotspot, type ImpactDependent, type ImpactReport, type ImpactResult, type IndexedFile, type InvertedIndex, type ProjectConfig, type ToolPromptOptions, type UnusedExport, analyzeImpact, analyzeProject, brandSignature, buildApplyPrompt, buildAsciiBanner, buildAuditPrompt, buildBuildPrompt, buildCeoPrompt, buildChatPrompt, buildContext, buildCreaPrompt, buildExplainPrompt, buildGitPrompt, buildIndex, buildIntelligencePrompt, buildPlayerPrompt, buildRefactorPrompt, buildReportPrompt, buildSearchPrompt, buildTasksPrompt, buildToolPrompt, callLocalLlm, chunkByTokens, clearConfigCache, cosineSimilarity, countTokens, disposeTreeSitter, embedIndex, ensureTreeSitterForExt, extractChunks, findProjectRoot, formatHealthReport, formatHealthReportMd, formatImpactReport, formatImpactReportMd, getCacheDir, getChangedFiles, getEmbedding, getEmbeddings, getExtractor, getIndex, globToRegExp, initTreeSitter, isLocalLlmEnabled, langInstruction, loadIndex, loadProjectConfig, matchesAnyGlob, normalizeLabels, resolveProjectPath, saveIndex, scoreChunks, selectChunks, styleInstruction, treeSitterReady, truncateToTokens };

package/dist/index.js

@@ -2777,6 +2777,68 @@ function buildToolPrompt(tool, opts) {
   if (!builder) throw new Error(`unknown prompt tool: ${tool}`);
   return normalizeLabels(builder(opts), opts.lang || "fr");
 }
+
+// src/local-llm.ts
+import { existsSync as existsSync2 } from "fs";
+import { join as join8 } from "path";
+var DEFAULT_MODEL_URI = "hf:Qwen/Qwen2.5-1.5B-Instruct-GGUF:Q4_K_M";
+var LOCAL_CONTEXT_SIZE = 8192;
+var LOCAL_MAX_TOKENS = 1024;
+function isLocalLlmEnabled() {
+  return !!(process.env.CODEBASE_LOCAL_LLM || "").trim();
+}
+function configuredModel() {
+  const v = (process.env.CODEBASE_LOCAL_LLM || "").trim();
+  if (!v || v === "1" || v.toLowerCase() === "true") return DEFAULT_MODEL_URI;
+  return v;
+}
+async function importLlama() {
+  try {
+    return await import("node-llama-cpp");
+  } catch {
+    throw new Error("Local LLM needs the optional dependency: npm i node-llama-cpp");
+  }
+}
+var modelPromise = null;
+function loadLocalModel() {
+  if (!modelPromise) {
+    modelPromise = (async () => {
+      const spec = configuredModel();
+      let modelPath = spec;
+      if (spec.startsWith("hf:")) {
+        const { createModelDownloader } = await importLlama();
+        const downloader = await createModelDownloader({
+          modelUri: spec,
+          dirPath: join8(getCacheDir(), "models")
+        });
+        modelPath = await downloader.download();
+      } else if (!existsSync2(spec)) {
+        throw new Error(`Local model not found: ${spec}`);
+      }
+      const { getLlama } = await importLlama();
+      const llama = await getLlama();
+      return llama.loadModel({ modelPath });
+    })();
+    modelPromise.catch(() => {
+      modelPromise = null;
+    });
+  }
+  return modelPromise;
+}
+async function callLocalLlm(prompt, lang = "fr") {
+  const model = await loadLocalModel();
+  const { LlamaChatSession } = await importLlama();
+  const context = await model.createContext({ contextSize: LOCAL_CONTEXT_SIZE });
+  try {
+    const session = new LlamaChatSession({
+      contextSequence: context.getSequence(),
+      systemPrompt: lang === "en" ? "You are a senior codebase analyst. Be precise and cite files with [source: path:line]." : "Tu es un analyste codebase senior. Sois precis et cite les fichiers avec [source: chemin:ligne]."
+    });
+    return await session.prompt(prompt, { temperature: 0.2, maxTokens: LOCAL_MAX_TOKENS });
+  } finally {
+    await context.dispose();
+  }
+}
 export {
   CONFIG_FILE,
   analyzeImpact,
@@ -2800,6 +2862,7 @@ export {
   buildSearchPrompt,
   buildTasksPrompt,
   buildToolPrompt,
+  callLocalLlm,
   chunkByTokens,
   clearConfigCache,
   cosineSimilarity,
@@ -2821,6 +2884,7 @@ export {
   getIndex,
   globToRegExp,
   initTreeSitter,
+  isLocalLlmEnabled,
   langInstruction,
   loadIndex,
   loadProjectConfig,