sk/Perplexica
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(UI): allow system prompts and persona prompts to be saved server side and individually included or excluded from messages

Browse source
This commit is contained in:
Willie Zutz 2025-05-27 12:53:30 -06:00
parent 8e6934bb64
commit 011d10df29
27 changed files with 1345 additions and 132 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/lib/chains/imageSearchAgent.ts
View file

@ -91,7 +91,14 @@ const outputParser = new LineOutputParser({
key: 'answer',
});
const createImageSearchChain = (llm: BaseChatModel) => {
const createImageSearchChain = (
llm: BaseChatModel,
systemInstructions?: string,
) => {
const systemPrompt = systemInstructions ? `${systemInstructions}\n\n` : '';
const fullPrompt = `${systemPrompt}${imageSearchChainPrompt}`;
return RunnableSequence.from([
RunnableMap.from({
chat_history: (input: ImageSearchChainInput) => {
@ -102,7 +109,7 @@ const createImageSearchChain = (llm: BaseChatModel) => {
},
date: () => formatDateForLLM(),
}),
PromptTemplate.fromTemplate(imageSearchChainPrompt),
PromptTemplate.fromTemplate(fullPrompt),
llm,
outputParser,
RunnableLambda.from(async (searchQuery: string) => {
@ -130,8 +137,9 @@ const createImageSearchChain = (llm: BaseChatModel) => {
const handleImageSearch = (
input: ImageSearchChainInput,
llm: BaseChatModel,
systemInstructions?: string,
) => {
const imageSearchChain = createImageSearchChain(llm);
const imageSearchChain = createImageSearchChain(llm, systemInstructions);
return imageSearchChain.invoke(input);
};