relative-to-root file paths for rag retrieval

continuedev · Jun 1, 2024 · 3d326cf · 3d326cf
1 parent 7217ab7
commit 3d326cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/context/retrieval/retrieval.ts b/core/context/retrieval/retrieval.ts
@@ -6,7 +6,7 @@ import {
 } from "../../index.js";
 import { LanceDbIndex } from "../../indexing/LanceDbIndex.js";
 
-import { deduplicateArray, getBasename } from "../../util/index.js";
+import { deduplicateArray, getRelativePath } from "../../util/index.js";
 import { RETRIEVAL_PARAMS } from "../../util/parameters.js";
 import { retrieveFts } from "./fullTextSearch.js";
 
@@ -149,7 +149,7 @@ export async function retrieveContextItemsFromEmbeddings(
 
   return [
     ...results.map((r) => {
-      const name = `${getBasename(r.filepath)} (${r.startLine}-${r.endLine})`;
+      const name = `${getRelativePath(r.filepath, workspaceDirs)} (${r.startLine}-${r.endLine})`;
       const description = `${r.filepath} (${r.startLine}-${r.endLine})`;
       return {
         name,