Fixed referencing between nodes of the same file

Author: preeesha

rocket-chatter-ingestion-server/src/core/llm.ts

@@ -5,6 +5,7 @@ const openai = new OpenAI({ apiKey: OPENAI_KEY })
 
 export namespace LLM {
 	export async function generateEmbeddings(data: string): Promise<number[]> {
+		return [1, 2, 3]
 		try {
 			const content = await openai.embeddings.create({
 				input: data,

rocket-chatter-ingestion-server/src/process/prepare/processor/file.ts

@@ -88,13 +88,21 @@ export class FileProcessor implements IFileProcessor {
 		}
 
 		// Replace import names with absolute paths
+		const nodesInFile: Record<string, string> = {} // { name: id }
+		for (const node of treeNodes) {
+			node.sourceFileRelativePath = sourceFile.getFullPath()
+			nodesInFile[node.name] = node.getID()
+		}
+
 		for (const treeNode of treeNodes) {
 			treeNode.sourceFileRelativePath = sourceFile.getFullPath()
 			treeNode.uses = treeNode.uses
 				.filter((x) => x.name)
 				.map((x) => {
-					if (parsedImports.has(x.name)) {
+					if (parsedImports.has(x.name))
 						x.name = `${parsedImports.get(x.name)!}.ts:${x.name}`
+					if (nodesInFile[x.name]) {
+						x.name = nodesInFile[x.name]
 					}
 					return x
 				})

rocket-chatter-ingestion-server/src/use.ts

@@ -3,11 +3,11 @@ import { Codebase } from "./process/prepare/codebase"
 import { FileProcessor } from "./process/prepare/processor/file"
 
 async function main() {
-	const codebase = new Codebase("./project", new FileProcessor(), 1)
+	const codebase = new Codebase("./project2", new FileProcessor(), 1)
 	await codebase.process()
 	await codebase.embed()
 
-	insertDataIntoDB(codebase.embeddingsDirPath, 1)
+	// insertDataIntoDB(codebase.embeddingsDirPath, 1)
 }
 
 main()