fix: pipelines

This commit is contained in:
Timothy J. Baek
2024-05-31 13:30:12 -07:00
parent fc2b314c4f
commit cb8c45d864
4 changed files with 11 additions and 8 deletions

View File

@@ -336,6 +336,7 @@ export const generateTitle = async (
template: string,
model: string,
prompt: string,
chat_id?: string,
url: string = OPENAI_API_BASE_URL
) => {
let error = null;
@@ -361,7 +362,9 @@ export const generateTitle = async (
],
stream: false,
// Restricting the max tokens to 50 to avoid long titles
max_tokens: 50
max_tokens: 50,
...(chat_id && { chat_id: chat_id }),
title: true
})
})
.then(async (res) => {

View File

@@ -1118,6 +1118,7 @@
) + ' {{prompt}}',
titleModelId,
userPrompt,
$chatId,
titleModel?.owned_by === 'openai' ?? false
? `${OPENAI_API_BASE_URL}`
: `${OLLAMA_API_BASE_URL}/v1`