From dfa4f9cfd0844522b36953abe0ad17afea25fac7 Mon Sep 17 00:00:00 2001
From: loks0n <22452787+loks0n@users.noreply.github.com>
Date: Fri, 17 Jan 2025 09:44:09 +0000
Subject: [PATCH] feat: custom system prompt

---
 src/main.js | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/main.js b/src/main.js
index 13610ed..ba835f1 100644
--- a/src/main.js
+++ b/src/main.js
@@ -34,7 +34,9 @@ app.post("/v1/models/assistant/prompt", async (req, res) => {
   const decoder = new TextDecoder();
   const text = decoder.decode(req.body);
 
-  let { prompt } = JSON.parse(text);
+  let { prompt, systemPrompt } = JSON.parse(text);
+  systemPrompt ??= SYSTEM_PROMPT;
+
   const relevantDocuments = await retriever.getRelevantDocuments(prompt);
 
   const chain = await getRagChain((token) => {
@@ -66,7 +68,8 @@ app.post("/v1/models/generic/prompt", async (req, res) => {
   const decoder = new TextDecoder();
   const text = decoder.decode(req.body);
 
-  let { prompt } = JSON.parse(text);
+  let { prompt, systemPrompt } = JSON.parse(text);
+  systemPrompt ??= SYSTEM_PROMPT;
 
   const chain = await getOpenAIChat((token) => {
     res.write(token);