Skip to content

Commit 2226168

Browse files
committed
Added source file as a node
1 parent fe569f1 commit 2226168

File tree

1 file changed

+24
-0
lines changed
  • rocket-chatter-ingestion-server/src/process/prepare/processor

1 file changed

+24
-0
lines changed

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

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,5 +119,29 @@ export class FileProcessor implements IFileProcessor {
119119
const dbNode = DBNode.fromTreeNode(treeNode)
120120
nodesRef[dbNode.id] = dbNode
121121
}
122+
123+
// Create file node
124+
const fileNode = new DBNode({
125+
id: sourceFile.getFullPath(),
126+
name: sourceFile.getFullPath(),
127+
type: "File",
128+
129+
code: fileContent,
130+
131+
filePath: sourceFile.getFullPath(),
132+
relations:
133+
[] ??
134+
treeNodes.map((x) => ({
135+
target: x.getID(),
136+
relation: "CONTAINS",
137+
})),
138+
139+
nameEmbeddings: [],
140+
codeEmbeddings: [],
141+
142+
isFile: true,
143+
descriptor: "Node",
144+
})
145+
nodesRef[fileNode.id] = fileNode
122146
}
123147
}

0 commit comments

Comments
 (0)