silentsilas
/
playground
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

potentially fix missing ebook in production

This commit is contained in:
silentsilas 2024-09-22 00:43:53 -04:00
parent 2c3a4be48c
commit bcd2a64e65
Signed by: silentsilas
GPG Key ID: 113DFB380F724A81
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/routes/api/ai/+server.ts
View File

@ -8,6 +8,7 @@ import type { RequestEvent } from '@sveltejs/kit';
import { EPubLoader } from "@langchain/community/document_loaders/fs/epub"; import { EPubLoader } from "@langchain/community/document_loaders/fs/epub";
import { join } from 'path'; import { join } from 'path';
import { getChatHistory, setChatHistory } from '$lib/store'; import { getChatHistory, setChatHistory } from '$lib/store';
import { dev } from '$app/environment';
export async function POST({ request, locals }: RequestEvent): Promise<Response> { export async function POST({ request, locals }: RequestEvent): Promise<Response> {
const { query } = await request.json(); const { query } = await request.json();
@ -15,8 +16,14 @@ export async function POST({ request, locals }: RequestEvent): Promise<Response>
const chatHistory = getChatHistory(sessionId); const chatHistory = getChatHistory(sessionId);
const directory = join(process.cwd(), 'static/book.epub'); let ebookPath;
const loader = new EPubLoader(directory); if (dev) {
ebookPath = join(process.cwd(), 'static', 'book.epub');
} else {
ebookPath = join(process.cwd(), 'build', 'book.epub');
}
const loader = new EPubLoader(ebookPath);
const docs = await loader.load(); const docs = await loader.load();
const context = docs.map(doc => doc.pageContent).join('\n\n'); const context = docs.map(doc => doc.pageContent).join('\n\n');