raj
/
anya
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: added relavent files as template option

This commit is contained in:
Raj Sharma 2024-10-31 13:01:03 +05:30
parent 3e1e269b33
commit 8039a18d82
2 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
assistant/system-prompts.ts
View File

@ -4,6 +4,7 @@ import { OpenAI } from "openai";
import { return_current_events } from "../tools/events"; import { return_current_events } from "../tools/events";
import { memory_manager_guide } from "../tools/memory-manager"; import { memory_manager_guide } from "../tools/memory-manager";
import { searchFilesByTagWithCache } from "../tools/notes"; import { searchFilesByTagWithCache } from "../tools/notes";
import { semantic_search_notes } from "../tools/notes-vectors";
const replaceTemplateStrings = ( const replaceTemplateStrings = (
template: string, template: string,
@ -30,10 +31,19 @@ export async function buildSystemPrompts(
desc: event.description, desc: event.description,
})); }));
const potentially_relavent_files = await semantic_search_notes(
context_message.content,
4
);
const potentially_relavent_files_paths = potentially_relavent_files.map(
(f) => f.metadata.filename
);
const data = { const data = {
memory_guide: memory_manager_guide("self", context_message.author.id), memory_guide: memory_manager_guide("self", context_message.author.id),
events, events,
user_id: context_message.author.id, user_id: context_message.author.id,
relavent_files: potentially_relavent_files_paths,
model, model,
}; };

1
tools/notes-vectors.ts
View File

@ -41,6 +41,7 @@ async function getAllFiles(
file.type === "file" && file.type === "file" &&
!file.basename.startsWith(".") && !file.basename.startsWith(".") &&
!file.filename.includes("/.obsidian/") && !file.filename.includes("/.obsidian/") &&
!file.filename.includes("prompts/") &&
(file.filename.endsWith(".txt") || file.filename.endsWith(".md")) (file.filename.endsWith(".txt") || file.filename.endsWith(".md"))
) { ) {
const content = await webdavClient.getFileContents(file.filename, { const content = await webdavClient.getFileContents(file.filename, {