From bcd2a64e65fc77350b941a7d492d8c26502b48a6 Mon Sep 17 00:00:00 2001 From: silentsilas Date: Sun, 22 Sep 2024 00:43:53 -0400 Subject: [PATCH] potentially fix missing ebook in production --- src/routes/api/ai/+server.ts | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/routes/api/ai/+server.ts b/src/routes/api/ai/+server.ts index 0369890..091fed0 100644 --- a/src/routes/api/ai/+server.ts +++ b/src/routes/api/ai/+server.ts @@ -8,6 +8,7 @@ import type { RequestEvent } from '@sveltejs/kit'; import { EPubLoader } from "@langchain/community/document_loaders/fs/epub"; import { join } from 'path'; import { getChatHistory, setChatHistory } from '$lib/store'; +import { dev } from '$app/environment'; export async function POST({ request, locals }: RequestEvent): Promise { const { query } = await request.json(); @@ -15,8 +16,14 @@ export async function POST({ request, locals }: RequestEvent): Promise const chatHistory = getChatHistory(sessionId); - const directory = join(process.cwd(), 'static/book.epub'); - const loader = new EPubLoader(directory); + let ebookPath; + 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 context = docs.map(doc => doc.pageContent).join('\n\n');