diff --git a/src/apis/trans.js b/src/apis/trans.js index e7273d4..232a8a2 100644 --- a/src/apis/trans.js +++ b/src/apis/trans.js @@ -554,6 +554,8 @@ const genOllama = ({ systemPrompt, userPrompt, model, + temperature, + maxTokens, customHeader, customBody, docInfo, @@ -565,9 +567,19 @@ const genOllama = ({ const data = { model, - system: systemPrompt, - prompt: userPrompt, - think: think, + messages: [ + { + role: "system", + content: systemPrompt, + }, + { + role: "user", + content: userPrompt, + }, + ], + temperature, + max_tokens: maxTokens, + think, stream: false, ...customBody, }; diff --git a/src/config/api.js b/src/config/api.js index ab37e5c..3483342 100644 --- a/src/config/api.js +++ b/src/config/api.js @@ -379,7 +379,7 @@ const defaultOpenaiApi = { }; const defaultOllamaApi = { ...defaultApi, - url: "http://localhost:11434/api/generate", + url: "http://localhost:11434/v1/chat/completions", model: "llama3.1", }; export const DEFAULT_TRANS_APIS = {